Module: ArAggregateByInterval::Utils

Extended by:
Utils
Included in:
Utils
Defined in:
lib/ar_aggregate_by_interval/utils.rb

Constant Summary collapse

DATE_ITERATOR_METHOD_MAP = 
{
  'monthly' => 'each_month_until',
  'weekly' => 'each_week_until',
  'daily' => 'each_day_until'
}

Instance Method Summary collapse

Instance Method Details

#args_to_hash(sum_or_count, daily_weekly_monthly, *args) ⇒ Object

support legacy arguments (as opposed to direct hash) can do: ArModel.count_weekly(:group_by_col, :from, :to, :options_hash) ArModel.count_weekly(:from, :to, :options_hash) # defaults to :created_at or ArModel.sum_weekly(:group_by_col, :aggregate_col, :from, :to, :options_hash)



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 18

def args_to_hash(sum_or_count, daily_weekly_monthly, *args)
  group_by_column, aggregate_column, from, to, options_hash = args

  group_by_column ||= 'created_at'

  if sum_or_count == 'count'
    if aggregate_column.present? && (aggregate_column.is_a?(Date) || aggregate_column.is_a?(Time))
      options_hash = to
      to = from
      from = aggregate_column
      aggregate_column = nil
    end
  elsif sum_or_count != 'count' && aggregate_column.nil? || !(aggregate_column.is_a?(String) || aggregate_column.is_a?(Symbol))
    raise ArgumentError, "aggregate_column cant be nil with #{sum_or_count}"
  end

  if !options_hash.nil? && !options_hash.is_a?(Hash)
    raise ArgumentError, 'last argument must be options hash'
  end

  return (options_hash || {}).merge({
    group_by_column: group_by_column.try(:intern),
    from: from,
    to: to,
    aggregate_column: aggregate_column.try(:intern)
  }).delete_if { |k, v| v.nil? }
end

#db_vendor ⇒ Object 



80
81
82
83
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 80

def db_vendor
  ActiveRecord::Base.connection_config.try(:symbolize_keys).try(:[], :adapter) ||
  ENV['DATABASE_URL']
end

#interval_inflector(interval, beg_or_end) ⇒ Object

converts args like: [weekly, beginning] to beginning_of_week



86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 86

def interval_inflector(interval, beg_or_end)
  raise "beginning or end, not #{beg_or_end}" unless beg_or_end.match(/\A(beginning|end)\z/)

  time_interval = {
    'monthly' => 'month',
    'weekly' => 'week',
    'daily' => 'day'
  }[interval] || raise("unknown interval #{interval}")

  "#{beg_or_end}_of_#{time_interval}"
end

#ruby_strftime_map ⇒ Object 



46
47
48
49
50
51
52
53
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 46

def ruby_strftime_map
  {
    'monthly' => '%Y-%m',
    # sqlite doesn't support ISO weeks
    'weekly' => Utils.db_vendor.match(/sqlite/i) ? '%Y-%U' : '%G-%V',
    'daily' => '%F'
  }
end

#select_for_grouping_column(grouping_col) ⇒ Object 



55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 55

def select_for_grouping_column(grouping_col)
  case db_vendor
  when /mysql/i
    {
      'monthly' => "date_format(#{grouping_col}, '%Y-%m')",
      'weekly' => "date_format(#{grouping_col}, '%x-%v')",
      'daily' => "date(#{grouping_col})"
    }
  when /postgres/i
    {
      'monthly' => "to_char(#{grouping_col}, 'YYYY-MM')",
      'weekly' => "to_char(#{grouping_col}, 'IYYY-IW')",
      'daily' => "date(#{grouping_col})"
    }
  when /sqlite/i
    {
      'monthly' => "strftime('%Y-%m', #{grouping_col})",
      'weekly' => "strftime('%Y-%W', #{grouping_col})", # sqlite doesn't support ISO weeks
      'daily' => "date(#{grouping_col})"
    }
  else
    raise "unknown database vendor #{db_vendor}"
  end
end

#to_f_or_i_or_s(v) ⇒ Object 



98
99
100
 
# File 'lib/ar_aggregate_by_interval/utils.rb', line 98

def to_f_or_i_or_s(v)
  ((float = Float(v)) && (float % 1.0 == 0) ? float.to_i : float) rescue v
end