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

---
 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 75edd8c446cb5..37f1c2932c4b9 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 {