From 9c908399da6c4c24cbacf18bfe4f90a0e987d830 Mon Sep 17 00:00:00 2001 From: crazycs520 Date: Wed, 18 Oct 2023 18:14:31 +0800 Subject: [PATCH] fix request source tag unset (#1026) 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 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)