我正在使用busboy,将我上传的文件写入缓冲区并对其执行一些验证(宽度,高度和文件大小)。我不能为我的生活弄清楚如果我发现上传有问题,如何中止/停止流。
例如,如果我想要允许的最大文件大小为500kb,我会在上传时跟踪缓冲区的大小,如果大小超过500kb,我想中止。
这是我的代码的简化版本。
var self = this;
var busboy = new Busboy({
headers: self.req.headers,
limits: {
files: 1
}
});
busboy.on('file', function(fieldname, file, filename, encoding, mimetype) {
file.fileRead = [];
var size = 0;
file.on('data', function(chunk) {
size += chunk.length;
/* DO VALIDATION HERE */
if( size > 500000) {
/*** ABORT HERE ***/
}
file.fileRead.push(chunk);
});
file.on('end', function() {
var data = Buffer.concat(file.fileRead, size);
// ... upload to S3
});
self.req.pipe(busboy);
});
答案 0 :(得分:6)
好的,所以我遇到了同样的问题,我用file.resume();
var fstream;
req.busboy.on('file', function (fieldname, file, filename, encoding, mimetype) {
// Validate file mimetype
if(mimetype != 'image/png'){
file.resume();
return res.json({
success: false,
message: 'Invalid file format'
});
}
// Upload
fstream = fs.createWriteStream(__dirname + '/tmp/' + timestamp + filename);
file.pipe(fstream);
fstream.on('close', function () {
return res.json({
success: true
});
});
});
req.pipe(req.busboy);
答案 1 :(得分:0)
我会尝试这样的事情:
var self = this;
var busboy = new Busboy({
headers: self.req.headers,
limits: {
files: 1,
fileSize: 500000
}
});
busboy.on('file', function(fieldname, file, filename, encoding, mimetype) {
file.fileRead = [];
file.on('limit', function() {
//Clear variables
});
file.on('data', function(chunk) {
file.fileRead.push(chunk);
});
file.on('end', function() {
var data = Buffer.concat(file.fileRead, size);
// ... upload to S3
});
self.req.pipe(busboy);
});
基本上我在Busboy的配置中添加了一个新限制:fileSize: 500 * 1024
我开始听极限事件:
file.on('limit', function() {
//Clear vars
});
答案 2 :(得分:0)
答案很简单。
// do the required validation like size check etc.
if( size > 500000)
{
self.req.unpipe();
return;
}
背景是这样的。我在busboy代码中看到busboy实现为WritableStream,下面使用Dicer
模块进行解析,也实现为WritableStream。流程是这样的:
req
stream ==> busboy ==> dicer ==> dicer引发事件==> busboy在文件上引发事件==>这些事件都会在您的代码中使用。
要停止整个代码流 - 上面的代码应该这样做。
答案 3 :(得分:0)
我认为正确的做法是关闭套接字并结束请求
busboy.on('file', function(fieldname, file, filename, encoding, mimetype) {
file.fileRead = [];
var size = 0;
file.on('data', function(chunk) {
size += chunk.length;
/* DO VALIDATION HERE */
if( size > 500000) {
self.req.socket.end();
self.res.end();
}
file.fileRead.push(chunk);
});
file.on('end', function() {
var data = Buffer.concat(file.fileRead, size);
// ... upload to S3
});
self.req.pipe(busboy);
});
答案 4 :(得分:-1)
我能够访问具有Dicer
方法的基础ignore
解析器,该方法基本上会停止上传文件。
以下是我的表现:busboy._parser.parser._ignore()
看起来确实非常hackish,但它确实有效,并且正是我想要的。