AbFileDownloader.java

/*
 * Copyright (C) 2012 www.amsoft.cn
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.andbase.util.download;

import java.io.File;
import java.net.HttpURLConnection;
import java.util.LinkedHashMap;
import java.util.Map;

import android.content.Context;
import android.os.Environment;

import com.ab.util.AbFileUtil;
import com.ab.util.AbLogUtil;

// TODO: Auto-generated Javadoc
/**
 * 描述：多线程支持断点续传下载器.
 *
 * @author 还如一梦中
 * @version v1.0
 * @date：2013-3-14 下午5:00:52
 */
public class AbFileDownloader {

	/** The context. */
	private Context context;
	
	/** The threads. */
	private AbDownloadThread threads;
	
	/** The m down file dao. */
	private DownFileDao mDownFileDao;
	
	/** 本地保存文件. */
	private File saveFile;
	
	/** The m down file. */
	private DownFile mDownFile = null;
	
	/** The m thread num. */
	private int mThreadNum = 1;
	
	/** The flag. */
	private boolean flag = true;
	
	/**
	 * 更新指定线程最后下载的位置.
	 *
	 * @param downFile the down file
	 */
	protected synchronized void update(DownFile downFile) {
		this.mDownFileDao.update(downFile);
	}

	/**
	 * 构建文件下载器.
	 *
	 * @param context the context
	 * @param downFile the down file
	 * @param threadNum 下载线程数
	 */
	public AbFileDownloader(Context context,DownFile downFile,int threadNum) {
		try {
			this.context = context;
			this.mDownFile = downFile;
			this.mThreadNum = threadNum;
			mDownFileDao = new DownFileDao(context);
			// 构建保存文件
			String fileName = AbFileUtil.getRealFileNameFromUrl(mDownFile.getDownUrl());
			saveFile = new File(AbFileUtil.getFileDownloadDir(context)+fileName);
			if (!saveFile.getParentFile().exists()){
				saveFile.getParentFile().mkdirs();
			}
			if (!saveFile.exists()){
				saveFile.createNewFile();
				//删除原来的下载数据
				mDownFileDao.delete(downFile.getDownUrl());
				
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 开始下载文件.
	 *
	 * @param listener 监听下载数量的变化,如果不需要了解实时下载的数量,可以设置为null
	 * @return 已下载文件大小
	 * @throws Exception the exception
	 */
	public void download(AbDownloadProgressListener listener) throws Exception {
		try {
			this.threads = new AbDownloadThread(this,mDownFile,saveFile);
			this.threads.setPriority(7);
			this.threads.start();
			this.mDownFileDao.save(mDownFile);
			// 循环判断所有线程是否完成下载
			while (flag && mDownFile.getDownLength() <= mDownFile.getTotalLength()) {
				Thread.sleep(2000);
				// 如果下载失败,再重新下载
				if (mDownFile.getDownLength() == -1) {
					//下载失败
					return;
				}
				
				// 没间隔几秒通知目前已经下载完成的数据长度
				if (listener != null){
					listener.onDownloadSize(mDownFile.getDownLength());
				}
				if(mDownFile.getDownLength() == mDownFile.getTotalLength()){
					break;
				}
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 获取Http响应头字段.
	 *
	 * @param http the http
	 * @return the http response header
	 */
	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 the 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() + ":" : "";
			AbLogUtil.i(AbFileDownloader.class, key + entry.getValue());
		}
	}

	/**
	 * Gets the threads.
	 *
	 * @return the threads
	 */
	public AbDownloadThread getThreads() {
		return threads;
	}

	/**
	 * Gets the save file.
	 *
	 * @return the save file
	 */
	public File getSaveFile() {
		return saveFile;
	}

	/**
	 * Gets the flag.
	 *
	 * @return the flag
	 */
	public boolean getFlag() {
		return flag;
	}

	/**
	 * Sets the flag.
	 *
	 * @param flag the new flag
	 */
	public void setFlag(boolean flag) {
		this.flag = flag;
	}
	
	
	
}