断点续传和分段上传

最新推荐文章于 2024-11-12 15:21:10 发布

fenglingcompany

最新推荐文章于 2024-11-12 15:21:10 发布

阅读量611

点赞数

文章标签： 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.net.HttpURLConnection;
import java.net.URL;
/**
 * 文件断点续传加分段上传线程
 * @author wzztestin
 *
 */
public class DownFileFetch extends Thread {
	DownFileInfoBean siteInfoBean = null; // 文件信息 Bean
	long[] nStartPos; // 开始位置
	long[] nEndPos; // 结束位置
	FileSplitterFetch[] fileSplitterFetch; // 子线程对象
	long nFileLength; // 文件长度
	boolean bFirst = true; // 是否第一次取文件
	boolean bStop = false; // 停止标志
	File tmpFile; // 文件下载的临时信息
	DataOutputStream output; // 输出到文件的输出流
	boolean fileflag;
	String downfile;

	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.getFilepath();
	}

	public void run() {
		// 获得文件长度
		// 分割文件
		// 实例 FileSplitterFetch
		// 启动 FileSplitterFetch 线程
		// 等待子线程返回
		try {
			if (bFirst) {
				nFileLength = getFileSize();
				if (nFileLength == -1) {
					System.err.println("File Length is not known!");
				} else if (nFileLength == -2) {
					System.err.println("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 FileSplitterFetch[nStartPos.length];
			for (int i = 0; i < nStartPos.length; i++) {
				fileSplitterFetch[i] = new FileSplitterFetch(
						siteInfoBean.getSSiteURL(), siteInfoBean.getSFilePath()
								+ File.separator + siteInfoBean.getSFileName(),
						nStartPos[i], nEndPos[i], i,fileflag,downfile);
				FileUtility.log("Thread " + i + " , nStartPos = " + nStartPos[i]
						+ ", nEndPos = " + nEndPos[i]);
				fileSplitterFetch[i].start();
			}
			// fileSplitterFetch[nPos.length-1] = new
			// FileSplitterFetch(siteInfoBean.getSSiteURL(),siteInfoBean.getSFilePath()
			// + File.separator +
			// siteInfoBean.getSFileName(),nPos[nPos.length-1],nFileLength,nPos.length-1);
			// FileUtility.log("Thread " +(nPos.length-1) +
			// ",nStartPos = "+nPos[nPos.length-1]+",nEndPos = " + nFileLength);
			// fileSplitterFetch[nPos.length-1].start();
			// 等待子线程结束
			// int count = 0;
			// 是否结束 while 循环
			boolean breakWhile = false;
			while (!bStop) {
				write_nPos();
				FileUtility.sleep(500);
				breakWhile = true;
				for (int i = 0; i < nStartPos.length; i++) {
					if (!fileSplitterFetch[i].bDownOver) {
						breakWhile = false;
						break;
					}
				}
				if (breakWhile)
					break;
				// count++;
				// if(count>4)
				// siteStop();
			}
			System.err.println("文件下载结束！");
		} catch (Exception e) {
			e.printStackTrace();
		}
	} // 获得文件长度

	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);
					return -2; // -2 represent access is error
				}
				String sHeader;
				for (int i = 1;; i++) {
					// DataInputStream in = new
					// DataInputStream(httpConnection.getInputStream ());
					// FileUtility.log(in.readLine());
					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();
			}
			FileUtility.log(nFileLength);
		}else{
			try{
				File myflie = new File(downfile);
				nFileLength = (int)myflie.length();
			}catch(Exception e){
				e.printStackTrace();
			}
			FileUtility.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(nPos[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();
		}
	}

	private void processErrorCode(int nErrorCode) {
		System.err.println("Error Code : " + nErrorCode);
	}

	// 停止文件下载
	public void siteStop() {
		bStop = true;
		for (int i = 0; i < nStartPos.length; i++)
			fileSplitterFetch[i].splitterStop();
	}
}

package com.test;

import java.io.*;
import java.net.*;
/**
 * 下载上传子线程
 * @author wzztestin
 *
 */
public class FileSplitterFetch extends Thread {
	String sURL; // 下载文件的地址
	long nStartPos; // 文件分段的开始位置
	long nEndPos; // 文件分段的结束位置
	int nThreadID; // 线程的 ID
	boolean bDownOver = false; // 是否下载完成
	boolean bStop = false; // 停止下载
	FileAccess fileAccessI = null; // 文件对象
	boolean fileflag; //是URL下载还是本地下载
	File file = null;//本地下载文件

