From f23e9dc2354d8b9b8086fa1c5bfd1ad08fd27f0f Mon Sep 17 00:00:00 2001
From: Juan Font Alonso <juanfontalonso@gmail.com>
Date: Sun, 4 Sep 2022 11:43:09 +0200
Subject: [PATCH] Pass the req context when pinging the DB

---
 api.go | 2 +-
 db.go  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/api.go b/api.go
index 18ac72fe..f5de5038 100644
--- a/api.go
+++ b/api.go
@@ -52,7 +52,7 @@ func (h *Headscale) HealthHandler(
 		}
 	}
 
-	if err := h.pingDB(); err != nil {
+	if err := h.pingDB(req.Context()); err != nil {
 		respond(err)
 
 		return
diff --git a/db.go b/db.go
index 17df384d..a1a4ef33 100644
--- a/db.go
+++ b/db.go
@@ -221,8 +221,8 @@ func (h *Headscale) setValue(key string, value string) error {
 	return nil
 }
 
-func (h *Headscale) pingDB() error {
-	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+func (h *Headscale) pingDB(ctx context.Context) error {
+	ctx, cancel := context.WithTimeout(ctx, time.Second)
 	defer cancel()
 	db, err := h.db.DB()
 	if err != nil {