Nodejs 应用gridfs-stream模块上传图片到mongodb数据库

Node.js 应用 gridfs-stream 模块上传图片到 MongoDB 数据库

在本教程中，我们将介绍如何使用 Node.js 和 gridfs-stream 模块将图片文件上传到 MongoDB 数据库。这将包括安装必要的依赖项、配置连接以及实现文件上传逻辑。

安装依赖项

首先，确保你已经安装了以下依赖项：

npm install mongoose gridfs-stream multer

• Mongoose：用于连接和操作 MongoDB。
• gridfs-stream：用于通过流式接口与 MongoDB 的 GridFS 进行交互。
• Multer：一个 Node.js 中间件，用于处理 multipart/form-data，主要用于上传文件。

配置连接

接下来，配置 Mongoose 连接到 MongoDB，并初始化 gridfs-stream：

const mongoose = require('mongoose');
const Grid = require('gridfs-stream');
const GridFile = require('./models/gridfs').GridFile; // 假设你有一个 GridFile 模型

// 连接 MongoDB
mongoose.connect('mongodb://localhost/mydatabase', { useNewUrlParser: true, useUnifiedTopology: true });

// 初始化 gridfs-stream
let gfs;
mongoose.connection.once('open', () => {
  gfs = Grid(mongoose.connection.db, mongoose.mongo);
});

实现文件上传逻辑

现在我们来实现文件上传逻辑。我们将使用 Multer 来处理文件上传请求：

const express = require('express');
const multer = require('multer');
const fs = require('fs');
const async = require('async');

const app = express();
const upload = multer({ dest: 'uploads/' });

app.post('/upload', upload.array('files'), (req, res) => {
  let fnames = Array.isArray(req.files) ? req.files : [req.files];

  async.map(fnames, processFile, (err, results) => {
    res.json({
      err: err,
      results: results
    });
  });
});

function processFile(file, callback) {
  const gfsOptions = {
    filename: file.originalname,
    mode: 'w',
    content_type: file.mimetype,
    metadata: {
      client: req.user.client.id,
      user: req.user.id
    }
  };

  const writeStream = gfs.createWriteStream(gfsOptions);
  fs.createReadStream(file.path).pipe(writeStream);

  writeStream.on('close', (file) => {
    fs.unlink(file.path, (err) => {
      if (err) {
        return callback(err, null);
      }
      callback(null, file);
    });
  });
}

app.listen(3000, () => console.log('Server running on port 3000'));

解释

1. 配置连接：

   • 使用 Mongoose 连接到 MongoDB。
   • 初始化 gridfs-stream 以便我们可以使用它来读写文件。

2. 文件上传逻辑：

   • 使用 Multer 处理文件上传。
   • 将上传的文件保存到临时目录（例如 uploads/）。
   • 使用 async.map 并行处理每个上传的文件。
   • 创建一个 writeStream 并将上传的文件流写入 MongoDB。
   • 在文件写入完成后删除临时文件。

通过上述步骤，你可以使用 Node.js 和 gridfs-stream 模块将图片文件上传到 MongoDB 数据库。