Java---多线程断点下载

http://blog.csdn.net/jwzhangjie/article/details/9772247

在上一章中我们实现了多线程下载功能，这里我们添加断点下载功能，防止下载过程中程序意外退出。具体代码如下：

[java] view plain copy

1. package com.jwzhangjie;  
3. /**
4.  * 说明：
5.  * 每一个线程下载的位置计算方式：
6.  * 开始位置：
7.  * (线程id - 1)*每一块大小
8.  * 结束位置：
9.  * (线程id*每一块大小) - 1 
10.  *  ---注意有时候不一定能够整除，所以最后一个线程的结束位置应该是文件的末尾
11.  *  
12.  *  步骤：
13.  *  1.本地创建一个大小跟服务器文件相同的临时文件
14.  *  2.计算分配几个线程去下载服务器上的资源，知道每个线程下载文件的位置
15.  *  3.开启三个线程，每一个线程下载对应位置的文件
16.  *  4.如果所有的线程，都把自己的数据下载完毕后，服务器上的资源都被下载到本地了
17.  *  
18.  *  断点下载：
19.  *  1.使用文件记录每一个线程的下载长度
20.  *  2.每一个下载开始之前，读取文件，如果文件存在并且长度大于0，则取出长度
21.  *  3.将每一个线程的起始位置+已经下载的长度
22.  *  4.所有的线程下载完毕后，删除保存下载长度的文件
23.  */
24. import java.io.File;  
25. import java.io.FileInputStream;  
26. import java.io.InputStream;  
27. import java.io.RandomAccessFile;  
28. import java.net.HttpURLConnection;  
29. import java.net.URL;  
31. public class Demo {  
32. public static String path = "http://192.168.5.103:8080/examples/downloader.exe";//"http://softdownload.hao123.com/hao123-soft-online-bcs/soft/Y/2013-07-18_YoudaoDict_baidu.alading.exe";  
33. public static int threadCount = 3;  
34. public static int runningThread = 3;  
35. public static void main(String[] args) throws Exception{  
36. //1.连接服务器，获取一个文件，获取文件的长度，在本地创建一个跟服务器一样大小的临时文件
37.         URL url = new URL(path);  
38.         HttpURLConnection conn = (HttpURLConnection)url.openConnection();  
39.         conn.setConnectTimeout(5000);  
40.         conn.setRequestMethod("GET");  
41. int code = conn.getResponseCode();  
42. if (code == 200) {  
43. //服务器端返回的数据的长度，实际上就是文件的长度
44. int length = conn.getContentLength();  
45.             System.out.println("文件总长度："+length);  
46. //在客户端本地创建出来一个大小跟服务器端一样大小的临时文件
47.             RandomAccessFile raf = new RandomAccessFile("setup.exe", "rwd");  
48. //指定创建的这个文件的长度
49.             raf.setLength(length);  
50.             raf.close();  
51. //假设是3个线程去下载资源。
52. //平均每一个线程下载的文件大小.
53. int blockSize = length / threadCount;  
54. for (int threadId = 1; threadId <= threadCount; threadId++) {  
55. //第一个线程下载的开始位置
56. int startIndex = (threadId - 1) * blockSize;  
57. int endIndex = threadId * blockSize - 1;  
58. if (threadId == threadCount) {//最后一个线程下载的长度要稍微长一点
59.                     endIndex = length;  
60.                 }  
61.                 System.out.println("线程："+threadId+"下载:---"+startIndex+"--->"+endIndex);  
62. new DownLoadThread(path, threadId, startIndex, endIndex).start();  
63.             }  
65.         }else {  
66.             System.out.printf("服务器错误!");  
67.         }  
68.     }  
70. /**
71.      * 下载文件的子线程  每一个线程下载对应位置的文件
72.      * @author jie
73.      *
74.      */
75. public static class DownLoadThread extends Thread{  
76. private int threadId;  
77. private int startIndex;  
78. private int endIndex;  
79. /**
80.          * @param path 下载文件在服务器上的路径
81.          * @param threadId 线程Id
82.          * @param startIndex 线程下载的开始位置
83.          * @param endIndex  线程下载的结束位置
84.          */
85. public DownLoadThread(String path, int threadId, int startIndex, int endIndex) {  
86. super();  
87. this.threadId = threadId;  
88. this.startIndex = startIndex;  
89. this.endIndex = endIndex;  
90.         }  
92. @Override
93. public void run() {  
94. try {  
95. //检查是否存在记录下载长度的文件，如果存在读取这个文件
96.                 File tmp_file = new File(threadId+".txt");  
97. if (tmp_file.exists() && tmp_file.length() > 0) {  
98.                     FileInputStream fio = new FileInputStream(tmp_file);  
99. byte[] temp = new byte[1024];  
100. int len = fio.read(temp);  
101.                     String downloadlen = new String(temp, 0, len);  
102. int downloadInt = Integer.parseInt(downloadlen);  
103.                     startIndex = downloadInt;//修改下载的真实的开始位置
104.                     System.out.println("线程："+threadId+"真实的下载位置："+startIndex+"--->"+endIndex);  
105.                     fio.close();  
106.                 }  
109.                 URL url = new URL(path);  
110.                 HttpURLConnection conn = (HttpURLConnection)url.openConnection();  
111.                 conn.setConnectTimeout(5000);  
112.                 conn.setRequestMethod("GET");  
113. //重要:请求服务器下载部分文件 指定文件的位置
114.                 conn.setRequestProperty("Range", "bytes="+startIndex+"-"+endIndex);  
115. //从服务器请求全部资源返回200 ok如果从服务器请求部分资源 返回 206 ok
116. int code = conn.getResponseCode();  
117.                 System.out.println("code:"+code);  
118.                 InputStream is = conn.getInputStream();//已经设置了请求的位置，返回的是当前位置对应的文件的输入流
119.                 RandomAccessFile raf = new RandomAccessFile("setup.exe", "rwd");  
120. //随机写文件的时候从哪个位置开始写
121.                 raf.seek(startIndex);//定位文件
123. int len = 0;  
124. byte[] buffer = new byte[1024];  
125. int total = 0;//已经下载的数据长度
126. while ((len = is.read(buffer)) != -1) {  
127.                     RandomAccessFile file = new RandomAccessFile(threadId+".txt", "rwd");  
128.                     raf.write(buffer, 0, len);  
129.                     total += len;  
130.                     file.write((""+(total+startIndex)).getBytes());  
131.                     file.close();  
132.                 }  
133.                 is.close();  
134.                 raf.close();  
135.                 System.out.println("线程："+threadId+"下载完毕");  
136.             } catch (Exception e) {  
137.                 e.printStackTrace();  
138.             }finally{  
139.                 runningThread--;  
140. if (runningThread == 0) {//所有的线程执行完毕
141. for (int i = 1; i <= threadCount; i++) {  
142.                         File file = new File(i+".txt");  
143.                         file.delete();  
144.                     }  
145.                     System.out.println("文件全部下载完毕!");  
146.                 }  
147.             }  
148.         }  
150.     }  
151. }  

