]> cat aescling's git repositories - mastodon.git/commitdiff
Revert support from loading Home timeline from database
authorClaire <claire.github-309c@sitedethib.com>
Tue, 19 Apr 2022 12:54:43 +0000 (14:54 +0200)
committersingle-right-quote <11325618-aescling@users.noreply.gitlab.com>
Thu, 5 May 2022 03:55:59 +0000 (23:55 -0400)
Unfortunately, the database query could turn out very inefficient and I did not
manage to find a way to improve that. Furthermore, there were still behavior
inconsistencies between fetching the timeline from Redis and fetching it from
Postgres.

app/models/home_feed.rb
spec/models/home_feed_spec.rb

index 53550b7db819e3d1db7260a66611b0f25bd59432..d6ebb5fa6bc47adc6fb874747077af869a21efcd 100644 (file)
@@ -9,52 +9,4 @@ class HomeFeed < Feed
   def regenerating?
     redis.exists?("account:#{@account.id}:regeneration")
   end
-
-  def get(limit, max_id = nil, since_id = nil, min_id = nil)
-    limit    = limit.to_i
-    max_id   = max_id.to_i if max_id.present?
-    since_id = since_id.to_i if since_id.present?
-    min_id   = min_id.to_i if min_id.present?
-
-    if min_id.present?
-      redis_min_id = fetch_min_redis_id
-      return from_redis(limit, max_id, since_id, min_id) if redis_min_id && min_id >= redis_min_id
-
-      statuses = from_database(limit, redis_min_id, since_id, min_id)
-      return statuses if statuses.size >= limit
-
-      remaining_limit = limit - statuses.size
-      min_id = statuses.first.id unless statuses.empty?
-      from_redis(remaining_limit, max_id, since_id, min_id) + statuses
-    else
-      statuses = from_redis(limit, max_id, since_id, min_id)
-      return statuses if statuses.size >= limit
-
-      if since_id.present?
-        redis_min_id = fetch_min_redis_id
-        return statuses if redis_min_id.present? && since_id >= redis_min_id
-      end
-
-      remaining_limit = limit - statuses.size
-      max_id = statuses.last.id unless statuses.empty?
-      statuses + from_database(remaining_limit, max_id, since_id, min_id)
-    end
-  end
-
-  protected
-
-  def from_database(limit, max_id, since_id, min_id)
-    scope = Status.where(account: @account).or(Status.where(account: @account.following))
-    scope = scope.where(visibility: %i(public unlisted private))
-    scope
-      .to_a_paginated_by_id(limit, min_id: min_id, max_id: max_id, since_id: since_id)
-      .reject { |status| FeedManager.instance.filter?(:home, status, @account) }
-      .sort_by { |status| -status.id }
-  end
-
-  private
-
-  def fetch_min_redis_id
-    redis.zrangebyscore(key, '(0', '(+inf', limit: [0, 1], with_scores: true).first&.first&.to_i
-  end
 end
index 20756633c47ea3a51a7f00d6d4b8ad691fb6b59f..ee7a83960a05271044680fc8e75a4163bacedd64 100644 (file)
@@ -1,72 +1,32 @@
 require 'rails_helper'
 
 RSpec.describe HomeFeed, type: :model do
-  let(:account)  { Fabricate(:account) }
-  let(:followed) { Fabricate(:account) }
-  let(:other)    { Fabricate(:account) }
+  let(:account) { Fabricate(:account) }
 
   subject { described_class.new(account) }
 
   describe '#get' do
     before do
-      account.follow!(followed)
-
-      Fabricate(:status, account: account,  id: 1)
-      Fabricate(:status, account: account,  id: 2)
-      status = Fabricate(:status, account: followed, id: 3)
-      Fabricate(:mention, account: account, status: status)
-      Fabricate(:status, account: account,  id: 10)
-      Fabricate(:status, account: other,    id: 11)
-      Fabricate(:status, account: followed, id: 12, visibility: :private)
-      Fabricate(:status, account: followed, id: 13, visibility: :direct)
+      Fabricate(:status, account: account, id: 1)
+      Fabricate(:status, account: account, id: 2)
+      Fabricate(:status, account: account, id: 3)
+      Fabricate(:status, account: account, id: 10)
     end
 
     context 'when feed is generated' do
       before do
-        FeedManager.instance.populate_home(account)
+        Redis.current.zadd(
+          FeedManager.instance.key(:home, account.id),
+          [[4, 4], [3, 3], [2, 2], [1, 1]]
+        )
       end
 
-      it 'gets statuses with ids in the range from redis with database' do
+      it 'gets statuses with ids in the range from redis' do
         results = subject.get(3)
 
-        expect(results.map(&:id)).to eq [12, 10, 3]
+        expect(results.map(&:id)).to eq [3, 2]
         expect(results.first.attributes.keys).to eq %w(id updated_at)
       end
-
-      it 'with since_id present' do
-        results = subject.get(3, nil, 3, nil)
-        expect(results.map(&:id)).to eq [12, 10]
-      end
-
-      it 'with min_id present' do
-        results = subject.get(3, nil, nil, 0)
-        expect(results.map(&:id)).to eq [3, 2, 1]
-      end
-    end
-
-    context 'when feed is only partial' do
-      before do
-        FeedManager.instance.populate_home(account)
-
-        Redis.current.zremrangebyrank(FeedManager.instance.key(:home, account.id), 0, -2)
-      end
-
-      it 'gets statuses with ids in the range from redis with database' do
-        results = subject.get(3)
-
-        expect(results.map(&:id)).to eq [12, 10, 3]
-        expect(results.first.attributes.keys).to eq %w(id updated_at)
-      end
-
-      it 'with since_id present' do
-        results = subject.get(3, nil, 3, nil)
-        expect(results.map(&:id)).to eq [12, 10]
-      end
-
-      it 'with min_id present' do
-        results = subject.get(3, nil, nil, 0)
-        expect(results.map(&:id)).to eq [3, 2, 1]
-      end
     end
 
     context 'when feed is being generated' do
@@ -74,20 +34,10 @@ RSpec.describe HomeFeed, type: :model do
         Redis.current.set("account:#{account.id}:regeneration", true)
       end
 
-      it 'returns from database' do
+      it 'returns nothing' do
         results = subject.get(3)
 
-        expect(results.map(&:id)).to eq [12, 10, 3]
-      end
-
-      it 'with since_id present' do
-        results = subject.get(3, nil, 3, nil)
-        expect(results.map(&:id)).to eq [12, 10]
-      end
-
-      it 'with min_id present' do
-        results = subject.get(3, nil, nil, 0)
-        expect(results.map(&:id)).to eq [3, 2, 1]
+        expect(results.map(&:id)).to eq []
       end
     end
   end