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;
 }
});