-
Notifications
You must be signed in to change notification settings - Fork 2
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add ability to limit outgoing message rate #142
base: dev-version-3
Are you sure you want to change the base?
Conversation
import java.util.concurrent.TimeUnit.SECONDS | ||
import java.util.concurrent.locks.LockSupport | ||
|
||
class RateLimiter(rate: Int) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This implementation isn't thread-safe, and we should remind about this if we want to use more than one thread for AMQP
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I can make it thread-safe
@@ -94,7 +95,7 @@ public void stop() throws IOException { | |||
private void handle(String consumerTag, RawMessageBatch messageBatch) { | |||
for (RawMessage protoMessage : messageBatch.getMessagesList()) { | |||
try { | |||
rateLimiter.acquire(); | |||
while (!rateLimiter.tryAcquire()) LockSupport.parkNanos(1_000); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's encapsulate the cycle in the tryAcquire function
9f748c6
to
3c78b1f
Compare
Co-authored-by: Nikita Smirnov <[email protected]>
This pull request sets up GitHub code scanning for this repository. Once the scans have completed and the checks have passed, the analysis results for this pull request branch will appear on this overview. Once you merge this pull request, the 'Security' tab will show more code scanning analysis results (for example, for the default branch). Depending on your configuration and choice of analysis tool, future pull requests will be annotated with code scanning analysis results. For more information about GitHub code scanning, check out the documentation. |
gradle.properties
Outdated
@@ -1 +1,5 @@ | |||
<<<<<<< HEAD |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Resolve conflicts
No description provided.