Update backend/pkg/commons/ratelimit/ratelimit.go

Co-authored-by: Lucca <[email protected]>
gobitfly · Jul 31, 2024 · f599336 · f599336
1 parent 5978301
commit f599336
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/backend/pkg/commons/ratelimit/ratelimit.go b/backend/pkg/commons/ratelimit/ratelimit.go
@@ -359,10 +359,7 @@ func updateWeights(firstRun bool) error {
 		if !firstRun && oldWeights[w.Endpoint] != weights[w.Endpoint] {
 			log.InfoWithFields(log.Fields{"endpoint": w.Endpoint, "weight": w.Weight, "oldWeight": oldWeights[w.Endpoint]}, "weight changed")
 		}
-		buckets[w.Endpoint] = strings.ReplaceAll(w.Bucket, ":", "_")
-		if buckets[w.Endpoint] == "" {
-			buckets[w.Endpoint] = defaultBucket
-		}
+		buckets[w.Endpoint] = cmp.Or(strings.ReplaceAll(w.Bucket, ":", "_"), defaultBucket)
 		if !firstRun && oldBuckets[w.Endpoint] != buckets[w.Endpoint] {
 			log.InfoWithFields(log.Fields{"endpoint": w.Endpoint, "bucket": w.Weight, "oldBucket": oldBuckets[w.Endpoint]}, "bucket changed")
 		}