close conn

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
 }

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()
+}