通常の日付のループ処理
RubyのDateクラスは先頭の日付と末尾の日付を繰り返し処理出来る仕組みがあります。
head = Date.new(2016, 1)
tail = Date.new(2016, 12)
# 1日ずつループ
(head..tail).each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20160102', '20160103', ...
# n日の間隔でループ
head.step(tail, 5).each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20160106', '20160111', ...
これはこれで、とても便利な機能なのですが、ループのstepの幅は日単位でしか指定することが出来ません。
年・月・週単位でループさせる
Enumeratorインスタンスを生成することで、任意の間隔でループさせる処理を書くことが出来るようになります。
e = Enumerator.new do |yielder|
head = Date.new 2016, 1
tail = Date.new 2016, 12
while head <= tail
yielder << head
head = head.next_month # ここをhead.next_weekやhead.next_yearに変えることで年毎や週毎のループに出来る
end
end
e.each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20160201', '20160301', ...
ほんの少し解説
yielderに対して<< argという形でメソッドを呼び出すとargがeachのブロック引数に渡されます。
上記のサンプルコードでは、yielder << headとなっているので、head（whileループの中でカウントアップされている日付）がeachのブロック引数に渡されていることになります。
便利に使う
下記のようなrefinementを利用して、Dateクラスを拡張すると便利です。
（active_supportを使うこと前提のコードです）
require 'date'
require 'active_support/all'
module DateExtender
refine Date do
def step_by_unit(tail, unit = :day, step = 1)
head = self.dup
Enumerator.new do |yielder|
while head <= tail
yielder << head
head += step.send(unit)
end
end
end
end
end
using DateExtender
# week by
head = Date.new(2016, 1)
tail = Date.new(2016, 1).end_of_month
head.step_by_unit(tail, :week).each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20160108', '20160115', ...
# month by
head = Date.new(2016, 1)
tail = Date.new(2016, 4)
head.step_by_unit(tail, :month).each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20160201', '20160301', ...
# year by
head = Date.new(2016)
tail = Date.new(2020)
head.step_by_unit(tail, :year).each { |date| p date.strftime('%Y%m%d') }
# => '20160101', '20170101', '20180101', ...