欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Java 服务器端支持断点续传的源代码【支持快车、迅雷】(仅支持 HTTP 协议)

程序员文章站 2022-07-12 16:48:23
...

  网上关于 Java 支持 HTTP 断点续传的文章不少,但关于 Java 服务器端支持 HTTP 断点续传的却比较少。
        本文是 Java 服务器端支持 HTTP 断点续传的源代码,支持快车、迅雷。
        本文使用一个简单的 Servlet 来作为支持断点续传的下载示例,在 Java Web 项目下部署好后,可以使用诸如 http://localhost/cds/http 的链接来调用 Servlet,进而被快车/迅雷监听进行下载。

        ArcSyncHttpDownloadServlet 源代码:

package com.defonds.cds.common;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.RandomAccessFile;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.defonds.cds.common.util.CommonUtil;

//HTTP 断点续传 demo(客户端测试工具:快车、迅雷)
public class ArcSyncHttpDownloadServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;
	final static Log log = LogFactory.getLog(ArcSyncHttpDownloadServlet.class);
	
	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws ServletException, IOException {
		this.doPost(req, resp);
	}
	
	@Override
	protected void doPost(HttpServletRequest request, HttpServletResponse response) {
		File downloadFile = new File("D:/defonds/book/pattern/SteveJobsZH.pdf");//要下载的文件
    	long fileLength = downloadFile.length();//记录文件大小
		long pastLength = 0;//记录已下载文件大小
		int rangeSwitch = 0;//0:从头开始的全文下载;1:从某字节开始的下载(bytes=27000-);2:从某字节开始到某字节结束的下载(bytes=27000-39000)
		long toLength = 0;//记录客户端需要下载的字节段的最后一个字节偏移量(比如bytes=27000-39000,则这个值是为39000)
		long contentLength = 0;//客户端请求的字节总量
		String rangeBytes = "";//记录客户端传来的形如“bytes=27000-”或者“bytes=27000-39000”的内容 
		RandomAccessFile raf = null;//负责读取数据
		OutputStream os = null;//写出数据
		OutputStream out = null;//缓冲
		byte b[] = new byte[1024];//暂存容器
		
		if (request.getHeader("Range") != null) {// 客户端请求的下载的文件块的开始字节
			response.setStatus(javax.servlet.http.HttpServletResponse.SC_PARTIAL_CONTENT);
			log.info("request.getHeader(\"Range\")=" + request.getHeader("Range"));
			rangeBytes = request.getHeader("Range").replaceAll("bytes=", "");
			if (rangeBytes.indexOf('-') == rangeBytes.length() - 1) {//bytes=969998336-
				rangeSwitch = 1;
				rangeBytes = rangeBytes.substring(0, rangeBytes.indexOf('-'));
				pastLength = Long.parseLong(rangeBytes.trim());
				contentLength = fileLength - pastLength + 1;//客户端请求的是 969998336 之后的字节
			} else {//bytes=1275856879-1275877358
				rangeSwitch = 2;
				String temp0 = rangeBytes.substring(0,rangeBytes.indexOf('-'));
				String temp2 = rangeBytes.substring(rangeBytes.indexOf('-') + 1, rangeBytes.length());
				pastLength = Long.parseLong(temp0.trim());//bytes=1275856879-1275877358,从第 1275856879 个字节开始下载
				toLength = Long.parseLong(temp2);//bytes=1275856879-1275877358,到第 1275877358 个字节结束
				contentLength = toLength - pastLength + 1;//客户端请求的是 1275856879-1275877358  之间的字节
			}
		} else {//从开始进行下载
			contentLength = fileLength;//客户端要求全文下载
		}
		
		/**
		 * 如果设设置了Content-Length,则客户端会自动进行多线程下载。如果不希望支持多线程,则不要设置这个参数。
		 * 响应的格式是:
		 * Content-Length: [文件的总大小] - [客户端请求的下载的文件块的开始字节]
		 * ServletActionContext.getResponse().setHeader("Content-Length",
		 * new Long(file.length() - p).toString());
		 */
		response.reset();//告诉客户端允许断点续传多线程连接下载,响应的格式是:Accept-Ranges: bytes
		response.setHeader("Accept-Ranges", "bytes");//如果是第一次下,还没有断点续传,状态是默认的 200,无需显式设置;响应的格式是:HTTP/1.1 200 OK
		if (pastLength != 0) {
			//不是从最开始下载,
			//响应的格式是:
			//Content-Range: bytes [文件块的开始字节]-[文件的总大小 - 1]/[文件的总大小]
			log.info("----------------------------不是从开始进行下载!服务器即将开始断点续传...");
			switch (rangeSwitch) {
				case 1 : {//针对 bytes=27000- 的请求
					String contentRange = new StringBuffer("bytes ").append(new Long(pastLength).toString()).append("-").append(new Long(fileLength - 1).toString()).append("/").append(new Long(fileLength).toString()).toString();
					response.setHeader("Content-Range", contentRange);
					break;
				}
				case 2 : {//针对 bytes=27000-39000 的请求
					String contentRange = rangeBytes + "/" + new Long(fileLength).toString();
					response.setHeader("Content-Range", contentRange);
					break;
				}
				default : {
					break;
				}
			}
		} else {
			//是从开始下载
			log.info("----------------------------是从开始进行下载!");
		}
		
		try {
	    	response.addHeader("Content-Disposition", "attachment; filename=\"" + downloadFile.getName() + "\"");
	    	response.setContentType( CommonUtil.setContentType(downloadFile.getName()));// set the MIME type.
	        response.addHeader("Content-Length", String.valueOf(contentLength));
			os = response.getOutputStream();
	    	out = new BufferedOutputStream(os);
	    	raf = new RandomAccessFile(downloadFile, "r");
	    	try {
	    		switch (rangeSwitch) {
		    		case 0 : {//普通下载,或者从头开始的下载
		    			//同1
		    		}
		    		case 1 : {//针对 bytes=27000- 的请求
		    			raf.seek(pastLength);//形如 bytes=969998336- 的客户端请求,跳过 969998336  个字节
				        int n = 0;
				        while ((n = raf.read(b, 0, 1024)) != -1) {
				        	out.write(b, 0, n);
				        }
		    			break;
		    		}
		    		case 2 : {//针对 bytes=27000-39000 的请求
		    			raf.seek(pastLength - 1);//形如 bytes=1275856879-1275877358 的客户端请求,找到第 1275856879 个字节
		    			int n = 0;
		    			long readLength = 0;//记录已读字节数
		    			while (readLength <= contentLength - 1024) {//大部分字节在这里读取
		    				n = raf.read(b, 0, 1024);
		    				readLength += 1024;
		    				out.write(b, 0, n);
		    			}
		    			if (readLength <= contentLength) {//余下的不足 1024 个字节在这里读取
		    				n = raf.read(b, 0, (int)(contentLength - readLength));
		    				out.write(b, 0, n);
		    			}
//		    			
//		    			raf.seek(pastLength);//形如 bytes=1275856879-1275877358 的客户端请求,找到第 1275856879 个字节
//			        	while (raf.getFilePointer() < toLength) {
//			        		out.write(raf.read());
//			        	}
		    			break;
		    		}
		    		default : {
		    			break;
		    		}
	    		}
		        out.flush();
	    	} catch(IOException ie) {
	    		/**
				 * 在写数据的时候,
				 * 对于 ClientAbortException 之类的异常,
				 * 是因为客户端取消了下载,而服务器端继续向浏览器写入数据时,
				 * 抛出这个异常,这个是正常的。
				 * 尤其是对于迅雷这种吸血的客户端软件,
				 * 明明已经有一个线程在读取 bytes=1275856879-1275877358,
				 * 如果短时间内没有读取完毕,迅雷会再启第二个、第三个。。。线程来读取相同的字节段,
				 * 直到有一个线程读取完毕,迅雷会 KILL 掉其他正在下载同一字节段的线程,
				 * 强行中止字节读出,造成服务器抛 ClientAbortException。
				 * 所以,我们忽略这种异常
				 */
				//ignore
	    	}
		} catch (Exception e) {
			log.error(e.getMessage(), e);
		} finally {
			if (out != null) {
				try {
					out.close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
			if (raf != null) {
				try {
					raf.close();
				} catch (IOException e) {
					log.error(e.getMessage(), e);
				}
			}
		}
	}
}

 

 ArcSyncHttpDownloadServlet 的 web.xml 配置清单:

		<!-- HTTP 断点续传 demo:127.0.0.1/cds/http -->
	<servlet>
		<servlet-name>httpServlet</servlet-name>
		<servlet-class>com.defonds.cds.common.ArcSyncHttpDownloadServlet</servlet-class>
	</servlet>	
	<servlet-mapping>
		<servlet-name>httpServlet</servlet-name>
		<url-pattern>/http</url-pattern>
	</servlet-mapping>	

  ArcSyncHttpDownloadServlet 调用到的工具类 com.defonds.cds.common.util.CommonUtil 源代码:

 

 

 

package com.defonds.cds.common.util;

public class CommonUtil {

	public static String setContentType(String returnFileName){
		String contentType = "application/octet-stream";
		if (returnFileName.lastIndexOf(".") < 0)
			return contentType;
		returnFileName = returnFileName.toLowerCase();
		returnFileName = returnFileName.substring(returnFileName.lastIndexOf(".")+1);
		
		if (returnFileName.equals("html") || returnFileName.equals("htm") || returnFileName.equals("shtml")){
			contentType = "text/html";
		} else if (returnFileName.equals("css")){
			contentType = "text/css";
		} else if (returnFileName.equals("xml")){
			contentType = "text/xml";
		} else if (returnFileName.equals("gif")){
			contentType = "image/gif";
		} else if (returnFileName.equals("jpeg") || returnFileName.equals("jpg")){
			contentType = "image/jpeg";
		} else if (returnFileName.equals("js")){
			contentType = "application/x-javascript";
		} else if (returnFileName.equals("atom")){
			contentType = "application/atom+xml";
		} else if (returnFileName.equals("rss")){
			contentType = "application/rss+xml";
		} else if (returnFileName.equals("mml")){
			contentType = "text/mathml"; 
		} else if (returnFileName.equals("txt")){
			contentType = "text/plain";
		} else if (returnFileName.equals("jad")){
			contentType = "text/vnd.sun.j2me.app-descriptor";
		} else if (returnFileName.equals("wml")){
			contentType = "text/vnd.wap.wml";
		} else if (returnFileName.equals("htc")){
			contentType = "text/x-component";
		} else if (returnFileName.equals("png")){
			contentType = "image/png";
		} else if (returnFileName.equals("tif") || returnFileName.equals("tiff")){
			contentType = "image/tiff";
		} else if (returnFileName.equals("wbmp")){
			contentType = "image/vnd.wap.wbmp";
		} else if (returnFileName.equals("ico")){
			contentType = "image/x-icon";
		} else if (returnFileName.equals("jng")){
			contentType = "image/x-jng";
		} else if (returnFileName.equals("bmp")){
			contentType = "image/x-ms-bmp";
		} else if (returnFileName.equals("svg")){
			contentType = "image/svg+xml";
		} else if (returnFileName.equals("jar") || returnFileName.equals("var") || returnFileName.equals("ear")){
			contentType = "application/java-archive";
		} else if (returnFileName.equals("doc")){
			contentType = "application/msword";
		} else if (returnFileName.equals("pdf")){
			contentType = "application/pdf";
		} else if (returnFileName.equals("rtf")){
			contentType = "application/rtf";
		} else if (returnFileName.equals("xls")){
			contentType = "application/vnd.ms-excel"; 
		} else if (returnFileName.equals("ppt")){
			contentType = "application/vnd.ms-powerpoint";
		} else if (returnFileName.equals("7z")){
			contentType = "application/x-7z-compressed";
		} else if (returnFileName.equals("rar")){
			contentType = "application/x-rar-compressed";
		} else if (returnFileName.equals("swf")){
			contentType = "application/x-shockwave-flash";
		} else if (returnFileName.equals("rpm")){
			contentType = "application/x-redhat-package-manager";
		} else if (returnFileName.equals("der") || returnFileName.equals("pem") || returnFileName.equals("crt")){
			contentType = "application/x-x509-ca-cert";
		} else if (returnFileName.equals("xhtml")){
			contentType = "application/xhtml+xml";
		} else if (returnFileName.equals("zip")){
			contentType = "application/zip";
		} else if (returnFileName.equals("mid") || returnFileName.equals("midi") || returnFileName.equals("kar")){
			contentType = "audio/midi";
		} else if (returnFileName.equals("mp3")){
			contentType = "audio/mpeg";
		} else if (returnFileName.equals("ogg")){
			contentType = "audio/ogg";
		} else if (returnFileName.equals("m4a")){
			contentType = "audio/x-m4a";
		} else if (returnFileName.equals("ra")){
			contentType = "audio/x-realaudio";
		} else if (returnFileName.equals("3gpp") || returnFileName.equals("3gp")){
			contentType = "video/3gpp";
		} else if (returnFileName.equals("mp4") ){
			contentType = "video/mp4";
		} else if (returnFileName.equals("mpeg") || returnFileName.equals("mpg") ){
			contentType = "video/mpeg";
		} else if (returnFileName.equals("mov")){
			contentType = "video/quicktime";
		} else if (returnFileName.equals("flv")){
			contentType = "video/x-flv";
		} else if (returnFileName.equals("m4v")){
			contentType = "video/x-m4v";
		} else if (returnFileName.equals("mng")){
			contentType = "video/x-mng";
		} else if (returnFileName.equals("asx") || returnFileName.equals("asf")){
			contentType = "video/x-ms-asf";
		} else if (returnFileName.equals("wmv")){
			contentType = "video/x-ms-wmv";
		} else if (returnFileName.equals("avi")){
			contentType = "video/x-msvideo";
		}
		
		return contentType;
	}
	
	
}

 

本文示例代码在客户端使用快车(FlashGet)3、迅雷7 进行断点续传的环境下测试 OK。

        本文仅为用来技术交流的演示 Demo,技术水平及知识面所限,其中错误和不当之处在所难免,敬请谅解。欢迎联系作者进行批评指正,技术交流 MSN:defonds(at)hotmail(dot)com,也可以给我发邮件,email:defonds(at)163(dot)com。

相关标签: 上传下载