NodeJS使用Range請求實現下載功能的方法示例

npm install mz

// 文件：server.js
const http = require("http");
const path = require("path");
const url = require("url");

// 引入 mz 模塊轉換成 Promise 的 fs 模塊
const fs = require("mz/fs");

// 請求處理函數
async function listener(req, res) {
 // 獲取 range 請求頭，格式為 Range:bytes=0-5
 let range = req.headers["range"];

 // 下載文件路徑
 let p = path.resovle(__dirname, url.parse(url, true).pathname);

 // 存在 range 請求頭將返回范圍請求的數據
 if (range) {
 // 獲取范圍請求的開始和結束位置
 let [, start, end] = range.match(/(\d*)-(\d*)/);

 // 錯誤處理
 try {
 let statObj = await fs.stat(p);
 } catch (e) {
 res.end("Not Found");
 }

 // 文件總字節數
 let total = statObj.size;

 // 處理請求頭中范圍參數不傳的問題
 start = start ? ParseInt(start) : 0;
 end = end ? ParseInt(end) : total - 1;

 // 響應客戶端
 res.statusCode = 206;
 res.setHeader("Accept-Ranges", "bytes");
 res.setHeader("Content-Range", `bytes ${start}-${end}/${total}`);
 fs.createReadStream(p, { start, end }).pipe(res);
 } else {
 // 沒有 range 請求頭時將整個文件內容返回給客戶端
 fs.createReadStream(p).pipe(res);
 }
}

// 創建服務器
const server = http.createServer(listener);

// 監聽端口
server.listen(3000, () => {
 console.log("server start 3000");
});

curl -v --header "Range:bytes=0-5" http://localhost:3000

// 文件：client.js
const http = require("http");
const fs = require("fs");
const path = require("path");

// 請求配置
let config = {
 host: "localhost",
 port: 3000,
 path: "/download.txt"
};

let start = 0; // 請求初始值
let step = 5; // 每次請求字符個數
let pause = false; // 暫停狀態
let total; // 文件總長度

// 創建可寫流
let ws = fs.createWriteStream(path.resolve(__dirname, config.path.slice(1)));

// 下載函數
function download() {
 // 配置，每次范圍請求 step 個字節
 config.headers = {
 "Range": `bytes=${start}-${start + step - 1}`;
 };

 // 維護下次 start 的值
 start += step;

 // 發送請求
 http.request(config, res => {
 // 獲取文件總長度
 if (typeof total !== "number") {
 total = res.headers["content-ranges"].match(/\/(\d*)/)[1];

 }

 // 讀取返回數據
 let buffers = [];
 res.on("data", data => buffers.push(data));
 res.on("end", () => {
 // 合并數據并寫入文件
 let buf = Buffer.concat(buffers);
 ws.write(buf);

 // 遞歸進行下一次請求
 if (!pause && start < total) {
 download();
 }
 });
 }).end();
}

// 監控輸入
process.stdin.on("data", data => {
 // 獲取指令
 let ins = data.toString().match(/(\w*)\/r/)[1];
 switch (ins) {
 case "s":
 case "r":
 pause = false;
 download();
 break;
 case "p":
 pause = true;
 break;
 }
});