Merge pull request #82 from ysyyork/remove_extra_db_query

remove the extra db query
davidcelis · May 25, 2017 · b26c72f · b26c72f
2 parents 8184db7 + 906b096
commit b26c72f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/api-pagination.rb b/lib/api-pagination.rb
@@ -81,7 +81,11 @@ def default_per_page_for_will_paginate(collection)
     end
 
     def detect_model(collection)
-      collection.first.class
+      if collection.respond_to?(:table_name)
+        collection.table_name.singularize.capitalize.constantize
+      else
+        collection.first.class
+      end
     end
   end
 end