milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix rate limit ut (#24096) 

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
pull/24063/head
wei liu 2023-05-15 11:13:21 +08:00 committed by GitHub
parent 41cb09473a
commit 4369915402
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/proxy/multi_rate_limiter_test.go
View File

@ -73,9 +73,9 @@ func TestMultiRateLimiter(t *testing.T) {
bak := Params.QuotaConfig.QuotaAndLimitsEnabled
paramtable.Get().Save(Params.QuotaConfig.QuotaAndLimitsEnabled.Key, "true")
multiLimiter := NewMultiRateLimiter()
multiLimiter.collectionLimiters[1] = newRateLimiter()
multiLimiter.collectionLimiters[2] = newRateLimiter()
multiLimiter.collectionLimiters[3] = newRateLimiter()
multiLimiter.collectionLimiters[1] = newRateLimiter(false)
multiLimiter.collectionLimiters[2] = newRateLimiter(false)
multiLimiter.collectionLimiters[3] = newRateLimiter(false)
for _, rt := range internalpb.RateType_value {
if IsDDLRequest(internalpb.RateType(rt)) {
multiLimiter.globalDDLLimiter.limiters.Insert(internalpb.RateType(rt), ratelimitutil.NewLimiter(ratelimitutil.Limit(5), 1))