From ae3d78bbc0489b58817afe790168d57fbe3e35d2 Mon Sep 17 00:00:00 2001 From: Ryan Leung Date: Wed, 31 Jan 2024 14:52:27 +0800 Subject: [PATCH] address comments Signed-off-by: Ryan Leung --- server/forward.go | 2 +- server/grpc_service.go | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/server/forward.go b/server/forward.go index 5b9a1a481ba..5be798f095d 100644 --- a/server/forward.go +++ b/server/forward.go @@ -143,7 +143,6 @@ func (s *GrpcServer) forwardTSO(stream pdpb.PD_TsoServer) error { cancelForward() } - s.closeDelegateClient(lastForwardedHost) clientConn, err := s.getDelegateClient(s.ctx, forwardedHost) if err != nil { tsoStreamErr = errors.WithStack(err) @@ -378,6 +377,7 @@ func (s *GrpcServer) closeDelegateClient(forwardedHost string) { return } client.(*grpc.ClientConn).Close() + log.Debug("close delegate client connection", zap.String("forwarded-host", forwardedHost)) } func (s *GrpcServer) isLocalRequest(host string) bool { diff --git a/server/grpc_service.go b/server/grpc_service.go index cc51588c9e7..54df54832f1 100644 --- a/server/grpc_service.go +++ b/server/grpc_service.go @@ -1332,7 +1332,6 @@ func (s *GrpcServer) RegionHeartbeat(stream pdpb.PD_RegionHeartbeatServer) error cancel() } - s.closeDelegateClient(lastForwardedSchedulingHost) client, err := s.getDelegateClient(s.ctx, forwardedSchedulingHost) if err != nil { errRegionHeartbeatClient.Inc()