【Koa】文件上传

主要使用两个 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')))