add missing keyToRegionIdCache

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

Author: marsishandsome

src/main/java/org/tikv/common/operation/RegionErrorHandler.java

@@ -180,7 +180,7 @@ public boolean handleRegionError(BackOffer backOffer, Errorpb.Error error) {
   // TiKV's due to slow appling.
   private boolean onRegionEpochNotMatch(BackOffer backOffer, List<Metapb.Region> currentRegions) {
     if (currentRegions.size() == 0) {
-      logger.warn("currentRegions.size() == 0");
+      // logger.warn("currentRegions.size() == 0");
       this.regionManager.onRegionStale(recv.getRegion());
       return false;
     }
@@ -213,12 +213,12 @@ private boolean onRegionEpochNotMatch(BackOffer backOffer, List<Metapb.Region> c
     }
 
     if (needInvalidateOld) {
-      logger.warn("needInvalidateOld, region=" + recv.getRegion());
+      // logger.warn("needInvalidateOld, region=" + recv.getRegion());
       this.regionManager.onRegionStale(recv.getRegion());
     }
 
     for (TiRegion region : newRegions) {
-      logger.warn("insertRegionToCache, region=" + region);
+      // logger.warn("insertRegionToCache, region=" + region);
       regionManager.insertRegionToCache(region);
     }
 

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

@@ -99,6 +99,7 @@ public synchronized void insertRegionToCache(TiRegion region) {
         keyToRegionIdCache.remove(makeRange(oldRegion.getStartKey(), oldRegion.getEndKey()));
       }
       regionCache.put(region.getId(), region);
+      keyToRegionIdCache.put(makeRange(region.getStartKey(), region.getEndKey()), region.getId());
     } catch (Exception ignore) {
     }
   }

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

@@ -100,7 +100,7 @@ public TiRegion getRegionByKey(ByteString key, BackOffer backOffer) {
       if (region == null) {
         logger.debug("Key not found in keyToRegionIdCache:" + formatBytesUTF8(key));
         Pair<Metapb.Region, Metapb.Peer> regionAndLeader = pdClient.getRegionByKey(backOffer, key);
-        logger.warn("getRegionByKey: regionAndLeader=" + regionAndLeader);
+        // logger.warn("getRegionByKey: regionAndLeader=" + regionAndLeader);
         region =
             cache.putRegion(createRegion(regionAndLeader.first, regionAndLeader.second, backOffer));
       }

src/main/java/org/tikv/raw/RawKVClient.java

@@ -133,7 +133,7 @@ private void put(ByteString key, ByteString value, long ttl, boolean atomic) {
       BackOffer backOffer = ConcreteBackOffer.newDeadlineBackOff(conf.getRawKVWriteTimeoutInMS());
       while (true) {
         RegionStoreClient client = clientBuilder.build(key, backOffer);
-        logger.warn("put region: " + client.getRegion().toString());
+        // logger.warn("put region: " + client.getRegion().toString());
         try {
           client.rawPut(backOffer, key, value, ttl, atomic);
           RAW_REQUEST_SUCCESS.labels(label).inc();
@@ -179,7 +179,7 @@ public ByteString putIfAbsent(ByteString key, ByteString value, long ttl) {
       BackOffer backOffer = ConcreteBackOffer.newDeadlineBackOff(conf.getRawKVWriteTimeoutInMS());
       while (true) {
         RegionStoreClient client = clientBuilder.build(key, backOffer);
-        logger.warn("putIfAbsent region: " + client.getRegion().toString());
+        // logger.warn("putIfAbsent region: " + client.getRegion().toString());
         try {
           ByteString result = client.rawPutIfAbsent(backOffer, key, value, ttl);
           RAW_REQUEST_SUCCESS.labels(label).inc();
@@ -265,7 +265,7 @@ public ByteString get(ByteString key) {
       BackOffer backOffer = ConcreteBackOffer.newDeadlineBackOff(conf.getRawKVReadTimeoutInMS());
       while (true) {
         RegionStoreClient client = clientBuilder.build(key, backOffer);
-        logger.warn("get region: " + client.getRegion().toString());
+        // logger.warn("get region: " + client.getRegion().toString());
         try {
           ByteString result = client.rawGet(backOffer, key);
           RAW_REQUEST_SUCCESS.labels(label).inc();