close listener when basecontext is done

DropMorePackets · Oct 6, 2023 · 43adb83 · 43adb83
1 parent 6188742
commit 43adb83
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/spop/agent.go b/spop/agent.go
@@ -34,6 +34,11 @@ func (a *Agent) Serve(l net.Listener) error {
 		a.BaseContext = context.Background()
 	}
 
+	go func() {
+		<-a.BaseContext.Done()
+		l.Close()
+	}()
+
 	for {
 		nc, err := l.Accept()
 		if err != nil {