store/tikv: set execution timeout for general TiKV requests (#24936)

pingcap · May 28, 2021 · 138692f · 138692f
1 parent ec12617
commit 138692f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/store/tikv/region_request.go b/store/tikv/region_request.go
@@ -259,6 +259,12 @@ func (s *RegionRequestSender) SendReqCtx(
 		}
 	})
 
+	// If the MaxExecutionDurationMs is not set yet, we set it to be the RPC timeout duration
+	// so TiKV can give up the requests whose response TiDB cannot receive due to timeout.
+	if req.Context.MaxExecutionDurationMs == 0 {
+		req.Context.MaxExecutionDurationMs = uint64(timeout.Milliseconds())
+	}
+
 	tryTimes := 0
 	for {
 		if (tryTimes > 0) && (tryTimes%1000 == 0) {