ImgScroll/src/main/java/com/rarchives/ripme/ripper/DownloadThreadPool.java

46 lines
1.3 KiB
Java
Raw Normal View History

package com.rarchives.ripme.ripper;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import org.apache.log4j.Logger;
import com.rarchives.ripme.utils.Utils;
2014-03-09 09:20:22 +01:00
/**
* Simple wrapper around a FixedThreadPool.
*/
public class DownloadThreadPool {
private static final Logger logger = Logger.getLogger(DownloadThreadPool.class);
private ThreadPoolExecutor threadPool = null;
public DownloadThreadPool() {
initialize("Main");
}
public DownloadThreadPool(String threadPoolName) {
initialize(threadPoolName);
}
private void initialize(String threadPoolName) {
int threads = Utils.getConfigInteger("threads.size", 10);
logger.debug("Initializing " + threadPoolName + " thread pool with " + threads + " threads");
threadPool = (ThreadPoolExecutor) Executors.newFixedThreadPool(threads);
}
public void addThread(Thread t) {
threadPool.execute(t);
}
public void waitForThreads() {
threadPool.shutdown();
try {
threadPool.awaitTermination(3600, TimeUnit.SECONDS);
} catch (InterruptedException e) {
2014-03-09 09:20:22 +01:00
logger.error("[!] Interrupted while waiting for threads to finish: ", e);
}
}
}