MongoDB聚合查询：如何为缺失月份补充值为0的空值

嘿，我完全懂你现在的痛点——聚合查询只会返回有数据的月份，但你需要把近四个月里哪怕没插入数据的月份也显示出来，值设为0对吧？别着急，我给你一套靠谱的解决方案，一步步来：

核心思路 #

要搞定这个问题，关键是先生成近四个月的完整月份列表（每个月份对应一个默认值0的文档），再把它和你原聚合得到的有数据的结果做合并，最后确保无数据的月份保留默认值。

完整聚合代码示例 #

先看完整的可运行代码，后面我再拆解关键步骤：

// 先计算近四个月的时间边界（包含当前月在内的四个月）
const now = new Date();
const fourMonthsAgo = new Date(now.setMonth(now.getMonth() - 3)); // 减3是因为要包含当前月，比如7月的话就是4、5、6、7月
const startYear = fourMonthsAgo.getFullYear();
const startMonth = fourMonthsAgo.getMonth() + 1; // MongoDB的$month返回1-12，所以要加1

db.collection.aggregate([
  // 1. 匹配近四个月的原始数据（和你原来的逻辑一致）
  {
    $match: {
      CREATE_DATE: {
        $gte: fourMonthsAgo,
        $lte: new Date()
      }
    }
  },
  // 2. 按年月分组计算平均值（一定要补全年份，避免跨年问题）
  {
    $group: {
      _id: {
        year: { $year: "$CREATE_DATE" },
        month: { $month: "$CREATE_DATE" }
      },
      avgValue: { $avg: "$YOUR_TARGET_FIELD" } // 替换成你要计算平均值的字段名
    }
  },
  // 3. 生成近四个月的完整月份模板（每个月份默认avgValue为0）
  {
    $unionWith: {
      coll: "collection", // 用同一个集合，或者MongoDB 5.0+可以直接用$documents生成
      pipeline: [
        { $project: { _id: 0, monthOffsets: { $range: [0, 4] } } }, // 生成0-3四个偏移量
        { $unwind: "$monthOffsets" },
        {
          $project: {
            _id: {
              // 处理跨年情况，比如12月+1变成1月，年份要加1
              $cond: {
                if: { $gt: [ { $add: [startMonth, "$monthOffsets"] }, 12 ] },
                then: {
                  year: { $add: [startYear, 1] },
                  month: { $mod: [ { $add: [startMonth, "$monthOffsets"] }, 12 ] }
                },
                else: {
                  year: startYear,
                  month: { $add: [startMonth, "$monthOffsets"] }
                }
              }
            },
            avgValue: 0 // 默认值设为0
          }
        }
      ]
    }
  },
  // 4. 按年月重新分组，用真实数据覆盖默认值
  {
    $group: {
      _id: "$_id",
      avgValue: { $max: "$avgValue" } // 有真实数据的话，平均值会覆盖0；如果你的值可能为0，改用$first+排序
    }
  },
  // 5. 可选：把月份转成英文名称，让结果更直观
  {
    $project: {
      _id: 0,
      monthName: {
        $arrayElemAt: [
          ["January", "February", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"],
          { $subtract: ["$_id.month", 1] }
        ]
      },
      avgValue: { $ifNull: ["$avgValue", 0] } // 双重保险，确保空值转0
    }
  },
  // 6. 按月份排序，保证结果顺序正确
  { $sort: { "monthName": 1 } }
])

关键步骤拆解 #

1. $unionWith的作用：这是核心！它把你原聚合得到的有数据的月份，和我们生成的四个默认月份（值为0）合并到一起，这样所有目标月份都被包含了。
2. 跨年处理：如果近四个月跨了年度（比如现在是1月，那就是去年10、11、12月和今年1月），代码里的$cond逻辑会自动调整年份，不会出现月份错误。
3. $max的小技巧：重新分组时用$max，因为真实的平均值（如果是正数的话）肯定大于0，所以会自动覆盖默认的0；如果你的业务场景中平均值可能为0，那可以先按avgValue倒序排序，再用$first取真实数据。

简化版（MongoDB 5.0+） #

如果你用的是MongoDB 5.0及以上版本，可以直接用$documents生成月份模板，不用依赖集合，代码更简洁：

// 替换上面的$unionWith部分
{
  $unionWith: {
    documents: [
      { _id: { year: 2024, month: 4 }, avgValue: 0 },
      { _id: { year: 2024, month: 5 }, avgValue: 0 },
      { _id: { year: 2024, month: 6 }, avgValue: 0 },
      { _id: { year: 2024, month: 7 }, avgValue: 0 }
    ]
  }
}

这种方式适合在后端代码里动态生成这四个文档对象，灵活性更高。

测试建议 #

你可以先单独运行生成月份模板的管道部分，确认四个月份都能正确生成（尤其是跨年的情况），再和原聚合结果合并，最后检查无数据的月份是否显示为0。

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