java实现文件的断点续传

所谓文件的断点续传，就是一个线程传输文件，另一个线程控制传输标识，以达到暂停文件效果、恢复文件上传的效果。

本demo使用最基本的线程之间的通信来实现一个简单的断点续传。

package com.test;
 
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
 
import java.io.*;
import java.nio.ByteBuffer;
import java.util.Scanner;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
 
public class Test {
    public static void main(String[] args) throws FileNotFoundException {
 
        File file = new File("d:\\12bb.eif");
        FileInputStream fis = new FileInputStream(file);
        FileOutputStream fos = new FileOutputStream("d:\\ddxc\\bqb.eif");
 
        BufferedInputStream bis = new BufferedInputStream(fis);
        BufferedOutputStream bos = new BufferedOutputStream(fos);
        String fileName = file.getName();
 
        ExecutorService pool = Executors.newFixedThreadPool(30);//创建线程池，可自行定义
        Task task = new Task(true, bis, bos, fileName);
        pool.execute(task);
        Scanner scanner = new Scanner(System.in);//TODO 输入文字并敲回车，开启一个新线程控制文件的上传、暂停（在上传和暂停间来回切换）
        while (scanner.hasNextLine()) {
            String s = scanner.nextLine();
            pool.execute(task);
        }
 
 
    }
 
}
 
@Data
@AllArgsConstructor
@NoArgsConstructor
class Task implements Runnable {
 
    private volatile Boolean flag = true;//控制文件继续、暂停上传的标识，true为继续上传，false暂停上传
    private BufferedInputStream bis;
    private BufferedOutputStream bos;
    private String flagName;//监听器，一般设置为用户编号+文件名，防止不同用户上传相同为文件。但本demo只以文件名作为监听器
 
    @Override
    public void run() {
        String threadName = Thread.currentThread().getName();
        //TODO 对线程的后缀名进行判断，如果是不是第一个线程，则判断为控制线程
        if (!threadName.endsWith("1")) {
            this.flag = !flag;
        }
        synchronized (flagName) {
            if (!threadName.endsWith("1")) {
 
                System.out.println(threadName + "控制线程开始运行");
 
                if (flag) {
                    flagName.notifyAll();
                    System.out.println("继续传输文件");
                }
 
            } else {
 
                try {
 
                    System.out.println(threadName + "传输线程开始运行");
                    int len = 0;
                    byte[] cbuf = new byte[1024];
                    while ((len = bis.read(cbuf, 0, cbuf.length)) != -1) {
                        if (!flag) {
                            System.out.println("文件通道阻塞中");
                            flagName.wait();
                        }
                        bos.write(cbuf, 0, len);
                        bos.flush();
 
                        Thread.sleep(10);
                    }
                    System.out.println("文件传输完毕");
                    bis.close();
                    bos.close();
                    System.exit(1);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
            System.out.println(threadName + "控制线程运行完毕");
        }
    }
}

在文件通道阻塞时，打开文件属性，观察文件大小是否停止增加。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持我们。