nodejs实现范围请求的实现代码
时间：2021-01-08 12:18:27|栏目：NodeJS|点击： 次
使服务器支持范围请求，允许客户端只请求文档的一部分。其流程是：
- 客户端向服务端发起请求
- 服务端响应，附上Accept-Ranges头（值表示表示范围的单位，通常是“bytes”），告诉客户端其接受范围请求
- 客户端发送新的请求，附上Ranges头，告诉服务端请求的是一个范围
- 服务端收到范围请求，分情况响应：
- 范围有效，服务端返回206 Partial Content，发送指定范围内内容，并在Content-Range头中指定该范围
- 范围无效，服务端返回416 Requested Range Not Satisfiable，并在Content-Range中指明可接受范围
请求中的Ranges头格式为（这里不考虑多范围请求了）：
Ranges: bytes=[start]-[end]
其中 start 和 end 并不是必须同时具有：
- 如果 end 省略，服务器应返回从 start 位置开始之后的所有字节
- 如果 start 省略，end 值指的就是服务器该返回最后多少个字节
- 如果均未省略，则服务器返回 start 和 end 之间的字节
响应中的Content-Range头有两种格式：
当范围有效返回 206 时：
Content-Range: bytes (start)-(end)/(total)
当范围无效返回 416 时：
Content-Range: bytes */(total)
//代码实现
res.setHeader('Content-Range', `bytes */${totalSize}`);
添加函数处理范围请求：
【注意】这里吧不合理的范围请求直接设置为返回全部，如果想设置范围错误就直接不返回可以设置http状态码为416
/**
* 范围请求
* 不在合理的范围内就直接返回所有
* 在合理规范的范围内就按照返回对应的部分
* @param {*} totleSize 总的范围
* @param {*} req
* @param {*} res
*/
module.exports = (totleSize, req, res) => {
const range = req.headers['range'];
if (!range) {
return {code: 200};
}
const sizes = range.match(/bytes=(\d*)-(\d*)/);
const end = sizes[2] || totleSize - 1;
const start = sizes[1] || totleSize - end;
//不规范的范围，是要直接一起返回，所以设置状态码200
if(start > end || start < 0 || end > totleSize) {
return {code: 200};
}
res.setHeader('Accept-Ranges', 'bytes');
res.setHeader('Content_Range', `bytes ${start}-${end}/${totleSize}`);
res.setHeader('Content_Length', end-start);
return {
code: 206,
start: parseInt(start),
end: parseInt(end)
};
};
范围读取：
let rs;
const { code, start, end } = range(stats.size, req, res);
if (code === 200) {
res.statusCode = 200;
rs = fs.createReadStream(filePath);
} else {
res.statusCode = 206;//表示是部分内容
//创建一个流读取文件, {start, end}表示文件读取的起始点和终点
rs = fs.createReadStream(filePath, { start, end });
}
//有这个pipe就不需要再去调用end了，这样直接调用end会导致res接受不到东西，因为pipe是异步的
rs.pipe(res);
