nodejs实现范围请求

墨蓝 2022-05-15 12:13 187阅读 0赞

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

1.  客户端向服务端发起请求
2.  服务端响应，附上`Accept-Ranges`头（值表示表示范围的单位，通常是“bytes”），告诉客户端其接受范围请求
3.  客户端发送新的请求，附上`Ranges`头，告诉服务端请求的是一个范围
4.  服务端收到范围请求，分情况响应：
    
     *  范围有效，服务端返回`206 Partial Content`，发送指定范围内内容，并在`Content-Range`头中指定该范围
     *  范围无效，服务端返回`416 Requested Range Not Satisfiable`，并在`Content-Range`中指明可接受范围

请求中的`Ranges`头格式为（这里不考虑多范围请求了）：

Ranges: bytes=[start]-[end]

其中 start 和 end 并不是必须同时具有：

1.  如果 end 省略，服务器应返回从 start 位置开始之后的所有字节
2.  如果 start 省略，end 值指的就是服务器该返回最后多少个字节
3.  如果均未省略，则服务器返回 start 和 end 之间的字节

响应中的`Content-Range`头有两种格式：

1.  当范围有效返回 206 时：
    
        Content-Range: bytes (start)-(end)/(total)
2.  当范围无效返回 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);