fix(coderd/batchstats): use debug log on query cancellation in flush (#9778)

mafredri · web-flow · commit 161a3cfa26a9 · 2023-09-19T18:16:07.000+03:00
Fixes #9772
diff --git a/coderd/batchstats/batcher.go b/coderd/batchstats/batcher.go
@@ -243,8 +243,8 @@ func (b *Batcher) flush(ctx context.Context, forced bool, reason string) {
 	err = b.store.InsertWorkspaceAgentStats(ctx, *b.buf)
 	elapsed := time.Since(start)
 	if err != nil {
-		if xerrors.Is(err, context.Canceled) {
-			b.log.Debug(ctx, "context canceled, skipping insert of workspace agent stats", slog.F("elapsed", elapsed))
+		if database.IsQueryCanceledError(err) {
+			b.log.Debug(ctx, "query canceled, skipping insert of workspace agent stats", slog.F("elapsed", elapsed))
 			return
 		}
 		b.log.Error(ctx, "error inserting workspace agent stats", slog.Error(err), slog.F("elapsed", elapsed))
