fix request source tag unset (#1025)

Signed-off-by: crazycs520 <[email protected]> Co-authored-by: disksing <[email protected]>
tikv · Oct 20, 2023 · 476ea22 · 476ea22
1 parent ec40555
commit 476ea22
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/locate/region_request.go b/internal/locate/region_request.go
@@ -1449,13 +1449,13 @@ func (s *RegionRequestSender) SendReqCtx(
 			}
 		}
 
-		if e := tikvrpc.SetContext(req, rpcCtx.Meta, rpcCtx.Peer); e != nil {
-			return nil, nil, retryTimes, err
-		}
 		rpcCtx.contextPatcher.applyTo(&req.Context)
 		if req.InputRequestSource != "" && s.replicaSelector != nil {
 			s.replicaSelector.patchRequestSource(req, rpcCtx)
 		}
+		if e := tikvrpc.SetContext(req, rpcCtx.Meta, rpcCtx.Peer); e != nil {
+			return nil, nil, retryTimes, err
+		}
 
 		var retry bool
 		resp, retry, err = s.sendReqToRegion(bo, rpcCtx, req, timeout)