Use count(:all) with Pagy

davidcelis · davidcelis · commit 08d764c74081 · 2018-06-21T11:10:59.000-07:00
diff --git a/lib/api-pagination.rb b/lib/api-pagination.rb
@@ -64,7 +64,8 @@ def paginate_with_pagy(collection, options)
     end
 
     def pagy_from(collection, options)
-      Pagy.new(count: collection.count, items: options[:per_page], page: options[:page])
+      count = collection.is_a?(Array) ? collection.count : collection.count(:all)
+      Pagy.new(count: count, items: options[:per_page], page: options[:page])
     end
 
     def pagy_pages_from(pagy)
