Java斷點續傳服務器代碼

jopen 11年前發布 | 15K 次閱讀 Java 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><br />
 本文由用戶 jopen 自行上傳分享,僅供網友學習交流。所有權歸原作者,若您的權利被侵害,請聯系管理員。
 轉載本站原創文章,請注明出處,并保留原始鏈接、圖片水印。
 本站是一個以用戶分享為主的開源技術平臺,歡迎各類分享!