]> cat aescling's git repositories - mastodon.git/commitdiff
Revert DM support in HomeFeed#from_database
authorClaire <claire.github-309c@sitedethib.com>
Sat, 16 Apr 2022 19:20:07 +0000 (21:20 +0200)
committersingle-right-quote <11325618-aescling@users.noreply.gitlab.com>
Thu, 5 May 2022 03:55:59 +0000 (23:55 -0400)
Fixes #1746

Queries could get prohibitively expensive.

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

index 44744509269cb9e830a8fc36bc9d8bd2f1f893c5..53550b7db819e3d1db7260a66611b0f25bd59432 100644 (file)
@@ -44,10 +44,8 @@ class HomeFeed < Feed
   protected
 
   def from_database(limit, max_id, since_id, min_id)
-    scope = Status.where(account: @account.following)
-    scope = scope.left_outer_joins(:mentions)
-    scope = scope.where(visibility: %i(public unlisted private)).or(scope.where(mentions: { account_id: @account.id })).group(Status.arel_table[:id])
-    scope = scope.or(Status.where(account: @account))
+    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) }
index 565357d7b586a0b76882d934487fbd86689d743d..20756633c47ea3a51a7f00d6d4b8ad691fb6b59f 100644 (file)
@@ -19,32 +19,23 @@ RSpec.describe HomeFeed, type: :model do
       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: 14, visibility: :direct)
-      dm = Fabricate(:status, account: followed, id: 15, visibility: :direct)
-      Fabricate(:mention, account: account, status: dm)
     end
 
     context 'when feed is generated' do
       before do
         FeedManager.instance.populate_home(account)
-
-        # Add direct messages because populate_home does not do that
-        Redis.current.zadd(
-          FeedManager.instance.key(:home, account.id),
-          [[14, 14], [15, 15]]
-        )
       end
 
       it 'gets statuses with ids in the range from redis with database' do
-        results = subject.get(5)
+        results = subject.get(3)
 
-        expect(results.map(&:id)).to eq [15, 14, 12, 10, 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(5, nil, 3, nil)
-        expect(results.map(&:id)).to eq [15, 14, 12, 10]
+        results = subject.get(3, nil, 3, nil)
+        expect(results.map(&:id)).to eq [12, 10]
       end
 
       it 'with min_id present' do
@@ -57,25 +48,19 @@ RSpec.describe HomeFeed, type: :model do
       before do
         FeedManager.instance.populate_home(account)
 
-        # Add direct messages because populate_home does not do that
-        Redis.current.zadd(
-          FeedManager.instance.key(:home, account.id),
-          [[14, 14], [15, 15]]
-        )
-
         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(5)
+        results = subject.get(3)
 
-        expect(results.map(&:id)).to eq [15, 14, 12, 10, 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(5, nil, 3, nil)
-        expect(results.map(&:id)).to eq [15, 14, 12, 10]
+        results = subject.get(3, nil, 3, nil)
+        expect(results.map(&:id)).to eq [12, 10]
       end
 
       it 'with min_id present' do
@@ -90,14 +75,14 @@ RSpec.describe HomeFeed, type: :model do
       end
 
       it 'returns from database' do
-        results = subject.get(5)
+        results = subject.get(3)
 
-        expect(results.map(&:id)).to eq [15, 14, 12, 10, 3]
+        expect(results.map(&:id)).to eq [12, 10, 3]
       end
 
       it 'with since_id present' do
-        results = subject.get(5, nil, 3, nil)
-        expect(results.map(&:id)).to eq [15, 14, 12, 10]
+        results = subject.get(3, nil, 3, nil)
+        expect(results.map(&:id)).to eq [12, 10]
       end
 
       it 'with min_id present' do