class Aggrobot::QueryPlanner::BucketedGroupsQueryPlanner

Public Class Methods

new(collection, group, opts = {}) click to toggle source
Calls superclass method
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 5
def initialize(collection, group, opts = {})
  required_params = [:buckets]
  optional_params = [:keep_empty]
  validate_options(opts, required_params, optional_params)
  raise_error 'Need to set group first' unless group
  super(collection, group)
  create_query_map(opts[:buckets])
  @keep_empty = opts[:keep_empty]
end

Public Instance Methods

query_results(extra_cols = []) click to toggle source
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 19
def query_results(extra_cols = [])
  return empty_buckets if collection_is_none?
  results = collect_query_results(extra_cols)
  results.reject! { |r| r[1] == 0 } unless @keep_empty
  results
end
sub_query(group_value) click to toggle source
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 15
def sub_query(group_value)
  @query_map[group_value]
end

Private Instance Methods

collect_query_results(extra_cols) click to toggle source
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 28
def collect_query_results(extra_cols)
  columns = [SQLFunctions.count] + extra_cols
  @query_map.collect do |group_value, query|
    results = query.limit(1).pluck(*columns).flatten
    if results[0] == 0
      @query_map[group_value] = @query_map[group_value].none
      results = [0]
    end
    results.unshift(group_value)
  end
end
create_query_map(buckets) click to toggle source
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 44
def create_query_map(buckets)
  @query_map = {}
  buckets.each do |bucket|
    @query_map[bucket] = @collection.where(group_condition(bucket))
  end
end
empty_buckets() click to toggle source
# File lib/aggrobot/query_planner/bucketed_groups_query_planner.rb, line 40
def empty_buckets
  @keep_empty ? @query_map.keys.collect { |k| [k, 0] } : []
end