mirror of
https://github.com/mastodon/mastodon.git
synced 2024-11-23 13:05:12 +01:00
cba2897108
* Cache relationships in API * Fetch relationships for search results in UI * Only save one account's maps in each cache item
72 lines
2.5 KiB
Ruby
72 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class AccountRelationshipsPresenter
|
|
attr_reader :following, :followed_by, :blocking,
|
|
:muting, :requested, :domain_blocking
|
|
|
|
def initialize(account_ids, current_account_id, **options)
|
|
@account_ids = account_ids.map { |a| a.is_a?(Account) ? a.id : a }
|
|
@current_account_id = current_account_id
|
|
|
|
@following = cached[:following].merge(Account.following_map(@uncached_account_ids, @current_account_id))
|
|
@followed_by = cached[:followed_by].merge(Account.followed_by_map(@uncached_account_ids, @current_account_id))
|
|
@blocking = cached[:blocking].merge(Account.blocking_map(@uncached_account_ids, @current_account_id))
|
|
@muting = cached[:muting].merge(Account.muting_map(@uncached_account_ids, @current_account_id))
|
|
@requested = cached[:requested].merge(Account.requested_map(@uncached_account_ids, @current_account_id))
|
|
@domain_blocking = cached[:domain_blocking].merge(Account.domain_blocking_map(@uncached_account_ids, @current_account_id))
|
|
|
|
cache_uncached!
|
|
|
|
@following.merge!(options[:following_map] || {})
|
|
@followed_by.merge!(options[:followed_by_map] || {})
|
|
@blocking.merge!(options[:blocking_map] || {})
|
|
@muting.merge!(options[:muting_map] || {})
|
|
@requested.merge!(options[:requested_map] || {})
|
|
@domain_blocking.merge!(options[:domain_blocking_map] || {})
|
|
end
|
|
|
|
private
|
|
|
|
def cached
|
|
return @cached if defined?(@cached)
|
|
|
|
@cached = {
|
|
following: {},
|
|
followed_by: {},
|
|
blocking: {},
|
|
muting: {},
|
|
requested: {},
|
|
domain_blocking: {},
|
|
}
|
|
|
|
@uncached_account_ids = []
|
|
|
|
@account_ids.each do |account_id|
|
|
maps_for_account = Rails.cache.read("relationship:#{@current_account_id}:#{account_id}")
|
|
|
|
if maps_for_account.is_a?(Hash)
|
|
@cached.merge!(maps_for_account)
|
|
else
|
|
@uncached_account_ids << account_id
|
|
end
|
|
end
|
|
|
|
@cached
|
|
end
|
|
|
|
def cache_uncached!
|
|
@uncached_account_ids.each do |account_id|
|
|
maps_for_account = {
|
|
following: { account_id => following[account_id] },
|
|
followed_by: { account_id => followed_by[account_id] },
|
|
blocking: { account_id => blocking[account_id] },
|
|
muting: { account_id => muting[account_id] },
|
|
requested: { account_id => requested[account_id] },
|
|
domain_blocking: { account_id => domain_blocking[account_id] },
|
|
}
|
|
|
|
Rails.cache.write("relationship:#{@current_account_id}:#{account_id}", maps_for_account, expires_in: 1.day)
|
|
end
|
|
end
|
|
end
|