nodejs实现范围请求的实现代码

时间:2020-05-16 huyao_road 人气:0

使服务器支持范围请求，允许客户端只请求文档的一部分。其流程是：

客户端向服务端发起请求
服务端响应，附上Accept-Ranges头（值表示表示范围的单位，通常是“bytes”），告诉客户端其接受范围请求
客户端发送新的请求，附上Ranges头，告诉服务端请求的是一个范围
服务端收到范围请求，分情况响应：
1. 范围有效，服务端返回206 Partial Content，发送指定范围内内容，并在Content-Range头中指定该范围
2. 范围无效，服务端返回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);

加载全部内容