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