3

我正在使用 node 和 express 来处理文件上传，并且我正在使用 multiparty/busboy 和 request 将它们直接流式传输到转换服务。

有没有办法在将流发送到相应的提供者之前验证流是否具有某些特定的文件类型？我尝试了https://github.com/mscdex/mmmagic从第一个块中获取 MIME 类型，并且效果很好。我想知道以下工作流程是否会以某种方式起作用：

  • 缓冲文件上传流并检查 Mime 类型的传入数据。
  • 当检查前几个块并且 MIME 类型正确时，将缓冲区清空到请求流中。
  • 当 mime 类型不正确时，发送错误消息并返回。

我试图让它工作，但我似乎有一些流兼容性问题（节点 0.8.x 与节点 0.10.x 流，请求库不支持）。

有没有解决这个问题的最佳实践？我看错了吗？

编辑：感谢保罗，我想出了这段代码：

https://gist.github.com/chmanie/8520572

4

1 回答 1

9

除了检查Content-Type客户端请求的标头之外，我不知道检查 MIME 类型的更好和更聪明的方法。

您可以使用Transform 流实现上述解决方案。在此示例中，转换流缓冲了一些任意数量的数据，然后将其发送到您的 MIME 检查库。如果一切正常，它会重新发送数据。随后的块将按原样发出。

var stream = require('readable-stream');
var mmm = require('mmmagic');
var mimeChecker = new stream.Transform();
mimeChecker.data = [];
mimeChecker.mimeFound = false;
mimeChecker._transform = function (chunk, encoding, done) {
  var self = this;

  if (self.mimeFound) {
    self.push(chunk);
    return done();
  }

  self.data.push(chunk);
  if (self.data.length < 10) {
    return done();
  }
  else if (self.data.length === 10) {
    var buffered = Buffer.concat(this.data);
    new mmm.Magic(mmm.MAGIC_MIME_TYPE).detect(buffered, function(err, result) {
      if (err) return self.emit('error', err);
      if (result !== 'text/plain') return self.emit('error', new Error('Wrong MIME'));
      self.data.map(self.push.bind(self));
      self.mimeFound = true;
      return done();
    });
  }
};

然后，您可以将此转换流通过管道传输到任何其他流，例如请求流（顺便说一下，它完全支持 Node 0.10 流）。

// Usage example
var fs = require('fs');
fs.createReadStream('input.txt').pipe(mimeChecker).pipe(fs.createWriteStream('output.txt'));

编辑：为了更清楚地了解 Node 0.8 和 0.10 流之间遇到的不兼容性，当您定义流并将.on('data')侦听器附加到它时，它将切换到流模式（又名 0.8 流），这意味着它会发出数据，即使目的地没有在听。如果您向以下位置发起异步请求，可能会发生这种情况Magic.detect()：数据仍然在流动，即使您在监听它。

于 2013-11-12T16:40:53.540 回答