mirror of
https://github.com/GravitLauncher/Launcher
synced 2024-11-15 11:39:11 +03:00
Filedownloader (#22)
* Added downloader runnable. * Added thread downloader. * Small fix. * Optimize downloading. * Remove program stop.
This commit is contained in:
parent
e4b8cf59bf
commit
3f2174689a
2 changed files with 118 additions and 0 deletions
|
@ -0,0 +1,83 @@
|
||||||
|
package ru.gravit.utils.downloader;
|
||||||
|
|
||||||
|
import java.io.BufferedInputStream;
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.net.HttpURLConnection;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.util.concurrent.atomic.AtomicBoolean;
|
||||||
|
import java.util.concurrent.atomic.AtomicInteger;
|
||||||
|
import java.util.concurrent.atomic.AtomicReference;
|
||||||
|
|
||||||
|
import ru.gravit.utils.helper.IOHelper;
|
||||||
|
import ru.gravit.utils.helper.LogHelper;
|
||||||
|
|
||||||
|
public class Downloader implements Runnable {
|
||||||
|
public static final int INTERVAL = 300;
|
||||||
|
|
||||||
|
public AtomicInteger writed = new AtomicInteger(0);
|
||||||
|
private final File file;
|
||||||
|
private final URL url;
|
||||||
|
public final AtomicBoolean interrupt = new AtomicBoolean(false);
|
||||||
|
public final AtomicBoolean interrupted = new AtomicBoolean(false);
|
||||||
|
private final int skip;
|
||||||
|
|
||||||
|
public AtomicReference<Throwable> ex = new AtomicReference<>(null);
|
||||||
|
|
||||||
|
public Downloader(URL url, File file) {
|
||||||
|
this.file = file;
|
||||||
|
this.url = url;
|
||||||
|
this.skip = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Downloader(URL url, File file, int skip) {
|
||||||
|
this.file = file;
|
||||||
|
this.url = url;
|
||||||
|
this.skip = skip;
|
||||||
|
}
|
||||||
|
|
||||||
|
public File getFile() {
|
||||||
|
return file;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void downloadFile() throws IOException {
|
||||||
|
if (!(url.getProtocol().equalsIgnoreCase("http") || url.getProtocol().equalsIgnoreCase("https"))) throw new IOException("Invalid protocol.");
|
||||||
|
HttpURLConnection connect = (HttpURLConnection) (url).openConnection();
|
||||||
|
connect.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.95 Safari/537.11"); // for stupid servers
|
||||||
|
connect.setInstanceFollowRedirects(true);
|
||||||
|
if (!(connect.getResponseCode() >= 200 && connect.getResponseCode() < 300)) throw new IOException(String.format("Invalid response of http server %d.", connect.getResponseCode()));
|
||||||
|
try (BufferedInputStream in = new BufferedInputStream(connect.getInputStream(), IOHelper.BUFFER_SIZE); FileOutputStream fout = new FileOutputStream(file, skip != 0)) {
|
||||||
|
final byte data[] = new byte[IOHelper.BUFFER_SIZE];
|
||||||
|
int count = -1;
|
||||||
|
long timestamp = System.currentTimeMillis();
|
||||||
|
int writed_local = 0;
|
||||||
|
in.skip(skip);
|
||||||
|
while ((count = in.read(data)) != -1) {
|
||||||
|
fout.write(data, 0, count);
|
||||||
|
writed_local += count;
|
||||||
|
if (System.currentTimeMillis() - timestamp > INTERVAL) {
|
||||||
|
writed.set(writed_local);
|
||||||
|
LogHelper.debug("Downloaded %d", writed_local);
|
||||||
|
if (interrupt.get()) {
|
||||||
|
interrupted.set(true);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
LogHelper.debug("Downloaded %d", writed_local);
|
||||||
|
writed.set(writed_local);
|
||||||
|
interrupted.set(true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
try {
|
||||||
|
downloadFile();
|
||||||
|
} catch (Throwable ex) {
|
||||||
|
this.ex.set(ex);
|
||||||
|
LogHelper.error(ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,35 @@
|
||||||
|
package ru.gravit.utils.downloader;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.net.URL;
|
||||||
|
|
||||||
|
public class DownloadingThread extends Thread {
|
||||||
|
private final Downloader runnable;
|
||||||
|
|
||||||
|
public DownloadingThread(File file, URL url, String name) {
|
||||||
|
super(name);
|
||||||
|
runnable = new Downloader(url, file);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Downloader getDownloader() {
|
||||||
|
return runnable;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void interrupt() {
|
||||||
|
runnable.interrupt.set(true);
|
||||||
|
while (!runnable.interrupted.get()) {
|
||||||
|
;
|
||||||
|
}
|
||||||
|
super.interrupt();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void hardInterrupt() {
|
||||||
|
super.interrupt();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void run() {
|
||||||
|
runnable.run();
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue