TheArchive/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2024-12-22 16:07:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass the req context when pinging the DB

Browse Source
This commit is contained in:
Juan Font Alonso 2022-09-04 11:43:09 +02:00
parent 4527801d48
commit f23e9dc235
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api.go
View File

@ -52,7 +52,7 @@ func (h *Headscale) HealthHandler(
} }
} }
if err := h.pingDB(); err != nil { if err := h.pingDB(req.Context()); err != nil {
respond(err) respond(err)
return return

4
db.go
View File

@ -221,8 +221,8 @@ func (h *Headscale) setValue(key string, value string) error {
return nil return nil
} }
func (h *Headscale) pingDB() error { func (h *Headscale) pingDB(ctx context.Context) error {
ctx, cancel := context.WithTimeout(context.Background(), time.Second) ctx, cancel := context.WithTimeout(ctx, time.Second)
defer cancel() defer cancel()
db, err := h.db.DB() db, err := h.db.DB()
if err != nil { if err != nil {