Skip to content

Commit bd1aded

Browse files
committed
Precompute overrides at loadOverrides time
1 parent 60b6c3b commit bd1aded

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

ratelimits/limit.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,6 @@ func parseOverrideLimits(newOverridesYAML overridesYAML) (Limits, error) {
258258
Comment: entry.Comment,
259259
isOverride: true,
260260
}
261-
lim.precompute()
262261

263262
err = ValidateLimit(lim)
264263
if err != nil {
@@ -402,13 +401,14 @@ func (l *limitRegistry) loadOverrides(ctx context.Context) error {
402401
return nil
403402
}
404403

405-
l.overrides = newOverrides
406-
407-
l.overridesTimestamp.SetToCurrentTime()
408404
newOverridesPerLimit := make(map[Name]float64)
409405
for _, override := range newOverrides {
406+
override.precompute()
410407
newOverridesPerLimit[override.Name]++
411408
}
409+
410+
l.overrides = newOverrides
411+
l.overridesTimestamp.SetToCurrentTime()
412412
for rlName, rlString := range nameToString {
413413
l.overridesPerLimit.WithLabelValues(rlString).Set(newOverridesPerLimit[rlName])
414414
}

ratelimits/limit_test.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,9 @@ func TestLoadOverrides(t *testing.T) {
262262
test.AssertNotError(t, err, "loading overrides from file")
263263
testOverrides, err := parseOverrideLimits(overridesData)
264264
test.AssertNotError(t, err, "parsing overrides")
265+
for _, override := range testOverrides {
266+
override.precompute()
267+
}
265268

266269
test.AssertDeepEquals(t, tb.limitRegistry.overrides, testOverrides)
267270

ratelimits/transaction.go

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,6 @@ func NewTransactionBuilderFromDatabase(defaults string, overrides GetOverridesFu
213213
continue
214214
}
215215

216-
newLimit.precompute()
217216
overrides[bucketKey] = newLimit
218217
}
219218
errorGauge.Set(errorCount)

0 commit comments

Comments
 (0)