Distrubute statuses as a fan-out-on-write system, with optional precomputing

2 files changed, 28 insertions, 0 deletions

diff --git a/app/models/feed.rb b/app/models/feed.rb
new file mode 100644
index 000000000..a063ad05b
--- /dev/null
+++ b/app/models/feed.rb
@@ -0,0 +1,27 @@
+class Feed
+  def initialize(type, account)
+    @type    = type
+    @account = account
+  end
+
+  def get(limit, offset = 0)
+    unhydrated = redis.zrevrange(key, offset, limit)
+    status_map = Hash.new
+
+    # If we're after most recent items and none are there, we need to precompute the feed
+    return PrecomputeFeedService.new.(@type, @account).take(limit) if unhydrated.empty? && offset == 0
+
+    Status.where(id: unhydrated).each { |status| status_map[status.id.to_s] = status }
+    return unhydrated.map { |id| status_map[id] }
+  end
+
+  private
+
+  def key
+    "feed:#{@type}:#{@account.id}"
+  end
+
+  def redis
+    $redis
+  end
+end
diff --git a/app/models/status.rb b/app/models/status.rb
index a346ac9b0..7e0c334ec 100644
--- a/app/models/status.rb
+++ b/app/models/status.rb
@@ -67,5 +67,6 @@ class Status < ActiveRecord::Base
 
   after_create do
     self.account.stream_entries.create!(activity: self)
+    FanOutOnWriteService.new.(self)
   end
 end