Java實現檔案斷點續傳

Evan1024發表於2024-03-08
Java斷點

檔案斷點續傳程式碼

import java.io.File;
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.URL;
import java.util.Date;


public class DT extends Thread {

    String urlt;//下載地址
    int startl;//子執行緒讀取/寫入起始位元組
    int end;//子執行緒寫入結束位元組長度
    String fileName;
    RandomAccessFile osf;

    public DT(int i, String url, String fileName, int start, int end) {
        this.setName("t" + i); //子執行緒名稱
        this.urlt = url; //下載地址
        this.fileName = fileName;
        this.startl = start; //子執行緒讀取/寫入起始位元組
        this.end = end;//子執行緒寫入結束位元組長度
    }

    public void run() {
        try {
            osf = new RandomAccessFile(fileName, "rw");
            URL url = new URL(urlt);
            HttpURLConnection http2 = (HttpURLConnection) url.openConnection();
            http2.setRequestProperty("User-Agent", "NetFox");

            /*
             * 斷點續傳和多執行緒下載的關鍵程式碼關鍵位置：即設定斷點
             * http2.setRequestProperty("RANGE", "bytes="+startl+"-");//設定斷點位置，向伺服器請求從檔案的哪個位元組開始讀取。
             * osf.seek(startl);//設定本地檔案從哪個位元組開始寫入
             * 如果是單執行緒，則首先要判斷下載檔案是否已經存在 及DownloadFile.java 裡的 fileName = "C:\eclipse.zip";
             * 如果存在則開始斷點續傳，方法同多執行緒：
             * 因為斷點續傳是從上次傳輸中斷的位元組開始，則首先要得到上次中斷的位置，既是檔案長度（針對單執行緒）f.length()
             * 然後設定HTTP請求頭屬性RANGE，該屬性告知伺服器從哪個自己開始讀取檔案。
             * 設定本地檔案寫入起始位元組，及接從上次傳輸斷點繼續寫入（斷點續傳）
             * osf.seek(offset) 該方法設定從offset後一個位元組開始寫入檔案
             * 注意：多執行緒不能用檔案長度做為寫檔案起始位元組，需有配置檔案記錄上次讀寫的位置，迅雷下載既是使用該種方法。
             *
             */
            http2.setRequestProperty("RANGE", "bytes=" + startl + "-");//設定斷點位置，向伺服器請求從檔案的哪個位元組開始讀取。
            osf.seek(startl);//設定本地檔案從哪個位元組開始寫入

            InputStream input = http2.getInputStream();
            byte b[] = new byte[1024];//設定緩衝池，每次只讀1024位元組
            Date d = new Date();//子執行緒開始下載時間
            int l;//計運算元執行緒讀取和寫入的檔案長度，當長度大於每個子執行緒平均下載長度則終止執行緒
            int i;
            l = 0;
            System.out.println(this.getName() + " 開始下載。。。");
            while ((i = input.read(b, 0, 1024)) != -1 && l < end) {    //執行緒下載位元組長度控制誤差小於緩衝池大小，本示例為緩衝池1024位元組
                osf.write(b, 0, i);
                b = new byte[1024];//重新賦值，避免重新讀入舊內容
                l += i;
            }
            Date d2 = new Date();//子執行緒結束下載時間
            System.out.println(this.getName() + " 執行緒耗時： " + (d2.getTime() - d.getTime()) / 1000 + " 秒,實際共下載：" + l + "位元組");//子執行緒下載耗時（秒）
        } catch (FileNotFoundException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        } catch (MalformedURLException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
}

測試程式碼

import java.io.*;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;


public class DownloadFile {


    /**
     * @param args
     */
    static int len;//執行緒平均下載檔案長度
    static int bn ;//每個執行緒寫入檔案的位元組數
    static int tn; //執行緒數
    static String urlt;//下載地址
    static String fileName;
    static RandomAccessFile osf; //檔案操作
    public static void main(String[] args) {
        // TODO Auto-generated method stub

        try {
            urlt = "http://192.168.16.77:8080/Stargreat/html/baidu player";
            fileName = "C:\" + urlt.split("//")[1].split("/")[urlt.split("//")[1].split("/").length-1];
            System.out.println(fileName);
            URL url = new URL(urlt);
            HttpURLConnection http = (HttpURLConnection) url.openConnection();

            /**
             * 此處設定5個執行緒下載一個檔案tn = 5;
             * 判斷平均每個執行緒需下載檔案長度：
             */
            System.out.println("file size:" + http.getContentLength());
            tn = 3;
            len = http.getContentLength() / tn ;//捨去餘數（餘數自動捨去）計算每個執行緒應下載平均長度，最後一個執行緒再加上餘數，則是整個檔案的長度,
            File f = new File(fileName);
            if (f.exists()){
                f.delete();
                osf = new RandomAccessFile(f, "rw");
                osf.seek(http.getContentLength()-1);
                osf.write(0);
            }else{
                osf = new RandomAccessFile(f, "rw");
                osf.seek(http.getContentLength()-1);
                osf.write(0);
            }
            System.out.println("temp 檔案長度：" + f.length());
            Thread t;//下載子執行緒，
            for (int j = 0; j < tn; j++) {
                if(j == tn - 1){//如果最後一個執行緒則加上餘數長度位元組
                    bn = len + (http.getContentLength() % tn);
                }else{
                    bn = len;
                }
                System.out.println("t"+ j + "執行緒下載長度：" + bn + "起始位元組：" + len*j);
                t = new DT(
                        j,
                        urlt,
                        fileName,
                        len*j,
                        bn

                );
                t.start();
            }

        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }
}

相關文章