35 lines
1014 B
Java
35 lines
1014 B
Java
|
package com.rarchives.ripme.ripper;
|
||
|
|
|
||
|
|
import java.util.concurrent.ExecutorService;
|
||
|
|
import java.util.concurrent.Executors;
|
||
|
|
import java.util.concurrent.TimeUnit;
|
||
|
|
|
||
|
|
import org.apache.log4j.Logger;
|
||
|
|
|
||
|
|
import com.rarchives.ripme.utils.Utils;
|
||
|
|
|
||
|
|
public class DownloadThreadPool {
|
||
|
|
|
||
|
|
private static final Logger logger = Logger.getLogger(DownloadThreadPool.class);
|
||
|
|
private ExecutorService threadPool = null;
|
||
|
|
|
||
|
|
public DownloadThreadPool() {
|
||
|
|
int threads = Utils.getConfigInteger("threads.size", 10);
|
||
|
|
logger.debug("Initializing thread pool with " + threads + " threads");
|
||
|
|
threadPool = Executors.newFixedThreadPool(threads);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void addThread(Thread t) {
|
||
|
|
threadPool.execute(t);
|
||
|
|
}
|
||
|
|
|
||
|
|
public void waitForThreads() {
|
||
|
|
threadPool.shutdown();
|
||
|
|
try {
|
||
|
|
threadPool.awaitTermination(60, TimeUnit.SECONDS);
|
||
|
|
} catch (InterruptedException e) {
|
||
|
|
logger.error("Interrupted while waiting for threads to finish: ", e);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|