Contents

class RangeGenerator
  attr_reader :from_date_str, :fetch_days

  def initialize(from_date_str, fetch_days)
    @from_date_str = from_date_str
    @fetch_days = fetch_days
  end

  def generate_range
    validate
    show_warnings
    range_only_past.map{|date| date.to_s}
  end

  private

  def from_date
    Date.parse(from_date_str)
  end

  def validate
    begin
      from_date
    rescue ArgumentError # invalid date
      raise Embulk::ConfigError.new "from_date '#{from_date_str}' is invalid date"
    end

    if fetch_days && fetch_days < 1
      # `days` only allowed nil or positive number
      raise Embulk::ConfigError.new "fetch_days '#{fetch_days}' is invalid. Please specify bigger number than 0."
    end
  end

  def show_warnings
    if from_date_too_early?
      Embulk.logger.warn "Mixpanel allow 2 days before to from_date, so no data is input."
    end

    if overdays?
      Embulk.logger.warn "These dates are too early access, ignored them: from #{overdays.first} to #{overdays.last}"
    end
  end

  def range
    if from_date_too_early?
      return []
    end

    if fetch_days
      from_date..(from_date + fetch_days - 1)
    else
      from_date..yesterday
    end
  end

  def range_only_past
    range.find_all{|date| date < today}
  end

  def overdays?
    ! overdays.empty?
  end

  def overdays
    range.to_a - range_only_past.to_a
  end

  def from_date_too_early?
    from_date > yesterday
  end

  def yesterday
    today - 1
  end

  def today
    @today ||= Date.today
  end
end

Version data entries

12 entries across 12 versions & 1 rubygems

Version	Path
embulk-input-mixpanel-0.5.1	lib/range_generator.rb
embulk-input-mixpanel-0.5.0	lib/range_generator.rb
embulk-input-mixpanel-0.4.7	lib/range_generator.rb
embulk-input-mixpanel-0.4.6	lib/range_generator.rb
embulk-input-mixpanel-0.4.5	lib/range_generator.rb
embulk-input-mixpanel-0.4.4	lib/range_generator.rb
embulk-input-mixpanel-0.4.3	lib/range_generator.rb
embulk-input-mixpanel-0.4.2	lib/range_generator.rb
embulk-input-mixpanel-0.4.1	lib/range_generator.rb
embulk-input-mixpanel-0.4.0	lib/range_generator.rb
embulk-input-mixpanel-0.3.4	lib/range_generator.rb
embulk-input-mixpanel-0.3.3	lib/range_generator.rb