node.js - 明确限制请求大小

Question

我想限制以下情况下的上传请求大小(简化):

app = Express();
router = Express.Router();
router.use('/upload', (req, res, next) => {
  const writeStream = fs.createWriteStream('filename');
  req.pipe(writeStream);
});
app.use(router);

如您所见，没有使用任何形式或类似形式(无法更改此设置)。因此，数据作为原始数据流提供。

由于我找不到适合这种情况的任何现有模块(body-parser 不适用于这种情况，raw-body 也不能使用，因为流应该保持不变)，我当前的方法是使用以下代码作为express的中间件:

function(req, res, next) {
  if (req.headers['content-length'] > limit) {
    res.set("Connection", "close");
    res.status(413).end();
    return;
  }
  let received = 0;
  req.on('data', (chunk) => {
    received += chunk.length;
    if (received > limit) {
      res.set("Connection", "close");
      res.status(413).end();
    }
  })
  next();
})

有没有更好的方法来做到这一点？是否有现有的 npm 包可以完成这项工作？

Answer 1

最后我通过使用@ambianBeing提供的答案解决了这个问题，并重写了随后处理缓冲区而不是流的函数。