add extended context to begin.Request

Signed-off-by: NikitaSkrynnik <[email protected]>
networkservicemesh · Aug 1, 2024 · abf4714 · abf4714
1 parent 515e6bc
commit abf4714
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/networkservice/common/begin/server.go b/pkg/networkservice/common/begin/server.go
@@ -102,7 +102,12 @@ func (b *beginServer) Request(ctx context.Context, request *networkservice.Netwo
 		}
 
 		withEventFactoryCtx := withEventFactory(ctx, eventFactoryServer)
-		conn, err = next.Server(withEventFactoryCtx).Request(withEventFactoryCtx, request)
+
+		newCloseCtx, cancel := context.WithTimeout(context.Background(), time.Minute*3)
+		defer cancel()
+		newCloseCtx = extend.WithValuesFromContext(newCloseCtx, withEventFactoryCtx)
+
+		conn, err = next.Server(newCloseCtx).Request(newCloseCtx, request)
 		if err != nil {
 			if eventFactoryServer.state != established {
 				eventFactoryServer.state = closed