16 lines
384 B
Ruby
16 lines
384 B
Ruby
# frozen_string_literal: true
|
|
|
|
class UnionFinder
|
|
# rubocop: disable CodeReuse/ActiveRecord
|
|
def find_union(segments, klass)
|
|
if segments.length > 1
|
|
union = Gitlab::SQL::Union.new(segments.map { |s| s.select(:id) })
|
|
|
|
klass.where("#{klass.table_name}.id IN (#{union.to_sql})")
|
|
else
|
|
segments.first
|
|
end
|
|
end
|
|
# rubocop: enable CodeReuse/ActiveRecord
|
|
end
|