amit-temporalio 0.3.0-x86_64-darwin

data/lib/temporalio/client/schedule_handle.rb

@@ -0,0 +1,126 @@
+# frozen_string_literal: true
+
+require 'temporalio/client/interceptor'
+require 'temporalio/client/schedule'
+
+module Temporalio
+  class Client
+    # Handle for interacting with a schedule. This is usually created via {Client.create_schedule} or
+    # {Client.schedule_handle}.
+    class ScheduleHandle
+      # @return [String] ID of the schedule.
+      attr_reader :id
+
+      # @!visibility private
+      def initialize(client:, id:)
+        @client = client
+        @id = id
+      end
+
+      # Backfill the schedule by going through the specified time periods as if they passed right now.
+      #
+      # @param backfills [Array<Schedule::Backfill>] Backfill periods.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def backfill(
+        *backfills,
+        rpc_options: nil
+      )
+        raise ArgumentError, 'At least one backfill required' if backfills.empty?
+
+        @client._impl.backfill_schedule(Interceptor::BackfillScheduleInput.new(
+                                          id:,
+                                          backfills:,
+                                          rpc_options:
+                                        ))
+      end
+
+      # Delete this schedule.
+      #
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def delete(rpc_options: nil)
+        @client._impl.delete_schedule(Interceptor::DeleteScheduleInput.new(
+                                        id:,
+                                        rpc_options:
+                                      ))
+      end
+
+      # Fetch this schedule's description.
+      #
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [Schedule::Description] Schedule description.
+      # @raise [Error::RPCError] RPC error from call.
+      def describe(rpc_options: nil)
+        @client._impl.describe_schedule(Interceptor::DescribeScheduleInput.new(
+                                          id:,
+                                          rpc_options:
+                                        ))
+      end
+
+      # Pause the schedule and set a note.
+      #
+      # @param note [String] Note to set on the schedule.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def pause(note: 'Paused via Ruby SDK', rpc_options: nil)
+        @client._impl.pause_schedule(Interceptor::PauseScheduleInput.new(
+                                       id:,
+                                       note:,
+                                       rpc_options:
+                                     ))
+      end
+
+      # Trigger an action on this schedule to happen immediately.
+      #
+      # @param overlap [Schedule::OverlapPolicy, nil] If set, overrides the schedule's overlap policy.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def trigger(overlap: nil, rpc_options: nil)
+        @client._impl.trigger_schedule(Interceptor::TriggerScheduleInput.new(
+                                         id:,
+                                         overlap:,
+                                         rpc_options:
+                                       ))
+      end
+
+      # Unpause the schedule and set a note.
+      #
+      # @param note [String] Note to set on the schedule.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def unpause(note: 'Unpaused via Ruby SDK', rpc_options: nil)
+        @client._impl.unpause_schedule(Interceptor::UnpauseScheduleInput.new(
+                                         id:,
+                                         note:,
+                                         rpc_options:
+                                       ))
+      end
+
+      # Update a schedule using a callback to build the update from the description.
+      #
+      # NOTE: In future versions, the callback may be invoked multiple times in a conflict-resolution loop.
+      #
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      # @yield Block called to obtain the update.
+      # @yieldparam [Schedule::Update::Input] Parameter to the block that contains a description with the schedule to be
+      #   updated.
+      # @yieldreturn [Schedule::Update, nil] The update to apply, or `nil` to not perform an update.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def update(rpc_options: nil, &updater)
+        @client._impl.update_schedule(Interceptor::UpdateScheduleInput.new(
+                                        id:,
+                                        updater:,
+                                        rpc_options:
+                                      ))
+      end
+    end
+  end
+end

data/lib/temporalio/client/workflow_execution.rb

