replaced enumerable #count for #size (suggested in #112)

ddnexus · Dec 10, 2018 · 01178cb · 01178cb
1 parent 0723228
commit 01178cb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/pagy/backend.rb b/lib/pagy/backend.rb
@@ -16,7 +16,7 @@ def pagy(collection, vars={})
 
     # Sub-method called only by #pagy: here for easy customization of variables by overriding
     def pagy_get_vars(collection, vars)
-      vars[:count] ||= (c = collection.count(:all)).is_a?(Hash) ? c.count : c
+      vars[:count] ||= (c = collection.count(:all)).is_a?(Hash) ? c.size : c
       vars[:page]  ||= params[ vars[:page_param] || VARS[:page_param] ]
       vars
     end

diff --git a/lib/pagy/extras/array.rb b/lib/pagy/extras/array.rb
@@ -12,7 +12,7 @@ def pagy_array(array, vars={})
 
     # Sub-method called only by #pagy_array: here for easy customization of variables by overriding
     def pagy_array_get_vars(array, vars)
-      vars[:count] ||= array.count
+      vars[:count] ||= array.size
       vars[:page]  ||= params[ vars[:page_param] || VARS[:page_param] ]
       vars
     end