在MongoDB中使用GridFS从express中保存缓冲区

const express = require('express');
const multer = require('multer');
const GridFsStorage = require('multer-gridfs-storage');
const mongoose = require('mongoose');
const Grid = require('gridfs-stream');
const crypto = require('crypto');

const mongoURI = 'mongodb://localhost:27017/mydatabase'; // 替换为你的MongoDB连接URI
const conn = mongoose.createConnection(mongoURI);

let gfs;
conn.once('open', () => {
  gfs = Grid(conn.db, mongoose.mongo);
  gfs.collection('uploads'); // 创建一个名为'uploads'的集合用于存储文件
});

const storage = new GridFsStorage({
  url: mongoURI,
  file: (req, file) => {
    return new Promise((resolve, reject) => {
      crypto.randomBytes(16, (err, buf) => {
        if (err) {
          return reject(err);
        }
        const filename = buf.toString('hex') + path.extname(file.originalname);
        const fileInfo = {
          filename: filename,
          bucketName: 'uploads' // 与上面创建的集合名称一致
        };
        resolve(fileInfo);
      });
    });
  }
});

const upload = multer({ storage });

const router = express.Router();

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

app.use('/api', router);