temporalio 0.4.0-aarch64-linux-musl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (183) hide show
  1. checksums.yaml +7 -0
  2. data/.yardopts +2 -0
  3. data/Gemfile +27 -0
  4. data/Rakefile +101 -0
  5. data/lib/temporalio/activity/complete_async_error.rb +11 -0
  6. data/lib/temporalio/activity/context.rb +123 -0
  7. data/lib/temporalio/activity/definition.rb +192 -0
  8. data/lib/temporalio/activity/info.rb +67 -0
  9. data/lib/temporalio/activity.rb +12 -0
  10. data/lib/temporalio/api/activity/v1/message.rb +25 -0
  11. data/lib/temporalio/api/batch/v1/message.rb +36 -0
  12. data/lib/temporalio/api/cloud/account/v1/message.rb +28 -0
  13. data/lib/temporalio/api/cloud/cloudservice/v1/request_response.rb +126 -0
  14. data/lib/temporalio/api/cloud/cloudservice/v1/service.rb +25 -0
  15. data/lib/temporalio/api/cloud/cloudservice.rb +3 -0
  16. data/lib/temporalio/api/cloud/identity/v1/message.rb +41 -0
  17. data/lib/temporalio/api/cloud/namespace/v1/message.rb +42 -0
  18. data/lib/temporalio/api/cloud/nexus/v1/message.rb +31 -0
  19. data/lib/temporalio/api/cloud/operation/v1/message.rb +28 -0
  20. data/lib/temporalio/api/cloud/region/v1/message.rb +24 -0
  21. data/lib/temporalio/api/cloud/resource/v1/message.rb +23 -0
  22. data/lib/temporalio/api/cloud/sink/v1/message.rb +24 -0
  23. data/lib/temporalio/api/cloud/usage/v1/message.rb +31 -0
  24. data/lib/temporalio/api/command/v1/message.rb +46 -0
  25. data/lib/temporalio/api/common/v1/grpc_status.rb +23 -0
  26. data/lib/temporalio/api/common/v1/message.rb +48 -0
  27. data/lib/temporalio/api/deployment/v1/message.rb +38 -0
  28. data/lib/temporalio/api/enums/v1/batch_operation.rb +22 -0
  29. data/lib/temporalio/api/enums/v1/command_type.rb +21 -0
  30. data/lib/temporalio/api/enums/v1/common.rb +26 -0
  31. data/lib/temporalio/api/enums/v1/deployment.rb +23 -0
  32. data/lib/temporalio/api/enums/v1/event_type.rb +21 -0
  33. data/lib/temporalio/api/enums/v1/failed_cause.rb +26 -0
  34. data/lib/temporalio/api/enums/v1/namespace.rb +23 -0
  35. data/lib/temporalio/api/enums/v1/nexus.rb +21 -0
  36. data/lib/temporalio/api/enums/v1/query.rb +22 -0
  37. data/lib/temporalio/api/enums/v1/reset.rb +23 -0
  38. data/lib/temporalio/api/enums/v1/schedule.rb +21 -0
  39. data/lib/temporalio/api/enums/v1/task_queue.rb +25 -0
  40. data/lib/temporalio/api/enums/v1/update.rb +22 -0
  41. data/lib/temporalio/api/enums/v1/workflow.rb +31 -0
  42. data/lib/temporalio/api/errordetails/v1/message.rb +44 -0
  43. data/lib/temporalio/api/export/v1/message.rb +24 -0
  44. data/lib/temporalio/api/failure/v1/message.rb +37 -0
  45. data/lib/temporalio/api/filter/v1/message.rb +27 -0
  46. data/lib/temporalio/api/history/v1/message.rb +92 -0
  47. data/lib/temporalio/api/namespace/v1/message.rb +31 -0
  48. data/lib/temporalio/api/nexus/v1/message.rb +41 -0
  49. data/lib/temporalio/api/operatorservice/v1/request_response.rb +49 -0
  50. data/lib/temporalio/api/operatorservice/v1/service.rb +23 -0
  51. data/lib/temporalio/api/operatorservice.rb +3 -0
  52. data/lib/temporalio/api/payload_visitor.rb +1581 -0
  53. data/lib/temporalio/api/protocol/v1/message.rb +23 -0
  54. data/lib/temporalio/api/query/v1/message.rb +28 -0
  55. data/lib/temporalio/api/replication/v1/message.rb +26 -0
  56. data/lib/temporalio/api/schedule/v1/message.rb +43 -0
  57. data/lib/temporalio/api/sdk/v1/enhanced_stack_trace.rb +25 -0
  58. data/lib/temporalio/api/sdk/v1/task_complete_metadata.rb +21 -0
  59. data/lib/temporalio/api/sdk/v1/user_metadata.rb +23 -0
  60. data/lib/temporalio/api/sdk/v1/workflow_metadata.rb +23 -0
  61. data/lib/temporalio/api/taskqueue/v1/message.rb +48 -0
  62. data/lib/temporalio/api/testservice/v1/request_response.rb +31 -0
  63. data/lib/temporalio/api/testservice/v1/service.rb +23 -0
  64. data/lib/temporalio/api/update/v1/message.rb +33 -0
  65. data/lib/temporalio/api/version/v1/message.rb +26 -0
  66. data/lib/temporalio/api/workflow/v1/message.rb +51 -0
  67. data/lib/temporalio/api/workflowservice/v1/request_response.rb +233 -0
  68. data/lib/temporalio/api/workflowservice/v1/service.rb +23 -0
  69. data/lib/temporalio/api/workflowservice.rb +3 -0
  70. data/lib/temporalio/api.rb +15 -0
  71. data/lib/temporalio/cancellation.rb +170 -0
  72. data/lib/temporalio/client/activity_id_reference.rb +32 -0
  73. data/lib/temporalio/client/async_activity_handle.rb +85 -0
  74. data/lib/temporalio/client/connection/cloud_service.rb +726 -0
  75. data/lib/temporalio/client/connection/operator_service.rb +201 -0
  76. data/lib/temporalio/client/connection/service.rb +42 -0
  77. data/lib/temporalio/client/connection/test_service.rb +111 -0
  78. data/lib/temporalio/client/connection/workflow_service.rb +1251 -0
  79. data/lib/temporalio/client/connection.rb +316 -0
  80. data/lib/temporalio/client/interceptor.rb +455 -0
  81. data/lib/temporalio/client/schedule.rb +991 -0
  82. data/lib/temporalio/client/schedule_handle.rb +126 -0
  83. data/lib/temporalio/client/with_start_workflow_operation.rb +115 -0
  84. data/lib/temporalio/client/workflow_execution.rb +119 -0
  85. data/lib/temporalio/client/workflow_execution_count.rb +36 -0
  86. data/lib/temporalio/client/workflow_execution_status.rb +18 -0
  87. data/lib/temporalio/client/workflow_handle.rb +389 -0
  88. data/lib/temporalio/client/workflow_query_reject_condition.rb +14 -0
  89. data/lib/temporalio/client/workflow_update_handle.rb +65 -0
  90. data/lib/temporalio/client/workflow_update_wait_stage.rb +17 -0
  91. data/lib/temporalio/client.rb +607 -0
  92. data/lib/temporalio/common_enums.rb +41 -0
  93. data/lib/temporalio/contrib/open_telemetry.rb +470 -0
  94. data/lib/temporalio/converters/data_converter.rb +99 -0
  95. data/lib/temporalio/converters/failure_converter.rb +202 -0
  96. data/lib/temporalio/converters/payload_codec.rb +26 -0
  97. data/lib/temporalio/converters/payload_converter/binary_null.rb +34 -0
  98. data/lib/temporalio/converters/payload_converter/binary_plain.rb +35 -0
  99. data/lib/temporalio/converters/payload_converter/binary_protobuf.rb +42 -0
  100. data/lib/temporalio/converters/payload_converter/composite.rb +66 -0
  101. data/lib/temporalio/converters/payload_converter/encoding.rb +35 -0
  102. data/lib/temporalio/converters/payload_converter/json_plain.rb +44 -0
  103. data/lib/temporalio/converters/payload_converter/json_protobuf.rb +41 -0
  104. data/lib/temporalio/converters/payload_converter.rb +71 -0
  105. data/lib/temporalio/converters/raw_value.rb +20 -0
  106. data/lib/temporalio/converters.rb +9 -0
  107. data/lib/temporalio/error/failure.rb +219 -0
  108. data/lib/temporalio/error.rb +156 -0
  109. data/lib/temporalio/internal/bridge/3.2/temporalio_bridge.so +0 -0
  110. data/lib/temporalio/internal/bridge/3.3/temporalio_bridge.so +0 -0
  111. data/lib/temporalio/internal/bridge/3.4/temporalio_bridge.so +0 -0
  112. data/lib/temporalio/internal/bridge/api/activity_result/activity_result.rb +34 -0
  113. data/lib/temporalio/internal/bridge/api/activity_task/activity_task.rb +31 -0
  114. data/lib/temporalio/internal/bridge/api/child_workflow/child_workflow.rb +33 -0
  115. data/lib/temporalio/internal/bridge/api/common/common.rb +27 -0
  116. data/lib/temporalio/internal/bridge/api/core_interface.rb +40 -0
  117. data/lib/temporalio/internal/bridge/api/external_data/external_data.rb +27 -0
  118. data/lib/temporalio/internal/bridge/api/nexus/nexus.rb +33 -0
  119. data/lib/temporalio/internal/bridge/api/workflow_activation/workflow_activation.rb +56 -0
  120. data/lib/temporalio/internal/bridge/api/workflow_commands/workflow_commands.rb +57 -0
  121. data/lib/temporalio/internal/bridge/api/workflow_completion/workflow_completion.rb +31 -0
  122. data/lib/temporalio/internal/bridge/api.rb +3 -0
  123. data/lib/temporalio/internal/bridge/client.rb +95 -0
  124. data/lib/temporalio/internal/bridge/runtime.rb +56 -0
  125. data/lib/temporalio/internal/bridge/testing.rb +69 -0
  126. data/lib/temporalio/internal/bridge/worker.rb +85 -0
  127. data/lib/temporalio/internal/bridge.rb +36 -0
  128. data/lib/temporalio/internal/client/implementation.rb +922 -0
  129. data/lib/temporalio/internal/metric.rb +122 -0
  130. data/lib/temporalio/internal/proto_utils.rb +165 -0
  131. data/lib/temporalio/internal/worker/activity_worker.rb +385 -0
  132. data/lib/temporalio/internal/worker/multi_runner.rb +213 -0
  133. data/lib/temporalio/internal/worker/workflow_instance/child_workflow_handle.rb +54 -0
  134. data/lib/temporalio/internal/worker/workflow_instance/context.rb +383 -0
  135. data/lib/temporalio/internal/worker/workflow_instance/details.rb +46 -0
  136. data/lib/temporalio/internal/worker/workflow_instance/external_workflow_handle.rb +32 -0
  137. data/lib/temporalio/internal/worker/workflow_instance/externally_immutable_hash.rb +22 -0
  138. data/lib/temporalio/internal/worker/workflow_instance/handler_execution.rb +25 -0
  139. data/lib/temporalio/internal/worker/workflow_instance/handler_hash.rb +41 -0
  140. data/lib/temporalio/internal/worker/workflow_instance/illegal_call_tracer.rb +97 -0
  141. data/lib/temporalio/internal/worker/workflow_instance/inbound_implementation.rb +62 -0
  142. data/lib/temporalio/internal/worker/workflow_instance/outbound_implementation.rb +400 -0
  143. data/lib/temporalio/internal/worker/workflow_instance/replay_safe_logger.rb +37 -0
  144. data/lib/temporalio/internal/worker/workflow_instance/replay_safe_metric.rb +40 -0
  145. data/lib/temporalio/internal/worker/workflow_instance/scheduler.rb +183 -0
  146. data/lib/temporalio/internal/worker/workflow_instance.rb +774 -0
  147. data/lib/temporalio/internal/worker/workflow_worker.rb +239 -0
  148. data/lib/temporalio/internal.rb +7 -0
  149. data/lib/temporalio/metric.rb +109 -0
  150. data/lib/temporalio/retry_policy.rb +74 -0
  151. data/lib/temporalio/runtime/metric_buffer.rb +94 -0
  152. data/lib/temporalio/runtime.rb +352 -0
  153. data/lib/temporalio/scoped_logger.rb +96 -0
  154. data/lib/temporalio/search_attributes.rb +356 -0
  155. data/lib/temporalio/testing/activity_environment.rb +160 -0
  156. data/lib/temporalio/testing/workflow_environment.rb +406 -0
  157. data/lib/temporalio/testing.rb +10 -0
  158. data/lib/temporalio/version.rb +5 -0
  159. data/lib/temporalio/worker/activity_executor/fiber.rb +49 -0
  160. data/lib/temporalio/worker/activity_executor/thread_pool.rb +46 -0
  161. data/lib/temporalio/worker/activity_executor.rb +55 -0
  162. data/lib/temporalio/worker/interceptor.rb +365 -0
  163. data/lib/temporalio/worker/thread_pool.rb +237 -0
  164. data/lib/temporalio/worker/tuner.rb +189 -0
  165. data/lib/temporalio/worker/workflow_executor/thread_pool.rb +235 -0
  166. data/lib/temporalio/worker/workflow_executor.rb +26 -0
  167. data/lib/temporalio/worker/workflow_replayer.rb +350 -0
  168. data/lib/temporalio/worker.rb +603 -0
  169. data/lib/temporalio/workflow/activity_cancellation_type.rb +20 -0
  170. data/lib/temporalio/workflow/child_workflow_cancellation_type.rb +21 -0
  171. data/lib/temporalio/workflow/child_workflow_handle.rb +43 -0
  172. data/lib/temporalio/workflow/definition.rb +598 -0
  173. data/lib/temporalio/workflow/external_workflow_handle.rb +41 -0
  174. data/lib/temporalio/workflow/future.rb +151 -0
  175. data/lib/temporalio/workflow/handler_unfinished_policy.rb +13 -0
  176. data/lib/temporalio/workflow/info.rb +104 -0
  177. data/lib/temporalio/workflow/parent_close_policy.rb +19 -0
  178. data/lib/temporalio/workflow/update_info.rb +20 -0
  179. data/lib/temporalio/workflow.rb +575 -0
  180. data/lib/temporalio/workflow_history.rb +47 -0
  181. data/lib/temporalio.rb +11 -0
  182. data/temporalio.gemspec +29 -0
  183. metadata +258 -0
