[英]Node Busboy abort upload
我正在使用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);
});
好的,所以我有同样的问题,我用file.resume();
解决了它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);
我会尝试这样的事情:
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
});
答案很简单。
// 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引发文件上的事件==>这些事件会在你的代码中消耗掉。
要停止这整个代码流 - 上面的代码应该这样做。
我认为正确的做法是关闭套接字并结束请求
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);
});
我能够访问底层的Dicer
解析器,它有一个基本上停止上传文件的ignore
方法。
我是这样做的: busboy._parser.parser._ignore()
它似乎非常hackish,但它工作,并完全符合我的要求。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.