From 96bb8b00d7e89813cb73b866cb482676e52bc28a Mon Sep 17 00:00:00 2001
From: husharp <ihusharp@gmail.com>
Date: Wed, 17 Jul 2024 11:16:13 +0800
Subject: [PATCH] fix datarace

Signed-off-by: husharp <ihusharp@gmail.com>
---
 pkg/utils/apiutil/serverapi/middleware.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/utils/apiutil/serverapi/middleware.go b/pkg/utils/apiutil/serverapi/middleware.go
index 0718702b5a5..9af8d234b34 100644
--- a/pkg/utils/apiutil/serverapi/middleware.go
+++ b/pkg/utils/apiutil/serverapi/middleware.go
@@ -270,7 +270,7 @@ func (h *redirector) waitForLeader(r *http.Request) (leader *pdpb.Member) {
 			}
 		case <-r.Context().Done():
 			return
-		case <-h.s.LoopContext().Done():
+		case <-h.s.Context().Done():
 			return
 		}
 	}