Improve ScheduleExceptions pagination.

cvut · Jun 8, 2015 · ae89be9 · ae89be9
1 parent f9944f6
commit ae89be9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/app/api/schedule_exceptions_endpoints.rb b/app/api/schedule_exceptions_endpoints.rb
@@ -10,8 +10,7 @@ class ScheduleExceptionsEndpoints < Grape::API
       params { use :pagination }
 
       get do
-        dataset = paginate(ScheduleException.dataset)
-        ScheduleExceptionsRepresenter.new(dataset)
+        represent_paginated(ScheduleException.dataset, ScheduleExceptionsRepresenter)
       end
 
       params do

diff --git a/app/helpers/api_helper.rb b/app/helpers/api_helper.rb
@@ -1,4 +1,7 @@
+require 'corefines'
 require 'sirius_api'
+
+using Corefines::Hash[:only, :rekey]
 ##
 # Helper methods used in Grape's API endpoints.
 module ApiHelper
@@ -41,4 +44,12 @@ def paginate(dataset)
       .limit(params[:limit] || DEFAULT_LIMIT)
       .offset(params[:offset] || DEFAULT_OFFSET)
   end
+
+  def represent_paginated(dataset, representer)
+    paginated_dataset = paginate(dataset)
+    meta = paginated_dataset.opts
+      .rekey(:total_count => :count)
+      .only(:count, :offset, :limit)
+    representer.for_collection.new(paginated_dataset).to_hash({'meta' => meta})
+  end
 end