java多线程分块上传并支持断点续传最新修正完整版本

package com.test;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
/**
 * 文件断点续传加分段上传线程
 * @author wzztestin
 *
 */
/**
 * 文件断点续传加分段上传线程
 * @author wzztestin
 *
 */
public class DownFileFetch extends Thread {
	DownFileInfoBean siteInfoBean = null; // 文件信息 Bean
	long[] nStartPos; // 开始位置
	long[] nEndPos; // 结束位置
	DownFileSplitterFetch[] fileSplitterFetch; // 子线程对象
	long nFileLength; // 文件长度
	boolean bFirst = true; // 是否第一次取文件
	boolean bStop = false; // 停止标志
	File tmpFile; // 文件下载的临时信息
	DataOutputStream output; // 输出到文件的输出流
	boolean fileflag; //是本地上传还是远程下载的标志
	File downfile; //本地文件下载
	int splitter = 0;
	
	/**
	 * 下载上传文件抓取初始化
	 * @param bean
	 * @throws IOException
	 */
	public DownFileFetch(DownFileInfoBean bean) throws IOException {
		siteInfoBean = bean;
		/**
		 * File.separator windows是\,unix是/
		 */
		tmpFile = new File(bean.getSFilePath() + File.separator
				+ bean.getSFileName() + ".info");
		if (tmpFile.exists()) {
			bFirst = false;
			//读取已下载的文件信息
			read_nPos();
		} else {
			nStartPos = new long[bean.getNSplitter()];
			nEndPos = new long[bean.getNSplitter()];
		}
		fileflag = bean.getFileflag();
		downfile = bean.getDownfile();
		this.splitter = bean.getNSplitter();
	}

