如何通过ID获取MongoDB GridFS文件的文件名并下载？

核心问题拆解 #

• 遍历所有文件获取文件名属于全表扫描，效率极低，应通过ID精准查询
• bucket.find({_id: id})失效是因为传入的id是字符串，GridFS的_id存储为ObjectId类型，需先转换
• 直接查询fs.files提示db未定义，是因为未正确使用已获取的数据库连接实例

优化后的代码实现 #

方式一：用GridFSBucket精准查询 #

修改控制器中的downloadFileByID函数，直接通过ObjectId定位文件，无需遍历：

const downloadFileByID = async (req, res) => {
  const { id } = req.params

  if (!mongoose.Types.ObjectId.isValid(id)){
      return res.status(404).json({error: 'No such file'})
  }
  const mongooseID = new mongoose.Types.ObjectId(id)

  try{
    // 精准查询单个文件，替代全表遍历
    const file = await bucket.find({ _id: mongooseID }).next();
    if (!file) {
      return res.status(404).json({error: "File not found, file ID does not exist"});
    }

    // 用查询到的filename生成写入流
    bucket.openDownloadStream(mongooseID).pipe(fs.createWriteStream('./' + file.filename));
    res.send({ filename: file.filename });
  } catch (error) {
    res.status(400).json({error: error.message || "Download failed"});
  }
}

方式二：直接操作fs.files集合 #

利用已获取的conn实例（mongoose连接成功后赋值）直接查询集合：

const downloadFileByID = async (req, res) => {
  const { id } = req.params

  if (!mongoose.Types.ObjectId.isValid(id)){
      return res.status(404).json({error: 'No such file'})
  }
  const mongooseID = new mongoose.Types.ObjectId(id)

  try{
    // 直接查询fs.files集合获取文件信息
    const file = await conn.collection('fs.files').findOne({ _id: mongooseID });
    if (!file) {
      return res.status(404).json({error: "File not found, file ID does not exist"});
    }

    bucket.openDownloadStream(mongooseID).pipe(fs.createWriteStream('./' + file.filename));
    res.send({ filename: file.filename });
  } catch (error) {
    res.status(400).json({error: error.message || "Download failed"});
  }
}

关键优化点 #

• 替换全表遍历为精准ID查询，查询效率从O(n)降至O(1)
• 确保查询条件的_id为ObjectId类型，匹配GridFS存储格式
• 复用已有的数据库连接实例conn，解决db未定义问题
• 增加文件不存在的前置判断，避免后续流操作报错

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