Node.js+Multer多类型文件上传处理问题求助

嘿，这个需求其实挺常见的，我来给你捋个清晰的实现思路，新手也能快速上手～

核心思路是：先用Multer的内存存储接收所有符合校验规则的文件，然后在路由处理函数里根据文件类型分情况处理——PDF直接写入磁盘，图片则流转到你的图片处理函数。这种方式比自定义存储引擎更直观，对新手友好。

步骤1：完善Multer配置（内存存储+校验逻辑） #

首先我们配置Multer使用内存存储，同时在fileFilter里完成类型和大小的校验：

const multer = require('multer');
const fs = require('fs');
const path = require('path');

// 定义允许的文件类型
const allowedTypes = {
  images: ['image/jpeg', 'image/png', 'image/gif'],
  pdfs: ['application/pdf']
};

// 配置内存存储：将文件暂存到内存buffer中
const storage = multer.memoryStorage();

// 完善fileFilter的类型与大小校验
const fileFilter = (req, file, cb) => {
  // 校验文件类型
  const isImage = allowedTypes.images.includes(file.mimetype);
  const isPdf = allowedTypes.pdfs.includes(file.mimetype);
  
  if (!isImage && !isPdf) {
    return cb(new Error('仅支持图片（JPG/PNG/GIF）和PDF文件'), false);
  }

  // 校验文件大小（示例：最大5MB）
  const maxSize = 5 * 1024 * 1024;
  if (file.size > maxSize) {
    return cb(new Error('文件大小不能超过5MB'), false);
  }

  // 给文件标记类型，方便后续快速判断
  file.isImage = isImage;
  file.isPdf = isPdf;
  cb(null, true);
};

// 创建Multer实例
const upload = multer({
  storage: storage,
  fileFilter: fileFilter
});

步骤2：编写路由处理函数（分类型处理文件） #

在路由里使用上面的upload中间件，然后根据之前标记的file.isImage/file.isPdf来分别处理：

const express = require('express');
const router = express.Router();

// 示例：你的图片处理函数（比如压缩、加水印等）
const processImage = async (imageBuffer) => {
  // 这里替换成你的实际图片处理逻辑，比如用sharp库：
  // const sharp = require('sharp');
  // const processedBuffer = await sharp(imageBuffer).resize(800).toBuffer();
  // return processedBuffer;
  
  // 先模拟返回原buffer，你可以按需修改
  return imageBuffer;
};

// 确保PDF存储目录存在（不存在则自动创建）
const pdfDir = path.join(__dirname, 'uploads/pdfs');
if (!fs.existsSync(pdfDir)) {
  fs.mkdirSync(pdfDir, { recursive: true });
}

// 上传接口
router.post('/upload', upload.single('file'), async (req, res) => {
  try {
    const file = req.file;
    if (!file) {
      return res.status(400).json({ message: '请选择要上传的文件' });
    }

    if (file.isPdf) {
      // PDF文件：直接写入磁盘
      const pdfPath = path.join(pdfDir, `${Date.now()}_${file.originalname}`);
      fs.writeFileSync(pdfPath, file.buffer);
      return res.json({
        message: 'PDF上传成功',
        filePath: pdfPath
      });
    }

    if (file.isImage) {
      // 图片文件：流转到处理函数
      const processedImage = await processImage(file.buffer);
      // 处理完成后可选择存储到磁盘或返回给前端
      const imageDir = path.join(__dirname, 'uploads/images');
      if (!fs.existsSync(imageDir)) {
        fs.mkdirSync(imageDir, { recursive: true });
      }
      const imagePath = path.join(imageDir, `${Date.now()}_${file.originalname}`);
      fs.writeFileSync(imagePath, processedImage);
      return res.json({
        message: '图片处理并上传成功',
        processedFilePath: imagePath
      });
    }
  } catch (error) {
    res.status(500).json({ message: '上传失败', error: error.message });
  }
});

module.exports = router;

关键细节说明 #

1. 内存存储的优势：用memoryStorage可以让所有文件先存在内存的buffer里，摆脱Multer默认直接存磁盘的限制，让我们能灵活决定后续处理逻辑。
2. 文件类型标记：在fileFilter里给file对象添加isImage/isPdf属性，避免后续重复解析mimetype，代码更简洁。
3. 目录自动创建：用fs.mkdirSync加recursive: true确保存储目录存在，避免写入文件时因目录不存在报错。
4. 多文件上传适配：如果需要一次上传多个文件，只需把upload.single('file')改成upload.array('files', 5)（最多5个文件），然后在处理函数里循环req.files数组即可，逻辑和单个文件一致。

内容的提问来源于stack exchange，提问作者Petr Fila