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);   
                }   
            }   
        }   
    }  

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/10495.html

(0)
上一篇 2021年7月19日
下一篇 2021年7月19日

相关推荐

发表回复

登录后才能评论