必威体育Betway必威体育官网
当前位置:首页 > IT技术

Java采用Http方式实现大文件下载

时间:2019-08-10 17:43:13来源:IT技术作者:seo实验室小编阅读:86次「手机版」
 

java下载文件

java采用Http方式实现大文件下载

java实现大文件下载,基于http方式,控件神马的就不说了。

思路:下载文件无非要读取文件然后写文件,主要这两个步骤,主要难点:

1.读文件,就是硬盘内存的过程,由于jdk内存限制,不能读的太大。

2.写文件,就是响应到浏览器端的过程,http协议是短链接,如果写文件太慢,时间过久,会造成浏览器死掉。

知识点:

1.org.apache.http.impl.client.Closeablehttpclient 模拟httpClient客户端发送http请求,可以控制到请求文件的字节位置。

2.BufferedInputStream都熟悉,用它接受请求来的流信息缓存。

3.RandomAccessFile文件随机类,可以向文件写入指定位置的流信息。

基于以上信息,我的实现思路就是首先判断下载文件大小,配合多线程分割定制http请求数量和请求内容,响应到写入到RandomAccessFile指定位置中。在俗点就是大的http分割成一个个小的http请求,相当于每次请求一个网页

废话不说,上代码


DownLoadManagerTest类:

package xxxx;

import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.concurrent.countdownlatch;
import org.apache.commons.lang.exception.ExceptionUtils;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.impl.conn.PoolingHttpClientConnectionManager;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.Springframework.beans.factory.annotation.Autowired;
import org.springframework.core.task.TaskExecutor;
import org.springframework.test.context.Activeprofiles;
import org.springframework.test.context.Contextconfiguration;
import org.springframework.test.context.junit4.AbstracttransactionalJUnit4SpringContextTests;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

/**
 * 
 * 文件下载管理类
 */

@RunWith(SpringJUnit4ClassRunner.class)
@ActiveProfiles("test")
@ContextConfiguration(locations={"classpath:test/APPlicationContext.xml"})
public class DownLoadManagerTest extends AbstractTransactionalJUnit4SpringContextTests{

private static final Logger LOGGER = LoggerFactory.getLogger(DownLoadManagerTest.class);

/**
 * 
 * 每个线程下载的字节数
 */

private long unitSize = 1000 * 1024;

@Autowired
private TaskExecutor taskExecutor;

private CloseableHttpClient httpClient;

private Long starttimes;

private Long endtimes;

@Before
public void setUp() throws Exception
{
    starttimes = system.currenttimemillis();
    System.out.println("测试开始....");
}

@After
public void tearDown() throws Exception
{
    endtimes = System.currentTimeMillis();
    System.out.println("测试结束!!");
    System.out.println("********************");
    System.out.println("下载总耗时:"+(endtimes-starttimes)/1000+"s");
    System.out.println("********************");
}

public DownLoadManagerTest() {

    System.out.println("初始化测试类....");
    PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager();
    cm.setMaxTotal(100);
    httpClient = HttpClients.custom().setConnectionManager(cm).build();

}

/**
 * 
 * 启动多个线程下载文件
 */
@Test
public void  doDownload() throws IOException {

    String remoteFileUrl="http://{host}:{port}/{project}/xx.xml";
    String localPath="E://test//";

    String fileName = new URL(remoteFileUrl).getFile();

    System.out.println("远程文件名称:"+fileName);
    fileName = fileName.substring(fileName.lastindexof("/") + 1,
            fileName.length()).replace("%20", " ");
    System.out.println("本地文件名称:"+fileName);
    long fileSize = this.getRemoteFileSize(remoteFileUrl);

    this.createFile(localPath+System.currentTimeMillis()+fileName, fileSize);

    Long threadCount = (fileSize/unitSize)+(fileSize % unitSize!=0?1:0);
    long offset = 0;

    CountDownLatch end = new CountDownLatch(threadCount.intValue());

    if (fileSize <= unitSize) {// 如果远程文件尺寸小于等于unitSize

        DownloadThreadTest downloadThread = new DownloadThreadTest(remoteFileUrl,

                localPath+fileName, offset, fileSize,end,httpClient);

        taskExecutor.execute(downloadThread);

    } else {// 如果远程文件尺寸大于unitSize

        for (int i = 1; i < threadCount; i++) {

            DownloadThreadTest downloadThread = new DownloadThreadTest(

            remoteFileUrl, localPath+fileName, offset, unitSize,end,httpClient);

            taskExecutor.execute(downloadThread);

            offset = offset + unitSize;

        }

        if (fileSize % unitSize != 0) {// 如果不能整除,则需要再创建一个线程下载剩余字节

            DownloadThreadTest downloadThread = new DownloadThreadTest(remoteFileUrl, localPath+fileName, offset, fileSize - unitSize * (threadCount-1),end,httpClient);
            taskExecutor.execute(downloadThread);
        }

    }
    try {
        end.await();
    } catch (InterruptedException e) {
        LOGGER.ERROR("DownLoadManager exception msg:{}",ExceptionUtils.getFullStackTrace(e));
        e.printstacktrace();
    }

// System.out.println(“111111”);

LOGGER.debug(“下载完成!{} “,localPath+fileName);

//return localPath+fileName;

}

/**
 * 
 * 获取远程文件尺寸
 */

private long getRemoteFileSize(String remoteFileUrl) throws IOException {

    long fileSize = 0;

    HttpURLConnection httpConnection = (HttpURLConnection) new URL(

    remoteFileUrl).openConnection();

    httpConnection.setrequestMethod("HEAD");

    int responseCode = httpConnection.getResponseCode();

    if (responseCode >= 400) {

        LOGGER.debug("Web服务器响应错误!");

        return 0;

    }

    String sHeader;

    for (int i = 1;; i++) {

        sHeader = httpConnection.getHeaderfieldKey(i);

        if (sHeader != null && sHeader.equals("content-Length")) {

            System.out.println("文件大小ContentLength:"
                    + httpConnection.getContentLength());

            fileSize = Long.parseLong(httpConnection
                    .getHeaderField(sHeader));

            break;

        }

    }

    return fileSize;

}

/**
 * 
 * 创建指定大小的文件
 */

private void createFile(String fileName, long fileSize) throws IOException {

    File newFile = new File(fileName);

    RandomAccessFile raf = new RandomAccessFile(newFile, "rw");

    raf.setLength(fileSize);

    raf.close();

}


public TaskExecutor getTaskExecutor() {
    return taskExecutor;
}

public void setTaskExecutor(TaskExecutor taskExecutor) {
    this.taskExecutor = taskExecutor;
}

}

