egg.js 通过 form 和 ajax 两种方式上传文件并自定义目录和文件名

egg.js 通过 form 和 ajax 两种方式上传文件并自定义目录和文件名

评论:10 · 阅读:8437· 喜欢:0

一、需求

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>

egg.js 通过 form 和 ajax 两种方式上传文件并自定义目录和文件名

三、通过 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);

最终结果:

egg.js 通过 form 和 ajax 两种方式上传文件并自定义目录和文件名

最终的 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;
}
上一篇:ABAP ALV单个单元格状态编辑


下一篇:利用"SQL"语句自动生成序号的两种方式