From 55d452e8cc8e782345ab36f2f5e57a45068067f1 Mon Sep 17 00:00:00 2001
From: radhitya <alif@radhitya.org>
Date: Sun, 14 Jun 2026 18:49:03 +0700
Subject: hmm

---
 internal/server/server.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'internal/server/server.go')

diff --git a/internal/server/server.go b/internal/server/server.go
index ec0dec9..8f991eb 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -20,11 +20,17 @@ type Server struct {
 	udp      *dns.Server
 	tcp      *dns.Server
 	doh      *http.Server
+	baseCtx context.Context
+	cancel context.CancelFunc
 }
 
 func New(udpAddr, tcpAddr, dohAddr string, logger *slog.Logger,
 r *resolver.Resolver, c *cache.Cache, b *blocklist.Blocklist) (*Server, error) {
-	s := &Server{logger: logger, resolver: r, cache: c, blocklist: b}
+	baseCtx, cancel := context.WithCancel(context.Background())
+
+	s := &Server{logger: logger, resolver: r, cache: c, blocklist: b,
+	baseCtx: baseCtx, cancel: cancel}
+
 	mux := dns.NewServeMux()
 	mux.HandleFunc(".", s.handleQuery)
 
-- 
cgit v1.2.3