Avoid the NPE in Heart.beat when shutting down

thelastpickle · May 31, 2018 · c5d3b1b · c5d3b1b
1 parent 5360688
commit c5d3b1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/server/src/main/java/io/cassandrareaper/service/RepairManager.java b/src/server/src/main/java/io/cassandrareaper/service/RepairManager.java
@@ -46,7 +46,7 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public final class RepairManager {
+public final class RepairManager implements AutoCloseable {
 
   private static final Logger LOG = LoggerFactory.getLogger(RepairManager.class);
 
@@ -372,4 +372,10 @@ private static void releaseLead(AppContext context, UUID leaderElectionId) {
       }
     }
   }
+
+  @Override
+  public void close() {
+    heart.close();
+    executor.shutdownNow();
+  }
 }
diff --git a/src/server/src/test/java/io/cassandrareaper/acceptance/ReaperTestJettyRunner.java b/src/server/src/test/java/io/cassandrareaper/acceptance/ReaperTestJettyRunner.java
@@ -104,6 +104,7 @@ public Application<ReaperApplicationConfiguration> newApplication() {
 
     @Override
     public void after() {
+      context.repairManager.close();
       context.isRunning.set(false);
       try {
         Thread.sleep(100);