MongoDB聚合改造：排序时将Null值置于末尾的实现方案

我来帮你改造这段Express路由代码，用MongoDB聚合查询解决Null值排序异常的问题，同时完整保留原有的文本搜索、多维度排序、分页等功能。核心思路是通过聚合的$addFields阶段添加一个辅助字段，标记目标排序字段是否为Null，然后先按这个辅助字段排序把Null值推到列表末尾，再按实际业务字段排序。

改造后的完整代码 #

router.get('/search/:page', async function (req, res) {
  const page = parseInt(req.params.page) || 0;
  const { search, sort } = req.query;
  const hasSearch = !!search;

  // 获取排序配置：包含辅助字段、排序规则、需要添加的isNull字段
  const { sortStage, addFieldsStage } = sortBy(sort, hasSearch);

  // 构建聚合管道
  const pipeline = [];

  // 1. 文本搜索匹配阶段
  if (hasSearch) {
    pipeline.push({
      $match: { $text: { $search: search } }
    });
    // 文本搜索需要添加score字段
    if (!addFieldsStage) {
      pipeline.push({ $addFields: { score: { $meta: 'textScore' } } });
    } else {
      // 合并score字段到addFields
      addFieldsStage.$addFields.score = { $meta: 'textScore' };
      pipeline.push(addFieldsStage);
    }
  } else if (addFieldsStage) {
    // 无搜索但有排序辅助字段时添加
    pipeline.push(addFieldsStage);
  }

  // 2. 字段投影阶段（对应原select）
  pipeline.push({
    $project: {
      _id: 0,
      extractor: 1,
      id: 1,
      title: 1,
      mediumResizedThumbnailFile: 1,
      directory: 1,
      uploader: 1,
      videoFile: 1,
      uploadDate: 1,
      duration: 1,
      width: 1,
      height: 1,
      viewCount: 1,
      // 按需包含排序相关字段和score
      ...(sortStage.likeCount && { likeCount: 1 }),
      ...(sortStage.dislikeCount && { dislikeCount: 1 }),
      ...(hasSearch && { score: 1 })
    }
  });

  // 3. 排序阶段
  pipeline.push({ $sort: sortStage });

  // 4. 分页阶段
  pipeline.push({ $skip: page * parsedEnv.PAGE_SIZE });
  pipeline.push({ $limit: parsedEnv.PAGE_SIZE });

  try {
    const videos = await Video.aggregate(pipeline).exec();
    res.json({ videos });
  } catch (err) {
    console.error('视频搜索出错:', err);
    return res.sendStatus(500);
  }
});

/**
 * 生成聚合所需的排序阶段和辅助字段阶段
 * @param {string} option 排序选项
 * @param {boolean} hasSearch 是否有文本搜索
 * @returns {Object} { sortStage, addFieldsStage }
 */
const sortBy = (option, hasSearch = false) => {
  let sortStage = {};
  let addFieldsStage = null;

  switch (option) {
    case 'relevance':
      if (hasSearch) {
        sortStage = { score: { $meta: 'textScore' }, uploadDate: -1 };
      } else {
        sortStage = { uploadDate: -1 };
      }
      break;
    case 'newest_date':
      sortStage = { uploadDate: -1 };
      break;
    case 'oldest_date':
      sortStage = { uploadDate: 1 };
      break;
    case 'longest_duration':
      sortStage = { duration: -1 };
      break;
    case 'shortest_duration':
      sortStage = { duration: 1 };
      break;
    case 'largest_size':
      sortStage = { 'videoFile.filesize': -1 };
      break;
    case 'smallest_size':
      sortStage = { 'videoFile.filesize': 1 };
      break;
    case 'most_views':
      // 添加isNull辅助字段，Null值排后面
      addFieldsStage = {
        $addFields: { isViewCountNull: { $eq: ["$viewCount", null] } }
      };
      sortStage = { isViewCountNull: 1, viewCount: -1 };
      break;
    case 'least_views':
      addFieldsStage = {
        $addFields: { isViewCountNull: { $eq: ["$viewCount", null] } }
      };
      sortStage = { isViewCountNull: 1, viewCount: 1 };
      break;
    case 'most_likes':
      addFieldsStage = {
        $addFields: { isLikeCountNull: { $eq: ["$likeCount", null] } }
      };
      sortStage = { isLikeCountNull: 1, likeCount: -1 };
      break;
    case 'least_likes':
      addFieldsStage = {
        $addFields: { isLikeCountNull: { $eq: ["$likeCount", null] } }
      };
      sortStage = { isLikeCountNull: 1, likeCount: 1 };
      break;
    case 'most_dislikes':
      addFieldsStage = {
        $addFields: { isDislikeCountNull: { $eq: ["$dislikeCount", null] } }
      };
      sortStage = { isDislikeCountNull: 1, dislikeCount: -1 };
      break;
    case 'least_dislikes':
      addFieldsStage = {
        $addFields: { isDislikeCountNull: { $eq: ["$dislikeCount", null] } }
      };
      sortStage = { isDislikeCountNull: 1, dislikeCount: 1 };
      break;
    default:
      sortStage = { uploadDate: -1 };
  }

  return { sortStage, addFieldsStage };
};

关键改造点说明 #

1. 辅助字段标记Null值
   对于可能存在Null值的排序字段（viewCount、likeCount、dislikeCount），我们通过$addFields添加一个形如isXXXNull的字段，用$eq判断该字段是否为Null。排序时先按这个字段升序（1），这样true（Null值）会排在false（有值）的后面。

2. 聚合管道分步处理

   • $match：处理文本搜索的匹配逻辑，和原find的pattern一致。
   • $addFields：动态添加Null标记字段和文本搜索的score字段。
   • $project：精确控制返回的字段，避免冗余数据，对应原select方法。
   • $sort：结合Null标记字段和业务字段排序，确保Null值始终在末尾。
   • $skip/$limit：保持原有的分页逻辑不变。

3. 兼容文本搜索排序
   当用户选择relevance排序且有搜索关键词时，依然优先按textScore排序，同时保留uploadDate作为次要排序字段，保证结果的合理性。

内容的提问来源于stack exchange，提问作者No stupid questions