replace deprecated method grpc.DialContext in favour of grpc.NewClient

Signed-off-by: Tariq Ibrahim <tibrahim@nvidia.com>
NVIDIA · Nov 20, 2024 · 1cae921 · 1cae921
1 parent b97b763
commit 1cae921
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/pkg/dcgmexporter/kubernetes.go b/pkg/dcgmexporter/kubernetes.go
@@ -105,13 +105,9 @@ func (p *PodMapper) Process(metrics MetricsByCounter, sysInfo SystemInfo) error
 }
 
 func connectToServer(socket string) (*grpc.ClientConn, func(), error) {
-	ctx, cancel := context.WithTimeout(context.Background(), connectionTimeout)
-	defer cancel()
-
-	conn, err := grpc.DialContext(ctx,
+	conn, err := grpc.NewClient(
 		socket,
 		grpc.WithTransportCredentials(insecure.NewCredentials()),
-		grpc.WithBlock(),
 		grpc.WithContextDialer(func(ctx context.Context, addr string) (net.Conn, error) {
 			d := net.Dialer{}
 			return d.DialContext(ctx, "unix", addr)