DownloadThreadTest类:

package xxxx;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.util.concurrent.CountDownLatch;
import org.apache.commons.lang.exception.ExceptionUtils;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.protocol.BasicHttpContext;
import org.apache.http.protocol.HttpContext;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 
 * 负责文件下载的类
 */

public class DownloadThreadTest extends Thread {

private static final Logger LOGGER = LoggerFactory
        .getLogger(DownloadThreadTest.class);

/**
 * 
 * 待下载的文件
 */

private String url = null;

/**
 * 
 * 本地文件名
 */

private String fileName = null;

/**
 * 
 * 偏移量
 */

private long offset = 0;

/**
 * 
 * 分配给本线程的下载字节数
 */

private long length = 0;

private CountDownLatch end;

private CloseableHttpClient httpClient;

private HttpContext context;

/**
 * 
 * @param url
 *            下载文件地址
 * 
 * @param fileName
 *            另存文件名
 * 
 * @param offset
 *            本线程下载偏移量
 * 
 * @param length
 *            本线程下载长度
 * 
 * 
 * 
 * @author Angus.wang
 * 
 * */

public DownloadThreadTest(String url, String file, long offset, long length,
        CountDownLatch end, CloseableHttpClient httpClient) {

    this.url = url;

    this.fileName = file;

    this.offset = offset;

    this.length = length;

    this.end = end;

    this.httpClient = httpClient;

    this.context = new BasicHttpContext();

    LOGGER.debug("偏移量=" + offset + ";字节数=" + length);

}

public void run() {

    try {

        HttpGet httpGet = new HttpGet(this.url);
        httpGet.addHeader("Range", "bytes=" + this.offset + "-"
                + (this.offset + this.length - 1));
        CloseableHttpResponse response = httpClient.execute(httpGet,
                context);
        ;
        BufferedInputStream bis = new BufferedInputStream(response
                .getEntity().getContent());

        byte[] buff = new byte[1024];

        int bytesRead;

        File newFile = new File(fileName);

        RandomAccessFile raf = new RandomAccessFile(newFile, "rw");

        while ((bytesRead = bis.read(buff, 0, buff.length)) != -1) {
            raf.seek(this.offset);
            raf.write(buff, 0, bytesRead);
            this.offset = this.offset + bytesRead;
        }
        raf.close();
        bis.close();
    } catch (ClientProtocolException e) {
        LOGGER.error("DownloadThread exception msg:{}",ExceptionUtils.getFullStackTrace(e));
    } catch (IOException e) {
        LOGGER.error("DownloadThread exception msg:{}",ExceptionUtils.getFullStackTrace(e));
    } finally {
        end.countDown();
        LOGGER.info(end.getCount() + " is go on!");
        System.out.println(end.getCount() + " is go on!");
    }
}

}

application.xml

<bean id="taskExecutor" 

class="org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor">

<!-- 线程池活跃的线程数 -->

<property name="corePoolSize" value="5" />

<!-- 线程池最大活跃的线程数 -->

<property name="maxPoolSize" value="10" />

<!-- 队列的最大容量 -->

<property name="queueCapacity" value="600" />

</bean>

<bean id="downLoadManager"

class="xx.DownLoadManagerTest">

<property name="taskExecutor" ref="taskExecutor" />

</bean>

测试运行,500M,我这网速得半个小时左右。要想下载更大的文件,只要jdk内存够大,就无限更改队列最大容量吧。

相关阅读

女人在家赚钱的项目:分享2018最适合女人的十种网络赚钱

现在的女人压力真大,出去上班就照顾不到家庭;回归家庭又赚不到钱,得不到大家的理解,一旦不上班了,马上会有人说与社会脱节,其实说与社会

【思维方式】如何判断一个产品是“好”产品?

很多场合下,产品经理都会被问到一个经典问题&#8211;什么样的产品才算是个好产品?今天我来尝试做下回答:个人觉得关于“好,坏”的判断

什么是javabean及其用法

一、什么是JavaBean JavaBean是一个遵循特定写法的Java类,它通常具有如下特点: 这个Java类必须具有一个无参的构造函数 属性必须

133道Java面试题及答案(面试必看)

Java 面试随着时间的改变而改变。在过去的日子里,当你知道 String 和 StringBuilder 的区别就能让你直接进入第二轮面试,但是现在问

斗鱼视频下载

#coding:utf-8 import urllib import urllib2 import requests import random import time import sys my_headers={ "Us

分享到:

栏目导航

推荐阅读

热门阅读