Code cleanup
This commit is contained in:
parent
93fcddd7a7
commit
cd2583a3be
|
@ -14,10 +14,18 @@ module Gitlab
|
|||
class ReferenceFilter < HTML::Pipeline::Filter
|
||||
LazyReference = Struct.new(:klass, :ids) do
|
||||
def self.load(refs)
|
||||
refs.group_by(&:klass).flat_map do |klass, refs|
|
||||
lazy_references, values = refs.partition { |ref| ref.is_a?(self) }
|
||||
|
||||
lazy_values = lazy_references.group_by(&:klass).flat_map do |klass, refs|
|
||||
ids = refs.flat_map(&:ids)
|
||||
klass.where(id: ids)
|
||||
end
|
||||
|
||||
values + lazy_values
|
||||
end
|
||||
|
||||
def load
|
||||
self.klass.where(id: self.ids)
|
||||
end
|
||||
end
|
||||
|
||||
|
|
|
@ -12,8 +12,7 @@ module Gitlab
|
|||
def initialize(*)
|
||||
super
|
||||
|
||||
result[:lazy_references] ||= Hash.new { |hash, type| hash[type] = [] }
|
||||
result[:references] ||= Hash.new { |hash, type| hash[type] = [] }
|
||||
result[:references] ||= Hash.new { |hash, type| hash[type] = [] }
|
||||
end
|
||||
|
||||
def call
|
||||
|
@ -41,23 +40,16 @@ module Gitlab
|
|||
|
||||
references.each do |type, values|
|
||||
Array.wrap(values).each do |value|
|
||||
refs =
|
||||
if value.is_a?(ReferenceFilter::LazyReference)
|
||||
result[:lazy_references]
|
||||
else
|
||||
result[:references]
|
||||
end
|
||||
|
||||
refs[type] << value
|
||||
result[:references][type] << value
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
# Will load all references of one type using one query.
|
||||
def load_lazy_references
|
||||
result[:lazy_references].each do |type, refs|
|
||||
values = ReferenceFilter::LazyReference.load(refs)
|
||||
result[:references][type].concat(values)
|
||||
refs = result[:references]
|
||||
refs.each do |type, values|
|
||||
refs[type] = ReferenceFilter::LazyReference.load(values)
|
||||
end
|
||||
end
|
||||
|
||||
|
|
|
@ -54,19 +54,13 @@ module Gitlab
|
|||
|
||||
pipeline = HTML::Pipeline.new([filter, Gitlab::Markdown::ReferenceGathererFilter], context)
|
||||
|
||||
values = []
|
||||
lazy_references = []
|
||||
|
||||
@texts.each do |text|
|
||||
values = @texts.flat_map do |text|
|
||||
result = pipeline.call(text)
|
||||
|
||||
values.concat(result[:references][filter_type])
|
||||
lazy_references.concat(result[:lazy_references][filter_type])
|
||||
result[:references][filter_type]
|
||||
end
|
||||
|
||||
lazy_values = Gitlab::Markdown::ReferenceFilter::LazyReference.load(lazy_references)
|
||||
values.concat(lazy_values)
|
||||
values
|
||||
Gitlab::Markdown::ReferenceFilter::LazyReference.load(values)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Reference in New Issue