fix: reconnect options (#374)

vesoft-inc · Oct 25, 2023 · e853862 · e853862
1 parent e5b9ae0
commit e853862
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pkg/nebula/meta_client.go b/pkg/nebula/meta_client.go
@@ -60,7 +60,8 @@ type (
 	}
 
 	metaClient struct {
-		client *meta.MetaServiceClient
+		client  *meta.MetaServiceClient
+		options []Option
 	}
 )
 
@@ -81,7 +82,10 @@ func newMetaConnection(endpoint string, options ...Option) (*metaClient, error)
 		return nil, err
 	}
 	metaServiceClient := meta.NewMetaServiceClientFactory(transport, pf)
-	mc := &metaClient{client: metaServiceClient}
+	mc := &metaClient{
+		client:  metaServiceClient,
+		options: options,
+	}
 	if err := mc.connect(); err != nil {
 		return nil, err
 	}
@@ -93,7 +97,7 @@ func (m *metaClient) reconnect(endpoint string) error {
 		return err
 	}
 
-	transport, pf, err := buildClientTransport(endpoint)
+	transport, pf, err := buildClientTransport(endpoint, m.options...)
 	if err != nil {
 		return err
 	}