FileDownloader

原创
2015/06/10 16:15
阅读数 3.2K

FileDownloader

 package cn.thewee.gourmetmeal.staff.download;
import java.io.File;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import android.content.Context;
import android.util.Log;

public class FileDownloader {
 private static final String TAG = "FileDownloader";
 private Context context;
 private FileService fileService;
 /* 停止下载 */
 private boolean exit;
 /* 已下载文件长度 */
 private int downloadSize = 0;
 /* 原始文件长度 */
 private int fileSize = 0;
 /* 线程数 */
 private DownloadThread[] threads;
 /* 本地保存文件 */
 private File saveFile;
 /* 缓存各线程下载的长度*/
 private Map<Integer, Integer> data = new ConcurrentHashMap<Integer, Integer>();
 /* 每条线程下载的长度 */
 private int block;
 /* 下载路径  */
 private String downloadUrl;
 /**
  * 获取线程数
  */
 public int getThreadSize() {
  return threads.length;
 }
 /**
  * 退出下载
  */
 public void exit(){
  this.exit = true;
 }
 public boolean getExit(){
  return this.exit;
 }
 /**
  * 获取文件大小
  * @return
  */
 public int getFileSize() {
  return fileSize;
 }
 /**
  * 累计已下载大小
  * @param size
  */
 protected synchronized void append(int size) {
  downloadSize += size;
 }
 /**
  * 更新指定线程最后下载的位置
  * @param threadId 线程id
  * @param pos 最后下载的位置
  */
 protected synchronized void update(int threadId, int pos) {
  this.data.put(threadId, pos);
  this.fileService.update(this.downloadUrl, threadId, pos);
 }
 /**
  * 构建文件下载器
  * @param downloadUrl 下载路径
  * @param fileSaveDir 文件保存目录
  * @param threadNum 下载线程数
  */
 public FileDownloader(Context context, String downloadUrl, File fileSaveDir, int threadNum) {
  try {
   this.context = context;
   this.downloadUrl = downloadUrl;
   fileService = new FileService(this.context);
   URL url = new URL(this.downloadUrl);
   if(!fileSaveDir.exists()) fileSaveDir.mkdirs();
   this.threads = new DownloadThread[threadNum];     
   HttpURLConnection conn = (HttpURLConnection) url.openConnection();
   conn.setConnectTimeout(5*1000);
   conn.setRequestMethod("GET");
   conn.setRequestProperty("Accept", "image/gif, image/jpeg, image/pjpeg, image/pjpeg, application/x-shockwave-flash, application/xaml+xml, application/vnd.ms-xpsdocument, application/x-ms-xbap, application/x-ms-application, application/vnd.ms-excel, application/vnd.ms-powerpoint, application/msword, */*");
   conn.setRequestProperty("Accept-Language", "zh-CN");
   conn.setRequestProperty("Referer", downloadUrl); 
   conn.setRequestProperty("Charset", "UTF-8");
   conn.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.2; Trident/4.0; .NET CLR 1.1.4322; .NET CLR 2.0.50727; .NET CLR 3.0.04506.30; .NET CLR 3.0.4506.2152; .NET CLR 3.5.30729)");
   conn.setRequestProperty("Connection", "Keep-Alive");
   conn.connect();
   printResponseHeader(conn);
   if (conn.getResponseCode()==200) {
    this.fileSize = conn.getContentLength();//根据响应获取文件大小
    if (this.fileSize <= 0) throw new RuntimeException("Unkown file size ");
      
    String filename = getFileName(conn);//获取文件名称
    this.saveFile = new File(fileSaveDir, filename);//构建保存文件
    Map<Integer, Integer> logdata = fileService.getData(downloadUrl);//获取下载记录
    if(logdata.size()>0){//如果存在下载记录
     for(Map.Entry<Integer, Integer> entry : logdata.entrySet())
      data.put(entry.getKey(), entry.getValue());//把各条线程已经下载的数据长度放入data中
    }
    if(this.data.size()==this.threads.length){//下面计算所有线程已经下载的数据总长度
     for (int i = 0; i < this.threads.length; i++) {
      this.downloadSize += this.data.get(i+1);
     }
     print("已经下载的长度"+ this.downloadSize);
    }
    //计算每条线程下载的数据长度
    this.block = (this.fileSize % this.threads.length)==0? this.fileSize / this.threads.length : this.fileSize / this.threads.length + 1;
   }else{
    throw new RuntimeException("server no response ");
   }
  } catch (Exception e) {
   print(e.toString());
   throw new RuntimeException("don't connection this url");
  }
 }
 /**
  * 获取文件名
  */
 private String getFileName(HttpURLConnection conn) {
  String filename = this.downloadUrl.substring(this.downloadUrl.lastIndexOf('/') + 1);
  if(filename==null || "".equals(filename.trim())){//如果获取不到文件名称
   for (int i = 0;; i++) {
    String mine = conn.getHeaderField(i);
    if (mine == null) break;
    if("content-disposition".equals(conn.getHeaderFieldKey(i).toLowerCase())){
     Matcher m = Pattern.compile(".*filename=(.*)").matcher(mine.toLowerCase());
     if(m.find()) return m.group(1);
    }
   }
   filename = UUID.randomUUID()+ ".tmp";//默认取一个文件名
  }
  return filename;
 }
 
 /**
  *  开始下载文件
  * @param listener 监听下载数量的变化,如果不需要了解实时下载的数量,可以设置为null
  * @return 已下载文件大小
  * @throws Exception
  */
 public int download(DownloadProgressListener listener) throws Exception{
  try {
   RandomAccessFile randOut = new RandomAccessFile(this.saveFile, "rw");
   if(this.fileSize>0) randOut.setLength(this.fileSize);
   randOut.close();
   URL url = new URL(this.downloadUrl);
   if(this.data.size() != this.threads.length){//如果原先未曾下载或者原先的下载线程数与现在的线程数不一致
    this.data.clear();
    for (int i = 0; i < this.threads.length; i++) {
     this.data.put(i+1, 0);//初始化每条线程已经下载的数据长度为0
    }
    this.downloadSize = 0;
   }
   for (int i = 0; i < this.threads.length; i++) {//开启线程进行下载
    int downLength = this.data.get(i+1);
    if(downLength < this.block && this.downloadSize<this.fileSize){//判断线程是否已经完成下载,否则继续下载 
     this.threads[i] = new DownloadThread(this, url, this.saveFile, this.block, this.data.get(i+1), i+1);
     this.threads[i].setPriority(7);
     this.threads[i].start();
    }else{
     this.threads[i] = null;
    }
   }
   fileService.delete(this.downloadUrl);//如果存在下载记录,删除它们,然后重新添加
   fileService.save(this.downloadUrl, this.data);
   boolean notFinish = true;//下载未完成
   while (notFinish) {// 循环判断所有线程是否完成下载
    Thread.sleep(900);
    notFinish = false;//假定全部线程下载完成
    for (int i = 0; i < this.threads.length; i++){
     if (this.threads[i] != null && !this.threads[i].isFinish()) {//如果发现线程未完成下载
      notFinish = true;//设置标志为下载没有完成
      if(this.threads[i].getDownLength() == -1){//如果下载失败,再重新下载
       this.threads[i] = new DownloadThread(this, url, this.saveFile, this.block, this.data.get(i+1), i+1);
       this.threads[i].setPriority(7);
       this.threads[i].start();
      }
     }
    }    
    if(listener!=null) listener.onDownloadSize(this.downloadSize);//通知目前已经下载完成的数据长度
   }
   if(downloadSize == this.fileSize) fileService.delete(this.downloadUrl);//下载完成删除记录
  } catch (Exception e) {
   print(e.toString());
   throw new Exception("file download error");
  }
  return this.downloadSize;
 }
 /**
  * 获取Http响应头字段
  * @param http
  * @return
  */
 public static Map<String, String> getHttpResponseHeader(HttpURLConnection http) {
  Map<String, String> header = new LinkedHashMap<String, String>();
  for (int i = 0;; i++) {
   String mine = http.getHeaderField(i);
   if (mine == null) break;
   header.put(http.getHeaderFieldKey(i), mine);
  }
  return header;
 }
 /**
  * 打印Http头字段
  * @param http
  */
 public static void printResponseHeader(HttpURLConnection http){
  Map<String, String> header = getHttpResponseHeader(http);
  for(Map.Entry<String, String> entry : header.entrySet()){
   String key = entry.getKey()!=null ? entry.getKey()+ ":" : "";
   print(key+ entry.getValue());
  }
 }
 private static void print(String msg){
  Log.i(TAG, msg);
 }
}

