java核心学习(二十九) 基本网络支持

package net;

import java.io.File;
import java.io.FileReader;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.channels.FileChannel;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;


public class DownUtil {
    private String path;  //下载资源的路径
    private String targetFile;  //
    private int threadNum;
    private DownThread[] threads;
    private int filesize;
    private final String infoFileSuffix = "INI.txt";

    public DownUtil(String path, String targetFile, int threadNum) {
        this.path = path;
        this.targetFile = targetFile;
        this.threadNum = threadNum;
    }

    public DownUtil(String path, String targetFile) {
        this.path = path;
        this.targetFile = targetFile;
        this.threadNum = Runtime.getRuntime().availableProcessors();//默认开启当前机器CPU数量的线程
    }
    public void download() throws Exception{
        //建立http请求取得文件大小
        URL url = new URL(path);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setRequestMethod("GET");
        conn.setRequestProperty("Accept","*/*");
        conn.setRequestProperty("Accept-Language","zh-CN");
        conn.setRequestProperty("Charset","UTF-8");
        conn.setRequestProperty("Connection","Keep-Alive");
        filesize = conn.getContentLength(); //取得文件大小
        conn.disconnect();
        //将下载任务分配给每个线程
        int currentPartSize = filesize / threadNum + 1; //每个线程负责的文件块的大小
        Path path = Paths.get(targetFile+infoFileSuffix);
        if(Files.exists(path)){

            //读取文件内容继续下载
            List<String> allLine = Files.readAllLines(path, Charset.forName("UTF-8"));
            for(int i = 0 ; i < threadNum ; i++){
                int startPos = i * currentPartSize;
                RandomAccessFile currentPart = new RandomAccessFile(targetFile,"rw");
                currentPart.seek(startPos);
                threads[i] = new DownThread(startPos,currentPartSize,currentPart,Integer.parseInt(allLine.get(i).toString()));
                threads[i].start();
            }

        }else {
            RandomAccessFile file = new RandomAccessFile(targetFile,"rw");
            file.setLength(filesize);
            file.close();
            List<String> infoStrings = new ArrayList<>();
            for(int i = 0 ; i < threadNum ; i++){
                infoStrings.add("0");
            }
            Files.write(path,infoStrings,Charset.forName("UTF-8"));
            for(int i = 0 ; i < threadNum ; i++){
                int startPos = i * currentPartSize;
                RandomAccessFile currentPart = new RandomAccessFile(targetFile,"rw");
                currentPart.seek(startPos);
                threads[i] = new DownThread(startPos,currentPartSize,currentPart);
                threads[i].start();
            }
            //重新写入文件
        }
    }

    private class DownThread extends Thread{

        private int startPos; //当前线程开始下载的位置
        private int currentPartSize;    //当前线程负责下载的大小
        private RandomAccessFile currntPart;  //当前线程需要下载的文件类
        public int length; //已下载的字节数，如果要实现断点下载，可以吧这个数据存在硬盘上，即存在文件里
        private final String acceptString = "*/*";
        private final String charSetString = "UTF-8";
        private final String acceptLanguageString = "zh-CN";


        public DownThread(int startPos, int currentPartSize, RandomAccessFile currntPart,int length) {
            this.startPos = startPos;
            this.currentPartSize = currentPartSize;
            this.currntPart = currntPart;
            this.length = length;
        }

        public DownThread(int startPos, int currentPartSize, RandomAccessFile currrntPart){
            this.startPos = startPos;
            this.currentPartSize = currentPartSize;
            this.currntPart = currrntPart;
            this.length = 0;
        }

        @Override
        public void run() {
            try {
                URL url = new URL(path);
                HttpURLConnection conn = (HttpURLConnection) url.openConnection();
                conn.setConnectTimeout(5000);
                conn.setRequestMethod("GET");
                conn.setRequestProperty("Accept",acceptString);
                conn.setRequestProperty("Charset",charSetString);
                conn.setRequestProperty("Accept-Language",acceptLanguageString);
                conn.setRequestProperty("Range","bytes=" + (startPos + length) + "-" + (startPos + currentPartSize - 1));
                InputStream inputStream = conn.getInputStream();

                byte[] buffer = new byte[1024];
                int hasRead = 0; //一次循环读取的字节数
                while (length < currentPartSize && (hasRead = inputStream.read(buffer)) != -1){
                    currntPart.write(buffer,0,hasRead);
                    length += hasRead;
                }
                currntPart.close();
                inputStream.close();
            }catch (Exception e){
                e.printStackTrace();
            }
        }
    }
}