NodeJS使用Range請(qǐng)求實(shí)現(xiàn)下載功能的方法示例

npm install mz

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

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

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

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

 // 存在 range 請(qǐng)求頭將返回范圍請(qǐng)求的數(shù)據(jù)
 if (range) {
 // 獲取范圍請(qǐng)求的開(kāi)始和結(jié)束位置
 let [, start, end] = range.match(/(\d*)-(\d*)/);

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

 // 文件總字節(jié)數(shù)
 let total = statObj.size;

 // 處理請(qǐng)求頭中范圍參數(shù)不傳的問(wèn)題
 start = start ? ParseInt(start) : 0;
 end = end ? ParseInt(end) : total - 1;

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

// 創(chuàng)建服務(wù)器
const server = http.createServer(listener);

// 監(jiān)聽(tīng)端口
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");

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

let start = 0; // 請(qǐng)求初始值
let step = 5; // 每次請(qǐng)求字符個(gè)數(shù)
let pause = false; // 暫停狀態(tài)
let total; // 文件總長(zhǎng)度

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

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

 // 維護(hù)下次 start 的值
 start += step;

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

 }

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

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

// 監(jiān)控輸入
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;
 }
});