Add comparators for sorting downloads.

[xudocci.git] / src / main / java / net / pterodactylus / xdcc / data / Download.java

diff --git a/src/main/java/net/pterodactylus/xdcc/data/Download.java b/src/main/java/net/pterodactylus/xdcc/data/Download.java
index 926adad..560e7f9 100644 (file)
--- a/src/main/java/net/pterodactylus/xdcc/data/Download.java
+++ b/src/main/java/net/pterodactylus/xdcc/data/Download.java
@@ -21,6 +21,9 @@ import static com.google.common.base.Preconditions.checkNotNull;
 
 import java.io.OutputStream;
 import java.net.InetAddress;
+import java.util.Comparator;
+import java.util.concurrent.atomic.AtomicLong;
+import java.util.concurrent.atomic.AtomicReference;
 
 import net.pterodactylus.irc.DccReceiver;
 
@@ -42,6 +45,24 @@ public class Download {
                }
        };
 
+       /** {@link Comparator} that sorts downloads by their name. */
+       public static final Comparator<Download> BY_NAME = new Comparator<Download>() {
+
+               @Override
+               public int compare(Download leftDownload, Download rightDownload) {
+                       return leftDownload.pack().name().compareToIgnoreCase(rightDownload.pack().name());
+               }
+       };
+
+       /**  {@link Comparator} that sorts running downloads first. */
+       public static final Comparator<Download> BY_RUNNING = new Comparator<Download>() {
+
+               @Override
+               public int compare(Download leftDownload, Download rightDownload) {
+                       return (leftDownload.dccReceiver() != null) ? -1 : (rightDownload.dccReceiver() != null) ? 1 : 0;
+               }
+       };
+
        /** The bot that is being downloaded from. */
        private final Bot bot;
 
@@ -49,19 +70,19 @@ public class Download {
        private final Pack pack;
 
        /** The name of the file being downloaded. */
-       private String filename;
+       private final AtomicReference<String> filename = new AtomicReference<String>();
 
        /** The size of the file being downloaded. */
-       private long filesize;
+       private final AtomicLong filesize = new AtomicLong();
 
        /** The remote address. */
-       private InetAddress remoteAddress;
+       private final AtomicReference<InetAddress> remoteAddress = new AtomicReference<InetAddress>();
 
        /** The output stream. */
-       private OutputStream outputStream;
+       private final AtomicReference<OutputStream> outputStream = new AtomicReference<OutputStream>();
 
        /** The DCC receiver. */
-       private DccReceiver dccReceiver;
+       private final AtomicReference<DccReceiver> dccReceiver = new AtomicReference<DccReceiver>();
 
        /**
         * Creates a new download.
@@ -104,7 +125,7 @@ public class Download {
         * @return The full name of the file
         */
        public String filename() {
-               return filename;
+               return filename.get();
        }
 
        /**
@@ -113,7 +134,7 @@ public class Download {
         * @return The size of the file
         */
        public long filesize() {
-               return filesize;
+               return filesize.get();
        }
 
        /**
@@ -122,7 +143,7 @@ public class Download {
         * @return The remote address to download from
         */
        public InetAddress remoteAddress() {
-               return remoteAddress;
+               return remoteAddress.get();
        }
 
        /**
@@ -131,7 +152,7 @@ public class Download {
         * @return The output stream
         */
        public OutputStream outputStream() {
-               return outputStream;
+               return outputStream.get();
        }
 
        /**
@@ -140,7 +161,7 @@ public class Download {
         * @return The DCC receiver
         */
        public DccReceiver dccReceiver() {
-               return dccReceiver;
+               return dccReceiver.get();
        }
 
        //
@@ -155,7 +176,7 @@ public class Download {
         * @return This download
         */
        public Download filename(String filename) {
-               this.filename = filename;
+               this.filename.set(filename);
                return this;
        }
 
@@ -167,7 +188,7 @@ public class Download {
         * @return This download
         */
        public Download filesize(long filesize) {
-               this.filesize = filesize;
+               this.filesize.set(filesize);
                return this;
        }
 
@@ -179,7 +200,7 @@ public class Download {
         * @return This download
         */
        public Download remoteAddress(InetAddress remoteAddress) {
-               this.remoteAddress = remoteAddress;
+               this.remoteAddress.set(remoteAddress);
                return this;
        }
 
@@ -191,7 +212,7 @@ public class Download {
         * @return This download
         */
        public Download outputStream(OutputStream outputStream) {
-               this.outputStream = outputStream;
+               this.outputStream.set(outputStream);
                return this;
        }
 
@@ -203,7 +224,7 @@ public class Download {
         * @return This download
         */
        public Download dccReceiver(DccReceiver dccReceiver) {
-               this.dccReceiver = dccReceiver;
+               this.dccReceiver.set(dccReceiver);
                return this;
        }