allow higher concurrency in keep-order mode

Signed-off-by: you06 <you1474600@gmail.com>
pingcap · Feb 16, 2023 · 1ab6790 · 1ab6790
1 parent 7494a8b
commit 1ab6790
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/store/copr/coprocessor.go b/store/copr/coprocessor.go
@@ -217,7 +217,14 @@ func (c *CopClient) BuildCopIterator(ctx context.Context, req *kv.Request, vars
 		// For ReqTypeAnalyze, we keep its concurrency to avoid slow analyze(see https://github.com/pingcap/tidb/issues/40162 for details).
 		if it.concurrency > 2 && it.req.Tp != kv.ReqTypeAnalyze {
 			oldConcurrency := it.concurrency
+			partitionNum := req.KeyRanges.PartitionNum()
+			if partitionNum > it.concurrency {
+				partitionNum = it.concurrency
+			}
 			it.concurrency = 2
+			if it.concurrency < partitionNum {
+				it.concurrency = partitionNum
+			}
 
 			failpoint.Inject("testRateLimitActionMockConsumeAndAssert", func(val failpoint.Value) {
 				if val.(bool) {