From 23bbb923acbabc0b77fb207c66551ba2648b2b07 Mon Sep 17 00:00:00 2001
From: Liangliang Gu <marsishandsome@gmail.com>
Date: Mon, 14 Feb 2022 17:43:10 +0800
Subject: [PATCH] [close #459] use target store context instead of proxy store
 context when calling grpc forward (#458)

Co-authored-by: Jian Zhang <zjsariel@gmail.com>
Co-authored-by: ti-srebot <66930949+ti-srebot@users.noreply.github.com>
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 7bccbb77d78..6b33a9a97c4 100644
--- a/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
+++ b/src/main/java/org/tikv/common/region/AbstractRegionStoreClient.java
@@ -316,7 +316,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 =