This is an automated email from the ASF dual-hosted git repository.
ycai pushed a commit to branch trunk
in repository https://gitbox.apache.org/repos/asf/cassandra-sidecar.git
The following commit(s) were added to refs/heads/trunk by this push:
new 30c04eb CASSANDRASC-45: Delegate methods to the RateLimiter
30c04eb is described below
commit 30c04eb38a796183643bdcbaff8f425d90ebf671
Author: Francisco Guerrero <[email protected]>
AuthorDate: Mon Oct 10 18:48:31 2022 -0700
CASSANDRASC-45: Delegate methods to the RateLimiter
Sidecar offers a `SidecarRateLimiter` class that internally uses the
`com.google.common.util.concurrent.RateLimiter`. In this commit, we expose
public methods
of the `RateLimiter` class using the delegate pattern. These methods will
allow us to tweak
the settings of the `RateLimiter` that are available to us
patch by Francisco Guerrero; reviewed by Yifan Cai, Dinesh Joshi for
CASSANDRASC-45
---
.../common/util/concurrent/SidecarRateLimiter.java | 75 ++++++++++++++++++++++
1 file changed, 75 insertions(+)
diff --git
a/src/main/java/com/google/common/util/concurrent/SidecarRateLimiter.java
b/src/main/java/com/google/common/util/concurrent/SidecarRateLimiter.java
index 01ee9ff..6e488ae 100644
--- a/src/main/java/com/google/common/util/concurrent/SidecarRateLimiter.java
+++ b/src/main/java/com/google/common/util/concurrent/SidecarRateLimiter.java
@@ -18,6 +18,10 @@
package com.google.common.util.concurrent;
+import java.util.concurrent.TimeUnit;
+
+import com.google.errorprone.annotations.CanIgnoreReturnValue;
+
/**
* Wrapper class over guava Rate Limiter, uses SmoothBursty Ratelimiter. This
class mainly exists to expose
* package protected method queryEarliestAvailable of guava RateLimiter
@@ -36,6 +40,8 @@ public class SidecarRateLimiter
return new SidecarRateLimiter(permitsPerSecond);
}
+ // Delegated methods
+
/**
* Returns earliest time permits will become available
*/
@@ -51,4 +57,73 @@ public class SidecarRateLimiter
{
return this.rateLimiter.tryAcquire();
}
+
+ /**
+ * Updates the stable rate of the internal {@code RateLimiter}, that is,
the {@code permitsPerSecond}
+ * argument provided in the factory method that constructed the {@code
RateLimiter}.
+ *
+ * @param permitsPerSecond the new stable rate of this {@code RateLimiter}
+ * @throws IllegalArgumentException if {@code permitsPerSecond} is
negative or zero
+ */
+ public void rate(double permitsPerSecond)
+ {
+ this.rateLimiter.setRate(permitsPerSecond);
+ }
+
+ public void doSetRate(double permitsPerSecond, long nowMicros)
+ {
+ rateLimiter.doSetRate(permitsPerSecond, nowMicros);
+ }
+
+ public double rate()
+ {
+ return rateLimiter.getRate();
+ }
+
+ public double doGetRate()
+ {
+ return rateLimiter.doGetRate();
+ }
+
+ @CanIgnoreReturnValue
+ public double acquire()
+ {
+ return rateLimiter.acquire();
+ }
+
+ @CanIgnoreReturnValue
+ public double acquire(int permits)
+ {
+ return rateLimiter.acquire(permits);
+ }
+
+ public long reserve(int permits)
+ {
+ return rateLimiter.reserve(permits);
+ }
+
+ public boolean tryAcquire(long timeout, TimeUnit unit)
+ {
+ return rateLimiter.tryAcquire(timeout, unit);
+ }
+
+ public boolean tryAcquire(int permits)
+ {
+ return rateLimiter.tryAcquire(permits);
+ }
+
+ public boolean tryAcquire(int permits, long timeout, TimeUnit unit)
+ {
+ return rateLimiter.tryAcquire(permits, timeout, unit);
+ }
+
+ public long reserveAndGetWaitLength(int permits, long nowMicros)
+ {
+ return rateLimiter.reserveAndGetWaitLength(permits, nowMicros);
+ }
+
+ public long reserveEarliestAvailable(int permits, long nowMicros)
+ {
+ return rateLimiter.reserveEarliestAvailable(permits, nowMicros);
+ }
}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
