如何在 Multer 中灵活处理不同文件类型的大小限制？

使用 Multer 灵活处理文件大小限制

引言

在处理文件上传时，限制文件大小至关重要，它可以防止服务器过载、保护用户免受恶意软件侵害，并确保符合存储限制。对于处理不同文件类型的应用程序，使用一个 multer 实例灵活设置不同的文件大小限制非常重要。这篇文章将深入探讨使用 multer 和自定义文件过滤器来实现此目标的方法。

问题陈述

如何在一个 multer 实例中处理不同文件类型（例如图像、视频）的不同大小限制？

解决方案

1. 使用 multer 的磁盘存储引擎

第一步是使用 multer 的 diskStorage() 函数设置存储位置和命名约定。这将允许我们根据文件类型将文件存储在不同的位置。

const storage = multer.diskStorage({
  destination: (req, file, cb) => {
    cb(null, "uploads/");
  },
  filename: (req, file, cb) => {
    const ext = path.extname(file.originalname);
    cb(null, Date.now() + ext);
  },
});

2. 设置自定义文件过滤器

接下来，我们需要创建一个自定义文件过滤器来验证文件的类型和大小。该过滤器将在 multer 处理文件之前运行。

const imageFilter = (req, file, cb) => {
  if (file.mimetype.startsWith("image")) {
    // 检查图像大小是否在限制范围内
    if (file.size > imageSizeLimit) {
      cb(new multer.MulterError("LIMIT_FILE_SIZE"), false);
    } else {
      cb(null, true);
    }
  } else {
    cb(new multer.MulterError("LIMIT_FILE_TYPE"), false);
  }
};

3. 配置 multer 实例

最后，使用自定义存储引擎和文件过滤器配置 multer 实例。

const upload = multer({
  storage,
  fileFilter: (req, file, cb) => {
    // 根据请求中的 "type" 字段选择过滤器
    if (req.body.type === "image") {
      imageFilter(req, file, cb);
    } else if (req.body.type === "video") {
      videoFilter(req, file, cb);
    } else {
      cb(new multer.MulterError("LIMIT_FILE_TYPE"), false);
    }
  },
});

使用示例

在你的路由中使用 multer 中间件：

const router = express.Router();

router.post("/upload", upload.single("file"), (req, res) => {
  res.json({
    file: req.file,
  });
});

结论

通过结合 multer 的磁盘存储引擎、自定义文件过滤器和灵活的配置选项，你可以轻松地在单个 multer 实例中处理不同文件类型和大小限制。这提供了更大的灵活性，同时保持安全性和效率。

常见问题解答

Q1：为什么使用磁盘存储引擎？
A：磁盘存储引擎允许根据文件类型将文件存储在不同位置，便于管理和组织。

Q2：如何指定文件类型？
A：使用 req.body.type 字段在请求中指定文件类型，并在 multer 配置中根据该字段选择过滤器。

Q3：如何自定义大小限制？
A：通过更新 imageSizeLimit 和 videoSizeLimit 变量可以自定义不同文件类型的大小限制。

Q4：我可以在multer中处理其他文件类型吗？
A：当然可以，只需创建一个类似 imageFilter 的自定义过滤器即可处理其他文件类型。

Q5：如何使用 multer 处理视频？
A：对于视频上传，请使用 multipart/form-data 编码类型，并确保视频大小限制合理。