mastofe

commit: 9fd3d7b6cdb97a8f0f47425a32f34fc5c54b94de
parent: a0f85774c482716505b0c8c8faf59616f06245d9
Author: Eugen Rochko <eugen@zeonfederated.com>
Date:   Sun,  2 Oct 2016 22:35:27 +0200

Add since_id param to feeds

Diffstat:
M
app/controllers/api/v1/accounts_controller.rb
2
+-
M
app/controllers/api/v1/statuses_controller.rb
4
++--
M
app/models/concerns/paginable.rb
7
++++++-
M
app/models/feed.rb
7
++++---

4 files changed, 13 insertions(+), 7 deletions(-)
diff --git a/app/controllers/api/v1/accounts_controller.rb b/app/controllers/api/v1/accounts_controller.rb
@@ -20,7 +20,7 @@ class Api::V1::AccountsController < ApiController
   end
 
   def statuses
-    @statuses = @account.statuses.with_includes.with_counters.paginate_by_max_id(20, params[:max_id] || nil).to_a
+    @statuses = @account.statuses.with_includes.with_counters.paginate_by_max_id(20, params[:max_id], params[:since_id]).to_a
   end
 
   def follow
diff --git a/app/controllers/api/v1/statuses_controller.rb b/app/controllers/api/v1/statuses_controller.rb
@@ -45,10 +45,10 @@ class Api::V1::StatusesController < ApiController
   end
 
   def home
-    @statuses = Feed.new(:home, current_user.account).get(20, params[:max_id]).to_a
+    @statuses = Feed.new(:home, current_user.account).get(20, params[:max_id], params[:since_id]).to_a
   end
 
   def mentions
-    @statuses = Feed.new(:mentions, current_user.account).get(20, params[:max_id]).to_a
+    @statuses = Feed.new(:mentions, current_user.account).get(20, params[:max_id], params[:since_id]).to_a
   end
 end
diff --git a/app/models/concerns/paginable.rb b/app/models/concerns/paginable.rb
@@ -2,6 +2,11 @@ module Paginable
   extend ActiveSupport::Concern
 
   included do
-    scope :paginate_by_max_id, -> (limit, max_id) { order('id desc').limit(limit).where(max_id.nil? ? '1=1' : ['id < ?', max_id]) }
+    def self.paginate_by_max_id(limit, max_id = nil, since_id = nil)
+      query = order('id desc').limit(limit)
+      query = query.where('id < ?', max_id) unless max_id.blank?
+      query = query.where('id > ?', since_id) unless since_id.blank?
+      query
+    end
   end
 end
diff --git a/app/models/feed.rb b/app/models/feed.rb
@@ -4,9 +4,10 @@ class Feed
     @account = account
   end
 
-  def get(limit, max_id = nil)
-    max_id     = '+inf' if max_id.nil?
-    unhydrated = redis.zrevrangebyscore(key, "(#{max_id}", '-inf', limit: [0, limit], with_scores: true).collect(&:last).map(&:to_i)
+  def get(limit, max_id = nil, since_id = nil)
+    max_id     = '+inf' if max_id.blank?
+    since_id   = '-inf' if since_id.blank?
+    unhydrated = redis.zrevrangebyscore(key, "(#{max_id}", "(#{since_id}", limit: [0, limit], with_scores: true).collect(&:last).map(&:to_i)
     status_map = {}
 
     # If we're after most recent items and none are there, we need to precompute the feed