fix: Close coordinator on context cancellation

mafredri · mafredri · commit 38c7ec1d2478 · 2022-09-23T11:07:38.000+03:00
diff --git a/agent/agent.go b/agent/agent.go
@@ -291,12 +291,12 @@ func (a *agent) runCoordinator(ctx context.Context) {
 		a.logger.Info(context.Background(), "connected to coordination server")
 		break
 	}
+	defer coordinator.Close()
 	select {
 	case <-ctx.Done():
 		return
 	default:
 	}
-	defer coordinator.Close()
 	sendNodes, errChan := tailnet.ServeCoordinator(coordinator, a.network.UpdateNodes)
 	a.network.SetNodeCallback(sendNodes)
 	select {

Original file line number	Diff line number	Diff line change
`@@ -291,12 +291,12 @@ func (a *agent) runCoordinator(ctx context.Context) {`
`291`	`291`	`a.logger.Info(context.Background(), "connected to coordination server")`
`292`	`292`	`break`
`293`	`293`	`}`
	`294`	`+ defer coordinator.Close()`
`294`	`295`	`select {`
`295`	`296`	`case <-ctx.Done():`
`296`	`297`	`return`
`297`	`298`	`default:`
`298`	`299`	`}`
`299`		`- defer coordinator.Close()`
`300`	`300`	`sendNodes, errChan := tailnet.ServeCoordinator(coordinator, a.network.UpdateNodes)`
`301`	`301`	`a.network.SetNodeCallback(sendNodes)`
`302`	`302`	`select {`