From f6d3b4f7ace91382e18f63ee17c404e6e22b6704 Mon Sep 17 00:00:00 2001
From: congqixia <congqi.xia@zilliz.com>
Date: Tue, 9 May 2023 11:00:40 +0800
Subject: [PATCH] Fix concurrent map read/write in rate limiter (#23958)

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
---
 internal/proxy/multi_rate_limiter.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/internal/proxy/multi_rate_limiter.go b/internal/proxy/multi_rate_limiter.go
index e1c8b9ca6..5c88debff 100644
--- a/internal/proxy/multi_rate_limiter.go
+++ b/internal/proxy/multi_rate_limiter.go
@@ -70,6 +70,9 @@ func (m *MultiRateLimiter) Check(collectionID int64, rt internalpb.RateType, n i
 		return commonpb.ErrorCode_Success
 	}
 
+	m.quotaStatesMu.RLock()
+	defer m.quotaStatesMu.RUnlock()
+
 	limiter := m.collectionLimiters[collectionID]
 	if limiter == nil {
 		return commonpb.ErrorCode_Success
-- 
GitLab