From 26fdf4600144e888cb40566b28214350eca659cd Mon Sep 17 00:00:00 2001
From: Msk233 <118117161+Mskxn@users.noreply.github.com>
Date: Wed, 26 Apr 2023 20:27:02 +0800
Subject: [PATCH] close the session to avoid leak goroutine

Signed-off-by: Mskxn <118117161+Mskxn@users.noreply.github.com>
---
 server/proxy/grpcproxy/register.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/proxy/grpcproxy/register.go b/server/proxy/grpcproxy/register.go
index 4fafb4810..98ba6c80b 100644
--- a/server/proxy/grpcproxy/register.go
+++ b/server/proxy/grpcproxy/register.go
@@ -69,10 +69,12 @@ func registerSession(lg *zap.Logger, c *clientv3.Client, prefix string, addr str
 
 	em, err := endpoints.NewManager(c, prefix)
 	if err != nil {
+		ss.Close()
 		return nil, err
 	}
 	endpoint := endpoints.Endpoint{Addr: addr, Metadata: getMeta()}
 	if err = em.AddEndpoint(c.Ctx(), prefix+"/"+addr, endpoint, clientv3.WithLease(ss.Lease())); err != nil {
+		ss.Close()
 		return nil, err
 	}