koa实现图片上传接口(当地生成)

目录结构

1. my-koa-app/
2. ├── config/               # 配置文件夹
3. │   └── database.js       # 数据库连接配置
4. ├── controllers/          # 控制器，包含具体的业务逻辑
5. │   └── uploadImageController.js # 用户相关的控制器
6. ├── models/               # Mongoose 数据模型
7. │   └── uploadImage.js           # 用户模型
8. ├── routes/               # 路由定义
9. │   └── routes.js     # 用户相关路由
10. ├── uploads/               # 上传图片目录
11. ├── app.js                # Koa 应用的入口文件
12. └── package.json          # 项目配置文件

复制代码

安装npm包

1. npm install koa koa-router koa-bodyparser mongoose koa-static koa-multer

复制代码

代码示例

app.js

1. const Koa = require('koa');
2. const bodyParser = require('koa-bodyparser');
3. const connectDB = require('./config/database');
4. const koaStatic = require('koa-static');
5. const userRoutes = require('./routes/routes');//引入路由
6. const path = require('path');
8. // 连接到数据库
9. connectDB();
11. const app = new Koa();
13. // 使用 bodyParser 解析请求体
14. app.use(bodyParser());
16. // 静态文件服务
17. app.use(koaStatic(path.join(__dirname, 'uploads'))); // 提供上传文件的静态资源服务
19. // 加载 user 路由
20. app.use(userRoutes.routes()); // 使用 userRoutes 路由
21. app.use(userRoutes.allowedMethods()); // 使用 allowedMethods
23. // 启动服务器
24. app.listen(3000, () => {
25.   console.log('Server is running on http://localhost:3000');
26. });

复制代码

config/database.js

1. // config/database.js
2. const mongoose = require('mongoose');
4. const connectDB = async () => {
5.   try {
6.     await mongoose.connect('mongodb://localhost:端口号/数据库名称');//例如:mongodb://localhost:27017/xxxx
7.     console.log('MongoDB connected');
8.   } catch (err) {
9.     console.error(err);
10.   }
11. };
13. module.exports = connectDB;

复制代码

models/uploadImage.js

1. const mongoose = require('mongoose');
2. // 创建图片存储的 Schema 和 Model
3. const ImgSchema = new mongoose.Schema({
4.     filename: String,//文件名称
5.     path: String,//文件目录
6.     url: String, // 用来存储图片的URL
7.     createdAt: { type: Date, default: Date.now },//创建时间
8.   });
9.   
10.   const photoImg = mongoose.model('photoImg', ImgSchema);
12.   module.exports = photoImg;

复制代码

controllers/uploadImageController.js

1. const multer = require('@koa/multer');
2. const path = require('path');
3. const photoImg = require('../models/uploadImage');  // 引用正确的模型
5. // 配置上传目录，确保 uploads 目录存在
6. const upload = multer({
7.   dest: path.join(__dirname, '/../uploads/'),  // 上传的文件存储路径
8.   limits: { fileSize: 5 * 1024 * 1024 }, // 限制文件大小为 5MB
9. });
11. // 上传图片接口的逻辑
12. const uploadImage = async (ctx) => {
14.     const { file } = ctx.request;  // 获取上传的文件
15.     console.log('File:', file);  // 打印文件信息
17.     if (!file) {
18.       ctx.status = 400;
19.       ctx.body = { error: 'No file uploaded' };  // 如果没有文件上传
20.       return;
21.     }
22.   
23.     const imageUrl = `http://localhost:3000/uploads/${file.filename}`; // 图片的 URL 地址
24.   
25.     // 保存文件信息到数据库
26.     const newPhoto = new photoImg({
27.       filename: file.originalname,
28.       path: file.path,
29.       url: imageUrl,
30.     });
31.   
32.     try {
33.       await newPhoto.save();
34.       ctx.status = 201;
35.       ctx.body = { success: true, url: imageUrl }; // 返回图片的 URL
36.     } catch (err) {
37.       ctx.status = 500;
38.       ctx.body = { error: 'Failed to save image', details: err.message };
39.     }
40.   };
41.   
42.   
44. // 获取所有图片信息
45. const getPhotos = async (ctx) => {
46.   try {
47.     const photos = await photoImg.find(); // 查询所有图片
48.     ctx.body = { success: true, data: photos };
49.   } catch (err) {
50.     ctx.status = 500;
51.     ctx.body = { error: 'Failed to fetch photos' };
52.   }
53. };
55. // 获取指定 ID 的图片信息
56. const getPhotoById = async (ctx) => {
57.   const { id } = ctx.params;
58.   try {
59.     const photo = await photoImg.findById(id); // 根据 ID 查找图片
60.     if (!photo) {
61.       ctx.status = 404;
62.       ctx.body = { success: false, message: 'Photo not found' };
63.     } else {
64.       ctx.status = 200;
65.       ctx.body = { success: true, data: photo };
66.     }
67.   } catch (err) {
68.     ctx.status = 500;
69.     ctx.body = { success: false, message: 'Network error' };
70.   }
71. };
73. module.exports = {
74.   uploadImage,
75.   getPhotos,
76.   getPhotoById,
77.   upload,  // 导出上传中间件
78. };

复制代码

routes/routes.js

1. // routes/routes.js
2. const Router = require('koa-router');
4. const  imgController = require('../controllers/uploadImageController')
6. const router = new Router();
8. // 图片上传接口
10. // 定义 Post 请求 - 获取用户信息
11. router.post('/upload', imgController.upload.single('image'), imgController.uploadImage);
13. // 定义 GET 请求 - 获取所有图片
14. router.get('/photos', imgController.getPhotos);
16. // 定义 GET 请求 -  获取指定 ID 的图片
17. router.get('/photos/:id', imgController.getPhotoById);
19. module.exports = router;

复制代码

测试

发起postman哀求，参数为image，格式类型为file

当地上传file打印结果：

哀求前uploads目录为空

哀求成功后uploads添加成功数据

查看数据库path，url目录保存图片，是否访问成功

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。