Java多线程下载的实现方法

 

package cn.me.test;

import java.io.InputStream;

import java.io.RandomAccessFile;

import java.net.HttpURLConnection;

import java.net.URL;

/**

 * 多线程下载

 * 1：使用RandomAccessFile在任意的位置写入数据。

 * 2：需要计算第一个线程下载的数据量，可以平均分配。如果不够平均时，

 *    则直接最后一个线程处理相对较少的数据

 * 3：必须要在下载之前准备好相同大小的文件，通过文件头获取

 */

public class MultiThreadDownload {

    public static void main(String[] args) throws Exception {

       //1：声明文件名和下载的地址

       String fileName = "aa.rar";

       String urlStr = "http://localhost:7777/day18";

       //2：声明Url

       URL url = new URL(urlStr+"/"+fileName);

       //3：获取连接

       HttpURLConnection con =

           (HttpURLConnection) url.openConnection();

       //4:设置请求方式

       con.setRequestMethod("GET");

       //5:获取请求头,即文件的长度

       int length = con.getContentLength();//获取下载文件的长度，以计算每个线程应该下载的数据量。

       //6：在指定的目录下，创建一个同等大小的文件

       RandomAccessFile file = new RandomAccessFile("d:/a/"+fileName, "rw");//创建一个相同大小的文件。

       //7:设置文件大小，占位

       file.setLength(length);//设置文件大小。

 

       file.close();

       //8：定义线程个数

       int size = 3;

       //9:计算每一个线程应该下载多少字节的数据，如果正好整除则最好，否则加1

       int block = length/size==0?length/size:length/size+1;//计算每个线程应该下载的数据量。

       System.err.println("每个线程应该下载："+block);

       //10：运行三个线程并计算从哪个字节开始到哪一个字节结束

       for(int i=0;i<size;i++){

           int start = i*block;

           int end = start+(block-1);//计算每一个线程的开始和结束字节。

 

         System.err.println(i+"="+start+","+end);

           new MyDownThread(fileName, start, end,url).start();

       }

    }

    static class MyDownThread extends Thread{

       //定义文件名

       private String fileName;

       //定义从何地开始下载

       private int start;

       //定义下载到哪一个字节

       private int end;

       private URL url;

       public MyDownThread(String fileName,int start,int end,URL url){

           this.fileName=fileName;

           this.start=start;

           this.end=end;

           this.url=url;

       }

       @Override

       public void run() {

           try{

              //11:开始下载

              HttpURLConnection con =

                     (HttpURLConnection) url.openConnection();

              con.setRequestMethod("GET");

              //12：设置分段下载的请求头

              con.setRequestProperty("Range","bytes="+start+"-"+end);//设置从服务器上读取的文件块。

 

              //13:开始下载，需要判断206

              if(con.getResponseCode()==206){//访问成功，则返回的状态码为206。

                  InputStream in = con.getInputStream();

                  //14:声明随机写文件对象，注意rwd是指即时将数据写到文件中，而不使用缓存区

                  RandomAccessFile out = new RandomAccessFile("d:/a/"+fileName,"rwd");

                  out.seek(start);//设置从文件的某个位置开始写数据。

                  byte[] b=new byte[1024];

                  int len = 0;

                  while((len=in.read(b))!=-1){

                     out.write(b,0,len);

                  }

                  out.close();

                  in.close();

              }

              System.err.println(this.getName()+"执行完成");

           }catch(Exception e){

              throw new RuntimeException(e);

           }

       }

    }

}