feat(throttle): add constructor for compatibility (#1164)

Signed-off-by: Shichao Nie <[email protected]>
AutoMQ · Apr 24, 2024 · 56e726b · 56e726b
1 parent c7370bd
commit 56e726b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/s3stream/src/main/java/com/automq/stream/s3/network/AsyncNetworkBandwidthLimiter.java b/s3stream/src/main/java/com/automq/stream/s3/network/AsyncNetworkBandwidthLimiter.java
@@ -42,11 +42,15 @@ public class AsyncNetworkBandwidthLimiter {
     private long availableExtraTokens;
 
     public AsyncNetworkBandwidthLimiter(Type type, long tokenSize, int refillIntervalMs) {
+        this(type, tokenSize, refillIntervalMs, tokenSize);
+    }
+
+    public AsyncNetworkBandwidthLimiter(Type type, long tokenSize, int refillIntervalMs, long maxTokens) {
         this.type = type;
         this.extraTokenSize = (long) (tokenSize * DEFAULT_EXTRA_TOKEN_RATIO);
         this.tokenSize = tokenSize - extraTokenSize;
         this.availableTokens = this.tokenSize;
-        this.maxTokens = tokenSize;
+        this.maxTokens = maxTokens;
         this.queuedCallbacks = new PriorityQueue<>();
         this.refillThreadPool = Executors.newSingleThreadScheduledExecutor(new DefaultThreadFactory("refill-bucket-thread"));
         this.callbackThreadPool = Executors.newFixedThreadPool(1, new DefaultThreadFactory("callback-thread"));