From 12ac97707997f11ebda9d7527aaff14d079a2460 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rafael=20Mendon=C3=A7a=20Fran=C3=A7a?= Date: Sun, 24 Jun 2012 06:42:41 -0700 Subject: [PATCH] Merge pull request #6842 from ernie/handle-non-strings-in-grouped-calculations Stop assuming strings for grouped calculations Conflicts: activerecord/lib/active_record/relation/calculations.rb --- .../active_record/relation/calculations.rb | 23 ++++++++++++++----- activerecord/test/cases/calculations_test.rb | 5 ++++ 2 files changed, 22 insertions(+), 6 deletions(-) diff --git a/activerecord/lib/active_record/relation/calculations.rb b/activerecord/lib/active_record/relation/calculations.rb index 0d60810cd2d91..42ee4961c03fe 100644 --- a/activerecord/lib/active_record/relation/calculations.rb +++ b/activerecord/lib/active_record/relation/calculations.rb @@ -244,10 +244,16 @@ def execute_simple_calculation(operation, column_name, distinct) #:nodoc: end def execute_grouped_calculation(operation, column_name, distinct) #:nodoc: - group_attr = @group_values - association = @klass.reflect_on_association(group_attr.first.to_sym) - associated = group_attr.size == 1 && association && association.macro == :belongs_to # only count belongs_to associations - group_fields = Array(associated ? association.foreign_key : group_attr) + group_attrs = @group_values + + if group_attrs.first.respond_to?(:to_sym) + association = @klass.reflect_on_association(group_attrs.first.to_sym) + associated = group_attrs.size == 1 && association && association.macro == :belongs_to # only count belongs_to associations + group_fields = Array(associated ? association.foreign_key : group_attrs) + else + group_fields = group_attrs + end + group_aliases = group_fields.map { |field| column_alias_for(field) } group_columns = group_aliases.zip(group_fields).map { |aliaz,field| [aliaz, column_for(field)] @@ -270,10 +276,14 @@ def execute_grouped_calculation(operation, column_name, distinct) #:nodoc: select_values += @select_values unless @having_values.empty? select_values.concat group_fields.zip(group_aliases).map { |field,aliaz| - "#{field} AS #{aliaz}" + if field.respond_to?(:as) + field.as(aliaz) + else + "#{field} AS #{aliaz}" + end } - relation = except(:group).group(group.join(',')) + relation = except(:group).group(group) relation.select_values = select_values calculated_data = @klass.connection.select_all(relation) @@ -303,6 +313,7 @@ def execute_grouped_calculation(operation, column_name, distinct) #:nodoc: # column_alias_for("count(*)") # => "count_all" # column_alias_for("count", "id") # => "count_id" def column_alias_for(*keys) + keys.map! {|k| k.respond_to?(:to_sql) ? k.to_sql : k} table_name = keys.join(' ') table_name.downcase! table_name.gsub!(/\*/, 'all') diff --git a/activerecord/test/cases/calculations_test.rb b/activerecord/test/cases/calculations_test.rb index 66c801ca752d7..e50de46f9edde 100644 --- a/activerecord/test/cases/calculations_test.rb +++ b/activerecord/test/cases/calculations_test.rb @@ -67,6 +67,11 @@ def test_should_group_by_field [1,6,2].each { |firm_id| assert c.keys.include?(firm_id) } end + def test_should_group_by_arel_attribute + c = Account.sum(:credit_limit, :group => Account.arel_table[:firm_id]) + [1,6,2].each { |firm_id| assert c.keys.include?(firm_id) } + end + def test_should_group_by_multiple_fields c = Account.count(:all, :group => ['firm_id', :credit_limit]) [ [nil, 50], [1, 50], [6, 50], [6, 55], [9, 53], [2, 60] ].each { |firm_and_limit| assert c.keys.include?(firm_and_limit) }