Update grpc_broker.go

hashicorp · Jun 4, 2023 · 7bb12ec · 7bb12ec
1 parent a88a423
commit 7bb12ec
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/grpc_broker.go b/grpc_broker.go
@@ -367,7 +367,10 @@ func (b *GRPCBroker) Close() error {
 }
 
 // Dial opens a connection by ID.
-func (b *GRPCBroker) Dial(id uint32) (conn *grpc.ClientConn, err error) {
+func (b *GRPCBroker) Dial(id uint32) (conn *grpc.ClientConn, err error) { return b.DialWithOptions(id) }
+
+// Dial opens a connection by ID with options.
+func (b *GRPCBroker) DialWithOptions(id uint32, opts ...grpc.DialOption) (conn *grpc.ClientConn, err error) {
 	var c *plugin.ConnInfo
 
 	// Open the stream
@@ -392,7 +395,7 @@ func (b *GRPCBroker) Dial(id uint32) (conn *grpc.ClientConn, err error) {
 		return nil, err
 	}
 
-	return dialGRPCConn(b.tls, netAddrDialer(addr))
+	return dialGRPCConn(b.tls, netAddrDialer(addr), opts...)
 }
 
 // NextId returns a unique ID to use next.