diff --git a/internal/locate/region_request.go b/internal/locate/region_request.go index ea4a62af13..461810941d 100644 --- a/internal/locate/region_request.go +++ b/internal/locate/region_request.go @@ -1197,13 +1197,13 @@ func (s *RegionRequestSender) SendReqCtx( } } - if e := tikvrpc.SetContext(req, rpcCtx.Meta, rpcCtx.Peer); e != nil { - return nil, nil, 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, err + } var retry bool resp, retry, err = s.sendReqToRegion(bo, rpcCtx, req, timeout)