多線程下載原理及核心代碼

原文出自：http://blog.csdn.net/howlaa/article/details/21875991

假如我們把一個服務器上的文件看作是一個水缸裏的水的話，那麼多線程下載就相當於從水缸上打了多個小孔，然後塞進去小管道進行抽水。呵呵，也許這個比喻不夠準確。多線程下載大致可分爲以下幾個步驟：

一、首先在本地創建一個與服務器文件大小相同的臨時文件(這個很好理解，如果我想下個2G的電影，我得給先在本地佔用2G的空間，不然不能下着下着沒空間了是吧)。

二、計算分配幾個線程去下載服務器上的資源，知道每個線程下載文件的起始位置。

  那麼這個起始位置怎麼計算呢？

文件長度/線程個數= 每個線程下載文件的大小。那麼

線程1下載的位置：0~每個線程下載文件的大小-1.

線程2：以此類推

那麼就是i線程的下載起始位置： (i-1)*每個線程下載文件的大小

三、開啓多個線程，每一個線程下載對應位置的文件。

四、如果所有的線程都把自己的數據下載完畢了，服務器上的資源就被下載到本地了。

五、當文件都下載到本地了，那麼還有一個文件就是把各個線程下載的文件如何串起來。那麼就要利用到一個類：RandomAccessFile 隨機文件訪問類。

代碼如下：

import java.io.InputStream;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;


public class Demo {
    public static int threadCount = 3;
    /**
     * @param args
     */
    public static void main(String[] args) throws Exception{
        //連接服務器，獲取文件長度，在本地創建一個大小和服務器一樣大的臨時文件
        String path ="http://192.168.1.100:8080/360.exe";
        URL url = new URL(path);
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        conn.setConnectTimeout(5000);
        conn.setRequestMethod("GET");
        int code = conn.getResponseCode();
        if(code==200){
            //服務器返回的數據的長度，實際上就是文件的長度
            int length = conn.getContentLength();
            System.out.println("文件總長度:"+length);
            RandomAccessFile raf = new RandomAccessFile("setup.exe", "rwd");
            //指定創建的文件的長度
            raf.setLength(length);
            raf.close();
            //在客戶端本地
            //假設3個線程去下載資源
            //平均每一個線程下載的文件的大小。
            int blockSize = length / threadCount;

            for(int threadId=1;threadId<=threadCount;++threadId){
                //第一個線程下載的開始位置
                int startIndex = (threadId-1)*blockSize;
                int endIndex = blockSize - 1;
                if(threadId==threadCount){
                    //最後一個線程下載的長度稍微長一點
                    endIndex = length;
                }
                System.out.println("線程:"+threadId+"下載:--"+startIndex+"-->"+endIndex);
                new DownLoadThread(threadId, startIndex, endIndex, path).start();
            }
        }else{
            System.out.println("訪問錯誤");
        }

    }

    /**
     * 下載文件的子線程，每個線程下載對應的文件
     *
     */
    public static class DownLoadThread extends Thread{
        private int threadId;
        private int startIndex;
        private int endIndex;
        private String path;
        /**
         * @param threadId線程ID
         * @param startIndex
         * @param endIndex
         * @param path 下載文件在服務器上的路徑
         */
        public DownLoadThread(int threadId, int startIndex, int endIndex,String path) {
            this.threadId = threadId;
            this.startIndex = startIndex;
            this.endIndex = endIndex;
            this.path = path;
        }

        @Override
        public void run() {
            try{
                URL url = new URL(path);
                HttpURLConnection conn = (HttpURLConnection) url.openConnection();
                conn.setRequestMethod("GET");
                //很重要：請求服務器下載部分的文件的指定的位置：
                conn.setRequestProperty("Range", "bytes="+startIndex+"-"+endIndex);
                conn.setConnectTimeout(5000);
                int code  = conn.getResponseCode();//從服務器請求全部資源 200ok ,如果請求部分資源 206 ok
                System.out.println("code="+code);

                InputStream is = conn.getInputStream();//返回資源
                RandomAccessFile raf = new RandomAccessFile("setup.exe", "rwd");
                //隨機寫文件的時候從哪個位置開始寫
                raf.seek(startIndex);//定位文件

                int len =0;
                byte[] buffer = new byte[1024];
                while((len = is.read(buffer)) != -1){
                    raf.write(buffer,0,len);
                }
                is.close();
                raf.close();
                System.out.println("線程"+threadId+"下載完畢");

            }catch(Exception e){
                e.printStackTrace();
            }
        }

    }

}