ruby-on-rails - Ruby 中的简单自定义范围

Question

我有一个名为_scope 的 Rails，它使用一个条件从表中提取一周中的特定日期，如下所示：

:conditions => [ 'EXTRACT(DOW FROM bookdate) IN (?)', (1..6).to_a ]

1..6 日期范围将是一个变量，具体取决于用户想要的日期，

产生这个 SQL

(EXTRACT(DOW FROM bookdate) IN (1,2,3,4,5,6)

我的问题是一周中的日子不是一个简单的范围......即1..6工作正常（周一......周六），但说6..2无法正常工作（周六至周二）......或者作为一个红宝石范围，或者它需要是 6,7,1,2 而不是 6,5,4,3,2（假设 6..2 在红宝石中工作，但它没有）。

如何为一周中的几天创建一个自定义范围，以适应这样的日期范围？

有任何想法吗？

谢谢，

score 7 · Accepted Answer

如果您想要一个函数来生成包含在其中的值范围:conditions，它将适用于 Sat-Tues 和 Mon-Sat 等范围，那么：

def day_range(from, to)
  if from <= to
    (from..to).to_a
  else
    (from..7).to_a + (1..to).to_a
  end
end

例如

irb(main):032:0> day_range(1, 6)
=> [1, 2, 3, 4, 5, 6]
irb(main):033:0> day_range(6, 2)
=> [6, 7, 1, 2]

然后，如果需要，只需将日期名称映射到日期编号即可。

1 回答 1