address code review

Signed-off-by: marsishandsome <marsishandsome@gmail.com>

Author: marsishandsome

src/main/java/org/tikv/common/TiSession.java

@@ -161,34 +161,31 @@ private static VersionInfo getVersionInfo() {
   }
 
   private synchronized void warmUp() {
-    long warmUpStartTime = System.currentTimeMillis();
+    long warmUpStartTime = System.nanoTime();
+    BackOffer backOffer = ConcreteBackOffer.newRawKVBackOff();
     try {
       this.client = getPDClient();
       this.regionManager = getRegionManager();
-      List<Metapb.Store> stores = this.client.getAllStores(ConcreteBackOffer.newGetBackOff());
+      List<Metapb.Store> stores = this.client.getAllStores(backOffer);
       // warm up store cache
       for (Metapb.Store store : stores) {
         this.regionManager.updateStore(
-            null,
-            new TiStore(this.client.getStore(ConcreteBackOffer.newGetBackOff(), store.getId())));
+            null, new TiStore(this.client.getStore(backOffer, store.getId())));
       }
 
       // use scan region to load region cache with limit
       ByteString startKey = ByteString.EMPTY;
       do {
         List<Pdpb.Region> regions =
             regionManager.scanRegions(
-                ConcreteBackOffer.newGetBackOff(),
-                startKey,
-                ByteString.EMPTY,
-                conf.getScanRegionsLimit());
+                backOffer, startKey, ByteString.EMPTY, conf.getScanRegionsLimit());
         if (regions == null || regions.isEmpty()) {
           // something went wrong, but the warm-up process could continue
           break;
         }
         for (Pdpb.Region region : regions) {
           regionManager.insertRegionToCache(
-              regionManager.createRegion(region.getRegion(), ConcreteBackOffer.newGetBackOff()));
+              regionManager.createRegion(region.getRegion(), backOffer));
         }
         startKey = regions.get(regions.size() - 1).getRegion().getEndKey();
       } while (!startKey.isEmpty());
@@ -211,7 +208,8 @@ private synchronized void warmUp() {
       logger.info("warm up fails, ignored ", e);
     } finally {
       logger.info(
-          String.format("warm up duration %d ms", System.currentTimeMillis() - warmUpStartTime));
+          String.format(
+              "warm up duration %d ms", (System.nanoTime() - warmUpStartTime) / 1000_000));
     }
   }
 

src/main/java/org/tikv/common/region/RegionStoreClient.java

@@ -274,7 +274,7 @@ public List<KvPair> scan(
     boolean forWrite = false;
     while (true) {
       // we should refresh region
-      region = regionManager.getRegionByKey(startKey);
+      region = regionManager.getRegionByKey(startKey, backOffer);
 
       Supplier<ScanRequest> request =
           () ->

src/main/java/org/tikv/common/util/ConcreteBackOffer.java

@@ -168,8 +168,6 @@ public boolean canRetryAfterSleep(BackOffFunction.BackOffFuncType funcType) {
   }
 
   public boolean canRetryAfterSleep(BackOffFunction.BackOffFuncType funcType, long maxSleepMs) {
-    SlowLogSpan slowLogSpan = getSlowLog().start("backoff " + funcType.name());
-    Histogram.Timer backOffTimer = BACKOFF_DURATION.labels(funcType.name()).startTimer();
     BackOffFunction backOffFunction =
         backOffFunctionMap.computeIfAbsent(funcType, this::createBackOffFunc);
 
@@ -185,6 +183,8 @@ public boolean canRetryAfterSleep(BackOffFunction.BackOffFuncType funcType, long
       }
     }
 
+    Histogram.Timer backOffTimer = BACKOFF_DURATION.labels(funcType.name()).startTimer();
+    SlowLogSpan slowLogSpan = getSlowLog().start("backoff " + funcType.name());
     try {
       Thread.sleep(sleep);
     } catch (InterruptedException e) {

src/test/java/org/tikv/BaseRawKVTest.java

@@ -14,6 +14,7 @@ protected TiConfiguration createTiConfiguration() {
     conf.setTest(true);
     conf.setEnableAtomicForCAS(true);
     conf.setEnableGrpcForward(false);
+    conf.setEnableAtomicForCAS(true);
     return conf;
   }
 }