MongoDB Atlas中GridFS音频文件通过ID播放失败问题排查

我在Heroku上部署了基于Next.js和Express的应用，用于处理MongoDB Atlas数据库GridFS中的音频文件。fs.files集合中有如下记录：

_id: 46e737001837a062ce85dd4c
length: 214023
chunkSize: 261120
uploadDate: 2023-08-12T05:09:22.521+00:00
filename: "Audio_2023-09-18:14:39:26"

访问链接https://myapp.herokuapp.com/download?afn=Audio_2023-09-18:14:39:26时音频可正常播放，但访问https://myapp.herokuapp.com/download?id=46e737001837a062ce85dd4c时无法播放且报错，尝试多种ID转换方式均无效。相关路由代码如下：

server.get('/download', (req, res) => {
  const gridFSBucket = new mongoose.mongo.GridFSBucket(conn.db);
  const dataType = "audio/mp3";

  if (req.query.afn) {
    const filename = req.query.afn;
    const downloadStream = gridFSBucket.openDownloadStreamByName(filename);
     res.set('Content-Type', dataType);
    res.set('Content-Disposition', `inline; filename="${filename}"`);
    downloadStream.pipe(res)
  }

  if (req.query.id) {
    // const recordID = req.query.id.toString(); // This does not work !
    // const recordID = new ObjectID(req.query.id); // This does not work !
    // const recordID = mongoose.Types.ObjectID(req.query.id); // This does not work !
    const recordID = mongoose.Types.ObjectID.from(req.query.id); // This does not work !
    const downloadStream = gridFSBucket.openDownloadStream(recordID);
    res.set('Content-Type', dataType)
    res.set('Content-Disposition', `inline; filename=BLABLAH`);
    downloadStream.pipe(res);
  }
});

问题分析 #

1. ObjectID类型不匹配：GridFSBucket.openDownloadStream需要的是MongoDB原生的ObjectId实例，你尝试的mongoose.Types.ObjectID是mongoose封装后的类型，在部分版本中与原生类型不兼容，导致无法正确匹配数据库中的_id。
2. 缺少错误处理：没有捕获ID格式错误和下载流异常，无法定位具体失败原因（比如ID无效、文件不存在或权限问题）。

解决方案 #

1. 使用MongoDB原生ObjectId #

从mongoose.mongo模块直接引入原生ObjectId，避免mongoose封装层的差异：

// 在文件顶部引入原生ObjectId
const { ObjectId } = mongoose.mongo;

修改ID分支的代码，添加错误捕获逻辑：

if (req.query.id) {
  try {
    // 用原生ObjectId构造函数创建实例
    const recordID = new ObjectId(req.query.id);
    const downloadStream = gridFSBucket.openDownloadStream(recordID);
    
    // 捕获下载流错误（比如文件不存在）
    downloadStream.on('error', (err) => {
      console.error('下载流错误:', err);
      res.status(404).send('音频文件不存在或ID无效');
    });
    
    res.set('Content-Type', dataType);
    res.set('Content-Disposition', `inline; filename=BLABLAH`);
    downloadStream.pipe(res);
  } catch (err) {
    // 捕获ID格式错误（比如非24字符十六进制字符串）
    console.error('ID格式错误:', err);
    res.status(400).send('无效的文件ID');
  }
}

2. 验证GridFSBucket初始化（可选） #

如果自定义了GridFS存储桶名称，需要显式指定bucketName参数，确保指向正确的存储桶：

const gridFSBucket = new mongoose.mongo.GridFSBucket(conn.db, {
  bucketName: 'fs' // 默认值为fs，若你的存储桶名不同需修改
});

3. 确认ID的有效性 #

检查传入的ID是否与fs.files集合中的_id完全一致，确保没有多余空格或特殊字符，MongoDB的ObjectId必须是24字符的十六进制字符串。

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