Fix: Update gRPC client connection to use DialContext

cmd/sdk-server/main.go

@@ -245,7 +245,8 @@ func runGrpc(grpcServer *grpc.Server, grpcEndpoint string) {
 
 // runGateway runs the grpc-gateway
 func runGateway(ctx context.Context, grpcEndpoint string, mux *gwruntime.ServeMux, httpServer *http.Server) {
-	conn, err := grpc.NewClient(grpcEndpoint, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	// nolint: staticcheck
+	conn, err := grpc.DialContext(ctx, grpcEndpoint, grpc.WithBlock(), grpc.WithTransportCredentials(insecure.NewCredentials()))
 	if err != nil {
 		logger.WithError(err).Fatal("Could not dial grpc server...")
 	}

examples/allocation-endpoint/server/main.go

@@ -271,7 +271,8 @@ func updateFailed(clusterName string, err error) {
 }
 
 func connectToAgonesCluster(ctx context.Context, clusterInfo *ClusterInfo) (*grpc.ClientConn, error) {
-	conn, err := grpc.NewClient(fmt.Sprintf("%s:443", clusterInfo.Endpoint), grpc.WithTransportCredentials(cred))
+	// nolint: staticcheck	
+	conn, err := grpc.DialContext(ctx, fmt.Sprintf("%s:443", clusterInfo.Endpoint), grpc.WithTransportCredentials(cred))
 	if err != nil {
 		return nil, errors.Wrapf(err, "could not connect to %s with endpoint %s", clusterInfo.Name, clusterInfo.Endpoint)
 	}

sdks/go/sdk.go

@@ -72,28 +72,16 @@ func NewSDK() (*SDK, error) {
 	// Block for at least 30 seconds.
 	ctx, cancel := context.WithTimeout(s.ctx, 30*time.Second)
 	defer cancel()
-	conn, err := grpc.NewClient(addr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	// nolint: staticcheck
+	conn, err := grpc.DialContext(ctx, addr, grpc.WithBlock(), grpc.WithTransportCredentials(insecure.NewCredentials()))
 	if err != nil {
 		return s, errors.Wrapf(err, "could not connect to %s", addr)
 	}
-
-	// Set up the connection within the 30-second context.
-	done := make(chan error, 1)
-	go func() {
-		s.client = sdk.NewSDKClient(conn)
-		s.health, err = s.client.Health(s.ctx)
-		s.alpha = newAlpha(conn)
-		s.beta = newBeta(conn)
-
-		done <- errors.Wrap(err, "could not set up health check")
-	}()
-
-	select {
-	case err := <-done:
-		return s, err
-	case <-ctx.Done():
-		return s, errors.New("timeout while setting up client")
-	}
+	s.client = sdk.NewSDKClient(conn)
+	s.health, err = s.client.Health(s.ctx)
+	s.alpha = newAlpha(conn)
+	s.beta = newBeta(conn)
+	return s, errors.Wrap(err, "could not set up health check")
 }
 
 // Alpha returns the Alpha SDK.