	public void run() {
		// 获得文件长度
		// 分割文件
		// 实例 FileSplitterFetch
		// 启动 FileSplitterFetch 线程
		// 等待子线程返回
		try {
			if (bFirst) {
				nFileLength = getFileSize();
				if (nFileLength == -1) {
					DownFileUtility.log("File Length is not known!");
				} else if (nFileLength == -2) {
					DownFileUtility.log("File is not access!");
				} else {
					for (int i = 0; i < nStartPos.length; i++) {
						nStartPos[i] = (long) (i * (nFileLength / nStartPos.length));
					}
					for (int i = 0; i < nEndPos.length - 1; i++) {
						nEndPos[i] = nStartPos[i + 1];
					}
					nEndPos[nEndPos.length - 1] = nFileLength;
				}
			}
			// 启动子线程
			fileSplitterFetch = new DownFileSplitterFetch[nStartPos.length];
			for (int i = 0; i < nStartPos.length; i++) {
				fileSplitterFetch[i] = new DownFileSplitterFetch(
						siteInfoBean.getSSiteURL(), siteInfoBean.getSFilePath()
								+ File.separator + siteInfoBean.getSFileName()+"_"+i,
						nStartPos[i], nEndPos[i], i,fileflag,downfile,bFirst);
				DownFileUtility.log("Thread " + i + " , nStartPos = " + nStartPos[i]
						+ ", nEndPos = " + nEndPos[i]);
				fileSplitterFetch[i].start();
			}
			//下载子线程是否完成标志
			boolean breakWhile = false;
			while (!bStop) {
				write_nPos();
				DownFileUtility.sleep(500);
				breakWhile = true;
				for (int i = 0; i < nStartPos.length; i++) {
					if (!fileSplitterFetch[i].bDownOver) {
						breakWhile = false;
						break;
					}else{
						write_nPos();
					}
				}
				if (breakWhile){
					break;
				}
			}
			hebinfile(siteInfoBean.getSFilePath()+ File.separator + siteInfoBean.getSFileName(),splitter);
			DownFileUtility.log("文件下载结束!");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 获得文件长度
	 * @return
	 */
	public long getFileSize() {
		int nFileLength = -1;
		if(fileflag){
			try {
				URL url = new URL(siteInfoBean.getSSiteURL());
				HttpURLConnection httpConnection = (HttpURLConnection) url
						.openConnection();
				httpConnection.setRequestProperty("User-Agent", "NetFox");
				int responseCode = httpConnection.getResponseCode();
				if (responseCode >= 400) {
					processErrorCode(responseCode);
					//represent access is error
					return -2; 
				}
				String sHeader;
				for (int i = 1;; i++) {
					sHeader = httpConnection.getHeaderFieldKey(i);
					if (sHeader != null) {
						if (sHeader.equals("Content-Length")) {
							nFileLength = Integer.parseInt(httpConnection
									.getHeaderField(sHeader));
							break;
						}
					} else {
						break;
					}
				}
			} catch (IOException e) {
				e.printStackTrace();
			} catch (Exception e) {
				e.printStackTrace();
			}
			DownFileUtility.log(nFileLength);
		}else{
			try{
				File myflie = downfile;
				nFileLength = (int)myflie.length();
			}catch(Exception e){
				e.printStackTrace();
			}
			DownFileUtility.log(nFileLength);
		}
		return nFileLength;
	}

	/**
	 * 保存下载信息(文件指针位置)
	 */
	private void write_nPos() {
		try {
			output = new DataOutputStream(new FileOutputStream(tmpFile));
			output.writeInt(nStartPos.length);
			for (int i = 0; i < nStartPos.length; i++) {
				output.writeLong(fileSplitterFetch[i].nStartPos);
				output.writeLong(fileSplitterFetch[i].nEndPos);
			}
			output.close();
		} catch (IOException e) {
			e.printStackTrace();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 读取保存的下载信息(文件指针位置)
	 */
	private void read_nPos() {
		try {
			DataInputStream input = new DataInputStream(new FileInputStream(
					tmpFile));
			int nCount = input.readInt();
			nStartPos = new long[nCount];
			nEndPos = new long[nCount];
			for (int i = 0; i < nStartPos.length; i++) {
				nStartPos[i] = input.readLong();
				nEndPos[i] = input.readLong();
			}
			input.close();
		} catch (IOException e) {
			e.printStackTrace();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 输出错误信息
	 * @param nErrorCode
	 */
	private void processErrorCode(int nErrorCode) {
		DownFileUtility.log("Error Code : " + nErrorCode);
	}

	/**
	 * 停止文件下载
	 */
	public void siteStop() {
		bStop = true;
		for (int i = 0; i < nStartPos.length; i++)
			fileSplitterFetch[i].splitterStop();
	}
	
	/**
	 * 合并文件
	 * @param sName
	 * @param splitternum
	 */
	private void hebinfile(String sName,int splitternum){
		try{
			File file = new File(sName);
			if(file.exists()){
				file.delete();
			}
			RandomAccessFile saveinput = new RandomAccessFile(sName,"rw");
			for(int i = 0;i 0) {
						write(saveinput,b, 0, nRead);
					}
					input.close();
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
			DownFileUtility.log("file size is "+saveinput.length());
		}catch(Exception e){
			e.printStackTrace();
		}
	}
	
	/**
	 * 写文件
	 * @param b
	 * @param nStart
	 * @param nLen
	 * @return
	 */
	private int write(RandomAccessFile oSavedFile,byte[] b, int nStart, int nLen) {
		int n = -1;
		try {
			oSavedFile.seek(oSavedFile.length());
			oSavedFile.write(b, nStart, nLen);
			n = nLen;
		} catch (IOException e) {
			e.printStackTrace();
		}
		return n;
	}
}

 

package com.test;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
/**
 * 下载上传子线程
 * @author wzztestin
 *
 */
public class DownFileSplitterFetch extends Thread {
	String sURL; // 下载文件的地址
	long nStartPos; // 文件分段的开始位置
	long nEndPos; // 文件分段的结束位置
	int nThreadID; // 线程的 ID
	boolean bDownOver = false; // 是否下载完成
	boolean bStop = false; // 停止下载
	DownFileAccess fileAccessI = null; // 文件对象
	boolean fileflag; //是URL下载还是本地下载
	File file = null;//本地下载文件
	boolean bFirst = true;
	
	/**
	 * 下载,上传子线程初始化
	 * @param sURL
	 * @param sName
	 * @param nStart
	 * @param nEnd
	 * @param id
	 * @param fileflag
	 * @param downfile
	 * @throws IOException
	 */
	public DownFileSplitterFetch(String sURL, String sName, long nStart, long nEnd,
			int id,boolean fileflag,File downfile,boolean bFirst) throws IOException {
		this.sURL = sURL;
		this.nStartPos = nStart;
		this.nEndPos = nEnd;
		nThreadID = id;
		fileAccessI = new DownFileAccess(sName, nStartPos,bFirst);
		this.fileflag = fileflag;
		this.file = downfile;
		this.bFirst = bFirst;
	}
	
	/**
	 * 线程执行
	 */
	public void run() {
		if(fileflag){
			this.urldownload();
		}else{
			this.filedownload();
		}
	}

	/**
	 * 打印回应的头信息
	 * @param con
	 */
	public void logResponseHead(HttpURLConnection con) {
		for (int i = 1;; i++) {
			String header = con.getHeaderFieldKey(i);
			if (header != null){
				DownFileUtility.log(header + " : " + con.getHeaderField(header));
			}else{
				break;
			}
		}
	}
	
	/**
	 * 地址下载
	 */
	private void urldownload(){
		DownFileUtility.log("Thread " + nThreadID + " url down filesize is "+(nEndPos-nStartPos));
		DownFileUtility.log("Thread " + nThreadID + " url start >> "+nStartPos +"------end >> "+nEndPos);
		while (nStartPos < nEndPos && !bStop) {
			try {
				URL url = new URL(sURL);
				HttpURLConnection httpConnection = (HttpURLConnection) url
						.openConnection();
				httpConnection.setRequestProperty("User-Agent", "NetFox");
				String sProperty = "bytes=" + nStartPos + "-";
				httpConnection.setRequestProperty("RANGE", sProperty);
				DownFileUtility.log(sProperty);
				InputStream input = httpConnection.getInputStream();
				byte[] b = new byte[1024];
				int nRead;
				while ((nRead = input.read(b, 0, 1024)) > 0
						&& nStartPos < nEndPos && !bStop) {
					if((nStartPos+nRead)>nEndPos)
					{
						nRead = (int)(nEndPos - nStartPos);
					}
					nStartPos += fileAccessI.write(b, 0, nRead);
				}
				DownFileUtility.log("Thread " + nThreadID + " nStartPos : "+nStartPos);
				fileAccessI.oSavedFile.close();
				DownFileUtility.log("Thread " + nThreadID + " is over!");
				input.close();
				bDownOver = true;
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
		if(!bDownOver){
			if(nStartPos >= nEndPos){
				bDownOver = true;
			}
		}
	}
	
	/**
	 * 文件下载
	 */
	private void filedownload(){
		DownFileUtility.log("Thread " + nThreadID + " down filesize is "+(nEndPos-nStartPos));
		DownFileUtility.log("Thread " + nThreadID + " start >> "+nStartPos +"------end >> "+nEndPos);
		while (nStartPos < nEndPos && !bStop) {
			try {
				RandomAccessFile input = new RandomAccessFile(file,"r");
				input.seek(nStartPos);
				byte[] b = new byte[1024];
				int nRead;
				while ((nRead = input.read(b, 0, 1024)) > 0
						&& nStartPos < nEndPos && !bStop) {
					if((nStartPos+nRead)>nEndPos)
					{
						nRead = (int)(nEndPos - nStartPos);
					}
					nStartPos += fileAccessI.write(b, 0, nRead);
				}
				fileAccessI.oSavedFile.close();
				DownFileUtility.log("Thread " + nThreadID + " is over!");
				input.close();
				bDownOver = true;
				input.close();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
		if(!bDownOver){
			if(nStartPos >= nEndPos){
				bDownOver = true;
			}
		}
		DownFileUtility.log("Thread " + nThreadID + "last start >> "+nStartPos );
	}
	
	/**
	 * 停止
	 */
	public void splitterStop() {
		bStop = true;
	}
}

 

package com.test;

import java.io.*;

/**
 * 文件对象
 * @author wzztestin
 *
 */
public class DownFileAccess implements Serializable {
	/**
	 * 
	 */
	private static final long serialVersionUID = -2518013155676212866L;
	//写入文件的流
	RandomAccessFile oSavedFile;
	//开始位置
	long nPos;
	boolean bFirst;

	public DownFileAccess() throws IOException {
		this("", 0,true);
	}
	
	/**
	 * 写入文件初始化
	 * @param sName
	 * @param nPos
	 * @throws IOException
	 */
	public DownFileAccess(String sName, long nPos,boolean bFirst) throws IOException {
		File wfile = new File(sName);
		oSavedFile = new RandomAccessFile(wfile,"rw");
		if(!bFirst){
			oSavedFile.seek(wfile.length());
		}
		this.nPos = nPos;
		this.bFirst = bFirst;
	}
	
	/**
	 * 写文件
	 * @param b
	 * @param nStart
	 * @param nLen
	 * @return
	 */
	public synchronized int write(byte[] b, int nStart, int nLen) {
		int n = -1;
		try {
			oSavedFile.write(b, nStart, nLen);
			n = nLen;
		} catch (IOException e) {
			e.printStackTrace();
		}
		return n;
	}
}

 

package com.test;

import java.io.File;

public class DownFileInfoBean {
	private String sSiteURL; // 文件的下载地址
	private String sFilePath; // 保存文件的路径
	private String sFileName; // 保存文件的名字
	private int nSplitter; // 文件分成几段,默认是5段
	private boolean fileflag; // 如果为FALSE则是本地下载,为TRUE则URL下载
	private File downfile;

	public File getDownfile() {
		return downfile;
	}

	public void setDownfile(File downfile) {
		this.downfile = downfile;
	}

	public boolean getFileflag() {
		return fileflag;
	}

	public void setFileflag(boolean fileflag) {
		this.fileflag = fileflag;
	}
	
	/**
	 * 默认初始化
	 */
	public DownFileInfoBean() {
		// default 5
		this("", "", "", 5,false,null);
	}
	
	/**
	 * 下载文件信息初始化
	 * @param sURL 下载的链接地址
	 * @param sPath 上传的保存路径
	 * @param sName 上传保存的文件名
	 * @param nSpiltter 文件分段个数
	 * @param fileflag 是本地文件上传下载还是链接上传下载的标志
	 * @param downfile 本地下载文件(FILE)
	 */
	public DownFileInfoBean(String sURL, String sPath, String sName, int nSpiltter,boolean fileflag,File downfile) {
		sSiteURL = sURL;
		sFilePath = sPath;
		sFileName = sName;
		this.nSplitter = nSpiltter;
		this.fileflag = fileflag;
		this.downfile = downfile;
	}

	public String getSSiteURL() {
		return sSiteURL;
	}

	public void setSSiteURL(String value) {
		sSiteURL = value;
	}

	public String getSFilePath() {
		return sFilePath;
	}

	public void setSFilePath(String value) {
		sFilePath = value;
	}

	public String getSFileName() {
		return sFileName;
	}

	public void setSFileName(String value) {
		sFileName = value;
	}

	public int getNSplitter() {
		return nSplitter;
	}

	public void setNSplitter(int nCount) {
		nSplitter = nCount;
	}
}

 

package com.test;

/**
 * 简单的工具类
 * @author wzztestin
 *
 */
public class DownFileUtility {
	public DownFileUtility() {
	}
	
	/**
	 * 休眠时长
	 * @param nSecond
	 */
	public static void sleep(int nSecond) {
		try {
			Thread.sleep(nSecond);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 打印日志信息
	 * @param sMsg
	 */
	public static void log(String sMsg) {
		System.err.println(sMsg);
	}
	
	/**
	 * 打印日志信息
	 * @param sMsg
	 */
	public static void log(int sMsg) {
		System.err.println(sMsg);
	}
}

 

package com.test;

public class TestMethod {
	public TestMethod() {
		try {
			DownFileInfoBean bean = new DownFileInfoBean(
					"http://cdn.market.hiapk.com/data/upload//2012/09_27/17/car.wu.wei.kyo.shandian_174928.apk", "D:\\temp",
					"shandian_174928.apk", 5,true,null);
			/*File file = new File("D:\\dan07.apk");
			DownFileInfoBean bean = new DownFileInfoBean(null, "D:\\temp",
					"dan07.apk", 3,false,file);*/
			DownFileFetch fileFetch = new DownFileFetch(bean);
			fileFetch.start();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static void main(String[] args) {
		new TestMethod();
	}
}

 

你可能感兴趣的:(java,EE,java,SE)