多线程断点下载

时间:2022-04-24
本文章向大家介绍多线程断点下载,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

多线程断点下载

  1. 多线程下载
public class MultiThreadDownloader {
	private URL url;        // 目标地址
	private File file;        // 本地文件
	private long threadLen;    // 每个线程下载多少
	private static final int THREAD_AMOUNT = 3;                // 线程数
	private static final String DIR_PATH = "F:/Download";    // 下载目录
	public MultiThreadDownloader(String address) throws IOException {    
		url = new URL(address);                                                            // 记住下载地址
		file = new File(DIR_PATH, address.substring(address.lastIndexOf("/") + 1));        // 截取地址中的文件名, 创建本地文件
		// 创建一个临时文件路径
	}
	public void download() throws IOException {
		HttpURLConnection conn = (HttpURLConnection) url.openConnection();
		conn.setConnectTimeout(3000);
		long totalLen = conn.getContentLength();                    // 获取文件总长度
		threadLen = (totalLen + THREAD_AMOUNT - 1) / THREAD_AMOUNT;    // 计算每个线程要下载的长度
		// 总长度 如果能整除 线程数, 每条线程下载的长度就是 总长度 / 线程数
		// 总长度 如果不能整除 线程数, 那么每条线程下载长度就是 总长度 / 线程数 + 1
		RandomAccessFile raf = new RandomAccessFile(file, "rw");    // 在本地创建一个和服务端大小相同的文件
		raf.setLength(totalLen);                                    // 设置文件的大小, 写入了若干个0
		raf.close();
		// 创建临时文件
		for (int i = 0; i < THREAD_AMOUNT; i++)        // 按照线程数循环
			new DownloadThread(i).start();            // 开启线程, 每个线程将会下载一部分数据到本地文件中
	}
	private class DownloadThread extends Thread {
		private int id;     // 用来标记当前线程是下载任务中的第几个线程
		public DownloadThread(int id) {
			this.id = id;
		}
		public void run() {
			// 从临时文件读取当前线程已完成的进度
			long start = id * threadLen;                    // 起始位置
			long end = id * threadLen + threadLen - 1;        // 结束位置
			System.out.println("线程" + id + ": " + start + "-" + end);
			try {
				HttpURLConnection conn = (HttpURLConnection) url.openConnection();
				conn.setConnectTimeout(3000);
				conn.setRequestProperty("Range", "bytes=" + start + "-" + end);    // 设置当前线程下载的范围(start和end都包含)
				InputStream in = conn.getInputStream();                            // 获取连接的输入流, 用来读取服务端数据
				RandomAccessFile raf = new RandomAccessFile(file, "rw");        // 随机读写文件, 用来向本地文件写出
				raf.seek(start);                                                // 设置保存数据的位置
				byte[] buffer = new byte[1024 * 100];    // 每次拷贝100KB
				int len;
				while ((len = in.read(buffer)) != -1) {
					raf.write(buffer, 0, len);            // 从服务端读取数据, 写到本地文件
					// 存储当前下载进度到临时文件
				}
				raf.close();
				System.out.println("线程" + id + "下载完毕");
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	public static void main(String[] args) throws IOException {
		new MultiThreadDownloader("http://192.168.1.240:8080/14.Web/android-sdk_r17-windows.zip").download();
	}
}
  1. 断点下载
public class BreakpointDownloader {
	private static final String DIR_PATH = "F:/Download";    // 下载目录
	private static final int THREAD_AMOUNT = 3;                // 总线程数
	private URL url;            // 目标下载地址
	private File dataFile;        // 本地文件
	private File tempFile;        // 用来存储每个线程下载的进度的临时文件
	private long threadLen;        // 每个线程要下载的长度
	private long totalFinish;    // 总共完成了多少
	private long totalLen;        // 服务端文件总长度
	private long begin;            // 用来记录开始下载时的时间
	public BreakpointDownloader(String address) throws IOException {    
		url = new URL(address);                                                            // 记住下载地址
		dataFile = new File(DIR_PATH, address.substring(address.lastIndexOf("/") + 1));    // 截取地址中的文件名, 创建本地文件
		tempFile = new File(dataFile.getAbsolutePath() + ".temp");                        // 在本地文件所在文件夹中创建临时文件
	}
	public void download() throws IOException {
		HttpURLConnection conn = (HttpURLConnection) url.openConnection();
		conn.setConnectTimeout(3000);
		totalLen = conn.getContentLength();                                    // 获取服务端发送过来的文件长度
		threadLen = (totalLen + THREAD_AMOUNT - 1) / THREAD_AMOUNT;            // 计算每个线程要下载的长度
		if (!dataFile.exists()) {                                            // 如果本地文件不存在
			RandomAccessFile raf = new RandomAccessFile(dataFile, "rws");    // 在本地创建文件
			raf.setLength(totalLen);                                        // 设置文件的大小和服务端相同
			raf.close();
		}
		if (!tempFile.exists()) {                                            // 如果临时文件不存在
			RandomAccessFile raf = new RandomAccessFile(tempFile, "rws");    // 创建临时文件, 用来记录每个线程已下载多少
			for (int i = 0; i < THREAD_AMOUNT; i++)                            // 按照线程数循环
				raf.writeLong(0);                                            // 写入每个线程的开始位置(都是从0开始)
			raf.close();
		}
		for (int i = 0; i < THREAD_AMOUNT; i++)    // 按照线程数循环
			new DownloadThread(i).start();        // 开启线程, 每个线程将会下载一部分数据到本地文件中
		begin = System.currentTimeMillis();        // 记录开始时间
	}
	private class DownloadThread extends Thread {
		private int id;     // 用来标记当前线程是下载任务中的第几个线程
		public DownloadThread(int id) {
			this.id = id;
		}
		public void run() {
			try {
				RandomAccessFile tempRaf = new RandomAccessFile(tempFile, "rws");        // 用来记录下载进度的临时文件
				tempRaf.seek(id * 8);                        // 将指针移动到当前线程的位置(每个线程写1个long值, 占8字节)
				long threadFinish = tempRaf.readLong();        // 读取当前线程已完成了多少
				synchronized(BreakpointDownloader.this) {    // 多个下载线程之间同步
					totalFinish += threadFinish;            // 统计所有线程总共完成了多少
				}
				long start = id * threadLen + threadFinish;        // 计算当前线程的起始位置
				long end = id * threadLen + threadLen - 1;        // 计算当前线程的结束位置
				System.out.println("线程" + id + ": " + start + "-" + end);
				HttpURLConnection conn = (HttpURLConnection) url.openConnection();
				conn.setConnectTimeout(3000);
				conn.setRequestProperty("Range", "bytes=" + start + "-" + end);        // 设置当前线程下载的范围
				InputStream in = conn.getInputStream();                                // 获取连接的输入流
				RandomAccessFile dataRaf = new RandomAccessFile(dataFile, "rws");    // 装载数据的本地文件(可以理解为输出流)
				dataRaf.seek(start);                                                // 设置当前线程保存数据的位置
				byte[] buffer = new byte[1024 * 100];            // 每次拷贝100KB
				int len;
				while ((len = in.read(buffer)) != -1) {
					dataRaf.write(buffer, 0, len);                // 从服务端读取数据, 写到本地文件
					threadFinish += len;                        // 每次写入数据之后, 统计当前线程完成了多少
					tempRaf.seek(id * 8);                        // 将临时文件的指针指向当前线程的位置
					tempRaf.writeLong(threadFinish);            // 将当前线程完成了多少写入到临时文件
					synchronized(BreakpointDownloader.this) {    // 多个下载线程之间同步
						totalFinish += len;                        // 统计所有线程总共完成了多少
					}
				}
				dataRaf.close();
				tempRaf.close();
				System.out.println("线程" + id + "下载完毕");
				if (totalFinish == totalLen) {                    // 如果已完成长度等于服务端文件长度(代表下载完成)
					System.out.println("下载完成, 耗时: " + (System.currentTimeMillis() - begin));
					tempFile.delete();                            // 删除临时文件
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	public static void main(String[] args) throws IOException {
		new BreakpointDownloader("http://192.168.1.240:8080/14.Web/android-sdk_r17-windows.zip").download();
	}
}