From 71f2474db4f0b21154ceb627df2abfeb51ac43b3 Mon Sep 17 00:00:00 2001 From: crazycs520 Date: Wed, 18 Oct 2023 17:22:31 +0800 Subject: [PATCH] fix request source tag unset Signed-off-by: crazycs520 --- internal/locate/region_request.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/internal/locate/region_request.go b/internal/locate/region_request.go index 9c9e613e0..a83b93dcc 100644 --- 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)