Merge pull request #13 from fbsobreira/grpc-dial-options

add dial option on grpc start
fbsobreira · Mar 16, 2021 · 8cb47d5 · 8cb47d5
2 parents 7ad5b92 + 092f1f0
commit 8cb47d5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/cmd/subcommands/root.go b/cmd/subcommands/root.go
@@ -22,6 +22,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/cobra/doc"
 	"golang.org/x/crypto/ssh/terminal"
+	"google.golang.org/grpc"
 )
 
 var (
@@ -55,7 +56,7 @@ var (
 				node = node + ":50051"
 			}
 			conn = client.NewGrpcClient(node)
-			if err := conn.Start(); err != nil {
+			if err := conn.Start(grpc.WithInsecure()); err != nil {
 				return err
 			}
 

diff --git a/pkg/client/client.go b/pkg/client/client.go
@@ -14,6 +14,7 @@ type GrpcClient struct {
 	Conn        *grpc.ClientConn
 	Client      api.WalletClient
 	grpcTimeout time.Duration
+	opts        []grpc.DialOption
 }
 
 // NewGrpcClient create grpc controller
@@ -40,12 +41,13 @@ func (g *GrpcClient) SetTimeout(timeout time.Duration) {
 }
 
 // Start initiate grpc  connection
-func (g *GrpcClient) Start() error {
+func (g *GrpcClient) Start(opts ...grpc.DialOption) error {
 	var err error
 	if len(g.Address) == 0 {
 		g.Address = "grp.trongrid.io:50051"
 	}
-	g.Conn, err = grpc.Dial(g.Address, grpc.WithInsecure())
+	g.opts = opts
+	g.Conn, err = grpc.Dial(g.Address, opts...)
 
 	if err != nil {
 		return fmt.Errorf("Connecting GRPC Client: %v", err)
@@ -67,7 +69,7 @@ func (g *GrpcClient) Reconnect(url string) error {
 	if len(url) > 0 {
 		g.Address = url
 	}
-	g.Start()
+	g.Start(g.opts...)
 	return nil
 }