主要使用两个 koa 插件,koa-body 里面自带文件上传功能,还有一个 koa-staitc 用于配置静态资源目录(可以通过路径直接访问图片)。
router
const Router = require('@koa/router');
const {upload} = require('../controller/user');
const router = new Router({prefix: '/user'});
router.post('/upload', upload)
controller
async upload(ctx, next) {
// 如果文件类型不是 jpg 则抛出异常
// 注意!!!这种情况下还是可以上传不合法的文件!
// 可行的方法是在 koaBody.formidable 中添加过滤器
// 但是在 koaBody.formidable 中处理 又需要我们手动写一个中间件处理异常
// 就不能使用如下方法捕获异常
if (ctx.request.files.file.mimetype !== 'image/jpeg') {
// koa 内部异常事件 手动抛出异常
ctx.app.emit('error', {massage: '只支持 jpg 文件'}, ctx);
ctx.body = {
code: 400,
message: '只支持 jpg 文件'
}
} else {
ctx.body = {
code: 200,
message: '上传成功',
data: {
// koa-body 将上传的文件信息保存在 ctx.request.files 中 file 指的是前端传输的 file(该字段名字和前端统一) 属性
// fileInfo: ctx.request.files.file
fileUrl: path.basename(ctx.request.files.file.filepath)
}
}
}
}
main.js
const { koaBody } = require('koa-body');
const koaStatic = require('koa-static')
const path = require("path");
const app = new Koa()
// console.log(path.join(__dirname, '../../public/upload'))
app.use(koaBody({
multipart: true,
formidable: {
// 这里最好不使用相对路径 因为这里的相对路径是相对 process.cwd()
// uploadDir: './public/upload',
uploadDir: path.join(__dirname, '../../public/upload'),
keepExtensions: true // 保留文件后缀
}
}))
app.use(koaStatic(path.join(__dirname, '../../public/upload')))