歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> Linux編程

Java實現多線程斷點下載(下載過程中可以暫停)

線程可以理解為下載的通道,一個線程就是一個文件的下載通道,多線程也就是同時開啟好幾個下載通道。當服務器提供下載服務時,使用下載者是共享帶寬的,在優先級相同的情況下,總服務器會對總下載線程進行平均分配。不難理解,如果你線程多的話,那下載的越快。

現流行的下載軟件都支持多線程,且支持中途暫停下載,再次開始時不會從頭開始下載。

兩種功能的實現步驟如下:

(1)連接到下載資源文件時,首先判斷資源文件大小,同步的在本地創建一個大小相同的臨時文件用於存儲下載數據。

(2)根據線程數量確定每個線程所需下載的文件大小

(3)根據每個線程下載的文件大小以及線程數量,決定每個線程的開始下載位置以及結束下載位置

(4)為實現斷點下載功能,需要對每個線程實時下載的位置做標記,方便下次開始時繼續之前進度下載

開啟3個線程下載視圖(默認下載目錄為java所轄項目):

整體實現代碼如下(優化空間還有很大哦,大家不要只是看看就完事喽)

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.RandomAccessFile;
import java.net.HttpURLConnection;
import java.net.URL;
public class DownLoad {
    // 聲明下載路徑“紅色警戒2”
    public static final String PATH = "http://www.linuxidc.com/linuxtest/linuxidc.rar";
    public static int threadCount = 0;// 聲明線程數量
    public static void main(String[] args) {
        try {
            URL url = new URL(PATH);
            // 獲取連接
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            // 通過獲取連接定義文件名
            String[] str = PATH.split("/");
            String fileName = str[5];
            // 獲取下載文件大小
            int fileLength = conn.getContentLength();
            System.out.println(fileName);
            // 在本地創建一個與服務器大小一致的可隨機寫入文件
            RandomAccessFile raf = new RandomAccessFile(fileName, "rwd");
            System.out.println(fileLength);// 測試用
            raf.setLength(fileLength);
            // 自定義線程數量
            threadCount = 3;
            // 計算每條線程下載數據的大小
            int blockSize = fileLength / threadCount;
            // 啟動線程下載
            for (int threadId = 1; threadId <= threadCount; threadId++) {
                // 核心代碼,定義每個線程開始以及結束的下載位置
                int startPos = (threadId - 1) * blockSize;// 開始下載的位置
                int endPos = (threadId * blockSize) - 1;// 結束下載的位置(不包含最後一塊)
                if (threadCount == threadId) {
                    endPos = fileLength;
                }
                new Thread(new DownLoadThread(threadId, startPos, endPos, PATH))
                        .start();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    // 實現下載線程
    static class DownLoadThread implements Runnable {
        private int threadId;
        private int startPos;
        private int endPos;
        private String path;
        public DownLoadThread(int threadId, int startPos, int endPos,
                String path) {
            super();
            this.threadId = threadId;
            this.startPos = startPos;
            this.endPos = endPos;
            this.path = path;
        }
        public void run() {
            try {
                URL url = new URL(path);
                String[] str = PATH.split("/");
                String fileName = str[5];
                HttpURLConnection conn = (HttpURLConnection) url
                        .openConnection();
                // 設置URL請求的方法(具體參考API)
                conn.setRequestMethod("GET");
                // 設置500毫秒為超時值
                conn.setReadTimeout(5000);
                File file = new File(threadId + ".txt");
                if (file.exists() && file.length() > 0) {
                    BufferedReader br = new BufferedReader(
                            new InputStreamReader(new FileInputStream(file)));
                    String saveStartPos = br.readLine();
                    if (saveStartPos != null && saveStartPos.length() > 0) {
                        startPos = Integer.parseInt(saveStartPos);
                    }
                }
                // 注意雙引號內的格式,不能包含空格(等其他字符),否則報416
                conn.setRequestProperty("Range", "bytes=" + startPos + "-"
                        + endPos);
                RandomAccessFile raf = new RandomAccessFile(fileName, "rwd");// 存儲下載文件的隨機寫入文件
                raf.seek(startPos);// 設置開始下載的位置
                System.out.println("線程" + threadId + ":" + startPos + "~~"
                        + endPos);
                InputStream is = conn.getInputStream();
                byte[] b = new byte[1024 * 1024 * 10];
                int len = -1;
                int newPos = startPos;
                while ((len = is.read(b)) != -1) {
                    RandomAccessFile rr = new RandomAccessFile(file, "rwd");// 存儲下載標記的文件
                    raf.write(b, 0, len);
                    // 將下載標記存入指定文檔
                    String savaPoint = String.valueOf(newPos += len);
                    rr.write(savaPoint.getBytes());
                    rr.close();
                }
                is.close();
                raf.close();
                System.out.println("下載完成");
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}

注意事項:

(1)理解每個線程下載的開始和結束位置的計算方法(不包含最後一個線程,從0開始)

開始位置=(線程數-1)*每個線程所要下載的文件長度(大小)

結束位置=(線程數*每個線程下載的文件大小)-1

(2)最後一個線程所分配的下載開始位置為上一個線程的結束位置,結束位置為文件長度。

所以最後一個線程所下載的文件長度一般不會與之前線程相同

(3)本次演示下載路徑默認為java項目所在目錄,而且目錄中不能包含與下載文件同名的文件,不然程序會報416網絡請求錯誤。

(4)我個人在實現時犯的一個錯誤conn.setRequestProperty("Range","bytes="+ startPos+"-"+endPos);一定要注意引號沒一定不能有多余的空格以及其他字符,我就是在引號內多了個空格,但錯誤定位一直是inputStream錯誤,找錯找的很痛苦

Copyright © Linux教程網 All Rights Reserved