如何通过Node.js的Multer将图片文件上传到MongoDB Atlas

1. 前置准备 #

• 已开通MongoDB Atlas集群，获取到数据库连接串，已替换连接串中的用户名、密码占位符，已配置Atlas的IP白名单（测试阶段可开允许所有IP访问）
• 本地已初始化Node.js项目
• 已安装所需依赖：
  npm install express multer mongoose dotenv

2. 适用场景说明 #

以下方案适合单张图片大小不超过16MB的场景，直接将图片二进制数据存在MongoDB普通文档中，新手易上手。如果需要存储大于16MB的图片，可改用MongoDB GridFS实现。

3. 分步实现 #

3.1 基础配置 #

在项目根目录新建.env文件，存储环境变量：

MONGO_URI=你的MongoDB Atlas连接串
PORT=3000

3.2 定义图片数据模型 #

新建models/Image.js文件，定义图片存储的Schema：

const mongoose = require('mongoose')

const imageSchema = new mongoose.Schema({
  // 图片名称
  name: String,
  img: {
    // 存储图片二进制数据
    data: Buffer,
    // 存储图片MIME类型，比如image/png、image/jpeg
    contentType: String
  }
})

module.exports = mongoose.model('Image', imageSchema)

3.3 编写服务入口代码 #

新建app.js作为项目入口文件：

require('dotenv').config()
const express = require('express')
const mongoose = require('mongoose')
const multer = require('multer')
const Image = require('./models/Image')
const app = express()

// 连接MongoDB Atlas
mongoose.connect(process.env.MONGO_URI)
  .then(() => console.log('数据库连接成功'))
  .catch(err => console.log('数据库连接失败：', err))

// 配置Multer使用内存存储，不生成本地临时文件
const storage = multer.memoryStorage()
// 可通过limits参数调整单文件大小限制，以下为10MB限制
const upload = multer({ 
  storage: storage,
  limits: { fileSize: 10 * 1024 * 1024 }
})

app.use(express.json())

// 单图上传接口，表单文件字段名为image
app.post('/upload', upload.single('image'), async (req, res) => {
  try {
    const saveImage = new Image({
      name: req.body.name || req.file.originalname,
      img: {
        data: req.file.buffer,
        contentType: req.file.mimetype
      }
    })
    await saveImage.save()
    res.status(200).json({
      code: 200,
      msg: '上传成功',
      imageId: saveImage._id
    })
  } catch (err) {
    res.status(500).json({
      code: 500,
      msg: '上传失败',
      error: err.message
    })
  }
})

// 启动服务
app.listen(process.env.PORT, () => {
  console.log(`服务运行在端口${process.env.PORT}`)
})

3.4 测试上传 #

启动服务后，用Postman发起测试请求：

• 请求方式：POST
• 请求地址：http://localhost:3000/upload
• 请求体格式：form-data
• 新增字段：image，类型选择File，选择本地要上传的图片；可选新增name字段自定义图片名称

4. 注意事项 #

• 不要将MongoDB Atlas连接串硬编码在业务代码中，必须通过环境变量存储，避免凭证泄露
• 生产环境不要开Atlas的允许所有IP访问配置，仅添加业务服务的固定IP到白名单
• 如果需要返回可直接访问的图片链接，可额外新增一个接口根据图片ID查询二进制数据，设置对应Content-Type后返回buffer即可

内容的提问来源于stack exchange，提问作者jayesh kaushal