MongoDB按日期单位聚合时如何为缺失日期生成零值？

针对你的需求，需要先生成start到end之间的完整日期序列，再和现有聚合结果做左连接补0，以下是修改后的聚合管道，同时我会逐段解释：

# 假设start、end、unit已提前定义（比如unit="day"）
MONGO_PIPE = [
    # 阶段1：生成完整的日期序列（从start到end，按unit间隔）
    {
        "$documents": [{"start": start, "end": end, "unit": unit}]
    },
    {
        "$addFields": {
            # 计算总间隔数：根据unit将end和start的差值转成对应单位的数量
            "totalIntervals": {
                "$dateDiff": {
                    "startDate": "$start",
                    "endDate": "$end",
                    "unit": "$unit"
                }
            }
        }
    },
    {
        "$addFields": {
            # 生成0到totalIntervals的偏移量数组（包含两端）
            "offsets": {"$range": [0, {"$add": ["$totalIntervals", 1]}]}
        }
    },
    {
        "$unwind": "$offsets"
    },
    {
        "$addFields": {
            # 根据偏移量生成每个日期点
            "fullDate": {
                "$dateAdd": {
                    "startDate": "$start",
                    "unit": "$unit",
                    "amount": "$offsets"
                }
            },
            # 格式化日期为需要的label格式
            "label": {
                "$dateToString": {"date": "$fullDate", "format": "%Y-%m-%dT%H:%M:%S"}
            }
        }
    },
    # 阶段2：获取原聚合的有数据的日期和值
    {
        "$lookup": {
            "from": "你的集合名称",  # 替换成实际的集合名
            "let": {"currentDate": "$fullDate"},
            "pipeline": [
                {
                    "$match": {
                        "$expr": {
                            "$and": [
                                {"$gte": ["$dt", start]},
                                {"$lte": ["$dt", end]},
                                # 匹配当前fullDate对应的unit截断后的日期
                                {"$eq": [
                                    {"$dateTrunc": {"date": "$dt", "unit": unit}},
                                    {"$dateTrunc": {"date": "$$currentDate", "unit": unit}}
                                ]}
                            ]
                        }
                    }
                },
                {
                    "$group": {
                        "_id": None,
                        "sumValue": {"$sum": "$value"}
                    }
                }
            ],
            "as": "aggResult"
        }
    },
    # 阶段3：处理左连接结果，补0
    {
        "$addFields": {
            "value": {
                "$cond": {
                    "if": {"$ne": [{"$size": "$aggResult"}, 0]},
                    "then": {"$arrayElemAt": ["$aggResult.sumValue", 0]},
                    "else": 0
                }
            }
        }
    },
    # 阶段4：整理成需要的dataset和labels格式
    {
        "$sort": {"fullDate": 1}
    },
    {
        "$group": {
            "_id": None,
            "dataset": {"$push": "$value"},
            "labels": {"$push": "$label"}
        }
    },
    {"$unset": "_id"}
]

关键步骤解释 #

• 阶段1：生成完整日期序列

  • $documents：创建临时文档存储start、end、unit参数
  • $dateDiff：计算start到end之间的总间隔数（比如day1到day5是4天间隔，加1后得到5个日期点）
  • $range：生成0到总间隔数的偏移量数组，对应每个日期的位置
  • $unwind：把偏移量数组拆成单个文档，每个文档对应一个日期点
  • $dateAdd：根据偏移量生成完整日期，$dateToString生成需求的label格式

• 阶段2：左连接原有聚合结果

  • $lookup：用生成的每个日期去关联原集合，匹配同一unit周期内的数据并求和
  • $dateTrunc确保匹配的是同一时间周期内的记录（比如unit=day时，匹配当天所有dt）

• 阶段3：补0处理

  • 用$cond判断：如果关联结果为空，就赋值0；否则取聚合得到的求和值

• 阶段4：整理输出格式

  • 按日期排序后，把数值和标签分别存入dataset和labels数组，最后移除_id字段

注意事项 #

• 替换代码中的"你的集合名称"为实际的集合名
• 确保start和end是ISODate类型（比如Python中用datetime.datetime(2024,1,1,tzinfo=pytz.utc)生成）
• 该方案支持$dateTrunc所有兼容的unit（hour、week、month等），只需对应传入unit参数即可

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