diff --git a/pkg/kubelet/cm/dra/plugin/client.go b/pkg/kubelet/cm/dra/plugin/client.go index 03e781f8a7e..74ece2c6ef8 100644 --- a/pkg/kubelet/cm/dra/plugin/client.go +++ b/pkg/kubelet/cm/dra/plugin/client.go @@ -22,6 +22,7 @@ import ( "fmt" "io" "net" + "time" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" @@ -131,6 +132,9 @@ func (r *draPluginClient) NodePrepareResource( ResourceHandle: resourceHandle, } + ctx, cancel := context.WithTimeout(ctx, 10*time.Second) + defer cancel() + return nodeClient.NodePrepareResource(ctx, req) } @@ -165,6 +169,9 @@ func (r *draPluginClient) NodeUnprepareResource( CdiDevices: cdiDevices, } + ctx, cancel := context.WithTimeout(ctx, 10*time.Second) + defer cancel() + return nodeClient.NodeUnprepareResource(ctx, req) }