@@ -0,0 +1,100 @@
+# frozen_string_literal: true
+
+require 'temporalio/api'
+require 'temporalio/client/workflow_execution_status'
+require 'temporalio/internal/proto_utils'
+require 'temporalio/search_attributes'
+
+module Temporalio
+  class Client
+    # Info for a single workflow execution run.
+    class WorkflowExecution
+      # @return [Api::Workflow::V1::WorkflowExecutionInfo] Underlying protobuf info.
+      attr_reader :raw_info
+
+      # @!visibility private
+      def initialize(raw_info, data_converter)
+        @raw_info = raw_info
+        @memo = Internal::ProtoUtils::LazyMemo.new(raw_info.memo, data_converter)
+        @search_attributes = Internal::ProtoUtils::LazySearchAttributes.new(raw_info.search_attributes)
+      end
+
+      # @return [Time, nil] When the workflow was closed if closed.
+      def close_time
+        Internal::ProtoUtils.timestamp_to_time(@raw_info.close_time)
+      end
+
+      # @return [Time, nil] When this workflow run started or should start.
+      def execution_time
+        Internal::ProtoUtils.timestamp_to_time(@raw_info.execution_time)
+      end
+
+      # @return [Integer] Number of events in the history.
+      def history_length
+        @raw_info.history_length
+      end
+
+      # @return [String] ID for the workflow.
+      def id
+        @raw_info.execution.workflow_id
+      end
+
+      # @return [Hash<String, Object>, nil] Memo for the workflow.
+      def memo
+        @memo.get
+      end
+
+      # @return [String, nil] ID for the parent workflow if this was started as a child.
+      def parent_id
+        @raw_info.parent_execution&.workflow_id
+      end
+
+      # @return [String, nil] Run ID for the parent workflow if this was started as a child.
+      def parent_run_id
+        @raw_info.parent_execution&.run_id
+      end
+
+      # @return [String] Run ID for this workflow run.
+      def run_id
+        @raw_info.execution.run_id
+      end
+
+      # @return [SearchAttributes, nil] Current set of search attributes if any.
+      def search_attributes
+        @search_attributes.get
+      end
+
+      # @return [Time] When the workflow was created.
+      def start_time
+        Internal::ProtoUtils.timestamp_to_time(@raw_info.start_time) || raise # Never nil
+      end
+
+      # @return [WorkflowExecutionStatus] Status for the workflow.
+      def status
+        Internal::ProtoUtils.enum_to_int(Api::Enums::V1::WorkflowExecutionStatus, @raw_info.status)
+      end
+
+      # @return [String] Task queue for the workflow.
+      def task_queue
+        @raw_info.task_queue
+      end
+
+      # @return [String] Type name for the workflow.
+      def workflow_type
+        @raw_info.type.name
+      end
+
+      # Description for a single workflow execution run.
+      class Description < WorkflowExecution
+        # @return [Api::WorkflowService::V1::DescribeWorkflowExecutionResponse] Underlying protobuf description.
+        attr_reader :raw_description
+
+        # @!visibility private
+        def initialize(raw_description, data_converter)
+          super(raw_description.workflow_execution_info, data_converter)
+          @raw_description = raw_description
+        end
+      end
+    end
+  end
+end

data/lib/temporalio/client/workflow_execution_count.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Temporalio
+  class Client
+    # Representation of a count from a count workflows call.
+    class WorkflowExecutionCount
+      # @return [Integer] Approximate number of workflows matching the original query. If the query had a group-by
+      # clause, this is simply the sum of all the counts in {groups}.
+      attr_reader :count
+
+      # @return [Array<AggregationGroup>] Groups if the query had a group-by clause, or empty if not.
+      attr_reader :groups
+
+      # @!visibility private
+      def initialize(count, groups)
+        @count = count
+        @groups = groups
+      end
+
+      # Aggregation group if the workflow count query had a group-by clause.
+      class AggregationGroup
+        # @return [Integer] Approximate number of workflows matching the original query for this group.
+        attr_reader :count
+
+        # @return [Array<Object>] Search attribute values for this group.
+        attr_reader :group_values
+
+        # @!visibility private
+        def initialize(count, group_values)
+          @count = count
+          @group_values = group_values
+        end
+      end
+    end
+  end
+end

data/lib/temporalio/client/workflow_execution_status.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require 'temporalio/api'
+
+module Temporalio
+  class Client
+    # Status of a workflow execution.
+    module WorkflowExecutionStatus
+      RUNNING = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_RUNNING
+      COMPLETED = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_COMPLETED
+      FAILED = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_FAILED
+      CANCELED = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_CANCELED
+      TERMINATED = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_TERMINATED
+      CONTINUED_AS_NEW = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_CONTINUED_AS_NEW
+      TIMED_OUT = Api::Enums::V1::WorkflowExecutionStatus::WORKFLOW_EXECUTION_STATUS_TIMED_OUT
+    end
+  end
+end

