test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-02 11:59:00 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix concurrent map read/write in rate limiter (#23958)

Browse Source 
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
This commit is contained in:
congqixia 2023-05-09 11:00:40 +08:00 committed by GitHub
parent 3827ac30bc
commit f6d3b4f7ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
internal/proxy/multi_rate_limiter.go
View File

@ -70,6 +70,9 @@ func (m *MultiRateLimiter) Check(collectionID int64, rt internalpb.RateType, n i
return commonpb.ErrorCode_Success return commonpb.ErrorCode_Success
} }
m.quotaStatesMu.RLock()
defer m.quotaStatesMu.RUnlock()
limiter := m.collectionLimiters[collectionID] limiter := m.collectionLimiters[collectionID]
if limiter == nil { if limiter == nil {
return commonpb.ErrorCode_Success return commonpb.ErrorCode_Success