Make DiskSize Comparable

1 files changed, 6 insertions, 1 deletions

diff --git a/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/file/DiskSize.java b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/file/DiskSize.java
index c76c872902d..2a41423f1e7 100644
--- a/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/file/DiskSize.java
+++ b/node-admin/src/main/java/com/yahoo/vespa/hosted/node/admin/task/util/file/DiskSize.java
@@ -7,7 +7,7 @@ import java.util.Objects;
 /**
  * @author freva
  */
-public class DiskSize {
+public class DiskSize implements Comparable<DiskSize> {
 
     public static final DiskSize ZERO = DiskSize.of(0);
     private static final char[] UNITS = "kMGTPE".toCharArray();
@@ -47,6 +47,11 @@ public class DiskSize {
     }
 
     @Override
+    public int compareTo(DiskSize rhs) {
+        return Long.compare(this.bytes, rhs.bytes);
+    }
+
+    @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;