store/tikv: remove use of IsStatenessReadOnly option in store/tikv

store/driver/txn/snapshot.go

@@ -78,6 +78,8 @@ func (s *tikvSnapshot) SetOption(opt int, val interface{}) {
  s.KVSnapshot.SetReplicaRead(val.(tikvstore.ReplicaReadType))
  case tikvstore.TaskID:
  s.KVSnapshot.SetTaskID(val.(uint64))
+ case tikvstore.IsStalenessReadOnly:
+ s.KVSnapshot.SetIsStatenessReadOnly(val.(bool))
  default:
  s.KVSnapshot.SetOption(opt, val)
  }

store/driver/txn/txn_driver.go

@@ -161,6 +161,8 @@ func (txn *tikvTxn) SetOption(opt int, val interface{}) {
  txn.SetEnable1PC(val.(bool))
  case tikvstore.TxnScope:
  txn.SetScope(val.(string))
+ case tikvstore.IsStalenessReadOnly:
+ txn.KVTxn.GetSnapshot().SetIsStatenessReadOnly(val.(bool))
  default:
  txn.KVTxn.SetOption(opt, val)
  }

store/tikv/snapshot.go

@@ -571,10 +571,6 @@ func (s *KVSnapshot) SetOption(opt int, val interface{}) {
  s.mu.Unlock()
  case kv.SampleStep:
  s.sampleStep = val.(uint32)
- case kv.IsStalenessReadOnly:
- s.mu.Lock()
- s.mu.isStaleness = val.(bool)
- s.mu.Unlock()
  case kv.MatchStoreLabels:
  s.mu.Lock()
  s.mu.matchStoreLabels = val.([]*metapb.StoreLabel)
@@ -628,6 +624,13 @@ func (s *KVSnapshot) SetTaskID(id uint64) {
  s.mu.taskID = id
 }
 
+// SetIsStatenessReadOnly indicates whether the transaction is staleness read only transaction
+func (s *KVSnapshot) SetIsStatenessReadOnly(b bool) {
+ s.mu.Lock()
+ defer s.mu.Unlock()
+ s.mu.isStaleness = b
+}
+
 // SnapCacheHitCount gets the snapshot cache hit count. Only for test.
 func (s *KVSnapshot) SnapCacheHitCount() int {
  return int(atomic.LoadInt64(&s.mu.hitCnt))