egg.js 框架写一个上传图片的接口

config.default.js 添加

config.multipart = {
    mode:'file'
  };

config.static = {
    // 静态化访问前缀,如:`http://127.0.0.1:7001/static/images/logo.png`
    prefix: '/static', 
    dir: path.join(appInfo.baseDir, 'app/public'), 
    dynamic: true, // 如果当前访问的静态资源没有缓存,则缓存静态文件,和`preload`配合使用;
    preload: false,
    maxAge: 31536000, // in prod env, 0 in other envs
    buffer: true, // in prod env, false in other envs
  };

上述配置参照博客

 https://www.jianshu.com/p/7e23f1c69279

Controller

// app/controller/uploadController.js

const Controller = require('egg').Controller;

class uploadFileController extends Controller {

  async uploadImage() {
    const { ctx } = this;
    const files = ctx.request.files; // 获取上传的文件
    console.log(files)
    // 调用 service 层的 saveImages 方法
    const result = await ctx.service.uploadFileService.saveImages(files);

    // 返回结果
    ctx.body = result;
  }
}

module.exports = uploadFileController;

Service 层

// app/service/uploadFileService.js

const Service = require('egg').Service;
const fs = require('fs');
const path = require('path');
const crypto = require('crypto'); // 引入 crypto 模块

class uploadFileService extends Service {
  async saveImages(files) {
    const ctx = this.ctx;
    const uploadDir = 'app/public/'; // 上传目录

    try {
      // 检查文件数量是否为1
      if (files.length !== 1) {
        return { success: false, message: 'Only one image can be uploaded at a time' };  
      }
      const file = files[0]; // 只处理第一个文件
      const fileName = file.filename;
      // 使用 MD5 对文件名进行加密
      const hash = crypto.createHash('md5').update(fileName).digest('hex');
      const timestamp = Date.now(); // 获取当前时间戳
      
      const newFileName = `${hash}_${timestamp}${path.extname(fileName)}`; // 新文件名
      
      const filePath = path.join(uploadDir, newFileName);
      console.log(newFileName, "====================");
      
      // 将文件保存到指定目录
      const writeStream = fs.createWriteStream(filePath);
      const readStream = fs.createReadStream(file.filepath);
      readStream.pipe(writeStream);

      // 可以在这里添加其他逻辑,如数据库记录保存等
      return { success: true, message: `http://127.0.0.1:7001/static/${newFileName}` };  // 通过链接访问
    } catch (err) {
      return { success: false, message: 'Failed to upload image' };
    }
  }
}

module.exports = uploadFileService;
posted @ 2025-02-23 15:55  我不喝鸡汤  阅读(40)  评论(0)    收藏  举报