使用 Koa2 轻松创建文件上传接口

2024-01-22 07:28:15

随着网络技术的蓬勃发展，文件上传功能已成为现代 Web 应用程序的基石。在 Koa2 框架的加持下，我们可以轻松创建强大的文件上传接口。本文将引导您逐步构建一个 Koa2 文件上传接口，并使用 Postman 进行全面测试。

构建 Koa2 文件上传接口

首先，让我们设置一个新的 Koa2 项目：

mkdir koa2-file-upload && cd koa2-file-upload
npm init -y
npm install koa2 koa2-bodyparser multer

安装完毕后，创建 server.js 文件并添加以下代码：

const Koa = require('koa2');
const Router = require('koa-router');
const multer = require('multer');
const app = new Koa();
const router = new Router();

// 设置文件上传目录
const storage = multer.diskStorage({
  destination: './uploads',
  filename: (req, file, cb) => {
    cb(null, Date.now() + '-' + file.originalname);
  },
});

// 创建 multer 实例
const upload = multer({ storage });

// 定义路由
router.post('/upload', upload.single('file'), async (ctx) => {
  const file = ctx.req.file;
  if (file) {
    ctx.body = {
      success: true,
      message: '文件上传成功',
      filename: file.filename,
    };
  } else {
    ctx.body = {
      success: false,
      message: '文件上传失败',
    };
  }
});

// 注册路由
app.use(router.routes());

// 启动服务器
app.listen(3000, () => {
  console.log('服务器已启动，监听端口 3000');
});