Commit f89f232d authored by Grzegorz Bizon's avatar Grzegorz Bizon

Simplify pipelines preloader implementation

parent 0b3cca56
......@@ -24,7 +24,7 @@ class Projects::PipelinesController < Projects::ApplicationController
@finished_count = limited_pipelines_count(project, 'finished')
@pipelines_count = limited_pipelines_count(project)
Gitlab::Ci::Pipeline::Preloader.new(@pipelines).preload!
Gitlab::Ci::Pipeline::Preloader.preload!(@pipelines)
respond_to do |format|
format.html
......
......@@ -6,20 +6,16 @@ module Gitlab
# Class for preloading data associated with pipelines such as commit
# authors.
class Preloader
def initialize(pipelines)
@pipelines = pipelines
def self.preload!(pipelines)
pipelines.each do |pipeline|
self.new(pipeline).tap do |preloader|
preloader.preload_commits
preloader.preload_pipeline_warnings
preloader.preload_stages_warnings
end
def preload!
@pipelines.each do |pipeline|
Pipeline::Preloader::Instance.new(pipeline)
.preload_commits
.preload_pipeline_warnings
.preload_stages_warnings
end
end
class Instance
def initialize(pipeline)
@pipeline = pipeline
end
......@@ -30,22 +26,21 @@ module Gitlab
#
# This also preloads the author of every commit. We're using "lazy_author"
# here since "author" immediately loads the data on the first call.
tap { @pipeline.commit.try(:lazy_author) }
@pipeline.commit.try(:lazy_author)
end
def preload_pipeline_warnings
# This preloads the number of warnings for every pipeline, ensuring
# that Ci::Pipeline#has_warnings? doesn't execute any additional
# queries.
tap { @pipeline.number_of_warnings }
@pipeline.number_of_warnings
end
def preload_stages_warnings
# This preloads the number of warnings for every stage, ensuring
# that Ci::Stage#has_warnings? doesn't execute any additional
# queries.
tap { @pipeline.stages.each { |stage| stage.number_of_warnings } }
end
@pipeline.stages.each { |stage| stage.number_of_warnings }
end
end
end
......
......@@ -10,13 +10,13 @@ describe Gitlab::Ci::Pipeline::Preloader do
double(:pipeline, commit: commit, stages: [stage])
end
describe '#preload!' do
describe '.preload!' do
it 'preloads commit authors and number of warnings' do
expect(commit).to receive(:lazy_author)
expect(pipeline).to receive(:number_of_warnings)
expect(stage).to receive(:number_of_warnings)
described_class.new([pipeline]).preload!
described_class.preload!([pipeline])
end
end
end
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment