Ruby基于计划起始日期的当月周期数据查询实现咨询

需求回顾 #

你需要根据固定的计划起始日（比如每月5号）动态计算查询周期：

• 若当前日期在当月起始日之前（比如12月3日，起始日是5号），则查询上个月起始日到当前日的记录（11月5日-12月3日）
• 若当前日期在当月起始日之后或当天（比如12月15日），则查询当月起始日到当前日的记录（12月5日-12月15日）

实现步骤 #

1. 编写时间计算辅助方法

先写一个方法计算当前周期的起始和结束时间，接收计划起始日（整数，比如5）作为参数：

def calculate_current_period(plan_start_day)
  today = DateTime.now
  current_year = today.year
  current_month = today.month

  # 判断当前日期是否大于等于计划起始日
  if today.day >= plan_start_day
    # 当月起始日
    period_start = DateTime.new(current_year, current_month, plan_start_day)
  else
    # 上个月起始日：处理跨年边界情况（比如1月5号，当前是12月3日）
    previous_month = current_month == 1 ? 12 : current_month - 1
    previous_year = current_month == 1 ? current_year - 1 : current_year
    period_start = DateTime.new(previous_year, previous_month, plan_start_day)
  end

  period_end = today
  [period_start, period_end]
end

2. 结合Mongoid进行数据查询

拿到计算好的周期时间后，用Mongoid的查询语法筛选数据。假设你要查询created_at在周期内的Customer记录：

# 设定计划起始日为5号
plan_start_day = 5
period_start, period_end = calculate_current_period(plan_start_day)

# 执行查询
filtered_customers = Customer.where(
  created_at: { :$gte => period_start, :$lte => period_end }
)

# 如果需要查询start_time字段，替换created_at即可
# filtered_customers = Customer.where(
#   start_time: { :$gte => period_start, :$lte => period_end }
# )

代码解释 #

• calculate_current_period方法：
  • 获取当前日期的年、月、日信息
  • 根据当前日与计划起始日的对比，确定周期起始是当月还是上个月的起始日，同时兼容跨年场景
  • 返回计算完成的周期起始和结束时间（结束时间为当前时间）
• Mongoid查询部分：
  • 使用where配合MongoDB的$gte（大于等于）和$lte（小于等于）操作符，筛选时间范围内的记录
  • 可根据业务需求替换为start_time或其他日期字段

测试示例 #

• 当今日是2022年12月3日：
  • 计算得到period_start为2022-11-05，period_end为2022-12-03
  • 查询结果为created_at在此区间内的Customer记录
• 当今日是2022年12月15日：
  • 计算得到period_start为2022-12-05，period_end为2022-12-15
  • 查询结果为created_at在此区间内的Customer记录

内容的提问来源于stack exchange，提问作者sagar verma