实验测试结果：

第一次下载：

[html] view plain copy

1. 文件总长度：5562040  
2. 线程：1下载:---0--->1854012  
3. 线程：2下载:---1854013--->3708025  
4. 线程：3下载:---3708026--->5562040  
5. code:206  
6. code:206  
7. code:206  

第二次下载：

[html] view plain copy

1. 文件总长度：5562040  
2. 线程：1下载:---0--->1854012  
3. 线程：2下载:---1854013--->3708025  
4. 线程：3下载:---3708026--->5562040  
5. 线程：1真实的下载位置：21504--->1854012  
6. 线程：2真实的下载位置：1871421--->3708025  
7. 线程：3真实的下载位置：3726458--->5562040  
8. code:206  
9. code:206  
10. code:206  

下载完毕：

[html] view plain copy

1. 文件总长度：5562040  
2. 线程：1下载:---0--->1854012  
3. 线程：2下载:---1854013--->3708025  
4. 线程：3下载:---3708026--->5562040  
5. 线程：2真实的下载位置：3512893--->3708025  
6. 线程：3真实的下载位置：5268602--->5562040  
7. 线程：1真实的下载位置：1677312--->1854012  
8. code:206  
9. code:206  
10. code:206  
11. 线程：1下载完毕  
12. 线程：2下载完毕  
13. 线程：3下载完毕  
14. 文件全部下载完毕!