From 90e874a3e38cdfdac9067d55e6829c048bcb459b Mon Sep 17 00:00:00 2001
From: chunhui-shi <cshi@maprtech.com>
Date: Thu, 1 Dec 2016 17:55:12 -0800
Subject: [PATCH] DRILL-5094: Comparator should guarantee transitive attribute.

---
 .../apache/drill/exec/store/schedule/AssignmentCreator.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/store/schedule/AssignmentCreator.java b/exec/java-exec/src/main/java/org/apache/drill/exec/store/schedule/AssignmentCreator.java
index eed200eb4b7..127264ab6ee 100644
--- a/exec/java-exec/src/main/java/org/apache/drill/exec/store/schedule/AssignmentCreator.java
+++ b/exec/java-exec/src/main/java/org/apache/drill/exec/store/schedule/AssignmentCreator.java
@@ -50,7 +50,8 @@ public class AssignmentCreator<T extends CompleteWork> {
   private static Comparator<Entry<DrillbitEndpoint,Long>> comparator = new Comparator<Entry<DrillbitEndpoint,Long>>() {
     @Override
     public int compare(Entry<DrillbitEndpoint, Long> o1, Entry<DrillbitEndpoint,Long> o2) {
-      return (int) (o1.getValue() - o2.getValue());
+      long ret = o1.getValue() - o2.getValue();
+      return ret > 0? 1 : ((ret < 0)? -1: 0);
     }
   };