add updatable timer sample

Author: chris-olszewski

test/updatable_timer/updatable_timer_workflow_test.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require 'test'
+require 'updatable_timer/updatable_timer_workflow'
+require 'securerandom'
+require 'temporalio/client'
+require 'temporalio/testing'
+require 'temporalio/worker'
+
+module UpdatableTimer
+  class UpdatableTimerWorkflowTest < Test
+    def test_workflow
+      Temporalio::Testing::WorkflowEnvironment.start_time_skipping do |env|
+        # Run workflow in a worker
+        worker = Temporalio::Worker.new(
+          client: env.client,
+          task_queue: "tq-#{SecureRandom.uuid}",
+          workflows: [UpdatableTimerWorkflow]
+        )
+        worker.run do
+          day_from_now = (Time.now(in: 'utc') + (24 * 60 * 60)).to_r
+          hour_from_now = (Time.now(in: 'utc') + (60 * 60)).to_r
+          handle = env.client.start_workflow(
+            UpdatableTimerWorkflow, day_from_now,
+            id: "wf-#{SecureRandom.uuid}", task_queue: worker.task_queue
+          )
+          assert_equal day_from_now, Rational(handle.query(UpdatableTimerWorkflow.wake_up_time))
+          handle.signal(UpdatableTimerWorkflow.update_wake_up_time, hour_from_now)
+          assert_equal hour_from_now, Rational(handle.query(UpdatableTimerWorkflow.wake_up_time))
+
+          handle.result
+        end
+      end
+    end
+  end
+end

updatable_timer/README.md

@@ -0,0 +1,25 @@
+# Updatable Timer Sample
+
+Demonstrates a helper class which relies on `Temporalio::Workflow.wait_condition` to implement a blocking sleep that can be updated at any moment.
+
+To run, first see [README.md](../README.md) for prerequisites. Then, in another terminal, start the Ruby worker from this directory:
+
+```bash
+  bundle exec ruby worker.rb
+```
+
+Then in another terminal, use the Ruby client to the workflow from this directory:
+
+```bash
+  bundle exec ruby starter.rb
+```
+
+The Ruby code will invoke the workflow which will create a timer that will resolve in a day.
+
+Finally run the updater to change the timer to 10 seconds from now:
+
+```bash
+  bundle exec ruby wake_up_timer_updater.rb
+```
+
+There is also a [test](../test/updatable_timer/updatable_timer_workflow_test.rb) that demonstrates querying the wake up time.

updatable_timer/starter.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+require 'temporalio/client'
+require_relative 'updatable_timer_workflow'
+
+# Create a Temporal client
+logger = Logger.new($stdout, level: Logger::INFO)
+client = Temporalio::Client.connect('localhost:7233', 'default', logger:)
+
+# Run workflow
+logger.info('Starting timer')
+client.execute_workflow(
+  UpdatableTimer::UpdatableTimerWorkflow, (Time.now(in: 'utc') + (24 * 60 * 60)).to_r,
+  id: 'updatable-timer-sample-workflow-id', task_queue: 'updatable-timer'
+)
+logger.info('Timer complete')

updatable_timer/updatable_timer.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require 'temporalio/workflow'
+require 'temporalio/error'
+
+module UpdatableTimer
+  class UpdatableTimer
+    def initialize(wake_up_time)
+      @wake_up_time = wake_up_time
+      @wake_up_time_updated = false
+    end
+
+    attr_reader :wake_up_time
+
+    def wake_up_time=(wake_up_time)
+      Temporalio::Workflow.logger.info("update_wake_up_time: #{wake_up_time}")
+      @wake_up_time = wake_up_time
+      @wake_up_time_updated = true
+    end
+
+    def sleep
+      Temporalio::Workflow.logger.info("sleep until: #{@wake_up_time}")
+      loop do
+        now = Temporalio::Workflow.now
+        sleep_interval = @wake_up_time - now
+
+        break if sleep_interval.negative?
+
+        Temporalio::Workflow.logger.info("going to sleep for: #{sleep_interval}")
+
+        begin
+          @wake_up_time_updated = false
+          Temporalio::Workflow.timeout(sleep_interval) do
+            Temporalio::Workflow.wait_condition { @wake_up_time_updated }
+          end
+        rescue Timeout::Error
+          next
+        end
+      end
+      Temporalio::Workflow.logger.info('sleep_until completed')
+    end
+  end
+end

updatable_timer/updatable_timer_workflow.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+require 'temporalio/workflow'
+require_relative 'updatable_timer'
+
+module UpdatableTimer
+  class UpdatableTimerWorkflow < Temporalio::Workflow::Definition
+    workflow_arg_hint Time
+
+    def execute(wake_up_time)
+      @timer = UpdatableTimer.new(Time.at(Rational(wake_up_time)))
+      @timer.sleep
+    end
+
+    workflow_query
+    def wake_up_time
+      Temporalio::Workflow.logger.info('get_wake_up_time')
+      @timer.wake_up_time.to_r
+    end
+
+    workflow_signal
+    def update_wake_up_time(wake_up_time)
+      wake_up_time = Time.at(Rational(wake_up_time))
+      Temporalio::Workflow.logger.info("update_wake_up_time: #{wake_up_time}")
+      @timer.wake_up_time = wake_up_time
+    end
+  end
+end

updatable_timer/wakeup_time_updater.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'temporalio/client'
+require_relative 'updatable_timer_workflow'
+
+# Create a Temporal client
+logger = Logger.new($stdout, level: Logger::INFO)
+client = Temporalio::Client.connect('localhost:7233', 'default', logger:)
+handle = client.workflow_handle('updatable-timer-sample-workflow-id')
+
+handle.signal(UpdatableTimer::UpdatableTimerWorkflow.update_wake_up_time, (Time.now(in: 'utc') + 10).to_r)
+logger.info('Updated wake up time to 10 seconds from now')

updatable_timer/worker.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative 'updatable_timer_workflow'
+require 'logger'
+require 'temporalio/client'
+require 'temporalio/worker'
+
+# Create a Temporal client
+client = Temporalio::Client.connect(
+  'localhost:7233',
+  'default',
+  logger: Logger.new($stdout, level: Logger::INFO)
+)
+
+# Create worker with the activities and workflow
+worker = Temporalio::Worker.new(
+  client:,
+  task_queue: 'updatable-timer',
+  workflows: [UpdatableTimer::UpdatableTimerWorkflow]
+)
+
+# Run the worker until SIGINT
+puts 'Starting worker (ctrl+c to exit)'
+worker.run(shutdown_signals: ['SIGINT'])