dynamic resource allocation: add timeouts for communiction with plugin (

kubernetes#114844) * add timeouts for communication with dra plugin * move timeout constant to k8s.io/kubernetes/pkg/kubelet/cm/util * move settings of timeout to pkg/kubelet/plugin/dra/plugin/client.go * remove timeout constant
ncdc · Mar 13, 2023 · d34b027 · d34b027
1 parent 185cd95
commit d34b027
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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)
 }