Add support for grouped results

lib/ajax-datatables-rails/base.rb

@@ -137,11 +137,15 @@ def retrieve_records
     end
 
     def records_total_count
-      fetch_records.count(:all)
+      numeric_count(fetch_records.count(:all))
     end
 
     def records_filtered_count
-      filter_records(fetch_records).count(:all)
+      numeric_count(filter_records(fetch_records).count(:all))
+    end
+
+    def numeric_count(count)
+      count.is_a?(Hash) ? count.values.size : count
     end
 
     def global_search_delimiter

spec/ajax-datatables-rails/orm/active_record_count_records_spec.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe AjaxDatatablesRails::ORM::ActiveRecord do
+
+  let(:datatable) { ComplexDatatable.new(sample_params) }
+  let(:records) { User.all }
+
+  describe '#records_total_count' do
+    context 'ungrouped results' do
+      it 'returns the count' do
+        expect(datatable.send(:records_total_count)).to eq records.count
+      end
+    end
+
+    context 'grouped results' do
+      let(:datatable) { GroupedDatatable.new(sample_params) }
+
+      it 'returns the count' do
+        expect(datatable.send(:records_total_count)).to eq records.count
+      end
+    end
+  end
+
+
+  describe '#records_filtered_count' do
+    context 'ungrouped results' do
+      it 'returns the count' do
+        expect(datatable.send(:records_filtered_count)).to eq records.count
+      end
+    end
+
+    context 'grouped results' do
+      let(:datatable) { GroupedDatatable.new(sample_params) }
+
+      it 'returns the count' do
+        expect(datatable.send(:records_filtered_count)).to eq records.count
+      end
+    end
+  end
+end

spec/support/datatables/grouped_datatable_array.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class GroupedDatatable < ComplexDatatable
+
+  def get_raw_records
+    User.all.group(:id)
+  end
+end