mirror of
https://github.com/mastodon/mastodon.git
synced 2024-08-20 21:08:15 -07:00
a872392cd9
* Add entity cache Use a caching layer for mentions and custom emojis that are dynamically extracted from text. Reduce duplicate text extractions * Fix code style issue
34 lines
1.2 KiB
Ruby
34 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'singleton'
|
|
|
|
class EntityCache
|
|
include Singleton
|
|
|
|
MAX_EXPIRATION = 7.days.freeze
|
|
|
|
def mention(username, domain)
|
|
Rails.cache.fetch(to_key(:mention, username, domain), expires_in: MAX_EXPIRATION) { Account.select(:username, :domain, :url).find_remote(username, domain) }
|
|
end
|
|
|
|
def emoji(shortcodes, domain)
|
|
shortcodes = [shortcodes] unless shortcodes.is_a?(Array)
|
|
cached = Rails.cache.read_multi(*shortcodes.map { |shortcode| to_key(:emoji, shortcode, domain) })
|
|
uncached_ids = []
|
|
|
|
shortcodes.each do |shortcode|
|
|
uncached_ids << shortcode unless cached.key?(to_key(:emoji, shortcode, domain))
|
|
end
|
|
|
|
unless uncached_ids.empty?
|
|
uncached = CustomEmoji.where(shortcode: shortcodes, domain: domain, disabled: false).select(:shortcode, :id, :image_file_name, :visible_in_picker).map { |item| [item.shortcode, item] }.to_h
|
|
uncached.each_value { |item| Rails.cache.write(to_key(:emoji, item.shortcode, domain), item, expires_in: MAX_EXPIRATION) }
|
|
end
|
|
|
|
shortcodes.map { |shortcode| cached[to_key(:emoji, shortcode, domain)] || uncached[shortcode] }.compact
|
|
end
|
|
|
|
def to_key(type, *ids)
|
|
"#{type}:#{ids.compact.map(&:downcase).join(':')}"
|
|
end
|
|
end
|