fix context early cancelation

gravitational · Oct 16, 2024 · be72ee1 · be72ee1
1 parent 3852e32
commit be72ee1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/srv/discovery/status.go b/lib/srv/discovery/status.go
@@ -413,9 +413,9 @@ func (s *Server) acquireSemaphoreForUserTask(userTaskName string) (releaseFn fun
 
 	// Once the lease parent context is canceled, the lease will be released.
 	ctxWithLease, cancel := context.WithCancel(lease)
-	defer cancel()
 
 	releaseFn = func() {
+		cancel()
 		lease.Stop()
 		if err := lease.Wait(); err != nil {
 			s.Log.WithError(err).WithField("semaphore", userTaskName).Warn("error cleaning up UserTask semaphore")