plumb keep-original-wf-id

Author: THardy98

temporalio/lib/temporalio/client/schedule.rb

@@ -716,7 +716,8 @@ def _to_proto
       Policy = Data.define(
         :overlap,
         :catchup_window,
-        :pause_on_failure
+        :pause_on_failure,
+        :keep_original_workflow_id
       )
 
       # Policies of a schedule.
@@ -728,6 +729,8 @@ def _to_proto
       # @!attribute pause_on_failure
       #   @return [Boolean] Whether to pause the schedule if an action fails or times out. Note: For workflows, this
       #     only applies after all retries have been exhausted.
+      # @!attribute keep_original_workflow_id
+      #   @return [Boolean] Whether to keep the original workflow ID without appending a timestamp for uniqueness.
       class Policy
         # @!visibility private
         def self._from_proto(raw_policies)
@@ -736,7 +739,8 @@ def self._from_proto(raw_policies)
                                                       raw_policies.overlap_policy,
                                                       zero_means_nil: true),
             catchup_window: Internal::ProtoUtils.duration_to_seconds(raw_policies.catchup_window) || raise, # Never nil
-            pause_on_failure: raw_policies.pause_on_failure
+            pause_on_failure: raw_policies.pause_on_failure,
+            keep_original_workflow_id: raw_policies.keep_original_workflow_id
           )
         end
 
@@ -747,10 +751,13 @@ def self._from_proto(raw_policies)
         #   missed actions.
         # @param pause_on_failure [Boolean] Whether to pause the schedule if an action fails or times out. Note: For
         #   workflows, this only applies after all retries have been exhausted.
+        # @param keep_original_workflow_id [Boolean] Whether to keep the original workflow ID without appending a
+        #   timestamp for uniqueness.
         def initialize(
           overlap: OverlapPolicy::SKIP,
           catchup_window: 365 * 24 * 60 * 60.0,
-          pause_on_failure: false
+          pause_on_failure: false,
+          keep_original_workflow_id: false
         )
           super
         end
@@ -760,7 +767,8 @@ def _to_proto
           Api::Schedule::V1::SchedulePolicies.new(
             overlap_policy: overlap,
             catchup_window: Internal::ProtoUtils.seconds_to_duration(catchup_window),
-            pause_on_failure:
+            pause_on_failure:,
+            keep_original_workflow_id:
           )
         end
       end

temporalio/sig/temporalio/client/schedule.rbs

@@ -239,13 +239,15 @@ module Temporalio
         attr_reader overlap: OverlapPolicy::enum
         attr_reader catchup_window: duration
         attr_reader pause_on_failure: bool
+        attr_reader keep_original_workflow_id: bool
 
         def self._from_proto: (untyped raw_policies) -> Policy
 
         def initialize: (
           ?overlap: OverlapPolicy::enum,
           ?catchup_window: duration,
-          ?pause_on_failure: bool
+          ?pause_on_failure: bool,
+          ?keep_original_workflow_id: bool
         ) -> void
 
         def _to_proto: -> untyped
@@ -332,4 +334,4 @@ module Temporalio
       end
     end
   end
-end
+end

temporalio/test/client_schedule_test.rb

@@ -53,7 +53,8 @@ def test_basics # rubocop:disable Metrics/AbcSize
       policy: Temporalio::Client::Schedule::Policy.new(
         overlap: Temporalio::Client::Schedule::OverlapPolicy::BUFFER_ONE,
         catchup_window: 5 * 60.0,
-        pause_on_failure: true
+        pause_on_failure: true,
+        keep_original_workflow_id: true
       ),
       state: Temporalio::Client::Schedule::State.new(
         note: 'sched note 1',
@@ -124,6 +125,7 @@ def test_basics # rubocop:disable Metrics/AbcSize
     new_schedule = Temporalio::Client::Schedule.new(
       action:,
       spec: Temporalio::Client::Schedule::Spec.new,
+      policy: Temporalio::Client::Schedule::Policy.new(keep_original_workflow_id: true),
       state: Temporalio::Client::Schedule::State.new(paused: true)
     )
     handle.update { Temporalio::Client::Schedule::Update.new(schedule: new_schedule) }
@@ -170,6 +172,7 @@ def test_basics # rubocop:disable Metrics/AbcSize
       # Check results
       exec = desc.info.recent_actions.first&.action #: Temporalio::Client::Schedule::ActionExecution::StartWorkflow
       assert_instance_of Temporalio::Client::Schedule::ActionExecution::StartWorkflow, exec
+      assert_equal action.id, exec.workflow_id
       assert_equal 'some-result',
                    env.client.workflow_handle(exec.workflow_id,
                                               first_execution_run_id: exec.first_execution_run_id).result