diff --git a/coderd/batchstats/batcher.go b/coderd/batchstats/batcher.go index d0b3e4fb6c..cc234c693e 100644 --- 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))