Java多线程下载实现

package com.test;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.ProtocolException;
import java.net.URL;

public class DownLoadUtil {

	private String urlStr;
	private String targetFile;

	private int threadNum;
	private DownloadThread[] threads;

	private int fileSize;
	
	
	
	public DownLoadUtil(String urlStr, String targetFile, int threadNum) {
		this.urlStr = urlStr;
		this.targetFile = targetFile;
		this.threadNum = threadNum;
		this.threads = new DownloadThread[threadNum];
	}

	public void download() {
		try {
			HttpURLConnection conn = this.getHttpURLConnection();
			conn.setRequestProperty("Connection", "Keep-Alive");
			fileSize = conn.getContentLength();
			conn.disconnect();
			int currentPartSize = fileSize/threadNum + 1;
			RandomAccessFile file = new RandomAccessFile(targetFile, "rw");
			file.setLength(fileSize);
			file.close();
			
			for(int i=0; i<threadNum; i++) {
				int startPosition = currentPartSize * i;
				RandomAccessFile currentFile = new RandomAccessFile(targetFile, "rw");
				currentFile.seek(startPosition);
				threads[i] = new DownloadThread(startPosition, currentFile, currentPartSize);
				new Thread(threads[i]).start();
			}
			
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	private class DownloadThread implements Runnable {

		private int startPosition;
		private RandomAccessFile downloadFile;
		private int currentPartSize;
		private int hasDownSize;

		public DownloadThread(int startPosition, RandomAccessFile downloadFile,
				int currentPartSize) {
			this.startPosition = startPosition;
			this.downloadFile = downloadFile;
			this.currentPartSize = currentPartSize;
		}

		@Override
		public void run() {
			try {
				HttpURLConnection conn = getHttpURLConnection();
				InputStream in = conn.getInputStream();
				in.skip(this.startPosition);
				byte[] buffer = new byte[1024];
				int hasRead = 0;
				while (hasDownSize < currentPartSize &&
						(hasRead = in.read(buffer)) != -1) {
					downloadFile.write(buffer, 0, hasRead);
					hasDownSize += hasRead;
				}
				downloadFile.close();
				in.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}

	}

	private HttpURLConnection getHttpURLConnection() {
		HttpURLConnection conn = null;
		try {
			URL url = new URL(urlStr);
			conn = (HttpURLConnection) url.openConnection();
			conn.setConnectTimeout(5 * 1000);
			conn.setRequestMethod("GET");
			conn.setRequestProperty(
					"Accept",
					"image/gif, image/jpeg, image/pjpeg, image/pjpeg, "
							+ "application/x-shockwave-flash, application/xaml+xml, "
							+ "application/vnd.ms-xpsdocument, application/x-ms-xbap, "
							+ "application/x-ms-application, application/vnd.ms-excel, "
							+ "application/vnd.ms-powerpoint, application/msword, */*");
			conn.setRequestProperty("Accept-Language", "zh-CN");
			conn.setRequestProperty("Charset", "UTF-8");
		} catch (MalformedURLException e) {
			e.printStackTrace();
		} catch (ProtocolException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return conn;
	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值