@@ -0,0 +1,32 @@
1
+ # frozen_string_literal: true
2
+
3
+ require 'temporalio/cancellation'
4
+ require 'temporalio/workflow'
5
+ require 'temporalio/workflow/external_workflow_handle'
6
+
7
+ module Temporalio
8
+ module Internal
9
+ module Worker
10
+ class WorkflowInstance
11
+ # Implementation of the external workflow handle.
12
+ class ExternalWorkflowHandle < Workflow::ExternalWorkflowHandle
13
+ attr_reader :id, :run_id
14
+
15
+ def initialize(id:, run_id:, instance:) # rubocop:disable Lint/MissingSuper
16
+ @id = id
17
+ @run_id = run_id
18
+ @instance = instance
19
+ end
20
+
21
+ def signal(signal, *args, cancellation: Workflow.cancellation)
22
+ @instance.context._signal_external_workflow(id:, run_id:, signal:, args:, cancellation:)
23
+ end
24
+
25
+ def cancel
26
+ @instance.context._cancel_external_workflow(id:, run_id:)
27
+ end
28
+ end
29
+ end
30
+ end
31
+ end
32
+ end
@@ -0,0 +1,22 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Temporalio
4
+ module Internal
5
+ module Worker
6
+ class WorkflowInstance
7
+ # Delegator to a hash that does not allow external mutations. Used for memo.
8
+ class ExternallyImmutableHash < SimpleDelegator
9
+ def initialize(initial_hash)
10
+ super(initial_hash.freeze)
11
+ end
12
+
13
+ def _update(&)
14
+ new_hash = __getobj__.dup
15
+ yield new_hash
16
+ __setobj__(new_hash.freeze)
17
+ end
18
+ end
19
+ end
20
+ end
21
+ end
22
+ end
@@ -0,0 +1,25 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Temporalio
4
+ module Internal
5
+ module Worker
6
+ class WorkflowInstance
7
+ # Representation of a currently-executing handler. Used to track whether any handlers are still running and warn
8
+ # on workflow complete as needed.
9
+ class HandlerExecution
10
+ attr_reader :name, :update_id, :unfinished_policy
11
+
12
+ def initialize(
13
+ name:,
14
+ update_id:,
15
+ unfinished_policy:
16
+ )
17
+ @name = name
18
+ @update_id = update_id
19
+ @unfinished_policy = unfinished_policy
20
+ end
21
+ end
22
+ end
23
+ end
24
+ end
25
+ end
@@ -0,0 +1,41 @@
1
+ # frozen_string_literal: true
2
+
3
+ module Temporalio
4
+ module Internal
5
+ module Worker
6
+ class WorkflowInstance
7
+ # Hash for handlers that notifies when one is added. Only `[]=` and `store` can be used to mutate it.
8
+ class HandlerHash < SimpleDelegator
9
+ def initialize(initial_frozen_hash, definition_class, &on_new_definition)
10
+ super(initial_frozen_hash)
11
+ @definition_class = definition_class
12
+ @on_new_definition = on_new_definition
13
+ end
14
+
15
+ def []=(name, definition)
16
+ store(name, definition)
17
+ end
18
+
19
+ # steep:ignore:start
20
+ def store(name, definition)
21
+ raise ArgumentError, 'Name must be a string or nil' unless name.nil? || name.is_a?(String)
22
+
23
+ unless definition.nil? || definition.is_a?(@definition_class)
24
+ raise ArgumentError,
25
+ "Value must be a #{@definition_class.name} or nil"
26
+ end
27
+ raise ArgumentError, 'Name does not match one in definition' if definition && name != definition.name
28
+
29
+ # Do a copy-on-write op on the underlying frozen hash
30
+ new_hash = __getobj__.dup
31
+ new_hash[name] = definition
32
+ __setobj__(new_hash.freeze)
33
+ @on_new_definition&.call(definition) unless definition.nil?
34
+ definition
35
+ end
36
+ # steep:ignore:end
37
+ end
38
+ end
39
+ end
40
+ end
41
+ end
@@ -0,0 +1,97 @@
1
+ # frozen_string_literal: true
2
+
3
+ require 'temporalio/workflow'
4
+
5
+ module Temporalio
6
+ module Internal
7
+ module Worker
8
+ class WorkflowInstance
9
+ # Class that installs {::TracePoint} to disallow illegal calls.
10
+ class IllegalCallTracer
11
+ def self.frozen_validated_illegal_calls(illegal_calls)
12
+ illegal_calls.to_h do |key, val|
13
+ raise TypeError, 'Invalid illegal call map, top-level key must be a String' unless key.is_a?(String)
14
+
15
+ # @type var fixed_val: :all | Hash[Symbol, bool]
16
+ fixed_val = case val
17
+ when Array
18
+ val.to_h do |sub_val|
19
+ unless sub_val.is_a?(Symbol)
20
+ raise TypeError,
21
+ 'Invalid illegal call map, each value must be a Symbol'
22
+ end
23
+
24
+ [sub_val, true]
25
+ end.freeze
26
+ when :all
27
+ :all
28
+ else
29
+ raise TypeError, 'Invalid illegal call map, top-level value must be an Array or :all'
30
+ end
31
+
32
+ [key.frozen? ? key : key.dup.freeze, fixed_val]
33
+ end.freeze
34
+ end
35
+
36
+ # Illegal calls are Hash[String, :all | Hash[Symbol, Bool]]
37
+ def initialize(illegal_calls)
38
+ @tracepoint = TracePoint.new(:call, :c_call) do |tp|
39
+ # Manual check for proper thread since we have seen issues in Ruby 3.2 where it leaks
40
+ next unless Thread.current == @enabled_thread
41
+
42
+ cls = tp.defined_class
43
+ next unless cls.is_a?(Module)
44
+
45
+ # Extract the class name from the defined class. This is more difficult than it seems because you have to
46
+ # resolve the attached object of the singleton class. But in older Ruby (at least <= 3.1), the singleton
47
+ # class of things like `Date` does not have `attached_object` so you have to fall back in these rare cases
48
+ # to parsing the string output. Reaching the string parsing component is rare, so this should not have
49
+ # significant performance impact.
50
+ cls_name = if cls.singleton_class?
51
+ if cls.respond_to?(:attached_object)
52
+ cls = cls.attached_object # steep:ignore
53
+ next unless cls.is_a?(Module)
54
+
55
+ cls.name.to_s
56
+ else
57
+ cls.to_s.delete_prefix('#<Class:').delete_suffix('>')
58
+ end
59
+ else
60
+ cls.name.to_s
61
+ end
62
+
63
+ # Check if the call is considered illegal
64
+ vals = illegal_calls[cls_name]
65
+ if vals == :all || vals&.[](tp.callee_id) # steep:ignore
66
+ raise Workflow::NondeterminismError,
67
+ "Cannot access #{cls_name} #{tp.callee_id} from inside a " \
68
+ 'workflow. If this is known to be safe, the code can be run in ' \
69
+ 'a Temporalio::Workflow::Unsafe.illegal_call_tracing_disabled block.'
70
+ end
71
+ end
72
+ end
73
+
74
+ def enable(&block)
75
+ # We've seen leaking issues in Ruby 3.2 where the TracePoint inadvertently remains enabled even for threads
76
+ # that it was not started on. So we will check the thread ourselves.
77
+ @enabled_thread = Thread.current
78
+ @tracepoint.enable do
79
+ block.call
80
+ ensure
81
+ @enabled_thread = nil
82
+ end
83
+ end
84
+
85
+ def disable(&block)
86
+ previous_thread = @enabled_thread
87
+ @tracepoint.disable do
88
+ block.call
89
+ ensure
90
+ @enabled_thread = previous_thread
91
+ end
92
+ end
93
+ end
94
+ end
95
+ end
96
+ end
97
+ end
@@ -0,0 +1,62 @@
1
+ # frozen_string_literal: true
2
+
3
+ require 'temporalio/internal/worker/workflow_instance'
4
+ require 'temporalio/worker/interceptor'
5
+ require 'temporalio/workflow'
6
+
7
+ module Temporalio
8
+ module Internal
9
+ module Worker
10
+ class WorkflowInstance
11
+ # Root implementation of the inbound interceptor.
12
+ class InboundImplementation < Temporalio::Worker::Interceptor::Workflow::Inbound
13
+ def initialize(instance)
14
+ super(nil) # steep:ignore
15
+ @instance = instance
16
+ end
17
+
18
+ def init(outbound)
19
+ @instance.context._outbound = outbound
20
+ end
21
+
22
+ def execute(input)
23
+ @instance.instance.execute(*input.args)
24
+ end
25
+
26
+ def handle_signal(input)
27
+ invoke_handler(input.signal, input)
28
+ end
29
+
30
+ def handle_query(input)
31
+ invoke_handler(input.query, input)
32
+ end
33
+
34
+ def validate_update(input)
35
+ invoke_handler(input.update, input, to_invoke: input.definition.validator_to_invoke)
36
+ end
37
+
38
+ def handle_update(input)
39
+ invoke_handler(input.update, input)
40
+ end
41
+
42
+ private
43
+
44
+ def invoke_handler(name, input, to_invoke: input.definition.to_invoke)
45
+ args = input.args
46
+ # Add name as first param if dynamic
47
+ args = [name] + args if input.definition.name.nil?
48
+ # Assume symbol or proc
49
+ case to_invoke
50
+ when Symbol
51
+ @instance.instance.send(to_invoke, *args)
52
+ when Proc
53
+ to_invoke.call(*args)
54
+ else
55
+ raise "Unrecognized invocation type #{to_invoke.class}"
56
+ end
57
+ end
58
+ end
59
+ end
60
+ end
61
+ end
62
+ end
@@ -0,0 +1,400 @@
1
+ # frozen_string_literal: true
2
+
3
+ require 'temporalio/activity/definition'
4
+ require 'temporalio/cancellation'
5
+ require 'temporalio/error'
6
+ require 'temporalio/internal/bridge/api'
7
+ require 'temporalio/internal/proto_utils'
8
+ require 'temporalio/internal/worker/workflow_instance'
9
+ require 'temporalio/worker/interceptor'
10
+ require 'temporalio/workflow'
11
+ require 'temporalio/workflow/child_workflow_handle'
12
+
13
+ module Temporalio
14
+ module Internal
15
+ module Worker
16
+ class WorkflowInstance
17
+ # Root implementation of the outbound interceptor.
18
+ class OutboundImplementation < Temporalio::Worker::Interceptor::Workflow::Outbound
19
+ def initialize(instance)
20
+ super(nil) # steep:ignore
21
+ @instance = instance
22
+ @activity_counter = 0
23
+ @timer_counter = 0
24
+ @child_counter = 0
25
+ @external_signal_counter = 0
26
+ @external_cancel_counter = 0
27
+ end
28
+
29
+ def cancel_external_workflow(input)
30
+ # Add command
31
+ seq = (@external_cancel_counter += 1)
32
+ cmd = Bridge::Api::WorkflowCommands::RequestCancelExternalWorkflowExecution.new(
33
+ seq:,
34
+ workflow_execution: Bridge::Api::Common::NamespacedWorkflowExecution.new(
35
+ namespace: @instance.info.namespace,
36
+ workflow_id: input.id,
37
+ run_id: input.run_id
38
+ )
39
+ )
40
+ @instance.add_command(
41
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(request_cancel_external_workflow_execution: cmd)
42
+ )
43
+ @instance.pending_external_cancels[seq] = Fiber.current
44
+
45
+ # Wait
46
+ resolution = Fiber.yield
47
+
48
+ # Raise if resolution has failure
49
+ return unless resolution.failure
50
+
51
+ raise @instance.failure_converter.from_failure(resolution.failure, @instance.payload_converter)
52
+ end
53
+
54
+ def execute_activity(input)
55
+ if input.schedule_to_close_timeout.nil? && input.start_to_close_timeout.nil?
56
+ raise ArgumentError, 'Activity must have schedule_to_close_timeout or start_to_close_timeout'
57
+ end
58
+
59
+ execute_activity_with_local_backoffs(local: false, cancellation: input.cancellation) do
60
+ seq = (@activity_counter += 1)
61
+ @instance.add_command(
62
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
63
+ schedule_activity: Bridge::Api::WorkflowCommands::ScheduleActivity.new(
64
+ seq:,
65
+ activity_id: input.activity_id || seq.to_s,
66
+ activity_type: input.activity,
67
+ task_queue: input.task_queue,
68
+ headers: ProtoUtils.headers_to_proto_hash(input.headers, @instance.payload_converter),
69
+ arguments: ProtoUtils.convert_to_payload_array(@instance.payload_converter, input.args),
70
+ schedule_to_close_timeout: ProtoUtils.seconds_to_duration(input.schedule_to_close_timeout),
71
+ schedule_to_start_timeout: ProtoUtils.seconds_to_duration(input.schedule_to_start_timeout),
72
+ start_to_close_timeout: ProtoUtils.seconds_to_duration(input.start_to_close_timeout),
73
+ heartbeat_timeout: ProtoUtils.seconds_to_duration(input.heartbeat_timeout),
74
+ retry_policy: input.retry_policy&._to_proto,
75
+ cancellation_type: input.cancellation_type,
76
+ do_not_eagerly_execute: input.disable_eager_execution
77
+ ),
78
+ user_metadata: ProtoUtils.to_user_metadata(input.summary, nil, @instance.payload_converter)
79
+ )
80
+ )
81
+ seq
82
+ end
83
+ end
84
+
85
+ def execute_local_activity(input)
86
+ if input.schedule_to_close_timeout.nil? && input.start_to_close_timeout.nil?
87
+ raise ArgumentError, 'Activity must have schedule_to_close_timeout or start_to_close_timeout'
88
+ end
89
+
90
+ execute_activity_with_local_backoffs(local: true, cancellation: input.cancellation) do |do_backoff|
91
+ seq = (@activity_counter += 1)
92
+ @instance.add_command(
93
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
94
+ schedule_local_activity: Bridge::Api::WorkflowCommands::ScheduleLocalActivity.new(
95
+ seq:,
96
+ activity_id: input.activity_id || seq.to_s,
97
+ activity_type: input.activity,
98
+ headers: ProtoUtils.headers_to_proto_hash(input.headers, @instance.payload_converter),
99
+ arguments: ProtoUtils.convert_to_payload_array(@instance.payload_converter, input.args),
100
+ schedule_to_close_timeout: ProtoUtils.seconds_to_duration(input.schedule_to_close_timeout),
101
+ schedule_to_start_timeout: ProtoUtils.seconds_to_duration(input.schedule_to_start_timeout),
102
+ start_to_close_timeout: ProtoUtils.seconds_to_duration(input.start_to_close_timeout),
103
+ retry_policy: input.retry_policy&._to_proto,
104
+ cancellation_type: input.cancellation_type,
105
+ local_retry_threshold: ProtoUtils.seconds_to_duration(input.local_retry_threshold),
106
+ attempt: do_backoff&.attempt || 0,
107
+ original_schedule_time: do_backoff&.original_schedule_time
108
+ )
109
+ )
110
+ )
111
+ seq
112
+ end
113
+ end
114
+
115
+ def execute_activity_with_local_backoffs(local:, cancellation:, &)
116
+ # We do not even want to schedule if the cancellation is already cancelled. We choose to use canceled
117
+ # failure instead of wrapping in activity failure which is similar to what other SDKs do, with the accepted
118
+ # tradeoff that it makes rescue more difficult (hence the presence of Error.canceled? helper).
119
+ raise Error::CanceledError, 'Activity canceled before scheduled' if cancellation.canceled?
120
+
121
+ # This has to be done in a loop for local activity backoff
122
+ last_local_backoff = nil
123
+ loop do
124
+ result = execute_activity_once(local:, cancellation:, last_local_backoff:, &)
125
+ return result unless result.is_a?(Bridge::Api::ActivityResult::DoBackoff)
126
+
127
+ # @type var result: untyped
128
+ last_local_backoff = result
129
+ # Have to sleep the amount of the backoff, which can be canceled with the same cancellation
130
+ # TODO(cretz): What should this cancellation raise?
131
+ Workflow.sleep(ProtoUtils.duration_to_seconds(result.backoff_duration), cancellation:)
132
+ end
133
+ end
134
+
135
+ # If this doesn't raise, it returns success | DoBackoff
136
+ def execute_activity_once(local:, cancellation:, last_local_backoff:, &)
137
+ # Add to pending activities (removed by the resolver)
138
+ seq = yield last_local_backoff
139
+ @instance.pending_activities[seq] = Fiber.current
140
+
141
+ # Add cancellation hook
142
+ cancel_callback_key = cancellation.add_cancel_callback do
143
+ # Only if the activity is present still
144
+ if @instance.pending_activities.include?(seq)
145
+ if local
146
+ @instance.add_command(
147
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
148
+ request_cancel_local_activity: Bridge::Api::WorkflowCommands::RequestCancelLocalActivity.new(seq:)
149
+ )
150
+ )
151
+ else
152
+ @instance.add_command(
153
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
154
+ request_cancel_activity: Bridge::Api::WorkflowCommands::RequestCancelActivity.new(seq:)
155
+ )
156
+ )
157
+ end
158
+ end
159
+ end
160
+
161
+ # Wait
162
+ resolution = Fiber.yield
163
+
164
+ # Remove cancellation callback
165
+ cancellation.remove_cancel_callback(cancel_callback_key)
166
+
167
+ case resolution.status
168
+ when :completed
169
+ @instance.payload_converter.from_payload(resolution.completed.result)
170
+ when :failed
171
+ raise @instance.failure_converter.from_failure(resolution.failed.failure, @instance.payload_converter)
172
+ when :cancelled
173
+ raise @instance.failure_converter.from_failure(resolution.cancelled.failure, @instance.payload_converter)
174
+ when :backoff
175
+ resolution.backoff
176
+ else
177
+ raise "Unrecognized resolution status: #{resolution.status}"
178
+ end
179
+ end
180
+
181
+ def initialize_continue_as_new_error(input)
182
+ # Do nothing
183
+ end
184
+
185
+ def signal_child_workflow(input)
186
+ _signal_external_workflow(
187
+ id: input.id,
188
+ run_id: nil,
189
+ child: true,
190
+ signal: input.signal,
191
+ args: input.args,
192
+ cancellation: input.cancellation,
193
+ headers: input.headers
194
+ )
195
+ end
196
+
197
+ def signal_external_workflow(input)
198
+ _signal_external_workflow(
199
+ id: input.id,
200
+ run_id: input.run_id,
201
+ child: false,
202
+ signal: input.signal,
203
+ args: input.args,
204
+ cancellation: input.cancellation,
205
+ headers: input.headers
206
+ )
207
+ end
208
+
209
+ def _signal_external_workflow(id:, run_id:, child:, signal:, args:, cancellation:, headers:)
210
+ raise Error::CanceledError, 'Signal canceled before scheduled' if cancellation.canceled?
211
+
212
+ # Add command
213
+ seq = (@external_signal_counter += 1)
214
+ cmd = Bridge::Api::WorkflowCommands::SignalExternalWorkflowExecution.new(
215
+ seq:,
216
+ signal_name: signal,
217
+ args: ProtoUtils.convert_to_payload_array(@instance.payload_converter, args),
218
+ headers: ProtoUtils.headers_to_proto_hash(headers, @instance.payload_converter)
219
+ )
220
+ if child
221
+ cmd.child_workflow_id = id
222
+ else
223
+ cmd.workflow_execution = Bridge::Api::Common::NamespacedWorkflowExecution.new(
224
+ namespace: @instance.info.namespace,
225
+ workflow_id: id,
226
+ run_id:
227
+ )
228
+ end
229
+ @instance.add_command(
230
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(signal_external_workflow_execution: cmd)
231
+ )
232
+ @instance.pending_external_signals[seq] = Fiber.current
233
+
234
+ # Add a cancellation callback
235
+ cancel_callback_key = cancellation.add_cancel_callback do
236
+ # Add the command but do not raise, we will let resolution do that
237
+ @instance.add_command(
238
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
239
+ cancel_signal_workflow: Bridge::Api::WorkflowCommands::CancelSignalWorkflow.new(seq:)
240
+ )
241
+ )
242
+ end
243
+
244
+ # Wait
245
+ resolution = Fiber.yield
246
+
247
+ # Remove cancellation callback
248
+ cancellation.remove_cancel_callback(cancel_callback_key)
249
+
250
+ # Raise if resolution has failure
251
+ return unless resolution.failure
252
+
253
+ raise @instance.failure_converter.from_failure(resolution.failure, @instance.payload_converter)
254
+ end
255
+
256
+ def sleep(input)
257
+ # If already cancelled, raise as such
258
+ if input.cancellation.canceled?
259
+ raise Error::CanceledError,
260
+ input.cancellation.canceled_reason || 'Timer canceled before started'
261
+ end
262
+
263
+ # Disallow negative durations
264
+ raise ArgumentError, 'Sleep duration cannot be less than 0' if input.duration&.negative?
265
+
266
+ # If the duration is infinite, just wait for cancellation
267
+ if input.duration.nil?
268
+ input.cancellation.wait
269
+ raise Error::CanceledError, input.cancellation.canceled_reason || 'Timer canceled'
270
+ end
271
+
272
+ # If duration is zero, we make it one millisecond. It was decided a 0 duration still makes a timer to ensure
273
+ # determinism if a timer's duration is altered from non-zero to zero or vice versa.
274
+ duration = input.duration
275
+ duration = 0.001 if duration.zero?
276
+
277
+ # Add command
278
+ seq = (@timer_counter += 1)
279
+ @instance.add_command(
280
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
281
+ start_timer: Bridge::Api::WorkflowCommands::StartTimer.new(
282
+ seq:,
283
+ start_to_fire_timeout: ProtoUtils.seconds_to_duration(duration)
284
+ ),
285
+ user_metadata: ProtoUtils.to_user_metadata(input.summary, nil, @instance.payload_converter)
286
+ )
287
+ )
288
+ @instance.pending_timers[seq] = Fiber.current
289
+
290
+ # Add a cancellation callback
291
+ cancel_callback_key = input.cancellation.add_cancel_callback do
292
+ # Only if the timer is still present
293
+ fiber = @instance.pending_timers.delete(seq)
294
+ if fiber
295
+ # Add the command for cancel then raise
296
+ @instance.add_command(
297
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
298
+ cancel_timer: Bridge::Api::WorkflowCommands::CancelTimer.new(seq:)
299
+ )
300
+ )
301
+ if fiber.alive?
302
+ fiber.raise(Error::CanceledError.new(input.cancellation.canceled_reason || 'Timer canceled'))
303
+ end
304
+ end
305
+ end
306
+
307
+ # Wait
308
+ Fiber.yield
309
+
310
+ # Remove cancellation callback (only needed on success)
311
+ input.cancellation.remove_cancel_callback(cancel_callback_key)
312
+ end
313
+
314
+ def start_child_workflow(input)
315
+ raise Error::CanceledError, 'Child canceled before scheduled' if input.cancellation.canceled?
316
+
317
+ # Add the command
318
+ seq = (@child_counter += 1)
319
+ @instance.add_command(
320
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
321
+ start_child_workflow_execution: Bridge::Api::WorkflowCommands::StartChildWorkflowExecution.new(
322
+ seq:,
323
+ namespace: @instance.info.namespace,
324
+ workflow_id: input.id,
325
+ workflow_type: input.workflow,
326
+ task_queue: input.task_queue,
327
+ input: ProtoUtils.convert_to_payload_array(@instance.payload_converter, input.args),
328
+ workflow_execution_timeout: ProtoUtils.seconds_to_duration(input.execution_timeout),
329
+ workflow_run_timeout: ProtoUtils.seconds_to_duration(input.run_timeout),
330
+ workflow_task_timeout: ProtoUtils.seconds_to_duration(input.task_timeout),
331
+ parent_close_policy: input.parent_close_policy,
332
+ workflow_id_reuse_policy: input.id_reuse_policy,
333
+ retry_policy: input.retry_policy&._to_proto,
334
+ cron_schedule: input.cron_schedule,
335
+ headers: ProtoUtils.headers_to_proto_hash(input.headers, @instance.payload_converter),
336
+ memo: ProtoUtils.memo_to_proto_hash(input.memo, @instance.payload_converter),
337
+ search_attributes: input.search_attributes&._to_proto_hash,
338
+ cancellation_type: input.cancellation_type
339
+ ),
340
+ user_metadata: ProtoUtils.to_user_metadata(
341
+ input.static_summary, input.static_details, @instance.payload_converter
342
+ )
343
+ )
344
+ )
345
+
346
+ # Set as pending start and register cancel callback
347
+ @instance.pending_child_workflow_starts[seq] = Fiber.current
348
+ cancel_callback_key = input.cancellation.add_cancel_callback do
349
+ # Send cancel if in start or pending
350
+ if @instance.pending_child_workflow_starts.include?(seq) ||
351
+ @instance.pending_child_workflows.include?(seq)
352
+ @instance.add_command(
353
+ Bridge::Api::WorkflowCommands::WorkflowCommand.new(
354
+ cancel_child_workflow_execution: Bridge::Api::WorkflowCommands::CancelChildWorkflowExecution.new(
355
+ child_workflow_seq: seq
356
+ )
357
+ )
358
+ )
359
+ end
360
+ end
361
+
362
+ # Wait for start
363
+ resolution = Fiber.yield
364
+
365
+ case resolution.status
366
+ when :succeeded
367
+ # Create handle, passing along the cancel callback key, and set it as pending
368
+ handle = ChildWorkflowHandle.new(
369
+ id: input.id,
370
+ first_execution_run_id: resolution.succeeded.run_id,
371
+ instance: @instance,
372
+ cancellation: input.cancellation,
373
+ cancel_callback_key:
374
+ )
375
+ @instance.pending_child_workflows[seq] = handle
376
+ handle
377
+ when :failed
378
+ # Remove cancel callback and handle failure
379
+ input.cancellation.remove_cancel_callback(cancel_callback_key)
380
+ if resolution.failed.cause == :START_CHILD_WORKFLOW_EXECUTION_FAILED_CAUSE_WORKFLOW_ALREADY_EXISTS
381
+ raise Error::WorkflowAlreadyStartedError.new(
382
+ workflow_id: resolution.failed.workflow_id,
383
+ workflow_type: resolution.failed.workflow_type,
384
+ run_id: nil
385
+ )
386
+ end
387
+ raise "Unknown child start fail cause: #{resolution.failed.cause}"
388
+ when :cancelled
389
+ # Remove cancel callback and handle cancel
390
+ input.cancellation.remove_cancel_callback(cancel_callback_key)
391
+ raise @instance.failure_converter.from_failure(resolution.cancelled.failure, @instance.payload_converter)
392
+ else
393
+ raise "Unknown resolution status: #{resolution.status}"
394
+ end
395
+ end
396
+ end
397
+ end
398
+ end
399
+ end
400
+ end