From f1ae4b4d847100250f53bdf5f7178aa9bf1818a7 Mon Sep 17 00:00:00 2001
From: Wallace <bupt2013211450@gmail.com>
Date: Wed, 23 Jun 2021 15:00:47 +0800
Subject: [PATCH] cherry pick #207 to release-3.1

Signed-off-by: ti-srebot <ti-srebot@pingcap.com>
---
 src/main/java/org/tikv/common/TiSession.java            | 6 ++++--
 src/main/java/org/tikv/common/region/RegionManager.java | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/tikv/common/TiSession.java b/src/main/java/org/tikv/common/TiSession.java
index 41410d64d3b..9892b68d44e 100644
--- a/src/main/java/org/tikv/common/TiSession.java
+++ b/src/main/java/org/tikv/common/TiSession.java
@@ -194,7 +194,7 @@ public Catalog getCatalog() {
     return res;
   }
 
-  public synchronized RegionManager getRegionManager() {
+  public RegionManager getRegionManager() {
     RegionManager res = regionManager;
     if (res == null) {
       synchronized (this) {
@@ -466,7 +466,9 @@ public synchronized void close() throws Exception {
     synchronized (sessionCachedMap) {
       sessionCachedMap.remove(conf.getPdAddrsString());
     }
-
+    if (regionManager != null) {
+      regionManager.close();
+    }
     if (tableScanThreadPool != null) {
       tableScanThreadPool.shutdownNow();
     }
diff --git a/src/main/java/org/tikv/common/region/RegionManager.java b/src/main/java/org/tikv/common/region/RegionManager.java
index 63a91c2ccd7..c3c1fd560d6 100644
--- a/src/main/java/org/tikv/common/region/RegionManager.java
+++ b/src/main/java/org/tikv/common/region/RegionManager.java
@@ -75,6 +75,12 @@ public RegionManager(ReadOnlyPDClient pdClient) {
     this.cacheInvalidateCallback = null;
   }
 
+  public synchronized void close() {
+    if (this.executor != null) {
+      this.executor.shutdownNow();
+    }
+  }
+
   public Function<CacheInvalidateEvent, Void> getCacheInvalidateCallback() {
     return cacheInvalidateCallback;
   }