data/lib/temporalio/client/workflow_handle.rb

@@ -0,0 +1,389 @@
+# frozen_string_literal: true
+
+require 'securerandom'
+require 'temporalio/api'
+require 'temporalio/client/interceptor'
+require 'temporalio/client/workflow_update_handle'
+require 'temporalio/client/workflow_update_wait_stage'
+require 'temporalio/error'
+require 'temporalio/workflow_history'
+
+module Temporalio
+  class Client
+    # Handle for interacting with a workflow. This is usually created via {Client.start_workflow} or
+    # {Client.workflow_handle}.
+    class WorkflowHandle
+      # @return [String] ID for the workflow.
+      attr_reader :id
+
+      # Run ID used for {signal}, {query}, and {start_update}/{execute_update} calls if present to ensure the
+      # signal/query/update happen on this exact run.
+      #
+      # This is only created via {Client.workflow_handle}. {Client.start_workflow} will not set this value.
+      #
+      # This cannot be mutated. If a different run ID is needed, {Client.workflow_handle} must be used instead.
+      #
+      # @return [String, nil] Run ID.
+      attr_reader :run_id
+
+      # Run ID used for {result} calls if present to ensure result is for a workflow starting from this run.
+      #
+      # When this handle is created via {Client.workflow_handle}, this is the same as {run_id}. When this handle is
+      # created via {Client.start_workflow}, this value will be the resulting run ID.
+      #
+      # This cannot be mutated. If a different run ID is needed, {Client.workflow_handle} must be used instead.
+      #
+      # @return [String, nil] Result run ID.
+      attr_reader :result_run_id
+
+      # Run ID used for some calls like {cancel} and {terminate} to ensure the cancel and terminate happen for a
+      # workflow ID on a chain started with this run ID.
+      #
+      # This can be set when using {Client.workflow_handle}. When {Client.start_workflow} is called without a start
+      # signal, this is set to the resulting run.
+      #
+      # This cannot be mutated. If a different first execution run ID is needed, {Client.workflow_handle} must be used
+      # instead.
+      #
+      # @return [String, nil] First execution run ID.
+      attr_reader :first_execution_run_id
+
+      # @!visibility private
+      def initialize(client:, id:, run_id:, result_run_id:, first_execution_run_id:)
+        @client = client
+        @id = id
+        @run_id = run_id
+        @result_run_id = result_run_id
+        @first_execution_run_id = first_execution_run_id
+      end
+
+      # Wait for the result of the workflow.
+      #
+      # This will use {result_run_id} if present to base the result on. To use another run ID, a new handle must be
+      # created via {Client.workflow_handle}.
+      #
+      # @param follow_runs [Boolean] If +true+, workflow runs will be continually fetched across retries and continue as
+      #   new until the latest one is found. If +false+, the first result is used.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [Object] Result of the workflow after being converted by the data converter.
+      #
+      # @raise [Error::WorkflowFailedError] Workflow failed with +cause+ as the cause.
+      # @raise [Error::WorkflowContinuedAsNewError] Workflow continued as new and +follow_runs+ is +false+.
+      # @raise [Error::RPCError] RPC error from call.
+      def result(follow_runs: true, rpc_options: nil)
+        # Wait on the close event, following as needed
+        hist_run_id = result_run_id
+        loop do
+          # Get close event
+          event = fetch_history_events(
+            wait_new_event: true,
+            event_filter_type: Api::Enums::V1::HistoryEventFilterType::HISTORY_EVENT_FILTER_TYPE_CLOSE_EVENT,
+            skip_archival: true,
+            specific_run_id: hist_run_id,
+            rpc_options:
+          ).next
+
+          # Check each close type'
+          case event.event_type
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_COMPLETED
+            attrs = event.workflow_execution_completed_event_attributes
+            hist_run_id = attrs.new_execution_run_id
+            next if follow_runs && hist_run_id && !hist_run_id.empty?
+
+            return @client.data_converter.from_payloads(attrs.result).first
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_FAILED
+            attrs = event.workflow_execution_failed_event_attributes
+            hist_run_id = attrs.new_execution_run_id
+            next if follow_runs && hist_run_id && !hist_run_id.empty?
+
+            raise Error::WorkflowFailedError.new, cause: @client.data_converter.from_failure(attrs.failure)
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_CANCELED
+            attrs = event.workflow_execution_canceled_event_attributes
+            raise Error::WorkflowFailedError.new, 'Workflow execution canceled', cause: Error::CanceledError.new(
+              'Workflow execution canceled',
+              details: @client.data_converter.from_payloads(attrs&.details)
+            )
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_TERMINATED
+            attrs = event.workflow_execution_terminated_event_attributes
+            raise Error::WorkflowFailedError.new, 'Workflow execution terminated', cause: Error::TerminatedError.new(
+              Internal::ProtoUtils.string_or(attrs.reason, 'Workflow execution terminated'),
+              details: @client.data_converter.from_payloads(attrs&.details)
+            )
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_TIMED_OUT
+            attrs = event.workflow_execution_timed_out_event_attributes
+            hist_run_id = attrs.new_execution_run_id
+            next if follow_runs && hist_run_id && !hist_run_id.empty?
+
+            raise Error::WorkflowFailedError.new, 'Workflow execution timed out', cause: Error::TimeoutError.new(
+              'Workflow execution timed out',
+              type: Api::Enums::V1::TimeoutType::TIMEOUT_TYPE_START_TO_CLOSE,
+              last_heartbeat_details: []
+            )
+          when :EVENT_TYPE_WORKFLOW_EXECUTION_CONTINUED_AS_NEW
+            attrs = event.workflow_execution_continued_as_new_event_attributes
+            hist_run_id = attrs.new_execution_run_id
+            next if follow_runs && hist_run_id && !hist_run_id.empty?
+
+            # TODO: Use more specific error and decode failure
+            raise Error::WorkflowContinuedAsNewError.new(new_run_id: attrs.new_execution_run_id)
+          else
+            raise Error, "Unknown close event type: #{event.event_type}"
+          end
+        end
+      end
+
+      # Get workflow details. This will get details for the {run_id} if present. To use a different run ID, create a new
+      # handle via {Client.workflow_handle}.
+      #
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [WorkflowExecution::Description] Workflow description.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of {Client.start_workflow} will describe the latest workflow with the same
+      #   workflow ID even if it is unrelated to the started workflow.
+      def describe(rpc_options: nil)
+        @client._impl.describe_workflow(Interceptor::DescribeWorkflowInput.new(
+                                          workflow_id: id,
+                                          run_id:,
+                                          rpc_options:
+                                        ))
+      end
+
+      # Get workflow history. This is a helper on top of {fetch_history_events}.
+      #
+      # @param event_filter_type [Api::Enums::V1::HistoryEventFilterType] Types of events to fetch.
+      # @param skip_archival [Boolean] Whether to skip archival.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [WorkflowHistory] Workflow history.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def fetch_history(
+        event_filter_type: Api::Enums::V1::HistoryEventFilterType::HISTORY_EVENT_FILTER_TYPE_ALL_EVENT,
+        skip_archival: false,
+        rpc_options: nil
+      )
+        WorkflowHistory.new(
+          fetch_history_events(
+            event_filter_type:,
+            skip_archival:,
+            rpc_options:
+          ).to_a
+        )
+      end
+
+      # Fetch an enumerator of history events for this workflow. Internally this is done in paginated form, but it is
+      # presented as an enumerator.
+      #
+      # @param wait_new_event [Boolean] If +true+, when the end of the current set of events is reached but the workflow
+      #   is not complete, this will wait for the next event. If +false+, the enumerable completes at the end of current
+      #   history.
+      # @param event_filter_type [Api::Enums::V1::HistoryEventFilterType] Types of events to fetch.
+      # @param skip_archival [Boolean] Whether to skip archival.
+      # @param specific_run_id [String, nil] Run ID to fetch events for. Default is the {run_id}. Most users will not
+      #   need to set this and instead use the one on the class.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [Enumerator<Api::History::V1::HistoryEvent>] Enumerable events.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      def fetch_history_events(
+        wait_new_event: false,
+        event_filter_type: Api::Enums::V1::HistoryEventFilterType::HISTORY_EVENT_FILTER_TYPE_ALL_EVENT,
+        skip_archival: false,
+        specific_run_id: run_id,
+        rpc_options: nil
+      )
+        @client._impl.fetch_workflow_history_events(Interceptor::FetchWorkflowHistoryEventsInput.new(
+                                                      workflow_id: id,
+                                                      run_id: specific_run_id,
+                                                      wait_new_event:,
+                                                      event_filter_type:,
+                                                      skip_archival:,
+                                                      rpc_options:
+                                                    ))
+      end
+
+      # Send a signal to the workflow. This will signal for {run_id} if present. To use a different run ID, create a new
+      # handle via {Client.workflow_handle}.
+      #
+      # @param signal [Workflow::Definition::Signal, Symbol, String] Signal definition or name.
+      # @param args [Array<Object>] Signal arguments.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of {Client.start_workflow} will signal the latest workflow with the same
+      #   workflow ID even if it is unrelated to the started workflow.
+      def signal(signal, *args, rpc_options: nil)
+        @client._impl.signal_workflow(Interceptor::SignalWorkflowInput.new(
+                                        workflow_id: id,
+                                        run_id:,
+                                        signal:,
+                                        args:,
+                                        headers: {},
+                                        rpc_options:
+                                      ))
+      end
+
+      # Query the workflow. This will query for {run_id} if present. To use a different run ID, create a new handle via
+      # {Client.workflow_handle}.
+      #
+      # @param query [Workflow::Definition::Query, Symbol, String] Query definition or name.
+      # @param args [Array<Object>] Query arguments.
+      # @param reject_condition [WorkflowQueryRejectCondition, nil] Condition for rejecting the query.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [Object, nil] Query result.
+      #
+      # @raise [Error::WorkflowQueryFailedError] The query on the workflow returned a failure.
+      # @raise [Error::WorkflowQueryRejectedError] A query reject condition was satisfied.
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of {Client.start_workflow} will query the latest workflow with the same
+      #   workflow ID even if it is unrelated to the started workflow.
+      def query(
+        query,
+        *args,
+        reject_condition: @client.options.default_workflow_query_reject_condition,
+        rpc_options: nil
+      )
+        @client._impl.query_workflow(Interceptor::QueryWorkflowInput.new(
+                                       workflow_id: id,
+                                       run_id:,
+                                       query:,
+                                       args:,
+                                       reject_condition:,
+                                       headers: {},
+                                       rpc_options:
+                                     ))
+      end
+
+      # Send an update request to the workflow and return a handle to it. This will target the workflow with {run_id} if
+      # present. To use a different run ID, create a new handle via {Client.workflow_handle}.
+      #
+      # @param update [Workflow::Definition::Update, Symbol, String] Update definition or name.
+      # @param args [Array<Object>] Update arguments.
+      # @param wait_for_stage [WorkflowUpdateWaitStage] Required stage to wait until returning. ADMITTED is not
+      #   currently supported. See https://docs.temporal.io/workflows#update for more details.
+      # @param id [String] ID of the update.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [WorkflowUpdateHandle] The update handle.
+      #
+      # @raise [Error::WorkflowUpdateRPCTimeoutOrCanceledError] This update call timed out or was canceled. This doesn't
+      #   mean the update itself was timed out or canceled.
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of {Client.start_workflow} will send updates the latest workflow with the same
+      #   workflow ID even if it is unrelated to the started workflow.
+      def start_update(
+        update,
+        *args,
+        wait_for_stage:,
+        id: SecureRandom.uuid,
+        rpc_options: nil
+      )
+        @client._impl.start_workflow_update(Interceptor::StartWorkflowUpdateInput.new(
+                                              workflow_id: self.id,
+                                              run_id:,
+                                              update_id: id,
+                                              update:,
+                                              args:,
+                                              wait_for_stage:,
+                                              headers: {},
+                                              rpc_options:
+                                            ))
+      end
+
+      # Send an update request to the workflow and wait for it to complete. This will target the workflow with {run_id}
+      # if present. To use a different run ID, create a new handle via {Client.workflow_handle}.
+      #
+      # @param update [Workflow::Definition::Update, Symbol, String] Update definition or name.
+      # @param args [Array<Object>] Update arguments.
+      # @param id [String] ID of the update.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @return [Object, nil] Update result.
+      #
+      # @raise [Error::WorkflowUpdateFailedError] If the update failed.
+      # @raise [Error::WorkflowUpdateRPCTimeoutOrCanceledError] This update call timed out or was canceled. This doesn't
+      #   mean the update itself was timed out or canceled.
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of {Client.start_workflow} will send updates the latest workflow with the same
+      #   workflow ID even if it is unrelated to the started workflow.
+      def execute_update(update, *args, id: SecureRandom.uuid, rpc_options: nil)
+        start_update(
+          update,
+          *args,
+          wait_for_stage: WorkflowUpdateWaitStage::COMPLETED,
+          id:,
+          rpc_options:
+        ).result
+      end
+
+      # Get a handle for an update. The handle can be used to wait on the update result.
+      #
+      # @param id [String] ID of the update.
+      # @param specific_run_id [String, nil] Workflow run ID to get update handle for. Default is the {run_id}. Most
+      #   users will not need to set this and instead use the one on the class.
+      #
+      # @return [WorkflowUpdateHandle] The update handle.
+      def update_handle(id, specific_run_id: run_id)
+        WorkflowUpdateHandle.new(
+          client: @client,
+          id:,
+          workflow_id: self.id,
+          workflow_run_id: specific_run_id,
+          known_outcome: nil
+        )
+      end
+
+      # Cancel the workflow. This will issue a cancellation for {run_id} if present. This call will make sure to use the
+      # run chain starting from {first_execution_run_id} if present. To create handles with these values, use
+      # {Client.workflow_handle}.
+      #
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of signal with start will cancel the latest workflow with the same workflow ID
+      #   even if it is unrelated to the started workflow.
+      def cancel(rpc_options: nil)
+        @client._impl.cancel_workflow(Interceptor::CancelWorkflowInput.new(
+                                        workflow_id: id,
+                                        run_id:,
+                                        first_execution_run_id:,
+                                        rpc_options:
+                                      ))
+      end
+
+      # Terminate the workflow. This will issue a termination for {run_id} if present. This call will make sure to use
+      # the run chain starting from {first_execution_run_id} if present. To create handles with these values, use
+      # {Client.workflow_handle}.
+      #
+      # @param reason [String, nil] Reason for the termination.
+      # @param details [Array<Object>] Details to store on the termination.
+      # @param rpc_options [RPCOptions, nil] Advanced RPC options.
+      #
+      # @raise [Error::RPCError] RPC error from call.
+      #
+      # @note Handles created as a result of signal with start will terminate the latest workflow with the same workflow
+      #   ID even if it is unrelated to the started workflow.
+      def terminate(reason = nil, details: [], rpc_options: nil)
+        @client._impl.terminate_workflow(Interceptor::TerminateWorkflowInput.new(
+                                           workflow_id: id,
+                                           run_id:,
+                                           first_execution_run_id:,
+                                           reason:,
+                                           details:,
+                                           rpc_options:
+                                         ))
+      end
+    end
+  end
+end

data/lib/temporalio/client/workflow_query_reject_condition.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require 'temporalio/api'
+
+module Temporalio
+  class Client
+    # Whether a query should be rejected in certain conditions.
+    module WorkflowQueryRejectCondition
+      NONE = Api::Enums::V1::QueryRejectCondition::QUERY_REJECT_CONDITION_NONE
+      NOT_OPEN = Api::Enums::V1::QueryRejectCondition::QUERY_REJECT_CONDITION_NOT_OPEN
+      NOT_COMPLETED_CLEANLY = Api::Enums::V1::QueryRejectCondition::QUERY_REJECT_CONDITION_NOT_COMPLETED_CLEANLY
+    end
+  end
+end