From 91fbac9b6a1cfaff8aa5837b0b6823b237e15780 Mon Sep 17 00:00:00 2001
From: DerekYRC <15521077528@163.com>
Date: Wed, 3 Aug 2022 20:15:27 +0800
Subject: [PATCH] shutdown thread pool

---
 .../cloud/polaris/registry/PolarisServiceRegistry.java    | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
index e92d81f1c..284c8113e 100644
--- a/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
+++ b/spring-cloud-starter-tencent-polaris-discovery/src/main/java/com/tencent/cloud/polaris/registry/PolarisServiceRegistry.java
@@ -38,6 +38,7 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import org.springframework.beans.BeanUtils;
+import org.springframework.beans.factory.DisposableBean;
 import org.springframework.cloud.client.serviceregistry.Registration;
 import org.springframework.cloud.client.serviceregistry.ServiceRegistry;
 
@@ -49,7 +50,7 @@ import static org.springframework.util.ReflectionUtils.rethrowRuntimeException;
  *
  * @author Haotian Zhang, Andrew Shan, Jie Cheng
  */
-public class PolarisServiceRegistry implements ServiceRegistry<Registration> {
+public class PolarisServiceRegistry implements ServiceRegistry<Registration>, DisposableBean {
 
 	private static final Logger LOGGER = LoggerFactory.getLogger(PolarisServiceRegistry.class);
 
@@ -216,4 +217,9 @@ public class PolarisServiceRegistry implements ServiceRegistry<Registration> {
 			}
 		}, polarisDiscoveryProperties.getHeartbeatInterval(), polarisDiscoveryProperties.getHeartbeatInterval(), MILLISECONDS);
 	}
+
+	@Override
+	public void destroy() throws Exception {
+		heartbeatExecutor.shutdown();
+	}
 }