9 Commits

Author SHA1 Message Date
marc barbier 379cfed69a bandwidth precision improvement 2021-07-10 17:49:12 +02:00
marc barbier f100e2d41b add version option 2021-07-10 17:11:33 +02:00
marc barbier a2a136a9fc refacto main 2021-07-10 16:59:44 +02:00
marc barbier 416f0b892e thread warning + better logging 2021-07-10 16:57:53 +02:00
marc barbier f726458f29 removed check 2021-07-10 16:53:47 +02:00
marc barbier 6799681c6b refactoring ModUpdater 2021-07-10 16:52:06 +02:00
marc barbier 155f7ada2e exception non blockante dans les threads 2021-07-09 23:28:33 +02:00
marc barbier c8c23cbe42 safer parse - direct download 2021-07-09 23:27:23 +02:00
marc barbier a93e180c60 safer parse 2021-07-09 23:26:09 +02:00
4 changed files with 103 additions and 58 deletions
+6 -2
View File
@@ -8,11 +8,15 @@ public class Log {
private static PrintStream lastStream = System.out; private static PrintStream lastStream = System.out;
public static void e(String who, String msg) { public static void e(String who, String msg) {
printNonStatic(String.valueOf(who) + " : " + msg, System.err); printNonStatic("[ERROR]" + who + " : " + msg, System.err);
} }
public static void w(String who, String msg) {
printNonStatic("[WARN]" + who + " : " + msg, System.out);
}
public static void i(String who, String msg) { public static void i(String who, String msg) {
if(Main.verbose) printNonStatic(String.valueOf(who) + " : " + msg, System.out); if(Main.verbose) printNonStatic("[INFO]" + who + " : " + msg, System.out);
} }
private static void printNonStatic(String line, PrintStream stream) { private static void printNonStatic(String line, PrintStream stream) {
+44 -20
View File
@@ -10,13 +10,19 @@ import java.util.Iterator;
public class Main { public class Main {
public static boolean verbose; public static boolean verbose;
public static boolean checkingonly;
public static Integer threadNb; public static Integer threadNb;
public static String mcVersion; public static String mcVersion;
public static void main(String[] args) { public static void main(String[] args) {
mcVersion = "1.12.2"; if(mcVersion == null) {
mcVersion = "1.12.2";
}
else {
if(!isVersionValid()) {
Log.e("main", "the version number provided is invalid");
System.exit(1);
}
}
checkModsDirectory(); checkModsDirectory();
new DatabaseVersionManager().updateIfNeeded(); new DatabaseVersionManager().updateIfNeeded();
interpretArgs(args); interpretArgs(args);
@@ -45,55 +51,73 @@ public class Main {
while (it.hasNext()) { while (it.hasNext()) {
String cmd = it.next(); String cmd = it.next();
switch (cmd) { switch (cmd) {
case "--check":
System.out.println("--check is not avaliable");
checkingonly = true;
continue;
case "-t": case "-t":
Log.w("main", "Warning using -t option can lead to corruption during download if you have a lot of cores or a slow internet");
if (threadNb != null) { if (threadNb != null) {
Log.e("main", "error conflicting arguments --thread and -t"); Log.e("main", "error conflicting arguments --thread and -t");
System.exit(1); System.exit(1);
} }
if (cores <= 0) { if (cores <= 0) {
Log.i("main","java think you have no cpu core... sooo lets go for 4"); Log.i("main","java think you have no cpu core... sooo lets go for 2");
threadNb = Integer.valueOf(4); threadNb = Integer.valueOf(2);
continue; break;
} }
Log.i("main", "running on " + cores + " thread"); Log.i("main", "running on " + cores + " thread");
threadNb = Integer.valueOf(cores); threadNb = Integer.valueOf(cores);
continue; break;
case "-v": case "-v":
verbose = true; verbose = true;
continue; break;
case "-version":
checkNext(it);
mcVersion = it.next();
break;
case "--thread": case "--thread":
if (threadNb != null) { if (threadNb != null) {
Log.e("main", "error conflicting arguments --thread and -t"); Log.e("main", "error conflicting arguments --thread and -t");
System.exit(1); System.exit(1);
} }
if (!it.hasNext()) { checkNext(it);
Log.e("main", "arguments invalid please refer to --help");
System.exit(1);
}
String threadNumber = it.next(); String threadNumber = it.next();
if (threadNumber.matches("[0-9]*")) { if (threadNumber.matches("[0-9]*")) {
threadNb = Integer.valueOf(Integer.parseInt(threadNumber)); threadNb = Integer.valueOf(Integer.parseInt(threadNumber));
continue; break;
} }
Log.e("main","arguments invalid please refer to --help"); Log.e("main","arguments invalid please refer to --help");
System.exit(1); System.exit(1);
continue; break;
case "--help": case "--help":
Log.i("main", "use --check to check without updating\nuse --help to see this help\nuse -v to see verbose\nuse --threads to specify the number of threads\nuse -t to set the number of thread automaticaly"); Log.i("main", "use --help to see this help\nuse -v to see verbose\nuse --threads to specify the number of threads\nuse -t to set the number of thread automaticaly");
System.exit(0); System.exit(0);
continue; break;
default: default:
Log.e("main", "unknown args " + cmd + "\nsee --help for help"); Log.e("main", "unknown args " + cmd + "\nsee --help for help");
System.exit(1); System.exit(1);
} }
} }
} }
private static boolean isVersionValid() {
switch(mcVersion.split(".").length) {
case 2:
return mcVersion.matches("[0-9]*\\.[0-9]*");
case 3:
return mcVersion.matches("[0-9]*\\.[0-9]*\\.[0-9]*");
default:
return false;
}
}
private static void checkNext(Iterator<String> it) {
if (!it.hasNext()) {
Log.e("main", "arguments invalid please refer to --help");
System.exit(1);
}
}
} }
@@ -16,7 +16,7 @@ public class ModUpdater {
private DirectUpdateManager directUpdateManager; private DirectUpdateManager directUpdateManager;
private CurseUpdateManager curseUpdateManager; private CurseUpdateManager curseUpdateManager;
private Database db; private Database db;
private final String ME = "ModUpdater"; private static final String ME = "ModUpdater";
public ModUpdater() { public ModUpdater() {
this.db = new Database(); this.db = new Database();
@@ -33,6 +33,14 @@ public class ModUpdater {
} }
//parse and download //parse and download
startProcessingThreads(modsList);
checkForFailures();
Log.i(ME, "finished");
curseUpdateManager.updateFile();
}
private void startProcessingThreads(File modsList) {
List<Thread> threads = new ArrayList<>(Main.threadNb); List<Thread> threads = new ArrayList<>(Main.threadNb);
try { try {
BufferedReader in = new BufferedReader(new FileReader(modsList)); BufferedReader in = new BufferedReader(new FileReader(modsList));
@@ -44,10 +52,8 @@ public class ModUpdater {
threads.add(t); threads.add(t);
} }
while(Main.threadNb == threads.size()) { while(Main.threadNb >= threads.size()) {
try { threadSleep();
Thread.sleep(10);
} catch (InterruptedException e) {}
} }
} }
in.close(); in.close();
@@ -55,23 +61,20 @@ public class ModUpdater {
e.printStackTrace(); e.printStackTrace();
System.exit(1); System.exit(1);
} }
waitThreadsEnd(threads);
}
//on attend la fin des threads private void checkForFailures() {
while( !threads.isEmpty() ) {
try {
Thread.sleep(100);
} catch (InterruptedException e) {}
}
if(! UpdaterThread.failedLines.isEmpty()) { if(! UpdaterThread.failedLines.isEmpty()) {
List<String> failures = new ArrayList<>(UpdaterThread.failedLines); List<String> failures = new ArrayList<>(UpdaterThread.failedLines);
UpdaterThread.failedLines.clear(); UpdaterThread.failedLines.clear();
Log.e(ME, "filed to download " + failures.size()); Log.e(ME, "filed to download " + failures.size());
Log.e(ME, "Retrying but slower"); Log.e(ME, "Retrying but slower");
for(String line : failures) { for(String line : failures) {
Thread t = new UpdaterThread(line, db, directUpdateManager, curseUpdateManager, threads); Thread t = new UpdaterThread(line, db, directUpdateManager, curseUpdateManager, null);
t.start(); t.start();
try { try {
t.join(); t.join();
@@ -85,10 +88,18 @@ public class ModUpdater {
} }
} }
} }
}
private void waitThreadsEnd(List<Thread> threads) {
while( !threads.isEmpty() ) {
threadSleep();
}
}
Log.i(ME, "finished"); private void threadSleep() {
curseUpdateManager.updateFile(); try {
Thread.sleep(10);
} catch (InterruptedException e) {}
} }
public static boolean isAComment(String line) public static boolean isAComment(String line)
@@ -105,7 +116,7 @@ final class UpdaterThread extends Thread {
private final CurseUpdateManager curseUpdateManager; private final CurseUpdateManager curseUpdateManager;
private final List<Thread> threads; private final List<Thread> threads;
private static final String ME = "ModUpdaterThread"; private static final String ME = "ModUpdaterThread";
public static final List<String> failedLines = new ArrayList<>(); protected static final List<String> failedLines = new ArrayList<>();
public UpdaterThread(String line, Database db, DirectUpdateManager directUpdateManager, CurseUpdateManager curseUpdateManager, List<Thread> threads) { public UpdaterThread(String line, Database db, DirectUpdateManager directUpdateManager, CurseUpdateManager curseUpdateManager, List<Thread> threads) {
super(line); super(line);
@@ -118,31 +129,36 @@ final class UpdaterThread extends Thread {
@Override @Override
public void run() { public void run() {
//used for post download checks try {
File downloadedFile = null; //used for post download checks
if(line.startsWith("direct=")) File downloadedFile = null;
{ if(line.startsWith("direct=") && line.contains(";") && line.contains("@"))
downloadedFile = handleDirectDownload(line); {
} else if (line.startsWith("del=")) { downloadedFile = handleDirectDownload(line);
String[] delete = line.split("del="); } else if (line.startsWith("del=")) {
if(new File("mods/" + delete[1]).delete()) { String[] delete = line.split("del=");
Log.i(ME, "Successfully removed " + delete[1]); if(new File("mods/" + delete[1]).delete()) {
Log.i(ME, "Successfully removed " + delete[1]);
}
} }
} else {
else { if (line.split("/").length >= 5 && line.contains("curseforge")) {
if (line.split("/").length >= 5) { downloadedFile = handleCurseDownload(line);
downloadedFile = handleCurseDownload(line); } else {
Log.e(ME, "unrecognized line: " + line);
}
} }
}
if(downloadedFile != null && !ArchiveHelper.checkJarIntegrity(downloadedFile)) { if(downloadedFile != null && !ArchiveHelper.checkJarIntegrity(downloadedFile)) {
downloadedFile.delete(); downloadedFile.delete();
failedLines.add(line); failedLines.add(line);
} }
} catch(Exception e) { e.printStackTrace(); }
super.run(); super.run();
//on se retire de la pool d'execution //on se retire de la pool d'execution
this.threads.remove(this); if(this.threads != null)this.threads.remove(this);
} }
private File handleDirectDownload(String line) { private File handleDirectDownload(String line) {
@@ -13,6 +13,7 @@ import java.nio.file.Paths;
import java.util.Locale; import java.util.Locale;
import downloader.Log; import downloader.Log;
import downloader.Main;
public class HttpHelper { public class HttpHelper {
private HttpHelper() {} private HttpHelper() {}
@@ -126,7 +127,7 @@ public class HttpHelper {
if(elapsed != 0) { if(elapsed != 0) {
long estimatedBandwith = (long) (bytesRead / elapsed); long estimatedBandwith = (long) (bytesRead / elapsed);
//display the estimate //display the estimate
Log.setStaticPrint(formatBandwidth(estimatedBandwith)); Log.setStaticPrint("estimated bandwidth :" + formatBandwidth(estimatedBandwith * Main.threadNb));
start = System.currentTimeMillis(); start = System.currentTimeMillis();
} }
downloadingWriter.write(buffer, 0, bytesRead); downloadingWriter.write(buffer, 0, bytesRead);