java 等待文件下载_多个文件开启固定线程等待下载－－java实现
今天群友的一个需求，无聊中，所以写了个小的demo。
需求：要下载多个文件，但同时下载的文件最多为3个，如果超过则显示“等待下载”，等有文件下载完成后再从“等待下载”中开始下载，这样的下载应该在很多地方见过。
思路：这里不考虑多线程下载单个文件的情况，一切从简出发。先想到的是利用线程池(池A)把所有要下载的文件封装成一个线程放在这里做为资源，再有一个线程池(池B)用来存放正在下载的线程，池B类似生产者消费者模型。下载完一个后就从池B中移除，再从池A中添加一个进来。直到池A的线程用完。
需要考虑的是怎么知道什么时候池B要移除，池A要添加进去呢？且看代码：
下载的线程DownloadThread.java
public DownloadThread(String url, String toPath) {
try {
this.url = url;
this.file = new File(toPath);
}catch (Exception e){
System.err.println("Stub!");
}
}
每个线程要下载的文件路径，和要保存的路径，当然也可以从url中解析出文件名，然后再默认一个存放路径，这样就只要传一个参数了，但现在：一切从简。
private volatile boolean isFinish = false;//是否下载完成
设置一个标志位，用来表示是否下载完成 ，在判断池B是否要移除当前线程很重要。
public void start(){
new Thread(new Download()).start();
}
开启下载。
class Download implements Runnable {
private void downloadURL(){
InputStream is = null;
FileOutputStream fos = null;
try {
URL uri = new URL(url);
URLConnection conn = uri.openConnection();
conn.setDoInput(true);
conn.setConnectTimeout(10 * 1000);
is = conn.getInputStream();
fos = new FileOutputStream(file);
byte[] buffer = new byte[1024];
int length = 0;
while(-1 != (length = is.read(buffer))){
fos.write(buffer, 0, length);
}
is.close();
fos.flush();
fos.close();
} catch (MalformedURLException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
} finally {
if(null != is) {
try {
is.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if(null != fos){
try {
fos.flush();
} catch (IOException e) {
e.printStackTrace();
}
try {
fos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
@Override
public void run() {
if(null != url && !("").equals(url) && null != file){
//开启下载到本地
System.out.println(System.currentTimeMillis() + "current thread start==>" + hashCode());
downloadURL();
System.out.println(System.currentTimeMillis() + "current thread end==>" + hashCode());
isFinish = true;
}
}
}
很简单，就是一个下载文件的线程。
线程池，其实包含了所有对线程的操作，没有再向上抽取一层了CustomThreadPool.java
private List customPool;// 线程池
private List startedPool; // 已开启的线程
定义两个线程池，也就是上面说的池A和池B
/** * 增加一个线程 * *@param thread */
public void addThread(DownloadThread thread) {
customPool.add(thread);
}
定义一个方法，把下载线程加到线程池中。
/** * 读取开启线程池 *@author Wicky * */
class StartPoolListener implements Runnable {
boolean isRun = true;
public void setIsRun(boolean isRun) {
this.isRun = isRun;
}
@Override
public void run() {
while (isRun) {
for (int i = 0; i < startedPool.size(); i++) {
DownloadThread thread = startedPool.get(i);
if (null != thread && thread.isFinish()) {
//手动保证集合并发安全
synchronized (thread) {
startedPool.remove(i);
}
}
}
}
}
}
另一个辅助线程，用来监听什么时候该移除线程，添加一个新线程了。
public void start() {
// 开启消费者监听
StartPoolListener lsn = new StartPoolListener();
Thread startPoolThread = new Thread(lsn);
startPoolThread.start();
// 不停的读取线程池中的线程，直到全部执行完成
while (customPool.size() > 0) {
// System.out.println("thread pool left==>" + customPool.size());
// 每次取最后一个线程
DownloadThread thread = customPool.get(customPool.size() - 1);
if (startedPool.size() < threadCount) {
// 开始线程
thread.start();
// 追加到已开启的线程中
startedPool.add(thread);
// 从原来的线程池中移除掉
customPool.remove(thread);
}
}
lsn.setIsRun(false);
startPoolThread.interrupt();
}
然后开始下载……这里分三块。
一、先开启一个监听线程
二、然后开始不停的读池A里的东西，如果发现池B中的数量小于指定的数量，就往里面添加一个新的线程
三、关闭监听线程
测试床MainTest.java
public static void main(String[] args){
String url1 = "http://f.hiphotos.baidu.com/image/pic/item/cc11728b4710b9125dcaf2ffc1fdfc03924522a3.jpg";
String url2 = "http://h.hiphotos.baidu.com/image/w%3D230/sign=f7bc78ce32adcbef013479059caf2e0e/5d6034a85edf8db1ec0d3d6e0a23dd54564e749c.jpg";
String url3 = "http://h.hiphotos.baidu.com/image/w%3D230/sign=56e13fbfaa014c08193b2fa63a7a025b/6a63f6246b600c3338816cde194c510fd9f9a128.jpg";
String url4 = "http://f.hiphotos.baidu.com/image/w%3D230/sign=e07d0317932397ddd6799f076983b216/8d5494eef01f3a29a8b0ef3e9b25bc315d607cc1.jpg";
String url5 = "http://c.hiphotos.baidu.com/image/pic/item/63d0f703918fa0ec4a43ba4c249759ee3c6ddbef.jpg";
String url6 = "http://h.hiphotos.baidu.com/image/pic/item/2e2eb9389b504fc2022d2904e7dde71190ef6d45.jpg";
String url7 = "http://g.hiphotos.baidu.com/image/w%3D230/sign=06cb213db8a1cd1105b675238912c8b0/d01373f082025aaf5a18bbc6f9edab64034f1a91.jpg";
DownloadThread thread1 = new DownloadThread(url1, "D:\\1.jpg");
DownloadThread thread2 = new DownloadThread(url2, "D:\\2.jpg");
DownloadThread thread3 = new DownloadThread(url3, "D:\\3.jpg");
DownloadThread thread4 = new DownloadThread(url4, "D:\\4.jpg");
DownloadThread thread5 = new DownloadThread(url5, "D:\\5.jpg");
DownloadThread thread6 = new DownloadThread(url6, "D:\\6.jpg");
DownloadThread thread7 = new DownloadThread(url7, "D:\\7.jpg");
CustomThreadPool pool = CustomThreadPool.getInstance(4);
pool.addThread(thread1);
pool.addThread(thread2);
pool.addThread(thread3);
pool.addThread(thread4);
pool.addThread(thread5);
pool.addThread(thread6);
pool.addThread(thread7);
//开启下载
pool.start();
}
线程操作还是有点不熟练，这里也只是提供下思路，肯定还会有更好的解决方案，大神勿喷