gitlab-org--gitlab-foss/lib/gitlab/markdown_cache/redis/extension.rb

79 lines
2 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module MarkdownCache
module Redis
module Extension
extend ActiveSupport::Concern
attr_reader :cached_markdown_version
class_methods do
def cache_markdown_field(markdown_field, context = {})
super
# define the `[field]_html` accessor
html_field = cached_markdown_fields.html_field(markdown_field)
define_method(html_field) do
load_cached_markdown unless markdown_data_loaded?
instance_variable_get("@#{html_field}")
end
end
end
prepended do
def self.preload_markdown_cache!(objects)
fields = Gitlab::MarkdownCache::Redis::Store.bulk_read(objects)
objects.each do |object|
fields[object.cache_key].value.each do |field_name, value|
object.write_markdown_field(field_name, value)
end
end
end
end
def write_markdown_field(field_name, value)
# The value read from redis is a string, so we're converting it back
# to an int.
value = value.to_i if field_name == :cached_markdown_version
instance_variable_set("@#{field_name}", value)
end
private
def save_markdown(updates)
markdown_store.save(updates)
end
def markdown_field_changed?(field_name)
false
end
def changed_attributes
{}
end
def cached_markdown
@cached_data ||= markdown_store.read
end
def load_cached_markdown
cached_markdown.each do |field_name, value|
write_markdown_field(field_name, value)
end
end
def markdown_data_loaded?
cached_markdown_version.present? || markdown_store.loaded?
end
def markdown_store
@store ||= Gitlab::MarkdownCache::Redis::Store.new(self)
end
end
end
end
end