	public FileSplitterFetch(String sURL, String sName, long nStart, long nEnd,
			int id,boolean fileflag,String filepath) throws IOException {
		this.sURL = sURL;
		this.nStartPos = nStart;
		this.nEndPos = nEnd;
		nThreadID = id;
		fileAccessI = new FileAccess(sName, nStartPos);
		this.fileflag = fileflag;
		if(filepath != null){
			this.file = new File(filepath);
		}
	}

	public void run() {
		if(fileflag){
			this.urldownload();
		}else{
			this.filedownload();
		}
	}

	// 打印回应的头信息
	public void logResponseHead(HttpURLConnection con) {
		for (int i = 1;; i++) {
			String header = con.getHeaderFieldKey(i);
			if (header != null)
				// responseHeaders.put(header,httpConnection.getHeaderField(header));
				FileUtility.log(header + " : " + con.getHeaderField(header));
			else
				break;
		}
	}
	
	/**
	 * 地址下载
	 */
	private void urldownload(){
		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);
				FileUtility.log(sProperty);
				InputStream input = httpConnection.getInputStream();
				// logResponseHead(httpConnection);
				byte[] b = new byte[1024];
				int nRead;
				while ((nRead = input.read(b, 0, 1024)) > 0
						&& nStartPos < nEndPos && !bStop) {
					nStartPos += fileAccessI.write(b, 0, nRead);
					// if(nThreadID == 1)
					// FileUtility.log("nStartPos = " + nStartPos + ", nEndPos = " +
					// nEndPos);
				}
				FileUtility.log("Thread " + nThreadID + " is over!");
				bDownOver = true;
				// nPos = fileAccessI.write (b,0,nRead);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
	
	/**
	 * 文件下载
	 */
	private void filedownload(){
		while (nStartPos < nEndPos && !bStop) {
			try {
				InputStream input = new FileInputStream(file);
				input.skip(nStartPos);
				byte[] b = new byte[1024];
				int nRead;
				while ((nRead = input.read(b, 0, 1024)) > 0
						&& nStartPos < nEndPos && !bStop) {
					nStartPos += fileAccessI.write(b, 0, nRead);
				}
				FileUtility.log("Thread " + nThreadID + " is over!");
				bDownOver = true;
				input.close();
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}

	public void splitterStop() {
		bStop = true;
	}
}

package com.test;

/**
 * 简单的工具类
 * @author wzztestin
 *
 */
public class FileUtility {
	public FileUtility() {
	}

	public static void sleep(int nSecond) {
		try {
			Thread.sleep(nSecond);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static void log(String sMsg) {
		System.err.println(sMsg);
	}

	public static void log(int sMsg) {
		System.err.println(sMsg);
	}
}

package com.test;

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

	public String getFilepath() {
		return filepath;
	}

	public void setFilepath(String filepath) {
		this.filepath = filepath;
	}

	public boolean getFileflag() {
		return fileflag;
	}

	public void setFileflag(boolean fileflag) {
		this.fileflag = fileflag;
	}

	public DownFileInfoBean() {
		// default 5
		this("", "", "", 5,false,null);
	}

	public DownFileInfoBean(String sURL, String sPath, String sName, int nSpiltter,boolean fileflag,String filepath) {
		sSiteURL = sURL;
		sFilePath = sPath;
		sFileName = sName;
		this.nSplitter = nSpiltter;
		this.fileflag = fileflag;
		this.filepath = filepath;
	}

	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;

import java.io.*;

public class FileAccess implements Serializable {
	/**
	 * 
	 */
	private static final long serialVersionUID = -2518013155676212866L;
	RandomAccessFile oSavedFile;
	long nPos;

	public FileAccess() throws IOException {
		this("", 0);
	}

	public FileAccess(String sName, long nPos) throws IOException {
		oSavedFile = new RandomAccessFile(sName, "rw");
		this.nPos = nPos;
		oSavedFile.seek(nPos);
	}

	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;

/**
 * 测试类
 * @author wzztestin
 *
 */
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);*/
			DownFileInfoBean bean = new DownFileInfoBean(null, "D:\\temp",
					"F-A2_20130923.apk", 10,false,"D:\\fengye-MA2013091200102_F-A2_20130923.apk");
			DownFileFetch fileFetch = new DownFileFetch(bean);
			fileFetch.start();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

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