fix: stop logging error when template schedule query is canceled

spikecurtis · spikecurtis · commit 9014afdffb78 · 2024-11-06T14:15:42.000+04:00
diff --git a/coderd/workspacestats/reporter.go b/coderd/workspacestats/reporter.go
@@ -154,17 +154,21 @@ func (r *Reporter) ReportAgentStats(ctx context.Context, now time.Time, workspac
 		templateSchedule, err := (*(r.opts.TemplateScheduleStore.Load())).Get(ctx, r.opts.Database, workspace.TemplateID)
 		// If the template schedule fails to load, just default to bumping
 		// without the next transition and log it.
-		if err != nil {
+		if err == nil {
+			next, allowed := schedule.NextAutostart(now, workspace.AutostartSchedule.String, templateSchedule)
+			if allowed {
+				nextAutostart = next
+			}
+		} else if database.IsQueryCanceledError(err) {
+			r.opts.Logger.Debug(ctx, "query canceled while loading template schedule",
+				slog.F("workspace_id", workspace.ID),
+				slog.F("template_id", workspace.TemplateID))
+		} else {
 			r.opts.Logger.Error(ctx, "failed to load template schedule bumping activity, defaulting to bumping by 60min",
 				slog.F("workspace_id", workspace.ID),
 				slog.F("template_id", workspace.TemplateID),
 				slog.Error(err),
 			)
-		} else {
-			next, allowed := schedule.NextAutostart(now, workspace.AutostartSchedule.String, templateSchedule)
-			if allowed {
-				nextAutostart = next
-			}
 		}
 	}
 
