optimize

pingcap · Mar 11, 2023 · be03c3b · be03c3b
1 parent f12d51a
commit be03c3b
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/core/src/main/scala/com/pingcap/tispark/safepoint/ServiceSafePoint.scala b/core/src/main/scala/com/pingcap/tispark/safepoint/ServiceSafePoint.scala
@@ -103,12 +103,17 @@ case class ServiceSafePoint(
   }
 
   def stopRegisterSafePoint(): Unit = {
-    minStartTs = Long.MaxValue
-    clientSession.getTiKVSession.getPDClient.updateServiceGCSafePoint(
-      serviceId,
-      ttl,
-      Long.MaxValue,
-      ConcreteBackOffer.newCustomBackOff(PD_UPDATE_SAFE_POINT_BACKOFF))
-    service.shutdownNow()
+    try {
+      minStartTs = Long.MaxValue
+      clientSession.getTiKVSession.getPDClient.updateServiceGCSafePoint(
+        serviceId,
+        ttl,
+        Long.MaxValue,
+        ConcreteBackOffer.newCustomBackOff(PD_UPDATE_SAFE_POINT_BACKOFF))
+    } catch {
+      case e: Exception => logger.error("Failed to stop register service GC safe point", e)
+    } finally {
+      service.shutdownNow()
+    }
   }
 }