Browse Source

functional-tester/rpcpb: support extra dial options

Signed-off-by: Gyuho Lee <gyuhox@gmail.com>
Gyuho Lee 7 years ago
parent
commit
b55a5a9771
1 changed files with 2 additions and 2 deletions
  1. 2 2
      tools/functional-tester/rpcpb/member.go

+ 2 - 2
tools/functional-tester/rpcpb/member.go

@@ -32,12 +32,12 @@ var dialOpts = []grpc.DialOption{
 }
 
 // DialEtcdGRPCServer creates a raw gRPC connection to an etcd member.
-func (m *Member) DialEtcdGRPCServer() (*grpc.ClientConn, error) {
+func (m *Member) DialEtcdGRPCServer(opts ...grpc.DialOption) (*grpc.ClientConn, error) {
 	if m.EtcdClientTLS {
 		// TODO: support TLS
 		panic("client TLS not supported yet")
 	}
-	return grpc.Dial(m.EtcdClientEndpoint, dialOpts...)
+	return grpc.Dial(m.EtcdClientEndpoint, append(dialOpts, opts...)...)
 }
 
 // CreateEtcdClient creates a client from member.