X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fdata%2FDownload.java;h=560e7f9ae572f9255eafaa198073001fb5a047c8;hb=1ca482cdf4956eb7dcdcfd75c4d0412cf0261fdf;hp=926adad3e598488af164b331517963005df80fd0;hpb=9d685ddca572f2300b8e463c420de6c4d5fa6eae;p=xudocci.git 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 --- 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 BY_NAME = new Comparator() { + + @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 BY_RUNNING = new Comparator() { + + @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 filename = new AtomicReference(); /** 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 remoteAddress = new AtomicReference(); /** The output stream. */ - private OutputStream outputStream; + private final AtomicReference outputStream = new AtomicReference(); /** The DCC receiver. */ - private DccReceiver dccReceiver; + private final AtomicReference dccReceiver = new AtomicReference(); /** * 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; }