ductwork 0.7.1 → 0.7.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6174d886b38c9a000eda9937e27efabd0f89a803c5c77fb5739c2b67d8c004c9
-  data.tar.gz: 6300c6f3d238e2f92aa74c5ebe01c37bcbf842693f186b9c56e8d8b2cbcf567d
+  metadata.gz: a3a59fb73965f325d3e23243f59e7e973419f3683d2de741cf5bbf115bccad8d
+  data.tar.gz: d66a5c291d197b7a8e679d23e9c68b6575d2e35100b06f7c412c97e5bc515ec3
 SHA512:
-  metadata.gz: fcda67f0378eb7e4108820ce87ac4b508698abcd02e71ef77b7ec8442dfeeea019a77e35bacba5a32fda4101004d2c0f349441d3921838deee97aea6ac93eed5
-  data.tar.gz: e74fef5cf5bbf80b8b112616636a556a6ec7742ade2fe44dae3679b3c679ef7b9eb96dc81ad016fb1d4285a06cef874f74d1f6c67a20062b5eaa0bdcfde6061c
+  metadata.gz: de5ca4b1c9dff4478992234183ae84d810cec930eede8d97dc7511959efad9521acb201ec5f39a2d96d430383ee3f240d3a3395bc55fc19ce272603000eb92a2
+  data.tar.gz: b86ea8a91f912ae55e509f2c059fc8db8d4ccff0055fe96db77593a98ce188255b2bbcb395709a0b672ded6b8df1a1537001fc56cfb3de7244a5dda9a4e48192

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Ductwork Changelog
 
+## [0.7.2]
+
+- chore: small DRY refactor in pipeline advancement
+
 ## [0.7.1]
 
 - chore: isolate on halt execution in `Ductwork::Job#execution_failed!`

data/lib/ductwork/models/pipeline.rb

@@ -113,9 +113,14 @@ module Ductwork
 
     private
 
-    def create_step_and_enqueue_job(klass:, step_type:, input_arg:)
+    def create_step_and_enqueue_job(edge:, input_arg:, klass: nil)
       status = :in_progress
       started_at = Time.current
+      # NOTE: "chain" is used by ActiveRecord so we have to call
+      # this enum value "default" :sad:
+      step_type = edge[:type] == "chain" ? "default" : edge[:type]
+      klass ||= edge.fetch(:to).sole
+
       next_step = steps.create!(klass:, status:, step_type:, started_at:)
       Ductwork::Job.enqueue(next_step, input_arg)
     end
@@ -157,11 +162,8 @@ module Ductwork
       else
         edges.each do |step_klass, step_edges|
           edge = step_edges[-1]
-          # NOTE: "chain" is used by ActiveRecord so we have to call
-          # this enum value "default" :sad:
-          step_type = edge[:type] == "chain" ? "default" : edge[:type]
 
-          if step_type == "collapse"
+          if edge[:type] == "collapse"
             conditionally_collapse_next_steps(step_klass, edge, advancing_ids)
           else
             advance_non_merging_steps(step_klass, edge, advancing_ids)
@@ -172,15 +174,13 @@ module Ductwork
     end
 
     def advance_non_merging_steps(step_klass, edge, advancing_ids)
-      # NOTE: "chain" is used by ActiveRecord so we have to call
-      # this enum value "default" :sad:
-      step_type = edge[:type] == "chain" ? "default" : edge[:type]
+      step_type = edge[:type]
 
       steps.where(id: advancing_ids, klass: step_klass).find_each do |step|
-        if step_type.in?(%w[default divide])
-          advance_to_next_steps(step_type, step.id, edge)
+        if step_type.in?(%w[chain divide])
+          advance_to_next_steps(step.id, edge)
         elsif step_type == "expand"
-          expand_to_next_steps(step_type, step.id, edge)
+          expand_to_next_steps(step.id, edge)
         else
           Ductwork.logger.error(
             msg: "Invalid step type",
@@ -192,7 +192,7 @@ module Ductwork
       end
     end
 
-    def advance_to_next_steps(step_type, step_id, edge)
+    def advance_to_next_steps(step_id, edge)
       too_many = edge[:to].tally.any? do |to_klass, count|
         depth = Ductwork
                 .configuration
@@ -204,15 +204,9 @@ module Ductwork
       if too_many
         halted!
       else
-        edge[:to].each do |to_klass|
-          next_step = steps.create!(
-            klass: to_klass,
-            status: :in_progress,
-            step_type: step_type,
-            started_at: Time.current
-          )
-          return_value = Ductwork::Job.find_by(step_id:).return_value
-          Ductwork::Job.enqueue(next_step, return_value)
+        edge[:to].each do |klass|
+          input_arg = Ductwork::Job.find_by(step_id:).return_value
+          create_step_and_enqueue_job(edge:, input_arg:, klass:)
         end
       end
     end
@@ -230,8 +224,7 @@ module Ductwork
     end
 
     def combine_next_steps(edges, advancing_ids)
-      klass = edges.values.sample.dig(-1, :to).sole
-      step_type = "combine"
+      edge = edges.values.sample[-1]
       groups = steps
                .where(id: advancing_ids)
                .group(:klass)
@@ -243,11 +236,11 @@ module Ductwork
         input_arg = Ductwork::Job
                     .where(step_id: group.map(&:id))
                     .map(&:return_value)
-        create_step_and_enqueue_job(klass:, step_type:, input_arg:)
+        create_step_and_enqueue_job(edge:, input_arg:)
       end
     end
 
-    def expand_to_next_steps(step_type, step_id, edge)
+    def expand_to_next_steps(step_id, edge)
       next_klass = edge[:to].sole
       return_value = Ductwork::Job
                      .find_by(step_id:)
@@ -258,18 +251,14 @@ module Ductwork
         halted!
       else
         Array(return_value).each do |input_arg|
-          create_step_and_enqueue_job(
-            klass: next_klass,
-            step_type: step_type,
-            input_arg: input_arg
-          )
+          create_step_and_enqueue_job(edge:, input_arg:)
         end
       end
     end
 
     def conditionally_collapse_next_steps(step_klass, edge, advancing_ids)
       if steps.where(status: %w[pending in_progress], klass: step_klass).none?
-        collapse_next_steps(edge[:to].sole, advancing_ids)
+        collapse_next_steps(edge, advancing_ids)
       else
         Ductwork.logger.debug(
           msg: "Not all expanded steps have completed; not collapsing",
@@ -279,15 +268,14 @@ module Ductwork
       end
     end
 
-    def collapse_next_steps(klass, advancing_ids)
-      step_type = "collapse"
+    def collapse_next_steps(edge, advancing_ids)
       input_arg = []
 
       Ductwork::Job.where(step_id: advancing_ids).find_each do |job|
         input_arg << job.return_value
       end
 
-      create_step_and_enqueue_job(klass:, step_type:, input_arg:)
+      create_step_and_enqueue_job(edge:, input_arg:)
     end
 
     def log_pipeline_advanced(edges)

data/lib/ductwork/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ductwork
-  VERSION = "0.7.1"
+  VERSION = "0.7.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ductwork
 version: !ruby/object:Gem::Version
-  version: 0.7.1
+  version: 0.7.2
 platform: ruby
 authors:
 - Tyler Ewing