基于Express和Multer实现文件本地服务器文件上传功能

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

const app = express();

// 配置文件上传目录和文件名
const storage = multer.diskStorage({
  destination: function (req, file, cb) {
    cb(null, 'uploads/'); // 指定文件上传目录
  },
  filename: function (req, file, cb) {
    const uniqueSuffix = Date.now() + '-' + Math.round(Math.random() * 1E9);
    cb(null, file.fieldname + '-' + uniqueSuffix + path.extname(file.originalname)); // 指定文件名
  }
});

// 创建文件上传中间件
const upload = multer({ storage: storage });

// 处理文件上传
app.post('/upload', upload.single('file'), (req, res) => {
  if (req.file) {
    // 文件上传成功
    res.status(200).send('File uploaded successfully');
  } else {
    // 没有选择上传文件
    res.status(400).send('No file selected');
  }
});

const fileType = require('file-type');

// 在文件上传处理逻辑中添加文件类型验证
const file = req.file;
const fileBuffer = fs.readFileSync(file.path);
const fileInfo = fileType(fileBuffer);

if (!fileInfo || !isAllowedFileType(fileInfo.ext)) {
  // 不允许的文件类型
  fs.unlinkSync(file.path); // 删除上传的文件
  return res.status(400).send('Invalid file type');
}

const upload = multer({
  storage: storage,
  fileFilter: fileFilter,
  limits: {
    fileSize: 10 * 1024 * 1024 // 10MB
  }
});