Java斷點續傳服務器代碼

jopen 8年前發布 | 5K 次閱讀 Java

    /**

 * 執行下載 
 *  
 * @param request 
 * @param response 
 */  
protected void doDownload(HttpServletRequest request,  
        HttpServletResponse response) throws Exception {  
    File downloadFile = new File(resourceFullPath);// 要下載的文件  

    String extName = resourceFullPath.substring(resourceFullPath  
            .lastIndexOf('.'));  
    // request.getHeaderNames()  
    // 文件不存在  
    if (!downloadFile.exists()) {  
        response.sendError(HttpServletResponse.SC_NOT_FOUND);  
        return;  
    }  

    if (cacheExts.contains(extName)) {// 被緩存的文件格式  
        // 從緩存中讀取文件  
        downloadByCache(downloadFile, request, response);  
        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;// 緩沖  
    int bsize = 1024;// 緩沖區大小  
    byte b[] = new byte[bsize];// 暫存容器  

    if (request.getParameter("showheader") != null) {  
        Enumeration paramNames = request.getHeaderNames();  

        while (paramNames.hasMoreElements()) {  
            String name = paramNames.nextElement().toString();  
            if (name != null && name.length() > 0) {  
                String value = request.getHeader(name);  
                logger.info("************" + name + ":" + value);  
            }  
        }  
    }  
    String range = request.getHeader("Range");  
    // if(range == null)  
    // range = "bytes=0-";  
    int responseStatus = 206;  
    if (range != null && range.trim().length() > 0 && !"null".equals(range)) {// 客戶端請求的下載的文件塊的開始字節  
        responseStatus = javax.servlet.http.HttpServletResponse.SC_PARTIAL_CONTENT;  
        logger.info("request.getHeader(\"Range\")=" + range);  
        rangeBytes = range.replaceAll("bytes=", "");  
        if (rangeBytes.endsWith("-")) {// bytes=969998336-  
            rangeSwitch = 1;  
            rangeBytes = rangeBytes.substring(0, rangeBytes.indexOf('-'));  
            pastLength = Long.parseLong(rangeBytes.trim());  
            contentLength = fileLength - pastLength;// 客戶端請求的是  
                                                    // 969998336之后的字節(包括bytes下標索引為969998336的字節)  
        } else {// bytes=1275856879-1275877358  
            rangeSwitch = 2;  
            String temp0 = rangeBytes.substring(0, rangeBytes.indexOf('-'));  
            String temp2 = rangeBytes.substring(  
                    rangeBytes.indexOf('-') + 1, rangeBytes.length());  
            // bytes=1275856879-1275877358,從第1275856879個字節開始下載  
            pastLength = Long.parseLong(temp0.trim());  
            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  

    // response.addHeader("Cache-Control", "max-age=1296000");  
    // response.addHeader("Expires", "Fri, 12 Oct 2012 03:43:01 GMT");  
    // response.addHeader("Last-Modified", "Tue, 31 Jul 2012 03:58:36 GMT");  
    // response.addHeader("Connection", "keep-alive");  
    // response.addHeader("ETag", downloadFile.getName() + "-" +  
    // downloadFile.lastModified());  
    // response.addHeader("Last-Modified", "Thu, 27 Sep 2012 05:24:44 GMT");  

    if (rangeSwitch != 0) {  
        response.setStatus(responseStatus);  
        // 不是從最開始下載,斷點下載響應號為206  
        // 響應的格式是:  
        // Content-Range: bytes [文件塊的開始字節]-[文件的總大小 - 1]/[文件的總大小]  
        logger.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 = range.replace("=", " ") + "/"  
                    + new Long(fileLength).toString();  
            response.setHeader("Content-Range", contentRange);  
            break;  
        }  
        default: {  
            break;  
        }  
        }  
    } else {  
        String contentRange = new StringBuffer("bytes ").append("0-")  
                .append(fileLength - 1).append("/").append(fileLength)  
                .toString();  
        response.setHeader("Content-Range", contentRange);  
        // 是從開始下載  
        logger.info("----------------------------是從開始到最后完整下載!");  
    }  

    try {  
        // /////////////////////////設置文件Content-Type///////////////////////////  

        String contentType = contentTypeMap.get(extName);  

        if (contentType != null) {  
            // /logger.debug("設置內容類型:" + contentType);  
            response.setContentType(contentType);// set the MIME type.  
        } else {  
            response.setContentType("application/octet-stream");// set the  
                                                                // MIME  
                                                                // type.  
        }  
        // /////////////////////////設置文件下載名稱Content-Disposition///////////////////////////  

        // 如果下載的文件需要重命名(因為物理文件存放的名稱都是數字串,已在上傳的時候被重命名過)  
        if (StringUtils.isNotBlank(downloadFileName)) {  
            String fileName = downloadFileName + extName;  
            fileName = encodingFileName(StringUtils.trim(fileName), request);  
            response.setHeader("Content-Disposition",  
                    "attachment; filename=" + fileName);  
        }  

        response.setHeader("Content-Length", String.valueOf(contentLength));  
        os = response.getOutputStream();  
        out = new BufferedOutputStream(os);  
        raf = new RandomAccessFile(downloadFile, "r");  
        try {  
            long outLength = 0;// 實際輸出字節數  
            switch (rangeSwitch) {  
            case 0: {// 普通下載,或者從頭開始的下載  
                // 同1,沒有break  
            }  
            case 1: {// 針對 bytes=27000- 的請求  
                raf.seek(pastLength);// 形如 bytes=969998336- 的客戶端請求,跳過  
                                        // 969998336 個字節  
                int n = 0;  
                while ((n = raf.read(b)) != -1) {  
                    out.write(b, 0, n);  
                    outLength += n;  
                }  
                // while ((n = raf.read(b, 0, 1024)) != -1) {  
                // out.write(b, 0, n);  
                // }  
                break;  
            }  
            case 2: {  
                // 針對 bytes=27000-39000 的請求,從27000開始寫數據  
                raf.seek(pastLength);  
                int n = 0;  
                long readLength = 0;// 記錄已讀字節數  
                while (readLength <= contentLength - bsize) {// 大部分字節在這里讀取  
                    n = raf.read(b);  
                    readLength += n;  
                    out.write(b, 0, n);  
                    outLength += n;  
                }  
                if (readLength <= contentLength) {// 余下的不足 1024 個字節在這里讀取  
                    n = raf.read(b, 0, (int) (contentLength - readLength));  
                    out.write(b, 0, n);  
                    outLength += n;  
                }  
                break;  
            }  
            default: {  
                break;  
            }  
            }  
            logger.info("Content-Length為:" + contentLength + ";實際輸出字節數:" + outLength);  
            out.flush();  
        } catch (IOException ie) {  
            /** 
             * 在寫數據的時候, 對于 ClientAbortException 之類的異常, 
             * 是因為客戶端取消了下載,而服務器端繼續向瀏覽器寫入數據時, 拋出這個異常,這個是正常的。 
             * 尤其是對于迅雷這種吸血的客戶端軟件, 明明已經有一個線程在讀取 bytes=1275856879-1275877358, 
             * 如果短時間內沒有讀取完畢,迅雷會再啟第二個、第三個。。。線程來讀取相同的字節段, 直到有一個線程讀取完畢,迅雷會 KILL 
             * 掉其他正在下載同一字節段的線程, 強行中止字節讀出,造成服務器拋 ClientAbortException。 
             * 所以,我們忽略這種異常 
             */  
            // ignore  
        }  
    } catch (Exception e) {  
        logger.error(e.getMessage(), e);  
    } finally {  
        if (out != null) {  
            try {  
                out.close();  
            } catch (IOException e) {  
                logger.error(e.getMessage(), e);  
            }  
        }  
        if (raf != null) {  
            try {  
                raf.close();  
            } catch (IOException e) {  
                logger.error(e.getMessage(), e);  
            }  
        }  
    }  
}  </pre> 


 本文由用戶 jopen 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!