diff --git a/internal/proxy/rate_limit_interceptor.go b/internal/proxy/rate_limit_interceptor.go index 01030fb8f1..0185237ea1 100644 --- a/internal/proxy/rate_limit_interceptor.go +++ b/internal/proxy/rate_limit_interceptor.go @@ -134,7 +134,7 @@ func getRequestInfo(ctx context.Context, req interface{}) (int64, map[int64][]in return dbID, collToPartIDs, internalpb.RateType_DMLInsert, proto.Size(r), err case *milvuspb.UpsertRequest: dbID, collToPartIDs, err := getCollectionAndPartitionID(ctx, req.(reqPartName)) - return dbID, collToPartIDs, internalpb.RateType_DMLUpsert, proto.Size(r), err + return dbID, collToPartIDs, internalpb.RateType_DMLInsert, proto.Size(r), err case *milvuspb.DeleteRequest: dbID, collToPartIDs, err := getCollectionAndPartitionID(ctx, req.(reqPartName)) return dbID, collToPartIDs, internalpb.RateType_DMLDelete, proto.Size(r), err diff --git a/internal/proxy/rate_limit_interceptor_test.go b/internal/proxy/rate_limit_interceptor_test.go index b046a3f8a3..d64b491333 100644 --- a/internal/proxy/rate_limit_interceptor_test.go +++ b/internal/proxy/rate_limit_interceptor_test.go @@ -92,7 +92,7 @@ func TestRateLimitInterceptor(t *testing.T) { PartitionName: "p1", DbName: "db1", }), size) - assert.Equal(t, internalpb.RateType_DMLUpsert, rt) + assert.Equal(t, internalpb.RateType_DMLInsert, rt) assert.Equal(t, database, int64(100)) assert.True(t, len(col2part) == 1) assert.Equal(t, int64(10), col2part[1][0])