0g-chain/client/grpc/query/connection.go

package query

import (
	"context"
	"crypto/tls"
	"fmt"
	"net/url"

	"github.com/0glabs/0g-chain/app"
	"github.com/cosmos/cosmos-sdk/codec"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials"
	"google.golang.org/grpc/credentials/insecure"
)

// newGrpcConnection parses a GRPC endpoint and creates a connection to it
func newGrpcConnection(ctx context.Context, endpoint string) (*grpc.ClientConn, error) {
	grpcUrl, err := url.Parse(endpoint)
	if err != nil {
		return nil, fmt.Errorf("failed to parse grpc connection \"%s\": %v", endpoint, err)
	}

	var creds credentials.TransportCredentials
	switch grpcUrl.Scheme {
	case "http":
		creds = insecure.NewCredentials()
	case "https":
		creds = credentials.NewTLS(&tls.Config{})
	default:
		return nil, fmt.Errorf("unknown grpc url scheme: %s", grpcUrl.Scheme)
	}

	// Ensure the encoding config is set up correctly with the query client
	// otherwise it will produce panics like:
	// invalid Go type math.Int for field ...
	encodingConfig := app.MakeEncodingConfig()
	protoCodec := codec.NewProtoCodec(encodingConfig.InterfaceRegistry)
	grpcCodec := protoCodec.GRPCCodec()

	secureOpt := grpc.WithTransportCredentials(creds)
	grpcConn, err := grpc.DialContext(
		ctx,
		grpcUrl.Host,
		secureOpt,
		grpc.WithDefaultCallOptions(grpc.ForceCodec(grpcCodec)),
	)
	if err != nil {
		return nil, err
	}

	return grpcConn, nil
}