Populate pipeline with objects before creating it

2 files changed, 32 insertions, 7 deletions

diff --git a/lib/gitlab/ci/pipeline/chain/create.rb b/lib/gitlab/ci/pipeline/chain/create.rb
index 753d0e89b1e..930d767b0f3 100644
--- a/lib/gitlab/ci/pipeline/chain/create.rb
+++ b/lib/gitlab/ci/pipeline/chain/create.rb
@@ -9,13 +9,6 @@ module Gitlab
             ::Ci::Pipeline.transaction do
               pipeline.save!
 
-              @command.seeds_block&.call(pipeline)
-
-              pipeline.stage_seeds.each do |seed|
-                seed.user = current_user
-                seed.to_resource.save!
-              end
-
               # TODO populate environments with find_or_initialize_by in the chain too.
 
               ##
diff --git a/lib/gitlab/ci/pipeline/chain/populate.rb b/lib/gitlab/ci/pipeline/chain/populate.rb
new file mode 100644
index 00000000000..5368ff15481
--- /dev/null
+++ b/lib/gitlab/ci/pipeline/chain/populate.rb
@@ -0,0 +1,32 @@
+module Gitlab
+  module Ci
+    module Pipeline
+      module Chain
+        class Populate < Chain::Base
+          PopulateError = Class.new(StandardError)
+
+          def perform!
+            ##
+            # Populate pipeline with seeds block.
+            #
+            # It comes from a block argument to CreatePipelineService#execute.
+            #
+            @command.seeds_block&.call(pipeline)
+
+            pipeline.stage_seeds.each do |seed|
+              seed.user = current_user
+
+              pipeline.stages << seed.to_resource
+            end
+
+            raise Populate::PopulateError if pipeline.persisted?
+          end
+
+          def break?
+            pipeline.persisted?
+          end
+        end
+      end
+    end
+  end
+end