From b7ecb7f56bdf37883db9aef9f18c213c63b1866f Mon Sep 17 00:00:00 2001
From: wayblink <anyang.wang@zilliz.com>
Date: Tue, 27 Jun 2023 17:08:45 +0800
Subject: [PATCH] Disable retryKeepAlive when LivenessCheck's Context close
 (#25161)

Signed-off-by: wayblink <anyang.wang@zilliz.com>
---
 internal/util/sessionutil/session_util.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/internal/util/sessionutil/session_util.go b/internal/util/sessionutil/session_util.go
index a55ecc9923..8b1796d2db 100644
--- a/internal/util/sessionutil/session_util.go
+++ b/internal/util/sessionutil/session_util.go
@@ -784,6 +784,7 @@ func (s *Session) LivenessCheck(ctx context.Context, callback func()) {
 				return
 			case <-ctx.Done():
 				log.Debug("liveness exits due to context done")
+				s.enableRetryKeepAlive = false
 				// cancel the etcd keepAlive context
 				if s.keepAliveCancel != nil {
 					s.keepAliveCancel()