fix context leak in timeout chain element

Signed-off-by: NikitaSkrynnik <nikita.skrynnik@xored.com>
networkservicemesh · Oct 14, 2024 · 7221c8c · 7221c8c
1 parent 76b397f
commit 7221c8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/networkservice/common/timeout/server.go b/pkg/networkservice/common/timeout/server.go
@@ -77,10 +77,11 @@ func (s *timeoutServer) Request(ctx context.Context, request *networkservice.Net
 	afterCh := timeClock.After(timeClock.Until(expirationTime) - requestTimeout)
 
 	go func(cancelCtx context.Context, afterCh <-chan time.Time) {
+		defer cancel()
 		select {
 		case <-cancelCtx.Done():
 		case <-afterCh:
-			eventFactory.Close(begin.CancelContext(cancelCtx))
+			<-eventFactory.Close(begin.CancelContext(cancelCtx))
 		}
 	}(cancelCtx, afterCh)