多线程实现下载文件

  1 /**
  2  * 
  3  */
  4 package com.hlcui.thread;
  5 
  6 import java.io.InputStream;
  7 import java.io.RandomAccessFile;
  8 import java.net.HttpURLConnection;
  9 import java.net.URL;
 10 
 11 /**
 12  * @author Administrator 多线程下载文件
 13  */
 14 public class DownloadUtil {
 15     // 定义成员变量
 16     private String path; // 远程资源路径
 17     private String targetPath; // 本地存储路径
 18     private DownFileThread[] threads; // 线程list
 19     private int threadNum; // 线程数量
 20     private long length; // 下载的文件大小
 21 
 22     // 构造初始化
 23     public DownloadUtil(String path, String targetPath, int threadNum) {
 24         super();
 25         this.path = path;
 26         this.targetPath = targetPath;
 27         this.threads = new DownFileThread[threadNum];
 28         this.threadNum = threadNum;
 29     }
 30 
 31     // 多线程下载文件资源
 32     public void download() {
 33         URL url;
 34         try {
 35             url = new URL(path);
 36             HttpURLConnection conn = (HttpURLConnection) url.openConnection();
 37             conn.setReadTimeout(5 * 1000); // 设置超时时间为5秒
 38             conn.setRequestMethod("GET");
 39             conn.setRequestProperty("connection", "keep-alive");
 40             conn.setRequestProperty("accept", "*/*");
 41 
 42             // 获取远程文件的大小
 43             length = conn.getContentLength();
 44             conn.disconnect();
 45 
 46             // 设置本地文件大小
 47             RandomAccessFile targetFile = new RandomAccessFile(targetPath, "rw");
 48             targetFile.setLength(length);
 49 
 50             // 每个线程下载大小
 51             long avgPart = length / threadNum + 1;
 52             // 下载文件
 53             for (int i = 0; i < threadNum; i++) {
 54                 long startPos = avgPart * i;
 55                 RandomAccessFile targetTmp = new RandomAccessFile(targetPath,
 56                         "rw");
 57                 targetTmp.seek(startPos); // 分段下载
 58                 threads[i] = new DownFileThread(startPos, targetTmp, avgPart);
 59                 threads[i].start();
 60             }
 61         } catch (Exception e) {
 62             e.printStackTrace();
 63         }
 64     }
 65 
 66     // 监控下载进度
 67     public double getDownRate() {
 68         int currentSize = 0;
 69         for (int i = 0; i < threadNum; i++) {
 70             currentSize += threads[i].length;
 71         }
 72         return currentSize * 1.0 / length;
 73     }
 74 
 75     // 定义线程类
 76     class DownFileThread extends Thread {
 77         private long startPos;
 78         private RandomAccessFile raf;
 79         private long size;
 80         private long length;
 81 
 82         public DownFileThread(long startPos, RandomAccessFile raf, long size) {
 83             super();
 84             this.startPos = startPos;
 85             this.raf = raf;
 86             this.size = size;
 87         }
 88 
 89         public void run() {
 90             URL url;
 91             try {
 92                 url = new URL(path);
 93                 HttpURLConnection conn = (HttpURLConnection) url
 94                         .openConnection();
 95                 conn.setReadTimeout(5 * 1000); // 设置超时时间为5秒
 96                 conn.setRequestMethod("GET");
 97                 conn.setRequestProperty("connection", "keep-alive");
 98                 conn.setRequestProperty("accept", "*/*");
 99 
100                 InputStream in = conn.getInputStream();
101                 in.skip(this.startPos);
102                 byte[] buf = new byte[1024];
103                 int hasRead = 0;
104                 while (length < size && (hasRead = in.read(buf)) != -1) {
105                     raf.write(buf, 0, hasRead);
106                     length += hasRead;
107                 }
108                 raf.close();
109                 in.close();
110             } catch (Exception e) {
111                 e.printStackTrace();
112             }
113         }
114     }
115 
116     // 测试
117     public static void main(String[] args) {
118         String path = "http://localhost:8080/healthcare/html/HealthCare.war";
119         String targetPath = "D:/healthcare.war";
120         final DownloadUtil download = new DownloadUtil(path, targetPath, 4);
121         download.download();
122         // 主线程负责下载文件，在启动一个线程负责监控下载的进度
123         new Thread(new Runnable() {
124 
125             @Override
126             public void run() {
127                 while (download.getDownRate() < 1) {
128                     System.out.println(download.getDownRate());
129                     try {
130                         Thread.sleep(200); // 200毫秒扫描一次
131                     } catch (InterruptedException e) {
132                         e.printStackTrace();
133                     }
134                 }
135             }
136 
137         }).start();
138     }
139 }