diff --git a/coderd/provisionerdserver/provisionerdserver.go b/coderd/provisionerdserver/provisionerdserver.go index f204cf2a728a4..2715ba6776e8d 100644 --- a/coderd/provisionerdserver/provisionerdserver.go +++ b/coderd/provisionerdserver/provisionerdserver.go @@ -242,10 +242,8 @@ func (s *server) heartbeatLoop() { } start := s.timeNow() hbCtx, hbCancel := context.WithTimeout(s.lifecycleCtx, s.heartbeatInterval) - if err := s.heartbeat(hbCtx); err != nil { - if !xerrors.Is(err, context.DeadlineExceeded) && !xerrors.Is(err, context.Canceled) { - s.Logger.Error(hbCtx, "heartbeat failed", slog.Error(err)) - } + if err := s.heartbeat(hbCtx); err != nil && !database.IsQueryCanceledError(err) { + s.Logger.Error(hbCtx, "heartbeat failed", slog.Error(err)) } hbCancel() elapsed := s.timeNow().Sub(start)