From ecae6f913577cc39706aac9faee2bca0434b6b82 Mon Sep 17 00:00:00 2001
From: Cian Johnston <cian@coder.com>
Date: Wed, 24 Jan 2024 18:42:05 +0000
Subject: [PATCH] fix(enterprise/tailnet): handle query canceled error in
 sendBeat() (#11794)

---
 enterprise/tailnet/pgcoord.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enterprise/tailnet/pgcoord.go b/enterprise/tailnet/pgcoord.go
index 75edd8c446..37f1c2932c 100644
--- a/enterprise/tailnet/pgcoord.go
+++ b/enterprise/tailnet/pgcoord.go
@@ -1494,7 +1494,7 @@ func (h *heartbeats) sendBeats() {
 
 func (h *heartbeats) sendBeat() {
 	_, err := h.store.UpsertTailnetCoordinator(h.ctx, h.self)
-	if xerrors.Is(err, context.Canceled) {
+	if database.IsQueryCanceledError(err) {
 		return
 	}
 	if err != nil {