luna/mastodon
0
0
Fork 0
mirror of https://github.com/lunaisnotaboy/mastodon.git synced 2024-11-17 12:24:31 +00:00
Code Issues Projects Releases Wiki Activity

Simplifying followers mappings

Browse source 
Deduplicating some logic and switching to pluck, to avoid pulling the entire model in memory.
This commit is contained in:
Kurtis Rainbolt-Greene 2017-01-07 18:09:00 -08:00 committed by GitHub
parent 6d097d559b
commit abe3ae1cc2
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
app/models/account.rb
View file

@ -175,19 +175,23 @@ class Account < ApplicationRecord
end end
def following_map(target_account_ids, account_id) def following_map(target_account_ids, account_id)
Follow.where(target_account_id: target_account_ids).where(account_id: account_id).map { |f| [f.target_account_id, true] }.to_h follow_mapping(Follow.where(target_account_id: target_account_ids, account_id: account_id), :target_account_id)
end end
def followed_by_map(target_account_ids, account_id) def followed_by_map(target_account_ids, account_id)
Follow.where(account_id: target_account_ids).where(target_account_id: account_id).map { |f| [f.account_id, true] }.to_h follow_mapping(Follow.where(account_id: target_account_ids, target_account_id: account_id), :account_id)
end end
def blocking_map(target_account_ids, account_id) def blocking_map(target_account_ids, account_id)
Block.where(target_account_id: target_account_ids).where(account_id: account_id).map { |b| [b.target_account_id, true] }.to_h follow_mapping(Block.where(target_account_id: target_account_ids, account_id: account_id), :target_account_id)
end end
def requested_map(target_account_ids, account_id) def requested_map(target_account_ids, account_id)
FollowRequest.where(target_account_id: target_account_ids).where(account_id: account_id).map { |r| [r.target_account_id, true] }.to_h follow_mapping(FollowRequest.where(target_account_id: target_account_ids, account_id: account_id), :target_account_id)
end
private def follow_mapping(query, field)
query.pluck(field).inject({}) { |mapping, id| mapping[id] = true }
end end
end end