NotAShelf/ncro
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cmd: fix discovery context lifecycle on shutdown

Browse source 
Signed-off-by: NotAShelf <raf@notashelf.dev>
Change-Id: I0e1c53e1fceee4a5f0e5b737e3a9ed576a6a6964
This commit is contained in:
raf 2026-04-05 15:12:29 +03:00
commit 8181b9f958
Signed by: NotAShelf
GPG key ID: 29D95B64378DB4BF
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
cmd/ncro/root.go
View file

@ -197,16 +197,18 @@ func runServer(_ *cobra.Command, _ []string) error {
// Start mDNS discovery in background // Start mDNS discovery in background
discoveryDone := make(chan struct{}) discoveryDone := make(chan struct{})
var discoveryCancel context.CancelFunc
if discoveryMgr != nil { if discoveryMgr != nil {
var ctx context.Context
ctx, discoveryCancel = context.WithCancel(context.Background())
go func() { go func() {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
if err := discoveryMgr.Start(ctx); err != nil { if err := discoveryMgr.Start(ctx); err != nil {
slog.Error("discovery error", "error", err) slog.Error("discovery error", "error", err)
} }
}() }()
go func() { go func() {
<-discoveryDone <-discoveryDone
discoveryCancel()
discoveryMgr.Stop() discoveryMgr.Stop()
}() }()
} }