DownloadThread

public class DownloadThread extends Thread {
 private static final String TAG = "DownloadThread";
 private File saveFile;
 private URL downUrl;
 private int block;
 /* 下载开始位置  */
 private int threadId = -1; 
 private int downLength;
 private boolean finish = false;
 private FileDownloader downloader;
 public DownloadThread(FileDownloader downloader, URL downUrl, File saveFile, int block, int downLength, int threadId) {
  this.downUrl = downUrl;
  this.saveFile = saveFile;
  this.block = block;
  this.downloader = downloader;
  this.threadId = threadId;
  this.downLength = downLength;
 }
  @Override
 public void run() {
  if(downLength < block){//未下载完成
   try {
    HttpURLConnection http = (HttpURLConnection) downUrl.openConnection();
    http.setConnectTimeout(5 * 1000);
    http.setRequestMethod("GET");
    http.setRequestProperty("Accept", "image/gif, image/jpeg, image/pjpeg, image/pjpeg, application/x-shockwave-flash, application/xaml+xml, application/vnd.ms-xpsdocument, application/x-ms-xbap, application/x-ms-application, application/vnd.ms-excel, application/vnd.ms-powerpoint, application/msword, */*");
    http.setRequestProperty("Accept-Language", "zh-CN");
    http.setRequestProperty("Referer", downUrl.toString()); 
    http.setRequestProperty("Charset", "UTF-8");
    int startPos = block * (threadId - 1) + downLength;//开始位置
    int endPos = block * threadId -1;//结束位置
    http.setRequestProperty("Range", "bytes=" + startPos + "-"+ endPos);//设置获取实体数据的范围
    http.setRequestProperty("User-Agent", "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.2; Trident/4.0; .NET CLR 1.1.4322; .NET CLR 2.0.50727; .NET CLR 3.0.04506.30; .NET CLR 3.0.4506.2152; .NET CLR 3.5.30729)");
    http.setRequestProperty("Connection", "Keep-Alive");    
    InputStream inStream = http.getInputStream();
    byte[] buffer = new byte[1024];
    int offset = 0;
    print("Thread " + this.threadId + " start download from position "+ startPos);
    RandomAccessFile threadfile = new RandomAccessFile(this.saveFile, "rwd");
    threadfile.seek(startPos);
    while (!downloader.getExit() && (offset = inStream.read(buffer, 0, 1024)) != -1) {
     threadfile.write(buffer, 0, offset);
     downLength += offset;
     downloader.update(this.threadId, downLength);
     downloader.append(offset);
    }
    threadfile.close();
    inStream.close();
    print("Thread " + this.threadId + " download finish");
    this.finish = true;
   } catch (Exception e) {
    this.downLength = -1;
    print("Thread "+ this.threadId+ ":"+ e);
   }  }}
 private static void print(String msg){
  Log.i(TAG, msg);
 }
 public boolean isFinish() {
  return finish;
 }
 public long getDownLength() {
  return downLength;
 }}
展开阅读全文
打赏
0
8 收藏
分享
加载中
更多评论
打赏
0 评论
8 收藏
0
分享
返回顶部
顶部