
1. 背景与挑战
在处理用户生日等数据时,我们经常需要查询特定日期范围内的生日,例如“查询所有5月1日到6月1日之间生日的员工”。这里的关键挑战在于,生日的查询通常只需要关注月份和日期,而忽略年份。传统的日期范围查询(如$gte和$lte直接应用于Date类型字段)会包含年份信息,导致无法满足仅按月日查询的需求。例如,将所有生日的年份统一设置为当前年份进行比较,虽然在某些情况下可行,但在处理跨年日期范围(如12月到次年1月)时会变得复杂且容易出错。
例如,如果一个员工的生日是1990-05-02,我们希望在2023-01-01到2023-06-01的范围内找到他,因为05-02确实位于01-01和06-01之间。而简单的年份设置方法,如:
if (birthDateStart) {
const currentYear = dayjs().year();
search.birthDate = {
$gte: dayjs(birthDateStart).set('year', currentYear).toDate(),
};
}
// ... 类似地设置 birthDateEnd这种方法在跨年查询(例如,查询12月15日到次年1月15日之间的生日)时会失效,因为它将所有日期强制拉到同一年份进行比较,无法正确处理日期“环绕”的情况。
2. 解决方案:MongoDB聚合框架
MongoDB的聚合框架提供了一种强大且灵活的方式来处理这类复杂查询。核心思想是:
- 从存储的birthDate字段中提取出月份和日期。
- 将提取出的月份和日期组合成一个单一的可比较数值。
- 根据用户输入的起始和结束日期,计算出对应的月份和日期的组合值。
- 使用$match阶段对这些组合值进行范围过滤。
2.1 提取月份和日期
首先,我们需要在聚合管道中使用$project阶段来提取birthDate字段的月份和日期。如果birthDate存储为ISO日期字符串,我们需要先用$dateFromString将其转换为日期对象。
// 示例输入日期
const birthDateStart = "2023-01-01"; // 用户输入的起始日期
const birthDateEnd = "2023-06-01"; // 用户输入的结束日期
// 在应用层处理输入日期,获取月份和日期
const dayjs = require('dayjs'); // 假设使用dayjs库
const startMonth = dayjs(birthDateStart).month() + 1; // 月份 (1-12)
const startDay = dayjs(birthDateStart).date(); // 日期 (1-31)
const endMonth = dayjs(birthDateEnd).month() + 1;
const endDay = dayjs(birthDateEnd).date();
// 将月份和日期转换为一个可比较的整数值,例如:月*100 + 日
// 示例:1月1日 -> 101, 5月2日 -> 502, 12月31日 -> 1231
const startMonthDayValue = startMonth * 100 + startDay;
const endMonthDayValue = endMonth * 100 + endDay;
// MongoDB 聚合管道
const pipeline = [
{
$project: {
_id: 1,
birthDate: 1,
// 从 birthDate 字段中提取月份和日期
// 假设 birthDate 已经是 BSON Date 类型或可解析的日期字符串
month: { "$month": "$birthDate" },
day: { "$dayOfMonth": "$birthDate" }
// 如果 birthDate 是字符串,可能需要先转换:
// month: { "$month": { "$dateFromString": { "date










