From fdf054f23f8e62bac62649391c29fd4a31868696 Mon Sep 17 00:00:00 2001
From: marsishandsome <marsishandsome@gmail.com>
Date: Thu, 30 Dec 2021 13:12:07 +0800
Subject: [PATCH] use target store context instead of proxy store context when
 calling grpc forward

Signed-off-by: marsishandsome <marsishandsome@gmail.com>
---
 .../java/org/tikv/common/region/AbstractRegionStoreClient.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java b/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
index cd0673dee0e..3467c974699 100644
--- a/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
+++ b/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
@@ -315,7 +315,7 @@ private TiStore switchProxyStore(BackOffer backOffer) {
       header.put(TiConfiguration.FORWARD_META_DATA_KEY, store.getStore().getAddress());
       Kvrpcpb.RawGetRequest rawGetRequest =
           Kvrpcpb.RawGetRequest.newBuilder()
-              .setContext(region.getReplicaContext(peer))
+              .setContext(region.getReplicaContext(region.getLeader()))
               .setKey(key)
               .build();
       ListenableFuture<Kvrpcpb.RawGetResponse> task =