From 32b07a05be261ede027e2070fbe1355e25748c9f Mon Sep 17 00:00:00 2001 From: marsishandsome Date: Sat, 11 Dec 2021 20:23:28 +0800 Subject: [PATCH] fix SmartRawKVClient close Signed-off-by: marsishandsome --- src/main/java/org/tikv/common/TiSession.java | 4 ++++ src/main/java/org/tikv/raw/SmartRawKVClient.java | 1 - 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/tikv/common/TiSession.java b/src/main/java/org/tikv/common/TiSession.java index c9d1aec0d1c..7e3adbf7a93 100644 --- a/src/main/java/org/tikv/common/TiSession.java +++ b/src/main/java/org/tikv/common/TiSession.java @@ -637,6 +637,10 @@ private synchronized void shutdown(boolean now) throws Exception { if (metricsServer != null) { metricsServer.close(); } + + if (circuitBreaker != null) { + circuitBreaker.close(); + } } if (now) { diff --git a/src/main/java/org/tikv/raw/SmartRawKVClient.java b/src/main/java/org/tikv/raw/SmartRawKVClient.java index 2c00b388e0d..fcca8945137 100644 --- a/src/main/java/org/tikv/raw/SmartRawKVClient.java +++ b/src/main/java/org/tikv/raw/SmartRawKVClient.java @@ -258,7 +258,6 @@ private void callWithCircuitBreaker(String funcName, Function0 func) { @Override public void close() throws Exception { - circuitBreaker.close(); client.close(); }