您的当前位置:首页正文

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

2020-11-27 来源:星星旅游

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

  1. 客户端向服务端发起请求
  2. 服务端响应,附上Accept-Ranges头(值表示表示范围的单位,通常是“bytes”),告诉客户端其接受范围请求
  3. 客户端发送新的请求,附上Ranges头,告诉服务端请求的是一个范围
  4. 服务端收到范围请求,分情况响应:
    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);