debian-mirror-gitlab/app/models/event_collection.rb

118 lines
3.3 KiB
Ruby
Raw Normal View History

2018-11-18 11:00:15 +05:30
# frozen_string_literal: true
2017-09-10 17:25:29 +05:30
# A collection of events to display in an event list.
#
# An EventCollection is meant to be used for displaying events to a user (e.g.
# in a controller), it's not suitable for building queries that are used for
# building other queries.
class EventCollection
2019-12-21 20:55:43 +05:30
include Gitlab::Utils::StrongMemoize
2017-09-10 17:25:29 +05:30
# To prevent users from putting too much pressure on the database by cycling
# through thousands of events we put a limit on the number of pages.
MAX_PAGE = 10
# projects - An ActiveRecord::Relation object that returns the projects for
# which to retrieve events.
# filter - An EventFilter instance to use for filtering events.
2019-12-21 20:55:43 +05:30
def initialize(projects, limit: 20, offset: 0, filter: nil, groups: nil)
2017-09-10 17:25:29 +05:30
@projects = projects
@limit = limit
@offset = offset
@filter = filter
2019-12-21 20:55:43 +05:30
@groups = groups
2017-09-10 17:25:29 +05:30
end
# Returns an Array containing the events.
def to_a
return [] if current_page > MAX_PAGE
2019-12-21 20:55:43 +05:30
relation = if groups
project_and_group_events
2017-09-10 17:25:29 +05:30
else
2020-03-13 15:44:24 +05:30
project_events
2017-09-10 17:25:29 +05:30
end
2020-04-22 19:07:51 +05:30
relation = apply_feature_flags(relation)
2019-12-21 20:55:43 +05:30
relation = paginate_events(relation)
2017-09-10 17:25:29 +05:30
relation.with_associations.to_a
end
2020-03-13 15:44:24 +05:30
def all_project_events
2020-04-22 19:07:51 +05:30
apply_feature_flags(Event.from_union([project_events]).recent)
2020-03-13 15:44:24 +05:30
end
2017-09-10 17:25:29 +05:30
private
2020-04-22 19:07:51 +05:30
def apply_feature_flags(events)
return events if ::Feature.enabled?(:wiki_events)
events.not_wiki_page
end
2020-03-13 15:44:24 +05:30
def project_events
relation_with_join_lateral('project_id', projects)
end
2019-12-21 20:55:43 +05:30
def project_and_group_events
group_events = relation_with_join_lateral('group_id', groups)
2017-09-10 17:25:29 +05:30
2019-12-21 20:55:43 +05:30
Event.from_union([project_events, group_events]).recent
2017-09-10 17:25:29 +05:30
end
# This relation is built using JOIN LATERAL, producing faster queries than a
# regular LIMIT + OFFSET approach.
2019-12-21 20:55:43 +05:30
def relation_with_join_lateral(parent_column, parents)
parents_for_lateral = parents.select(:id).to_sql
2017-09-10 17:25:29 +05:30
lateral = filtered_events
.limit(limit_for_join_lateral)
2019-12-21 20:55:43 +05:30
.where("events.#{parent_column} = parents_for_lateral.id") # rubocop:disable GitlabSecurity/SqlInjection
2017-09-10 17:25:29 +05:30
.to_sql
# The outer query does not need to re-apply the filters since the JOIN
# LATERAL body already takes care of this.
2019-12-21 20:55:43 +05:30
base_relation
.from("(#{parents_for_lateral}) parents_for_lateral")
2017-09-10 17:25:29 +05:30
.joins("JOIN LATERAL (#{lateral}) AS #{Event.table_name} ON true")
end
def filtered_events
@filter ? @filter.apply_filter(base_relation) : base_relation
end
def paginate_events(events)
events.limit(@limit).offset(@offset)
end
def base_relation
# We want to have absolute control over the event queries being built, thus
# we're explicitly opting out of any default scopes that may be set.
Event.unscoped.recent
end
def limit_for_join_lateral
# Applying the OFFSET on the inside of a JOIN LATERAL leads to incorrect
# results. To work around this we need to increase the inner limit for every
# page.
#
# This means that on page 1 we use LIMIT 20, and an outer OFFSET of 0. On
# page 2 we use LIMIT 40 and an outer OFFSET of 20.
@limit + @offset
end
def current_page
(@offset / @limit) + 1
end
def projects
@projects.except(:order)
end
2019-12-21 20:55:43 +05:30
def groups
strong_memoize(:groups) do
groups.except(:order) if @groups
end
end
2017-09-10 17:25:29 +05:30
end