delete config when QPS and concurrency set to 0

Signed-off-by: Ryan Leung <[email protected]>
tikv · Sep 24, 2024 · d03c2f1 · d03c2f1
1 parent edb43c0
commit d03c2f1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/ratelimit/option.go b/pkg/ratelimit/option.go
@@ -78,7 +78,11 @@ func UpdateDimensionConfig(cfg *DimensionConfig) Option {
 			return InAllowList
 		}
 		lim, _ := l.limiters.LoadOrStore(label, newLimiter())
-		return lim.(*limiter).updateDimensionConfig(cfg)
+		status := lim.(*limiter).updateDimensionConfig(cfg)
+		if status&QPSDeleted != 0 && status&ConcurrencyDeleted != 0 {
+			l.limiters.Delete(label)
+		}
+		return status
 	}
 }