From 4e1417e3fb47a0688844ebf33a66d2dca7fb503d Mon Sep 17 00:00:00 2001
From: lysu <sulifx@gmail.com>
Date: Thu, 22 Jul 2021 17:59:20 +0800
Subject: [PATCH] tikv: use resolve lock lite for snapshot.Get() (#247)

Signed-off-by: lysu <sulifx@gmail.com>
---
 tikv/client_helper.go | 3 ++-
 tikv/snapshot.go      | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/tikv/client_helper.go b/tikv/client_helper.go
index a1a5d392..95ea8189 100644
--- a/tikv/client_helper.go
+++ b/tikv/client_helper.go
@@ -58,12 +58,13 @@ type ClientHelper struct {
 }
 
 // NewClientHelper creates a helper instance.
-func NewClientHelper(store *KVStore, resolvedLocks *util.TSSet) *ClientHelper {
+func NewClientHelper(store *KVStore, resolvedLocks *util.TSSet, resolveLite bool) *ClientHelper {
 	return &ClientHelper{
 		lockResolver:  store.GetLockResolver(),
 		regionCache:   store.GetRegionCache(),
 		resolvedLocks: resolvedLocks,
 		client:        store.GetTiKVClient(),
+		resolveLite:   resolveLite,
 	}
 }
 
diff --git a/tikv/snapshot.go b/tikv/snapshot.go
index 866ca195..7ba075f0 100644
--- a/tikv/snapshot.go
+++ b/tikv/snapshot.go
@@ -328,7 +328,7 @@ func (s *KVSnapshot) batchGetKeysByRegions(bo *Backoffer, keys [][]byte, collect
 }
 
 func (s *KVSnapshot) batchGetSingleRegion(bo *Backoffer, batch batchKeys, collectF func(k, v []byte)) error {
-	cli := NewClientHelper(s.store, &s.resolvedLocks)
+	cli := NewClientHelper(s.store, &s.resolvedLocks, false)
 	s.mu.RLock()
 	if s.mu.stats != nil {
 		cli.Stats = make(map[tikvrpc.CmdType]*locate.RPCRuntimeStats)
@@ -497,7 +497,7 @@ func (s *KVSnapshot) get(ctx context.Context, bo *Backoffer, k []byte) ([]byte,
 		}
 	})
 
-	cli := NewClientHelper(s.store, &s.resolvedLocks)
+	cli := NewClientHelper(s.store, &s.resolvedLocks, true)
 
 	s.mu.RLock()
 	if s.mu.stats != nil {