Fix max stored permits

jd-opensource · Jan 23, 2025 · f7328fc · f7328fc
1 parent 8d202af
commit f7328fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...in/java/com/jd/live/agent/governance/invoke/ratelimit/leakybucket/LeakyBucketLimiter.java b/...in/java/com/jd/live/agent/governance/invoke/ratelimit/leakybucket/LeakyBucketLimiter.java
@@ -42,7 +42,7 @@ public LeakyBucketLimiter(RateLimitPolicy limitPolicy, SlidingWindow slidingWind
     @Override
     protected double getMaxStoredPermits() {
         // Keep a constant rate and prevent excessive token accumulation.
-        return Math.max(getPermits(option.getPositive(KEY_FLOW_WINDOW_SECONDS, 1)), 1D);
+        return getPermits(option.getPositive(KEY_FLOW_WINDOW_SECONDS, 1));
     }
 
     @Override

diff --git a/...in/java/com/jd/live/agent/governance/invoke/ratelimit/tokenbucket/TokenBucketLimiter.java b/...in/java/com/jd/live/agent/governance/invoke/ratelimit/tokenbucket/TokenBucketLimiter.java
@@ -102,7 +102,7 @@ protected boolean doAcquire(int permits, long nowMicros, long timeoutMicros) {
      * Initializes the rate limiter by setting the maximum number of permits.
      */
     protected void initialize() {
-        this.maxStoredPermits = getMaxStoredPermits();
+        this.maxStoredPermits = Math.max(getMaxStoredPermits(), 1D);
     }
 
     /**