使用Mongoose .populate()时处理不存在的关联ObjectId

Mongoose原生的populate()方法会自动过滤掉未匹配到的文档，无法直接保留无效的ObjectId并标记。以下是两种可行的实现方案：

方法一：手动对比原始ID与Populate结果（逻辑直观，前端友好） #

先获取用户的原始ID数组，单独查询有效歌曲后，合并结果并标记无效条目：

实现步骤： #

1. 查询用户原始数据（保留未处理的ID数组）：

// 使用lean()返回纯JSON对象，避免Mongoose文档对象的干扰
const userRaw = await User.findById(userId).lean();

2. 批量查询所有关联的有效歌曲：

// 收集所有需要校验的歌曲ID
const allSongIds = [
  ...userRaw.likedSongs,
  ...userRaw.playlists.flatMap(pl => pl.songs),
  ...userRaw.creator.songs
];
const validSongs = await Song.find({ _id: { $in: allSongIds } });

3. 将有效歌曲转为Map，提升查找效率：

const songMap = new Map(validSongs.map(song => [song._id.toString(), song]));

4. 遍历处理各个数组，标记无效ID：

// 处理likedSongs
userRaw.likedSongs = userRaw.likedSongs.map(id => {
  const idStr = id.toString();
  return songMap.get(idStr) || { _id: idStr, empty: true };
});

// 处理playlists中的songs
userRaw.playlists = userRaw.playlists.map(playlist => ({
  ...playlist,
  songs: playlist.songs.map(id => {
    const idStr = id.toString();
    return songMap.get(idStr) || { _id: idStr, empty: true };
  })
}));

// 处理creator.songs
userRaw.creator.songs = userRaw.creator.songs.map(id => {
  const idStr = id.toString();
  return songMap.get(idStr) || { _id: idStr, empty: true };
});

5. 返回处理后的userRaw给前端，前端可通过empty: true识别无效条目并提示用户清理。

方法二：使用MongoDB聚合查询（数据库层面处理，减少服务器内存消耗） #

利用MongoDB的$lookup（保留原始条目）和$map、$cond操作符，直接在数据库完成匹配与标记：

const [user] = await User.aggregate([
  // 匹配目标用户
  { $match: { _id: mongoose.Types.ObjectId(userId) } },
  // 一次性查询所有关联的有效歌曲
  {
    $lookup: {
      from: 'songs',
      localField: '$likedSongs',
      foreignField: '_id',
      as: 'validSongs'
    }
  },
  // 处理likedSongs数组，标记无效ID
  {
    $addFields: {
      likedSongs: {
        $map: {
          input: '$likedSongs',
          as: 'songId',
          in: {
            $cond: [
              { $in: ['$$songId', '$validSongs._id'] },
              { $arrayElemAt: ['$validSongs', { $indexOfArray: ['$validSongs._id', '$$songId'] }] },
              { _id: '$$songId', empty: true }
            ]
          }
        }
      }
    }
  },
  // 处理playlists中的songs数组
  {
    $addFields: {
      playlists: {
        $map: {
          input: '$playlists',
          as: 'playlist',
          in: {
            ...'$$playlist',
            songs: {
              $map: {
                input: '$$playlist.songs',
                as: 'songId',
                in: {
                  $cond: [
                    { $in: ['$$songId', '$validSongs._id'] },
                    { $arrayElemAt: ['$validSongs', { $indexOfArray: ['$validSongs._id', '$$songId'] }] },
                    { _id: '$$songId', empty: true }
                  ]
                }
              }
            }
          }
        }
      }
    }
  },
  // 处理creator.songs数组
  {
    $addFields: {
      'creator.songs': {
        $map: {
          input: '$creator.songs',
          as: 'songId',
          in: {
            $cond: [
              { $in: ['$$songId', '$validSongs._id'] },
              { $arrayElemAt: ['$validSongs', { $indexOfArray: ['$validSongs._id', '$$songId'] }] },
              { _id: '$$songId', empty: true }
            ]
          }
        }
      }
    }
  },
  // 移除临时字段
  { $project: { validSongs: 0 } }
]).exec();

用户确认清理后的后端处理 #

前端收集用户确认删除的无效ID后，用MongoDB的$pull操作符批量移除：

// 清理likedSongs中的无效ID
await User.findByIdAndUpdate(userId, {
  $pull: { likedSongs: { $in: invalidSongIds } }
});

// 清理所有playlists中的无效ID
await User.findByIdAndUpdate(userId, {
  $pull: { 'playlists.$[].songs': { $in: invalidSongIds } }
});

// 清理creator.songs中的无效ID
await User.findByIdAndUpdate(userId, {
  $pull: { 'creator.songs': { $in: invalidSongIds } }
});

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