import org.springframework.core.io.Resource;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;

import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutionException;

public class Client {
    public static void main(String[] args) throws IOException, ExecutionException, InterruptedException {

       merge(2,985);

    }

    static void multidownload() throws IOException, ExecutionException, InterruptedException {
        List<CompletableFuture<Integer> > list = new ArrayList<>();
        for (int i = 50; i < 986; i+=10) {
            final int index = i;
            CompletableFuture<Integer> future = CompletableFuture.supplyAsync(() -> {
                try {
                    download(index, Math.max(index + 10,985));
                } catch (IOException e) {
                    e.printStackTrace();
                }
                return 1;
            });
            list.add(future);
        }

        for (CompletableFuture<Integer> future : list) {
            future.get();
        }
    }
    static void merge(int start,int end) throws IOException {
        BufferedOutputStream outputStream = new BufferedOutputStream(new FileOutputStream("/Users/liyh/Movies/gjkc/one.ts"));
        byte[] buffer = new byte[1024];

        for (int i = start; i <= end; i ++) {
            BufferedInputStream inputStream = new BufferedInputStream(new FileInputStream("/Users/liyh/Movies/gjkc/gjkc" + String.format("%03d", i) +".ts"));
            int readcount;
            while ((readcount = inputStream.read(buffer)) > 0) {
                outputStream.write(buffer, 0, readcount);
            }
            inputStream.close();
        }
        outputStream.close();
        System.out.println("merge suc");
    }
    static RestTemplate restTemplate = new RestTemplate();
    static void download(int start,int end) throws IOException {

        for (int i = start; i < end; i++) {
            String num = String.format("%03d", i);
            String url = "https://m001.m1m2m3u8mp4.com/G0107/gjkc/gjkc" + num + ".ts";

            HttpHeaders headers = new HttpHeaders();
            HttpEntity<Resource> httpEntity = new HttpEntity<>(headers);
            ResponseEntity<byte[]> response = restTemplate.exchange(url, HttpMethod.GET,httpEntity, byte[].class);

//得到的是一个二进制的byte[]  将它转成流
            InputStream in = new ByteArrayInputStream(response.getBody());
            Path path = Paths.get("/Users/liyh/Movies/gjkc/gjkc"+num+".ts");
            Files.copy(in,path);
            System.out.println("suc download:" + url);
        }


    }
}