Merge pull request kubernetes-client#1115 from nurkiewicz/patch-1

k8s-ci-robot · web-flow · commit 351f1803e767 · 2020-07-28T07:57:48.000-07:00
Provide standard method to cleanup schedulers
diff --git a/extended/src/main/java/io/kubernetes/client/extended/leaderelection/LeaderElector.java b/extended/src/main/java/io/kubernetes/client/extended/leaderelection/LeaderElector.java
@@ -25,7 +25,7 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class LeaderElector {
+public class LeaderElector implements AutoCloseable {
 
   private static final Logger log = LoggerFactory.getLogger(LeaderElector.class);
 
@@ -43,12 +43,12 @@ public class LeaderElector {
   private String reportedLeader;
   private Consumer<String> onNewLeaderHook;
 
-  private ScheduledExecutorService scheduledWorkers =
+  private final ScheduledExecutorService scheduledWorkers =
       Executors.newSingleThreadScheduledExecutor(
           makeThreadFactory("leader-elector-scheduled-worker-%d"));
-  private ExecutorService leaseWorkers =
+  private final ExecutorService leaseWorkers =
       Executors.newSingleThreadExecutor(makeThreadFactory("leader-elector-lease-worker-%d"));
-  private ExecutorService hookWorkers =
+  private final ExecutorService hookWorkers =
       Executors.newSingleThreadExecutor(makeThreadFactory("leader-elector-hook-worker-%d"));
 
   public LeaderElector(LeaderElectionConfig config) {
@@ -337,4 +337,11 @@ private void maybeReportTransition() {
   private ThreadFactory makeThreadFactory(String nameFormat) {
     return new ThreadFactoryBuilder().setDaemon(true).setNameFormat(nameFormat).build();
   }
+
+  @Override
+  public void close() {
+    scheduledWorkers.shutdownNow();
+    leaseWorkers.shutdownNow();
+    hookWorkers.shutdownNow();
+  }
 }
