From 2249ef3b792e19b02aedcd0b4f0a6332dab997ea Mon Sep 17 00:00:00 2001 From: abelanger5 Date: Mon, 17 Nov 2025 15:45:14 -0500 Subject: [PATCH] fix: small scheduler optimizations (#2426) * fix: actually increment snapshot count * add a context with timeout to wrap replenish --- pkg/scheduling/v1/scheduler.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pkg/scheduling/v1/scheduler.go b/pkg/scheduling/v1/scheduler.go index 2668e60c9..03e7d2f1e 100644 --- a/pkg/scheduling/v1/scheduler.go +++ b/pkg/scheduling/v1/scheduler.go @@ -380,11 +380,13 @@ func (s *Scheduler) loopReplenish(ctx context.Context) { case <-ctx.Done(): return case <-ticker.C: - err := s.replenish(ctx, true) + innerCtx, cancel := context.WithTimeout(ctx, 2*time.Second) + err := s.replenish(innerCtx, true) if err != nil { s.l.Error().Err(err).Msg("error replenishing slots") } + cancel() } } } @@ -410,6 +412,8 @@ func (s *Scheduler) loopSnapshot(ctx context.Context) { } s.exts.ReportSnapshot(sqlchelpers.UUIDToStr(s.tenantId), in) + + count++ } } }