java斷點續傳服務端代碼

此類核心續傳servlet
|||||||||||||||||||


import java.io.BufferedOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.RandomAccessFile;
import java.util.concurrent.ConcurrentSkipListMap;
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 org.springframework.util.Assert;

//HTTP 斷點續傳 demo（客戶端測試工具：快車、迅雷）
public class MyHttpDownloadServlet 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) {
           String name = request. getParameter("filename");

            try {
                 log.info( "請求下載的連接地址為：" + request.getRequestURL() + "?" + request.getQueryString());
                Assert. hasText(name);
           } catch (IllegalArgumentException e) {
                 log.error( "請求下載的文件名參數為空！" );
                 return;
           }
           String path = "D:/Install/" + name;
            log.info( "文件拼裝路徑：" + path);
           File downloadFile = new File(path);
            if (downloadFile.exists()) {
                 if (downloadFile.isFile()) {
                      if (downloadFile.length() > 0) {
                     } else {
                            log.info( "請求下載的文件是一個空文件" );
                            return;
                     }
                      if (!downloadFile.canRead()) {
                            log.info( "請求下載的文件不是一個可讀的文件" );
                            return;
                     } else {
                     }
                } else {
                      log.info( "請求下載的文件是一個文件夾" );
                      return;
                }
           } else {
                 log.info( "請求下載的文件不存在！" );
                 return;
           }

            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; // 客戶端請求的是 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; // 客戶端請求的是 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); // 形如 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);
                           }
                            break;
                     }
                      default: {
                            break;
                     }
                     }
                     out.flush();
                      log.info( "------------------------------下載結束" );
                } catch (IOException ie) {
                      /**
                      * 在寫數據的時候， 對於 ClientAbortException 之類的異常， 是因為客戶端取消了下載，而服務器端繼續向瀏覽器寫入數據時， 拋出這個異常，這個是正常的。 尤其是對於迅雷這種吸血的客戶端軟件， 明明已經有一個線程在讀取 bytes=1275856879-1275877358， 如果短時間內沒有讀取完畢，迅雷會再啟第二個、第三個。。。線程來讀取相同的字節段， 直到有一個線程讀取完畢，迅雷會 KILL 掉其他正在下載同一字節段的線程， 強行中止字節讀出，造成服務器拋 ClientAbortException。 所以，我們忽略這種異常
                      */
                      // ignore
                      log.info( "#提醒# 向客戶端傳輸時出現IO異常，但此異常是允許的的，有可能客戶端取消了下載，導致此異常，不用關心！" );
                }
           } 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);
                     }
                }
           }
     }
}

 

此類是根據擴展名返回ContentType類型
||||||||||||||||||||||||||||||||||||||


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("apk")) {
                contentType = "application/vnd.android.package-archive" ;
           } else if ( returnFileName.equals("sis")) {
                contentType = "application/vnd.symbian.install";
           } else if ( returnFileName.equals("sisx")) {
                contentType = "application/vnd.symbian.install";
           } else if ( returnFileName.equals("exe")) {
                contentType = "application/x-msdownload";
           } else if ( returnFileName.equals("msi")) {
                contentType = "application/x-msdownload";
           } 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;
     }

}

 

web.xml中servlet標簽
||||||||||||||||||||


               <servlet>
             <description></ description>
             <display-name> ArcSyncHttpDownloadServlet</display-name >
             <servlet-name> ArcSyncHttpDownloadServlet</servlet-name >
             <servlet-class> com.MyHttpDownloadServlet</servlet-class >
       </servlet>
       <servlet-mapping>
             <servlet-name> ArcSyncHttpDownloadServlet</servlet-name >
             <url-pattern> /http</url-pattern >
       </servlet-mapping>

 

此類為測試類
|||||||||||||


import java.io.BufferedInputStream;
import java.io.File;
import java.io.RandomAccessFile;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLDecoder;

public class TestDownload {

    /**
     * @param args
     */
    public static void main(String[] args) {
        HttpURLConnection httpURLConnection = null;
        URL url = null;
        BufferedInputStream bis = null;
        byte[] buf = new byte[10240];
        int size = 0;
        String fileName = "a.exe";
        String filePath = "C:/Documents and Settings/Administrator/桌面";
        StringremoteUrl= "http://10.10.129.99:8080/project/http?filename=a.exe" ;
        
        // 檢查本地文件
        RandomAccessFile rndFile = null;
        File file = new File(filePath + "\\" + fileName);
//        long remoteFileSize = getRemoteFileSzie(remoteUrl);
//        long nPos = 0;
        long localFileSzie =0;
        if (file.exists()) {                      
           localFileSzie = file.length();
        } else {
            // 建立文件
            try {
                file.createNewFile();
            } catch (Exception e) {
                e.printStackTrace();
            }          
        }
      
        // 下載文件
        try {
            url = new URL(remoteUrl);      
            httpURLConnection = (HttpURLConnection)url.openConnection();
            // 設置User-Agent
            httpURLConnection.setRequestProperty( "User-Agent", "Net");
            // 設置續傳開始
            System. out.println( "本地文件大小：" +localFileSzie);
            httpURLConnection.setRequestProperty( "Range", "bytes=" + localFileSzie  + "-" );
            // 獲取輸入流
            bis = new BufferedInputStream(httpURLConnection.getInputStream());  
           
            long remoteFileSize = httpURLConnection.getContentLength();
            System. out.println( "返回結果碼：" +httpURLConnection.getHeaderField("result"));
            //System.out.println("文件名："+httpURLConnection.getHeaderField("Content-Disposition"));
                 if (localFileSzie < remoteFileSize) {
                      if (localFileSzie==0) {
                           System. out.println( "文件不存在，開始下載..." );
                     } else{
                           System. out.println( "文件續傳..." );
                     }
                } else {
                     System. out.println( "文件存在，重新下載..." );
                     file.delete();
                      try {
                           file.createNewFile();
                     } catch (Exception e) {
                           e.printStackTrace();
                     }
                }
           
            rndFile = new RandomAccessFile(filePath + "\\" + fileName, "rw");
       
            rndFile.seek(localFileSzie);
            int i = 0;
            while ((size = bis.read(buf)) != -1) {
                //if (i > 500) break;              
                rndFile.write(buf, 0, size);
                i++;
            }
            System. out.println( "i=" + i);
            httpURLConnection.disconnect();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static long getRemoteFileSzie(String url) {
        long size = 0;
        try {
            HttpURLConnection httpUrl = (HttpURLConnection)( new URL(url)).openConnection();
            size = httpUrl.getContentLength();
            httpUrl.disconnect();          
        } catch (Exception e) {
            e.printStackTrace();
        }
        return size;
    }
}