能否通过Multer-GridFS-Storage下载MongoDB集合中的文件？

嘿，恭喜你已经搞定了文件上传的部分！接下来咱们来解决从MongoDB的GridFS中下载文件的问题，其实思路很清晰，我给你一步步拆解，附上具体的代码示例：

1. 准备必要的模块 #

首先确保你已经引入了mongodb包（Multer-GridFS-Storage依赖它），如果用mongoose的话也可以直接用mongoose的连接实例：

const mongoose = require('mongoose');
const { GridFSBucket } = require('mongodb');
// 如果你用原生MongoDB客户端的话：
// const { MongoClient, GridFSBucket } = require('mongodb');

2. 创建GridFSBucket实例 #

这个实例是和你上传时的存储桶关联的，必须保证bucketName和你上传配置里的一致：

// 假设你已经通过mongoose成功连接了数据库
const db = mongoose.connection.db;
// 替换成你上传时设置的bucketName，比如你代码里的bucketName:...
const bucket = new GridFSBucket(db, { bucketName: 'your-bucket-name' });

// 如果是原生MongoDB客户端的话，连接后获取db：
// let db;
// MongoClient.connect(gridfsDBString)
//   .then(client => {
//     db = client.db();
//     bucket = new GridFSBucket(db, { bucketName: 'your-bucket-name' });
//   })
//   .catch(err => console.error('数据库连接失败:', err));

3. 编写下载路由 #

这里提供两种常见的下载方式，你可以根据需求选择：

方式一：根据文件名下载 #

适合文件名唯一的场景：

app.get('/download/:filename', (req, res) => {
  const targetFilename = req.params.filename;

  // 先查询文件是否存在
  bucket.find({ filename: targetFilename }).toArray((err, files) => {
    if (err) {
      return res.status(500).json({ message: '查询文件失败', error: err.message });
    }
    if (!files || files.length === 0) {
      return res.status(404).json({ message: '未找到该文件' });
    }

    // 设置响应头，让浏览器触发下载（如果想在线预览，可以把attachment改成inline）
    res.setHeader('Content-Disposition', `attachment; filename="${files[0].filename}"`);
    res.setHeader('Content-Type', files[0].contentType);

    // 创建下载流并pipe到响应对象
    const downloadStream = bucket.openDownloadStreamByName(targetFilename);
    downloadStream.pipe(res);

    // 处理流的错误
    downloadStream.on('error', (streamErr) => {
      res.status(500).json({ message: '下载失败', error: streamErr.message });
    });
  });
});

方式二：根据文件ID下载 #

如果文件名可能重复，用文件的_id更准确：

app.get('/download-by-id/:fileId', (req, res) => {
  const fileId = req.params.fileId;

  try {
    // 把字符串ID转换成MongoDB的ObjectId类型
    const objectId = new mongoose.Types.ObjectId(fileId);
    const downloadStream = bucket.openDownloadStream(objectId);

    // 监听file事件获取文件元信息，设置响应头
    downloadStream.on('file', (file) => {
      res.setHeader('Content-Disposition', `attachment; filename="${file.filename}"`);
      res.setHeader('Content-Type', file.contentType);
    });

    downloadStream.pipe(res);

    // 处理文件不存在的情况
    downloadStream.on('error', () => {
      res.status(404).json({ message: '未找到该文件或下载失败' });
    });
  } catch (err) {
    res.status(400).json({ message: '无效的文件ID格式' });
  }
});

4. 测试下载 #

启动你的服务后，直接在浏览器访问对应的路由即可，比如：

• 根据文件名下载：http://localhost:3000/download/your-file-name.jpg
• 根据ID下载：http://localhost:3000/download-by-id/60d21b4667d0d8992e610c85

这样就能触发文件下载啦！

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