From 37aab966dff25e2e0a21f921ac6b6b0eea636090 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Wed, 21 Jun 2023 10:49:26 +0800 Subject: [PATCH] close conn --- pkg/discoveryregistry/discovery_register.go | 1 + pkg/discoveryregistry/zookeeper/discover.go | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/pkg/discoveryregistry/discovery_register.go b/pkg/discoveryregistry/discovery_register.go index 880e31e20..e9e706ab4 100644 --- a/pkg/discoveryregistry/discovery_register.go +++ b/pkg/discoveryregistry/discovery_register.go @@ -11,6 +11,7 @@ type Conn interface { GetConns(ctx context.Context, serviceName string, opts ...grpc.DialOption) ([]*grpc.ClientConn, error) GetConn(ctx context.Context, serviceName string, opts ...grpc.DialOption) (*grpc.ClientConn, error) AddOption(opts ...grpc.DialOption) + CloseConn(conn *grpc.ClientConn) // do not use this method for call rpc GetClientLocalConns() map[string][]resolver.Address } diff --git a/pkg/discoveryregistry/zookeeper/discover.go b/pkg/discoveryregistry/zookeeper/discover.go index 9103b884a..1a5c29659 100644 --- a/pkg/discoveryregistry/zookeeper/discover.go +++ b/pkg/discoveryregistry/zookeeper/discover.go @@ -104,3 +104,7 @@ func (s *ZkClient) GetConn(ctx context.Context, serviceName string, opts ...grpc newOpts := append(s.options, grpc.WithDefaultServiceConfig(fmt.Sprintf(`{"LoadBalancingPolicy": "%s"}`, s.balancerName))) return grpc.DialContext(ctx, fmt.Sprintf("%s:///%s", s.scheme, serviceName), append(newOpts, opts...)...) } + +func (s *ZkClient) CloseConn(conn *grpc.ClientConn) { + conn.Close() +}