Contents

module Aggrobot
  module QueryPlanner
    class GroupLimitQueryPlanner < DefaultQueryPlanner

      def initialize(collection, group, opts)
        ParametersValidator.validate_options(opts, [:limit_to, :sort_by], [:always_include, :other_group, :order])
        raise_error 'limit_to has to be a number' unless opts[:limit_to].is_a?(Fixnum)
        super(collection, group)
        @query_map = {}
        process_top_groups_options(opts)
      end

      def sub_query(group_name)
        group_name == @other_group ? @collection.where.not(@top_groups_conditions) : @collection.where(@group => group_name)
      end

      def query_results(extra_cols = [])
        return [] if collection_is_none?
        columns = [@group, SqlFunctions.count] + extra_cols
        top_group_results = results_query.where(@top_groups_conditions).pluck(*columns)
        top_group_results + other_group_results(columns)
      end

      protected

      def other_group_results(columns)
        if @other_group
          columns[0] = SqlFunctions.sanitize(@other_group)
          @collection.where.not(@top_groups_conditions).group(columns[0]).pluck(*columns)
        else
          []
        end
      end

      def results_query
        @results_query ||= @collection.group(@group)
      end

      def calculate_top_groups(opts)
        @collection.group(@group).order("#{opts[:sort_by]} #{opts[:order]}").limit(opts[:limit_to]).pluck(@group).flatten
      end

      def process_top_groups_options(opts)
        opts[:order] ||= 'desc'
        top_groups = calculate_top_groups(opts)
        if opts[:always_include] && !top_groups.include?(opts[:always_include])
          top_groups.pop
          top_groups << opts[:always_include]
        end
        @top_groups_conditions = {@group => top_groups}
        @other_group = opts[:other_group]
      end

    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
aggrobot-0.0.2	lib/aggrobot/query_planner/group_limit_query_planner.rb