一、需求
egg.js
的文件上传个人觉得很一般,内置的 multipart 插件并不怎么好用。
egg-multipart
也是基于 co-busboy
实现的。
egg 官方给的文件上传的示例地址:
二、CSRF 校验
egg
文件的上传需要进行 csrf
校验,而且这个校验默认只支持放在 ctx.query._csrf
字段中,很不方便,每次文件上传都会携带。
习惯 laravel 或者某些 PHP 框架的人都知道,直接放在 <input type="hidden" name="_csrf" >
的字段中方便的多。(不过我默认试过放在 input 字段中,是不起作用的,其他方法我暂时没有尝试)
至于生成 csrf
的方式,可以在 cookie 中获取,cookie 中存放了 csrfToken=token
,因此是可以通过前端获取的,这为通过 ajax 实现文件上传也提供了帮助。
如果在模板文件中使用的话,可以通过 {{ctx.csrf | safe}}
生成 token。
因为 token 必须放在 query._csrf
中,所以 form 的 html 如下:
<form action="/post/add?_csrf={{ctx.csrf | safe}}" method="POST" enctype="multipart/form-data">
</form>
三、通过 form 表单上传文件
表单 html 不过多描述,默认使用上面的 url 进行表单提交,路由也不具体说明。
文件上传的时候使用了额外的4个依赖,依赖如下:
依赖 | 说明 |
---|---|
stream-wormhole | 将 stream 流消耗掉 |
await-stream-ready | 文件读写流 ready 库,能够使用 await |
await-stream-ready
主要是方便的使用 await
进行文件上传,而 stream-wormhole
是因为如果上传失败出现异常,那会导致浏览器响应崩溃,因此需要将 stream 消耗掉。
重点说下 controller 中的代码:
app/controller/post.js
:
// 添加文章
async add(){
await this.ctx.render('post/add');
}
// 添加文章操作
async addAction(){
const {ctx} = this;
// 获取 steam
const stream = await ctx.getFileStream();
// 生成文件名
const filename = Date.now() + '' + Number.parseInt(Math.random() * 10000) + path.extname(stream.filename);
// 写入路径
const target = path.join(this.config.baseDir, 'app/public/upload/', filename);
const writeStream = fs.createWriteStream(target);
try {
// 写入文件
await awaitStreamReady(stream.pipe(writeStream));
} catch (err) {
// 必须将上传的文件流消费掉,要不然浏览器响应会卡死
await sendToWormhole(stream);
throw err;
}
ctx.body = stream.fields;
}
上面代码中,通过 await ctx.getFileStream()
获取到流,stream 是一个 FileStream 对象,有许多有用的属性,比如下面这些属性,其中 stream.filename
能获取文件的原始名称,上面 controller 代码中,便是使用 filename 获取到了文件的后缀。
当表单的 enctype 设置成 multipart/form-data
之后,便不能使用 ctx.request.body
获取其他字段,现在是个空对象。
要获取其他字段的名字,需要使用 stream.fields
来获取其他字段的值。
FileStream {
fieldname: 'photo', // 字段名
filename: '1.jpg', // 文件名
encoding: '7bit', // 编码
transferEncoding: '7bit',
mime: 'image/jpeg', // 类型
mimeType: 'image/jpeg',
fields: { title: '', description: '', author: '', content: '' } }
上传结果:
{"title":"","description":"","author":"","content":""}
四、通过 ajax 上传文件
ajax 文件上传后端代码基本不用变动,只是前端代码这边构建一个 formData 即可。
下面的代码是官方示例的给的,因为用 formData 进行 ajax 文件上传很多示例,不再重复,即使使用其他插件也是无所谓的,本质上 HTTP 请求的 Content-type
还是 multipart/form-data
。
关键点在于 _csrf
的获取,方法 getCsrf()
是通过 cookie 获取 token 的方式,当然如果页面还是在模板中,那我觉得通过一个隐藏的值存放 csrf token
,然后再去获取更加方便。
后端代码没什么变动的,本质是一样的。
$('form').submit(function(e) {
e.preventDefault();
var formData = new FormData();
formData.append('name', $('input[type=text]').val());
// Attach file
formData.append('image', $('input[type=file]')[0].files[0]);
// console.log(formData);
$.ajax({
url: '/ajax?_csrf=' + getCsrf(),
data: formData,
method: 'POST',
contentType: false, // NEEDED, DON'T OMIT THIS (requires jQuery 1.6+)
processData: false, // NEEDED, DON'T OMIT THIS
success: function(result) {
console.log(result);
},
error: function(responseStr) {
alert("error", responseStr);
}
});
// 通过 cookie 获取 csrf token
function getCsrf() {
var keyValue = document.cookie.match('(^|;) ?csrfToken=([^;]*)(;|$)');
return keyValue ? keyValue[2] : null;
}
});
五、自定义文件上传目录
按照我自己的习惯,我在上传文件的时候,自然是希望文件上传到 app/public/upload/20180707/xxxx.jpg
,但是由于 writeStream
的限制,20180707
必然是需要确定文件目录存在的。
因此如果使用这种方式上传,则需要增加几行代码用来判断并且生成文件夹。
为了方便生成日期如 20180707
的目录,我用了 dayjs
库来格式化时间。
为了省事我直接用了
mkdirSync()
// 上传基础目录
const uplaodBasePath = 'app/public/upload/';
// 生成文件名
const filename = Date.now() + '' + Number.parseInt(Math.random() * 10000) + path.extname(stream.filename);
// 生成文件夹
const dirName = dayjs(Date.now()).format('YYYYMMDD');
// 判断文件夹是否存在,不存在则直接创建文件夹
if(! fs.existsSync()) fs.mkdirSync(path.join(this.config.baseDir,uplaodBasePath,dirName));
// 生成写入路径
const target = path.join(this.config.baseDir, uplaodBasePath, dirName, filename);
// 写入流
const writeStream = fs.createWriteStream(target);
最终结果:
最终的 controller 代码如下:
// 添加文章操作
async addAction(){
const {ctx} = this;
// 获取 steam
const stream = await ctx.getFileStream();
// 上传基础目录
const uplaodBasePath = 'app/public/upload/';
// 生成文件名
const filename = Date.now() + '' + Number.parseInt(Math.random() * 10000) + path.extname(stream.filename);
// 生成文件夹
const dirName = dayjs(Date.now()).format('YYYYMMDD');
if(! fs.existsSync()) fs.mkdirSync(path.join(this.config.baseDir,uplaodBasePath,dirName));
// 生成写入路径
const target = path.join(this.config.baseDir, uplaodBasePath, dirName, filename);
// 写入流
const writeStream = fs.createWriteStream(target);
try {
// 写入文件
await awaitStreamReady(stream.pipe(writeStream));
} catch (err) {
// 必须将上传的文件流消费掉,要不然浏览器响应会卡死
await sendToWormhole(stream);
throw err;
}
ctx.body = stream.fields;
}