fix request source tag unset (#1026)

Signed-off-by: crazycs520 <crazycs520@gmail.com>
tikv · Oct 18, 2023 · 44449c0 · 44449c0
1 parent 916bb20
commit 44449c0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/locate/region_request.go b/internal/locate/region_request.go
@@ -1341,13 +1341,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)