Java多线程断点下载

Java多线程断点下载

获取响应码：getResponseCode()   200 请求成功；

获取请求数据的长度：getContentLength()

RandomAccessFile设置文件长度：setLength(length)

获取部分文件conn.setRequestProperty("Range", "bytes="+startIndex+"-"+endIndex);，状态码正常为206

 

• 算法概述：

1.建连接

//分段下载

2. 获取网络文件长度length

3. 创建length长度的文件

4. 对下载进行分段，不同的线程对应自己的分段，分段之间不交叉，假设线程T 对应着文件段 t

5. 启动多个线程，每个线程请求并下载网络文件的对应段，如线程T下载网络文件的段t

//实现断点

6.每个线程创建一个对应的txt文件，保存本段已经下载的到文件的长度（相对于整个文件的起始位置）

7.每次本线程开始下载的时候先读取本线程对应的txt文件设置一下更新的起始位置。

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * 多线程断点下载示例
 * @author YUANYUAN
 *
 */
public class Demo {
    //下载所使用的线程数
    private static int threadCount=3;
    //当前活动的线程数
    private static int activeThread;

    public static void main(String[] args) throws Exception{
        //请求服务器的路径
        String path="http://192.168.2.114:8080/sqlite.exe";
        //构造URL地址
        URL url=new URL(path);
        //打开连接
        HttpURLConnection conn=(HttpURLConnection) url.openConnection();
        //设置请求超时的时间
        conn.setConnectTimeout(5000);
        //设置请求方式
        conn.setRequestMethod("GET");
        //获取相应码
        int code=conn.getResponseCode();
        if (code==200) {//请求成功
            //获取请求数据的长度
            int length=conn.getContentLength();
            //在客户端创建一个跟服务器文件大小相同的临时文件
            RandomAccessFile raf=new RandomAccessFile("setup.exe", "rwd");
            //指定临时文件的长度
            raf.setLength(length);
            raf.close();
            //假设3个线程去下载资源
            //平均每一个线程要下载的文件的大小
            int blockSize=length/threadCount;
            for (int threadId = 1; threadId <= threadCount; threadId++) {
                //当前线程下载数据的开始位置
                int startIndex=blockSize*(threadId-1);
                //当前线程下载数据的结束位置
                int endIndex=blockSize*threadId-1;
                //确定最后一个线程要下载数据的最大位置
                if (threadId==threadCount) {
                    endIndex=length;
                }
                //显示下载数据的区间
                System.out.println("线程【"+threadId+"】开始下载："+startIndex+"---->"+endIndex);
                //开启下载的子线程
                new DownloadThread(path, threadId, startIndex, endIndex).start();
                activeThread++;
                System.out.println("当前活动的线程数："+activeThread);
            }
            
        }else{//请求失败
            System.out.println("服务器异常，下载失败！");
        }
    }
    
    /**
     * 下载文件的子线程 每一个文件都下载对应的数据
     * @author YUANYUAN
     *
     */
    public static class DownloadThread extends Thread{
        private String path;
        private int threadId;
        private int startIndex;
        private int endIndex;
        
        /**
         * 构造方法
         * @param path 下载文件的路径
         * @param threadId 下载文件的线程
         * @param startIndex 下载文件开始的位置
         * @param endIndex 下载文件结束的位置
         */
        public DownloadThread(String path, int threadId, int startIndex,
                int endIndex) {
            this.path = path;
            this.threadId = threadId;
            this.startIndex = startIndex;
            this.endIndex = endIndex;
        }

        @Override
        public void run() {
            //构造URL地址
            try {
                
                File tempFile=new File(threadId+".txt");
                //检查记录是否存在,如果存在读取数据
                if (tempFile.exists()) {
                    FileInputStream fis=new FileInputStream(tempFile);
                    byte[] temp=new byte[1024];
                    int length=fis.read(temp);
                    //读取到已经下载的位置
                    int downloadNewIndex=Integer.parseInt(new String(temp, 0, length));
                    //设置重新开始下载的开始位置
                    startIndex=downloadNewIndex;
                    fis.close();
                    //显示真实下载数据的区间
                    System.out.println("线程【"+threadId+"】真实开始下载数据区间："+startIndex+"---->"+endIndex);
                }
                
                URL url = new URL(path);
                HttpURLConnection conn=(HttpURLConnection) url.openConnection();
                conn.setConnectTimeout(5000);
                conn.setRequestMethod("GET");
                //设置请求属性，请求部分资源
                conn.setRequestProperty("Range", "bytes="+startIndex+"-"+endIndex);
                int code=conn.getResponseCode();
                if (code==206) {//下载部分资源，正常返回的状态码为206
                    InputStream is=conn.getInputStream();//已经设置了请求的位置，所以返回的是对应的部分资源
                    //构建随机访问文件
                    RandomAccessFile raf=new RandomAccessFile("setup.exe", "rwd");
                    //设置 每一个线程随机写文件开始的位置
                    raf.seek(startIndex);
                    //开始写文件
                    int len=0;
                    byte[] buffer=new byte[1024];
                    //该线程已经下载数据的长度
                    int total=0;
                    
                    while((len=is.read(buffer))!=-1){//读取输入流
                        //记录当前线程已下载数据的长度
                        RandomAccessFile file=new RandomAccessFile(threadId+".txt","rwd");
                        raf.write(buffer,0,len);//写文件
                        total+=len;//更新该线程已下载数据的总长度
                        System.out.println("线程【"+threadId+"】已下载数据："+(total+startIndex));
                        //将已下载数据的位置记录写入到文件
                        file.write((startIndex+total+"").getBytes());
                        file.close();
                    }
                    is.close();
                    raf.close();
                    //提示下载完毕
                    System.out.println("线程【"+threadId+"】下载完毕");
                }
            } catch (Exception e) {
                e.printStackTrace();
                System.out.println("线程【"+threadId+"】下载出现异常！！");
            }finally{
                //活动的线程数减少
                activeThread--;
                if (activeThread==0) {
                    for (int i = 1; i <= threadCount; i++) {
                        File tempFile=new File(i+".txt");
                        tempFile.delete();
                    }
                    System.out.println("下载完毕，已清除全部临时文件");
                }
            }
            
        }
    }
}