fix request source tag unset (#1025)

Signed-off-by: crazycs520 <crazycs520@gmail.com>
Co-authored-by: disksing <i@disksing.com>
This commit is contained in:
crazycs 2023-10-20 14:26:25 +08:00 committed by GitHub
parent ec40555fac
commit 476ea22ffb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions

View File

@ -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)