使用mongoose在mongo db中存储图像

npm install mongoose multer

const mongoose = require('mongoose');

const imageSchema = new mongoose.Schema({
  fileName: String,
  fileType: String,
  imageData: Buffer
});

const Image = mongoose.model('Image', imageSchema);

module.exports = Image;

const express = require('express');
const multer = require('multer');
const Image = require('../models/image');

const router = express.Router();

// 创建multer实例，配置图像存储路径和文件名
const storage = multer.memoryStorage();
const upload = multer({ storage });

// 处理图像上传
router.post('/upload', upload.single('image'), async (req, res) => {
  try {
    const { originalname, mimetype, buffer } = req.file;

    // 创建图像对象并保存到数据库
    const image = new Image({
      fileName: originalname,
      fileType: mimetype,
      imageData: buffer
    });
    await image.save();

    res.status(200).json({ message: '图像上传成功' });
  } catch (error) {
    res.status(500).json({ message: '图像上传失败' });
  }
});

// 获取图像
router.get('/:id', async (req, res) => {
  try {
    const image = await Image.findById(req.params.id);

    if (!image) {
      return res.status(404).json({ message: '找不到图像' });
    }

    res.set('Content-Type', image.fileType);
    res.send(image.imageData);
  } catch (error) {
    res.status(500).json({ message: '获取图像失败' });
  }
});

module.exports = router;

const express = require('express');
const mongoose = require('mongoose');
const imageRoutes = require('./routes/image');

const app = express();

// 连接到MongoDB数据库
mongoose.connect('mongodb://localhost/mydatabase', {
  useNewUrlParser: true,
  useUnifiedTopology: true
}).then(() => {
  console.log('已连接到数据库');
}).catch((error) => {
  console.error('数据库连接失败', error);
});

// 添加图像路由
app.use('/images', imageRoutes);

// 启动应用程序
app.listen(3000, () => {
  console.log('应用程序已启动');
});