support optional grpc.DialOption to GetServerConnection (#132)

Author: mattkim

app/connection.go

@@ -12,10 +12,10 @@ import (
 	"google.golang.org/grpc/metadata"
 )
 
-func GetServerConnection(c *cli.Context) (context.Context, *grpc.ClientConn, error) {
+func GetServerConnection(c *cli.Context, opts ...grpc.DialOption) (context.Context, *grpc.ClientConn, error) {
 
 	serverAddr := c.String(ServerFlagName)
-	var option grpc.DialOption
+	var credentialOption grpc.DialOption
 	parts := strings.Split(serverAddr, ":")
 
 	if len(parts) != 2 {
@@ -25,16 +25,16 @@ func GetServerConnection(c *cli.Context) (context.Context, *grpc.ClientConn, err
 	hostname := parts[0]
 	switch hostname {
 	case "localhost":
-		option = grpc.WithInsecure()
+		credentialOption = grpc.WithInsecure()
 	default:
-		option = grpc.WithTransportCredentials(credentials.NewTLS(&tls.Config{
+		credentialOption = grpc.WithTransportCredentials(credentials.NewTLS(&tls.Config{
 			MinVersion: tls.VersionTLS12,
 			ServerName: hostname,
 		}))
 	}
 	conn, err := grpc.Dial(
 		serverAddr,
-		option,
+		append(opts, credentialOption)...,
 	)
 	if err != nil {
 		return nil, nil, err