From 74776b122837ce152c4d33074377c3cb3004f3b8 Mon Sep 17 00:00:00 2001
From: Alexander Dejanovski <alex@thelastpickle.com>
Date: Fri, 28 Jul 2017 18:51:21 +0200
Subject: [PATCH] Fix coordinator node being nulled out by abort() for
 incremental repairs.

---
 src/main/java/com/spotify/reaper/service/SegmentRunner.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/spotify/reaper/service/SegmentRunner.java b/src/main/java/com/spotify/reaper/service/SegmentRunner.java
index 6a63cbace..e6fe5ec24 100644
--- a/src/main/java/com/spotify/reaper/service/SegmentRunner.java
+++ b/src/main/java/com/spotify/reaper/service/SegmentRunner.java
@@ -120,7 +120,7 @@ public static void postpone(AppContext context, RepairSegment segment, Optional<
   }
 
   public static void abort(AppContext context, RepairSegment segment, JmxProxy jmxConnection) {
-    postpone(context, segment, Optional.fromNullable((RepairUnit) null));
+    postpone(context, segment, context.storage.getRepairUnit(segment.getRepairUnitId()));
     LOG.info("Aborting repair on segment with id {} on coordinator {}",
         segment.getId(), segment.getCoordinatorHost());
     jmxConnection.cancelAllRepairs();