From 694a54d196223ec58e15ce78f3b3ec8ed1c5b395 Mon Sep 17 00:00:00 2001 From: Pascal Fischer Date: Tue, 20 May 2025 18:13:29 +0200 Subject: [PATCH] make interval configurable --- management/server/grpcserver.go | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/management/server/grpcserver.go b/management/server/grpcserver.go index 6ad6d8d2c..cac8149be 100644 --- a/management/server/grpcserver.go +++ b/management/server/grpcserver.go @@ -84,11 +84,16 @@ func NewServer( } } - syncRatePerS, err := strconv.Atoi(os.Getenv("NB_SYNC_RATE_PER_M")) - if syncRatePerS == 0 || err != nil { - syncRatePerS = 200 + syncTokenPerInterval, err := strconv.Atoi(os.Getenv("NB_SYNC_RATE_PER_M")) + if syncTokenPerInterval == 0 || err != nil { + syncTokenPerInterval = 200 + } + log.WithContext(ctx).Infof("sync rate limit set to %d/min", syncTokenPerInterval) + + syncTokenInterval, err := time.ParseDuration(os.Getenv("NB_SYNC_RATE_INTERVAL")) + if syncTokenInterval == 0 || err != nil { + syncTokenInterval = time.Minute } - log.WithContext(ctx).Infof("sync rate limit set to %d/min", syncRatePerS) syncBurst, err := strconv.Atoi(os.Getenv("NB_SYNC_BURST")) if syncBurst == 0 || err != nil { @@ -107,7 +112,7 @@ func NewServer( authManager: authManager, appMetrics: appMetrics, ephemeralManager: ephemeralManager, - syncRate: rate.Every(time.Minute / time.Duration(syncRatePerS)), + syncRate: rate.Every(syncTokenInterval / time.Duration(syncTokenPerInterval)), syncBurst: syncBurst, }, nil }