continue fix

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

Author: marsishandsome

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

@@ -159,23 +159,46 @@ private static VersionInfo getVersionInfo() {
 
   private synchronized void warmUp() {
     long warmUpStartTime = System.currentTimeMillis();
+    warmUpStore();
+    warmUpRegion();
+    warmUpGrpc();
+    logger.info(
+        String.format("warm up duration %d ms", System.currentTimeMillis() - warmUpStartTime));
+  }
+
+  private void warmUpStore() {
     try {
+      BackOffer backOffer = ConcreteBackOffer.newRawKVBackOff();
       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())));
       }
-      ByteString startKey = ByteString.EMPTY;
+    } catch (Exception e) {
+      // ignore error
+      logger.info("warm up store fails, ignored ", e);
+    }
+  }
 
+  private void warmUpRegion() {
+    try {
+      BackOffer backOffer = ConcreteBackOffer.newRawKVBackOff();
+      ByteString startKey = ByteString.EMPTY;
       do {
-        TiRegion region = regionManager.getRegionByKey(startKey);
+        TiRegion region = regionManager.getRegionByKey(startKey, backOffer);
         startKey = region.getEndKey();
       } while (!startKey.isEmpty());
+    } catch (Exception e) {
+      // ignore error
+      logger.info("warm up region fails, ignored ", e);
+    }
+  }
 
+  private void warmUpGrpc() {
+    try {
       RawKVClient rawKVClient = createRawClient();
       ByteString exampleKey = ByteString.EMPTY;
       Optional<ByteString> prev = rawKVClient.get(exampleKey);
@@ -190,10 +213,7 @@ private synchronized void warmUp() {
       }
     } catch (Exception e) {
       // ignore error
-      logger.info("warm up fails, ignored ", e);
-    } finally {
-      logger.info(
-          String.format("warm up duration %d ms", System.currentTimeMillis() - warmUpStartTime));
+      logger.info("warm up grpc fails, ignored ", e);
     }
   }
 

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,7 +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 +184,7 @@ public boolean canRetryAfterSleep(BackOffFunction.BackOffFuncType funcType, long
       }
     }
 
+    SlowLogSpan slowLogSpan = getSlowLog().start("backoff " + funcType.name());
     try {
       Thread.sleep(sleep);
     } catch (InterruptedException e) {

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

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