temporalio 1.10.1 → 1.11.0

data/lib/temporalio/contrib/aws/s3_driver/s3_storage_driver.rb

@@ -0,0 +1,211 @@
+# frozen_string_literal: true
+
+require 'digest'
+require 'temporalio/api'
+require 'temporalio/cancellation'
+require 'temporalio/contrib/aws/s3_driver/s3_storage_driver_client'
+require 'temporalio/converters/storage_driver'
+require 'temporalio/converters/storage_driver_activity_info'
+require 'temporalio/converters/storage_driver_claim'
+require 'temporalio/converters/storage_driver_workflow_info'
+
+module Temporalio
+  module Contrib
+    module Aws
+      module S3Driver
+        # Storage driver that stores and retrieves payloads in Amazon S3.
+        #
+        # Payloads are content addressed by a SHA-256 hash of their serialized bytes, so storing the same payload twice
+        # for the same execution uploads it once. Object keys are built from the namespace and the workflow or activity
+        # the payload is stored for, percent-encoded to S3's safe character set.
+        #
+        # @note WARNING: This API is experimental and may change in the future.
+        #
+        # @!visibility private
+        class S3StorageDriver < Converters::StorageDriver
+          # Driver type shared by every S3 driver instance across SDK languages.
+          DRIVER_TYPE = 'aws.s3driver'
+
+          # Default maximum serialized payload size, in bytes.
+          DEFAULT_MAX_PAYLOAD_SIZE = 50 * 1024 * 1024
+
+          CLAIM_BUCKET = 'bucket'
+          CLAIM_KEY = 'key'
+          CLAIM_HASH_ALGORITHM = 'hash_algorithm'
+          CLAIM_HASH_VALUE = 'hash_value'
+          NULL_SEGMENT = 'null'
+          private_constant :CLAIM_BUCKET, :CLAIM_KEY, :CLAIM_HASH_ALGORITHM, :CLAIM_HASH_VALUE, :NULL_SEGMENT
+
+          StoredObject = Data.define(:payload, :bucket, :key, :hash_value, :data)
+          private_constant :StoredObject
+
+          # @return [String] Driver instance name.
+          attr_reader :name
+
+          # Create an S3 storage driver.
+          #
+          # @param client [S3StorageDriverClient] Client that performs the S3 requests, for example an
+          #   {AwsSdkS3StorageDriverClient}.
+          # @param bucket [String, Proc] Bucket name, or a proc called with the {Converters::StorageDriverStoreContext}
+          #   and the payload that returns the bucket name for that payload.
+          # @param driver_name [String] Driver instance name written into history. Only change this when registering
+          #   multiple S3 drivers with different configurations on the same external storage.
+          # @param max_payload_size [Integer] Maximum serialized payload size, in bytes. Storing a larger payload
+          #   raises.
+          def initialize(client:, bucket:, driver_name: DRIVER_TYPE, max_payload_size: DEFAULT_MAX_PAYLOAD_SIZE)
+            super()
+            unless max_payload_size.is_a?(Integer) && max_payload_size.positive?
+              raise ArgumentError, "max_payload_size must be a positive integer, got #{max_payload_size.inspect}"
+            end
+
+            @client = client
+            @bucket = bucket.is_a?(String) ? proc { bucket } : bucket
+            @name = driver_name
+            @max_payload_size = max_payload_size
+          end
+
+          # @return [String] Always {DRIVER_TYPE}.
+          def type
+            DRIVER_TYPE
+          end
+
+          # (see Converters::StorageDriver#store)
+          def store(context, payloads)
+            context_segments = context_key_segments(context.target)
+            objects = payloads.map { |payload| stored_object(context, payload, context_segments) }
+            uploads = objects.uniq { |object| [object.bucket, object.key] }
+            run_concurrently(uploads, context.cancellation) do |object, cancellation|
+              context.limiter.permit(object.payload) do
+                cancellation.check!
+                upload_if_absent(object)
+              end
+            end
+            objects.map do |object|
+              Converters::StorageDriverClaim.new(
+                claim_data: {
+                  CLAIM_BUCKET => object.bucket,
+                  CLAIM_KEY => object.key,
+                  CLAIM_HASH_ALGORITHM => 'sha256',
+                  CLAIM_HASH_VALUE => object.hash_value
+                }
+              )
+            end
+          end
+
+          # (see Converters::StorageDriver#retrieve)
+          def retrieve(context, claims)
+            claims.each { |claim| validate_claim(claim) }
+            run_concurrently(claims.to_a, context.cancellation) do |claim, cancellation|
+              bucket = claim.claim_data.fetch(CLAIM_BUCKET)
+              key = claim.claim_data.fetch(CLAIM_KEY)
+              expected_hash = claim.claim_data.fetch(CLAIM_HASH_VALUE)
+              data = context.limiter.permit(claim) do
+                cancellation.check!
+                download(bucket, key)
+              end
+              actual_hash = Digest::SHA256.hexdigest(data)
+              if actual_hash != expected_hash
+                raise "S3StorageDriver integrity check failed [bucket=#{bucket}, key=#{key}]: " \
+                      "expected sha256:#{expected_hash}, got sha256:#{actual_hash}"
+              end
+              Api::Common::V1::Payload.decode(data)
+            end
+          end
+
+          private
+
+          def stored_object(context, payload, context_segments)
+            bucket = @bucket.call(context, payload)
+            data = payload.to_proto
+            if data.bytesize > @max_payload_size
+              raise ArgumentError, "Payload size #{data.bytesize} bytes exceeds the configured max_payload_size " \
+                                   "of #{@max_payload_size} bytes"
+            end
+            hash_value = Digest::SHA256.hexdigest(data)
+            StoredObject.new(payload:, bucket:, key: "v0#{context_segments}/d/sha256/#{hash_value}", hash_value:, data:)
+          end
+
+          def upload_if_absent(object)
+            return if @client.object_exists?(bucket: object.bucket, key: object.key)
+
+            @client.put_object(bucket: object.bucket, key: object.key, data: object.data)
+          rescue StandardError
+            raise "S3StorageDriver store failed [bucket=#{object.bucket}, key=#{object.key}#{client_context}]"
+          end
+
+          def download(bucket, key)
+            @client.get_object(bucket:, key:)
+          rescue StandardError
+            raise "S3StorageDriver retrieve failed [bucket=#{bucket}, key=#{key}#{client_context}]"
+          end
+
+          def validate_claim(claim)
+            bucket, key, hash_algorithm, hash_value =
+              claim.claim_data.values_at(CLAIM_BUCKET, CLAIM_KEY, CLAIM_HASH_ALGORITHM, CLAIM_HASH_VALUE)
+            if bucket.nil? || bucket.empty? || key.nil? || key.empty?
+              raise ArgumentError, 'S3StorageDriver claim is missing required location information: ' \
+                                   "claim_data must contain 'bucket' and 'key'"
+            end
+            if hash_algorithm.nil? || hash_algorithm.empty? || hash_value.nil? || hash_value.empty?
+              raise ArgumentError, 'S3StorageDriver claim is missing required content hash information ' \
+                                   "[bucket=#{bucket}, key=#{key}]: claim_data must contain 'hash_algorithm' and " \
+                                   "'hash_value'"
+            end
+            return if hash_algorithm == 'sha256'
+
+            raise ArgumentError, "S3StorageDriver unsupported hash algorithm [bucket=#{bucket}, key=#{key}]: " \
+                                 "expected sha256, got #{hash_algorithm}"
+          end
+
+          # Each request blocks on I/O, so every item gets its own thread. The first failure cancels the rest so they
+          # stop before making further requests. Requests already in flight run to completion, since a blocking S3
+          # call cannot be safely interrupted.
+          def run_concurrently(items, parent_cancellation, &block)
+            cancellation, cancel_proc = Cancellation.new(parent_cancellation)
+            mutex = Mutex.new
+            first_error = nil #: StandardError?
+            threads = items.map do |item|
+              Thread.new do
+                block.call(item, cancellation)
+              rescue StandardError => e
+                mutex.synchronize { first_error ||= e }
+                cancel_proc.call
+                nil
+              end
+            end
+            results = threads.map(&:value)
+            raise first_error if first_error
+
+            results
+          end
+
+          def context_key_segments(target)
+            case target
+            when Converters::StorageDriverWorkflowInfo
+              "/ns/#{encode_key_segment(target.namespace)}/wt/#{encode_key_segment(target.type)}" \
+              "/wi/#{encode_key_segment(target.id)}/ri/#{encode_key_segment(target.run_id)}"
+            when Converters::StorageDriverActivityInfo
+              "/ns/#{encode_key_segment(target.namespace)}/at/#{encode_key_segment(target.type)}" \
+              "/ai/#{encode_key_segment(target.id)}/ri/#{encode_key_segment(target.run_id)}"
+            else
+              ''
+            end
+          end
+
+          # Escapes every UTF-8 byte outside S3's safe character set so keys are portable across S3 implementations
+          # and identical to the keys other SDK languages build.
+          def encode_key_segment(value)
+            return NULL_SEGMENT if value.nil? || value.empty?
+
+            encoded = value.b.gsub(/[^A-Za-z0-9!\-_.*'()]/n) { |byte| format('%%%02X', byte.ord) }
+            encoded.force_encoding(Encoding::UTF_8)
+          end
+
+          def client_context
+            @client.describe.map { |key, value| ", #{key}=#{value}" }.join
+          end
+        end
+      end
+    end
+  end
+end

data/lib/temporalio/contrib/aws/s3_driver/s3_storage_driver_client.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module Temporalio
+  module Contrib
+    module Aws
+      module S3Driver
+        # S3 object operations used by {S3StorageDriver}.
+        #
+        # This is kept separate from any particular S3 library so the driver does not depend on one. Use
+        # {AwsSdkS3StorageDriverClient} for the official +aws-sdk-s3+ gem, or subclass this to bring another client.
+        #
+        # Implementations are called concurrently from multiple threads and must be thread safe.
+        #
+        # @note WARNING: This API is experimental and may change in the future.
+        #
+        # @!visibility private
+        class S3StorageDriverClient
+          # Upload data to the given bucket and key.
+          #
+          # @param bucket [String] Bucket name.
+          # @param key [String] Object key.
+          # @param data [String] Binary data to upload.
+          def put_object(bucket:, key:, data:)
+            raise NotImplementedError
+          end
+
+          # Check whether an object exists at the given bucket and key.
+          #
+          # @param bucket [String] Bucket name.
+          # @param key [String] Object key.
+          # @return [Boolean] Whether the object exists.
+          def object_exists?(bucket:, key:)
+            raise NotImplementedError
+          end
+
+          # Download the object at the given bucket and key.
+          #
+          # @param bucket [String] Bucket name.
+          # @param key [String] Object key.
+          # @return [String] Binary object data.
+          def get_object(bucket:, key:)
+            raise NotImplementedError
+          end
+
+          # Client-specific diagnostic information, such as the region, that the driver appends to error messages to
+          # help diagnose misconfiguration.
+          #
+          # @return [Hash<String, String>] Diagnostic information. Empty by default.
+          def describe
+            {}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/temporalio/contrib/aws/s3_driver.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'temporalio/contrib/aws/s3_driver/s3_storage_driver'
+require 'temporalio/contrib/aws/s3_driver/s3_storage_driver_client'
+
+module Temporalio
+  module Contrib
+    # Integrations with Amazon Web Services.
+    module Aws
+      # External storage driver for Amazon S3.
+      #
+      # This does not load the +aws-sdk-s3+ gem, so a custom {S3StorageDriverClient} can be used without it. Require
+      # +temporalio/contrib/aws/s3_driver/aws_sdk_s3_storage_driver_client+ to use the official AWS client.
+      #
+      # @note WARNING: This API is experimental and may change in the future.
+      #
+      # @!visibility private
+      module S3Driver
+      end
+    end
+  end
+end

data/lib/temporalio/contrib/open_telemetry.rb

@@ -549,7 +549,7 @@ module Temporalio
               return nil
             end
 
-            span = root.tracer.start_span(name, attributes:, links:, start_timestamp: time, kind:) # steep:ignore
+            span = root.tracer.start_span(name, attributes:, links:, start_timestamp: time, kind:)
             # Record exception and set status if present
             if exception
               span.record_exception(exception)

data/lib/temporalio/converters/external_storage_concurrency.rb

@@ -14,11 +14,11 @@ module Temporalio
     #
     # @!attribute [r] max_driver_operations
     #   @return [Integer] Maximum requests in flight at once across every driver registered on the owning
-    #     {ExternalStorage}. Each worker and each client gets its own budget of this size.
+    #     {ExternalStorage}. Each worker and each client gets its own concurrency limit of this size.
     # @!attribute [r] max_operations_per_message
     #   @return [Integer] Maximum requests in flight at once on behalf of a single message, such as one workflow task
-    #     activation, activity task, Nexus operation, or client request. Every message gets its own budget of this
-    #     size, which stops a message carrying many large payloads from starving the rest.
+    #     activation, activity task, Nexus operation, or client request. Every message gets its own concurrency
+    #     limit of this size, which stops a message carrying many large payloads from starving the rest.
     #
     # @!visibility private
     class ExternalStorageConcurrency
@@ -28,7 +28,7 @@ module Temporalio
       # Shared with every other SDK.
       DEFAULT_MAX_OPERATIONS_PER_MESSAGE = 8
 
-      # Create concurrency limits.
+      # Create concurrency limits. Each must be at least 1.
       #
       # @param max_driver_operations [Integer] Maximum requests in flight across the configuration.
       # @param max_operations_per_message [Integer] Maximum requests in flight for one message.
@@ -36,8 +36,6 @@ module Temporalio
         max_driver_operations: DEFAULT_MAX_DRIVER_OPERATIONS,
         max_operations_per_message: DEFAULT_MAX_OPERATIONS_PER_MESSAGE
       )
-        # A budget of zero would stop every storage operation rather than disabling the limit, so it is rejected
-        # here instead of deadlocking the first offload.
         unless max_driver_operations.is_a?(Integer) && max_driver_operations >= 1
           raise ArgumentError, 'max_driver_operations must be an Integer of at least 1'
         end

data/lib/temporalio/converters/payload_converter/binary_protobuf.rb

@@ -24,7 +24,7 @@ module Temporalio
           Api::Common::V1::Payload.new(
             metadata: {
               'encoding' => ENCODING,
-              'messageType' => value.class.descriptor.name # steep:ignore NoMethod
+              'messageType' => value.class.descriptor.name
             },
             data: value.to_proto
           )

data/lib/temporalio/converters/payload_converter/json_protobuf.rb

@@ -23,7 +23,7 @@ module Temporalio
           Api::Common::V1::Payload.new(
             metadata: {
               'encoding' => ENCODING,
-              'messageType' => value.class.descriptor.name # steep:ignore NoMethod
+              'messageType' => value.class.descriptor.name
             },
             data: value.to_json.b
           )

data/lib/temporalio/converters/storage_driver_limiter.rb

@@ -6,9 +6,8 @@ module Temporalio
     # {StorageDriverStoreContext} and {StorageDriverRetrieveContext}; a driver calls it rather than
     # implementing it.
     #
-    # A driver is handed a whole batch and decides for itself how many requests to make of it, so the SDK cannot
-    # bound that fan out on the driver's behalf. The limit is cooperative: a driver that never calls {#permit} makes
-    # requests that no budget counts, and the SDK warns when it detects that.
+    # The SDK cannot see the requests a driver makes, so the limit works only if the driver takes a permit for
+    # each request.
     #
     # @note WARNING: This API is experimental and may change in the future.
     #
@@ -17,12 +16,11 @@ module Temporalio
       # Hold a permit for the duration of the given block, waiting for one to be available before the block runs
       # and releasing it when the block returns or raises.
       #
-      # Take one permit per request, not one around a whole batch: wrapping a whole fan out in a single permit
-      # satisfies this signature while quietly defeating the limit. Taking a permit while already holding one can
-      # deadlock once the nesting reaches the configured limit, so a driver must not nest these.
+      # Wrapping a whole fan out in a single permit satisfies this signature while quietly defeating the limit.
+      # Do not take more than one permit for a given request: at the limit, every holder would be waiting for a
+      # permit only another holder can release.
       #
-      # Waiting for a permit observes the cancellation on the context this limiter came from, so a driver that is
-      # waiting when the SDK abandons the operation stops waiting.
+      # Waiting for a permit observes the cancellation on the context this limiter came from.
       #
       # @param item [Api::Common::V1::Payload, StorageDriverClaim] What this one request covers: a payload when
       #   storing, a claim when retrieving.

data/lib/temporalio/converters/storage_driver_retrieve_context.rb

@@ -21,8 +21,7 @@ module Temporalio
     # @!attribute [r] cancellation
     #   @return [Cancellation] Cancelled when the SDK abandons this retrieve operation.
     # @!attribute [r] limiter
-    #   @return [StorageDriverLimiter] Bounds how many retrieve requests this driver may have in flight. Take one
-    #     permit per request.
+    #   @return [StorageDriverLimiter] Limiter for this driver's retrieve requests.
     #
     # @!visibility private
     class StorageDriverRetrieveContext; end # rubocop:disable Lint/EmptyClass

data/lib/temporalio/converters/storage_driver_store_context.rb

@@ -22,8 +22,7 @@ module Temporalio
     # @!attribute [r] cancellation
     #   @return [Cancellation] Cancelled when the SDK abandons this store operation.
     # @!attribute [r] limiter
-    #   @return [StorageDriverLimiter] Bounds how many store requests this driver may have in flight. Take one permit
-    #     per request.
+    #   @return [StorageDriverLimiter] Limiter for this driver's store requests.
     #
     # @!visibility private
     class StorageDriverStoreContext; end # rubocop:disable Lint/EmptyClass

data/lib/temporalio/env_config.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'pathname'
+require 'temporalio/error'
 require 'temporalio/internal/bridge'
 
 module Temporalio
@@ -136,7 +137,12 @@ module Temporalio
     #   @return [Boolean, ClientConfigTLS, nil] TLS configuration
     # @!attribute [r] grpc_meta
     #   @return [Hash] gRPC metadata
-    ClientConfigProfile = Data.define(:address, :namespace, :api_key, :tls, :grpc_meta)
+    # @!attribute [r] ext
+    #   @return [Hash<String, Hash<String, Object>>] Raw extension config from +profile.<profile>.ext+, keyed by
+    #     extension name. Each extension decodes its own table. TOML datetimes are ISO 8601 strings. Not modified by
+    #     environment variables. This should be treated as read-only, since copies of a
+    #     profile may share it. WARNING: Extensions are experimental.
+    ClientConfigProfile = Data.define(:address, :namespace, :api_key, :tls, :grpc_meta, :ext)
 
     # A client configuration profile loaded from environment and files.
     #
@@ -153,7 +159,8 @@ module Temporalio
           namespace: hash[:namespace],
           api_key: hash[:api_key],
           tls: ClientConfigTLS.from_h(hash[:tls]),
-          grpc_meta: hash[:grpc_meta] || {}
+          grpc_meta: hash[:grpc_meta] || {},
+          ext: hash[:ext] || {}
         )
       end
 
@@ -185,7 +192,7 @@ module Temporalio
       end
 
       # Create a ClientConfigProfile instance with defaults
-      def initialize(address: nil, namespace: nil, api_key: nil, tls: nil, grpc_meta: {})
+      def initialize(address: nil, namespace: nil, api_key: nil, tls: nil, grpc_meta: {}, ext: {})
         super
       end
 
@@ -196,8 +203,9 @@ module Temporalio
           address: address,
           namespace: namespace,
           api_key: api_key,
-          tls: tls&.to_h&.then { |tls_hash| tls_hash.empty? ? nil : tls_hash }, # steep:ignore
-          grpc_meta: grpc_meta && grpc_meta.empty? ? nil : grpc_meta
+          tls: tls&.to_h&.then { |tls_hash| tls_hash.empty? ? nil : tls_hash },
+          grpc_meta: grpc_meta && grpc_meta.empty? ? nil : grpc_meta,
+          ext: ext && ext.empty? ? nil : ext
         }.compact
       end
 
@@ -222,6 +230,46 @@ module Temporalio
       end
     end
 
+    # Extension mixin to include for extending client configuration with extra TOML properties and environment
+    # variables, and applying them to the client created from the configuration.
+    #
+    # Each extension owns a TOML table under +profile.<profile>.ext.<name>+, e.g.:
+    #
+    #   [profile.default.ext.my_extension]
+    #   some_field = "value"
+    #
+    # Extensions are recommended to use a +TEMPORAL_EXT_<NAME>_+ prefix for any environment variables they read.
+    #
+    # WARNING: Extensions are experimental.
+    module ClientConfigExtension
+      # @abstract
+      # @return [String] The extension's table key under +profile.<profile>.ext+. Must be unique among loaded
+      #   extensions.
+      def name
+        raise NotImplementedError
+      end
+
+      # Read this extension's config from the profile and env, and return a plugin that applies it.
+      #
+      # @abstract
+      # @param request [ClientConfigExtensionRequest] Loaded profile and env.
+      # @return [Client::Plugin, nil] Plugin, or nil if there's nothing to apply.
+      def client_plugin(request)
+        raise NotImplementedError
+      end
+    end
+
+    # Input to {ClientConfigExtension#client_plugin}.
+    #
+    # WARNING: Extensions are experimental.
+    #
+    # @!attribute [r] profile
+    #   @return [ClientConfigProfile] Loaded profile, with environment variable overrides applied.
+    # @!attribute [r] env
+    #   @return [Hash<String, String>, nil] Environment variables to read extension env vars from, or nil if
+    #     environment variable loading is disabled.
+    ClientConfigExtensionRequest = Data.define(:profile, :env)
+
     # Client configuration loaded from TOML and environment variables.
     #
     # This contains a mapping of profile names to client profiles.
@@ -273,7 +321,7 @@ module Temporalio
       # Load a single client profile and convert to connect config
       #
       # This is a convenience function that combines loading a profile and
-      # converting it to a connect config hash.
+      # converting it to a connect config hash, plus applying extensions.
       #
       # @param profile [String, nil] The profile to load from the config
       # @param config_source [Pathname, String, nil] Configuration source
@@ -281,6 +329,8 @@ module Temporalio
       # @param disable_env [Boolean] If true, environment variable loading and overriding is disabled
       # @param config_file_strict [Boolean] If true, will error on unrecognized keys
       # @param override_env_vars [Hash, nil] Environment variables to use for loading and overrides
+      # @param extensions [Array<ClientConfigExtension>] Extensions to call, in order, with the loaded profile. Plugins
+      #   they return are included as +plugins+ in the keyword args. WARNING: Extensions are experimental.
       # @return [Array] Tuple of [positional_args, keyword_args] that can be splatted to Client.connect
       def self.load_client_connect_options(
         profile: nil,
@@ -288,8 +338,10 @@ module Temporalio
         disable_file: false,
         disable_env: false,
         config_file_strict: false,
-        override_env_vars: nil
+        override_env_vars: nil,
+        extensions: []
       )
+        _validate_extensions!(extensions)
         prof = ClientConfigProfile.load(
           profile: profile,
           config_source: config_source,
@@ -298,7 +350,36 @@ module Temporalio
           config_file_strict: config_file_strict,
           override_env_vars: override_env_vars
         )
-        prof.to_client_connect_options
+        positional_args, keyword_args = prof.to_client_connect_options
+        return [positional_args, keyword_args] if extensions.empty?
+
+        request = ClientConfigExtensionRequest.new(
+          profile: prof,
+          env: disable_env ? nil : override_env_vars || ENV.to_h
+        )
+        plugins = extensions.filter_map do |extension|
+          extension.client_plugin(request)
+        rescue StandardError => e
+          raise Error, "client config extension \"#{extension.name}\": #{e.message}"
+        end
+        keyword_args[:plugins] = plugins unless plugins.empty?
+        [positional_args, keyword_args]
+      end
+
+      # @!visibility private
+      def self._validate_extensions!(extensions)
+        names = {}
+        extensions.each_with_index do |extension, index|
+          unless extension.is_a?(ClientConfigExtension)
+            raise ArgumentError, "#{extension.class} does not implement EnvConfig::ClientConfigExtension"
+          end
+
+          name = extension.name
+          raise ArgumentError, "client config extension at index #{index} has empty name" if name.empty?
+          raise ArgumentError, "duplicate client config extension name \"#{name}\"" if names.key?(name)
+
+          names[name] = true
+        end
       end
 
       # Create a ClientConfig instance with defaults

data/lib/temporalio/internal/bridge/api/activity_result/activity_result.rb

@@ -7,10 +7,11 @@ require 'google/protobuf'
 require 'google/protobuf/duration_pb'
 require 'google/protobuf/timestamp_pb'
 require 'temporalio/api/common/v1/message'
+require 'temporalio/api/enums/v1/failed_cause'
 require 'temporalio/api/failure/v1/message'
 
 
-descriptor_data = "\n7temporal/sdk/core/activity_result/activity_result.proto\x12\x17\x63oresdk.activity_result\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a$temporal/api/common/v1/message.proto\x1a%temporal/api/failure/v1/message.proto\"\x95\x02\n\x17\x41\x63tivityExecutionResult\x12\x35\n\tcompleted\x18\x01 \x01(\x0b\x32 .coresdk.activity_result.SuccessH\x00\x12\x32\n\x06\x66\x61iled\x18\x02 \x01(\x0b\x32 .coresdk.activity_result.FailureH\x00\x12:\n\tcancelled\x18\x03 \x01(\x0b\x32%.coresdk.activity_result.CancellationH\x00\x12I\n\x13will_complete_async\x18\x04 \x01(\x0b\x32*.coresdk.activity_result.WillCompleteAsyncH\x00\x42\x08\n\x06status\"\xfc\x01\n\x12\x41\x63tivityResolution\x12\x35\n\tcompleted\x18\x01 \x01(\x0b\x32 .coresdk.activity_result.SuccessH\x00\x12\x32\n\x06\x66\x61iled\x18\x02 \x01(\x0b\x32 .coresdk.activity_result.FailureH\x00\x12:\n\tcancelled\x18\x03 \x01(\x0b\x32%.coresdk.activity_result.CancellationH\x00\x12\x35\n\x07\x62\x61\x63koff\x18\x04 \x01(\x0b\x32\".coresdk.activity_result.DoBackoffH\x00\x42\x08\n\x06status\":\n\x07Success\x12/\n\x06result\x18\x01 \x01(\x0b\x32\x1f.temporal.api.common.v1.Payload\"}\n\x07\x46\x61ilure\x12\x31\n\x07\x66\x61ilure\x18\x01 \x01(\x0b\x32 .temporal.api.failure.v1.Failure\x12?\n\x05\x63\x61use\x18\x02 \x01(\x0e\x32\x30.coresdk.activity_result.ActivityTaskFailedCause\"A\n\x0c\x43\x61ncellation\x12\x31\n\x07\x66\x61ilure\x18\x01 \x01(\x0b\x32 .temporal.api.failure.v1.Failure\"\x13\n\x11WillCompleteAsync\"\x8d\x01\n\tDoBackoff\x12\x0f\n\x07\x61ttempt\x18\x01 \x01(\r\x12\x33\n\x10\x62\x61\x63koff_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.Duration\x12:\n\x16original_schedule_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*\xf3\x01\n\x17\x41\x63tivityTaskFailedCause\x12*\n&ACTIVITY_TASK_FAILED_CAUSE_UNSPECIFIED\x10\x00\x12\x31\n-ACTIVITY_TASK_FAILED_CAUSE_PAYLOADS_TOO_LARGE\x10\x01\x12\x37\n3ACTIVITY_TASK_FAILED_CAUSE_EXTERNAL_STORAGE_FAILURE\x10\x02\x12@\n<ACTIVITY_TASK_FAILED_CAUSE_ACTIVITY_WORKER_UNHANDLED_FAILURE\x10\x03\x42\x34\xea\x02\x31Temporalio::Internal::Bridge::Api::ActivityResultb\x06proto3"
+descriptor_data = "\n7temporal/sdk/core/activity_result/activity_result.proto\x12\x17\x63oresdk.activity_result\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a$temporal/api/common/v1/message.proto\x1a(temporal/api/enums/v1/failed_cause.proto\x1a%temporal/api/failure/v1/message.proto\"\x95\x02\n\x17\x41\x63tivityExecutionResult\x12\x35\n\tcompleted\x18\x01 \x01(\x0b\x32 .coresdk.activity_result.SuccessH\x00\x12\x32\n\x06\x66\x61iled\x18\x02 \x01(\x0b\x32 .coresdk.activity_result.FailureH\x00\x12:\n\tcancelled\x18\x03 \x01(\x0b\x32%.coresdk.activity_result.CancellationH\x00\x12I\n\x13will_complete_async\x18\x04 \x01(\x0b\x32*.coresdk.activity_result.WillCompleteAsyncH\x00\x42\x08\n\x06status\"\xfc\x01\n\x12\x41\x63tivityResolution\x12\x35\n\tcompleted\x18\x01 \x01(\x0b\x32 .coresdk.activity_result.SuccessH\x00\x12\x32\n\x06\x66\x61iled\x18\x02 \x01(\x0b\x32 .coresdk.activity_result.FailureH\x00\x12:\n\tcancelled\x18\x03 \x01(\x0b\x32%.coresdk.activity_result.CancellationH\x00\x12\x35\n\x07\x62\x61\x63koff\x18\x04 \x01(\x0b\x32\".coresdk.activity_result.DoBackoffH\x00\x42\x08\n\x06status\":\n\x07Success\x12/\n\x06result\x18\x01 \x01(\x0b\x32\x1f.temporal.api.common.v1.Payload\"{\n\x07\x46\x61ilure\x12\x31\n\x07\x66\x61ilure\x18\x01 \x01(\x0b\x32 .temporal.api.failure.v1.Failure\x12=\n\x05\x63\x61use\x18\x02 \x01(\x0e\x32..temporal.api.enums.v1.ActivityTaskFailedCause\"A\n\x0c\x43\x61ncellation\x12\x31\n\x07\x66\x61ilure\x18\x01 \x01(\x0b\x32 .temporal.api.failure.v1.Failure\"\x13\n\x11WillCompleteAsync\"\x8d\x01\n\tDoBackoff\x12\x0f\n\x07\x61ttempt\x18\x01 \x01(\r\x12\x33\n\x10\x62\x61\x63koff_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.Duration\x12:\n\x16original_schedule_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB4\xea\x02\x31Temporalio::Internal::Bridge::Api::ActivityResultb\x06proto3"
 
 pool = ::Google::Protobuf::DescriptorPool.generated_pool
 pool.add_serialized_file(descriptor_data)
@@ -27,7 +28,6 @@ module Temporalio
           Cancellation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_result.Cancellation").msgclass
           WillCompleteAsync = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_result.WillCompleteAsync").msgclass
           DoBackoff = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_result.DoBackoff").msgclass
-          ActivityTaskFailedCause = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_result.ActivityTaskFailedCause").enummodule
         end
       end
     end

data/lib/temporalio/internal/bridge/api/activity_task/activity_task.rb

@@ -7,10 +7,11 @@ require 'google/protobuf'
 require 'google/protobuf/duration_pb'
 require 'google/protobuf/timestamp_pb'
 require 'temporalio/api/common/v1/message'
+require 'temporalio/api/nexus/v1/message'
 require 'temporalio/internal/bridge/api/common/common'
 
 
-descriptor_data = "\n3temporal/sdk/core/activity_task/activity_task.proto\x12\x15\x63oresdk.activity_task\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a$temporal/api/common/v1/message.proto\x1a%temporal/sdk/core/common/common.proto\"\x8d\x01\n\x0c\x41\x63tivityTask\x12\x12\n\ntask_token\x18\x01 \x01(\x0c\x12-\n\x05start\x18\x03 \x01(\x0b\x32\x1c.coresdk.activity_task.StartH\x00\x12/\n\x06\x63\x61ncel\x18\x04 \x01(\x0b\x32\x1d.coresdk.activity_task.CancelH\x00\x42\t\n\x07variant\"\xb1\x07\n\x05Start\x12\x1a\n\x12workflow_namespace\x18\x01 \x01(\t\x12\x15\n\rworkflow_type\x18\x02 \x01(\t\x12\x45\n\x12workflow_execution\x18\x03 \x01(\x0b\x32).temporal.api.common.v1.WorkflowExecution\x12\x13\n\x0b\x61\x63tivity_id\x18\x04 \x01(\t\x12\x15\n\ractivity_type\x18\x05 \x01(\t\x12\x45\n\rheader_fields\x18\x06 \x03(\x0b\x32..coresdk.activity_task.Start.HeaderFieldsEntry\x12.\n\x05input\x18\x07 \x03(\x0b\x32\x1f.temporal.api.common.v1.Payload\x12:\n\x11heartbeat_details\x18\x08 \x03(\x0b\x32\x1f.temporal.api.common.v1.Payload\x12\x32\n\x0escheduled_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x42\n\x1e\x63urrent_attempt_scheduled_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x30\n\x0cstarted_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07\x61ttempt\x18\x0c \x01(\r\x12<\n\x19schedule_to_close_timeout\x18\r \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x39\n\x16start_to_close_timeout\x18\x0e \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x34\n\x11heartbeat_timeout\x18\x0f \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x39\n\x0cretry_policy\x18\x10 \x01(\x0b\x32#.temporal.api.common.v1.RetryPolicy\x12\x32\n\x08priority\x18\x12 \x01(\x0b\x32 .temporal.api.common.v1.Priority\x12\x10\n\x08is_local\x18\x11 \x01(\x08\x12\x0e\n\x06run_id\x18\x13 \x01(\t\x1aT\n\x11HeaderFieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12.\n\x05value\x18\x02 \x01(\x0b\x32\x1f.temporal.api.common.v1.Payload:\x02\x38\x01\"\x8a\x01\n\x06\x43\x61ncel\x12;\n\x06reason\x18\x01 \x01(\x0e\x32+.coresdk.activity_task.ActivityCancelReason\x12\x43\n\x07\x64\x65tails\x18\x02 \x01(\x0b\x32\x32.coresdk.activity_task.ActivityCancellationDetails\"\xa0\x01\n\x1b\x41\x63tivityCancellationDetails\x12\x14\n\x0cis_not_found\x18\x01 \x01(\x08\x12\x14\n\x0cis_cancelled\x18\x02 \x01(\x08\x12\x11\n\tis_paused\x18\x03 \x01(\x08\x12\x14\n\x0cis_timed_out\x18\x04 \x01(\x08\x12\x1a\n\x12is_worker_shutdown\x18\x05 \x01(\x08\x12\x10\n\x08is_reset\x18\x06 \x01(\x08*o\n\x14\x41\x63tivityCancelReason\x12\r\n\tNOT_FOUND\x10\x00\x12\r\n\tCANCELLED\x10\x01\x12\r\n\tTIMED_OUT\x10\x02\x12\x13\n\x0fWORKER_SHUTDOWN\x10\x03\x12\n\n\x06PAUSED\x10\x04\x12\t\n\x05RESET\x10\x05\x42\x32\xea\x02/Temporalio::Internal::Bridge::Api::ActivityTaskb\x06proto3"
+descriptor_data = "\n3temporal/sdk/core/activity_task/activity_task.proto\x12\x15\x63oresdk.activity_task\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a$temporal/api/common/v1/message.proto\x1a#temporal/api/nexus/v1/message.proto\x1a%temporal/sdk/core/common/common.proto\"\x8d\x01\n\x0c\x41\x63tivityTask\x12\x12\n\ntask_token\x18\x01 \x01(\x0c\x12-\n\x05start\x18\x03 \x01(\x0b\x32\x1c.coresdk.activity_task.StartH\x00\x12/\n\x06\x63\x61ncel\x18\x04 \x01(\x0b\x32\x1d.coresdk.activity_task.CancelH\x00\x42\t\n\x07variant\"\xdb\x08\n\x05Start\x12\x1a\n\x12workflow_namespace\x18\x01 \x01(\t\x12\x15\n\rworkflow_type\x18\x02 \x01(\t\x12\x45\n\x12workflow_execution\x18\x03 \x01(\x0b\x32).temporal.api.common.v1.WorkflowExecution\x12\x13\n\x0b\x61\x63tivity_id\x18\x04 \x01(\t\x12\x15\n\ractivity_type\x18\x05 \x01(\t\x12\x45\n\rheader_fields\x18\x06 \x03(\x0b\x32..coresdk.activity_task.Start.HeaderFieldsEntry\x12.\n\x05input\x18\x07 \x03(\x0b\x32\x1f.temporal.api.common.v1.Payload\x12:\n\x11heartbeat_details\x18\x08 \x03(\x0b\x32\x1f.temporal.api.common.v1.Payload\x12\x32\n\x0escheduled_time\x18\t \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x42\n\x1e\x63urrent_attempt_scheduled_time\x18\n \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x30\n\x0cstarted_time\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0f\n\x07\x61ttempt\x18\x0c \x01(\r\x12<\n\x19schedule_to_close_timeout\x18\r \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x39\n\x16start_to_close_timeout\x18\x0e \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x34\n\x11heartbeat_timeout\x18\x0f \x01(\x0b\x32\x19.google.protobuf.Duration\x12\x39\n\x0cretry_policy\x18\x10 \x01(\x0b\x32#.temporal.api.common.v1.RetryPolicy\x12\x32\n\x08priority\x18\x12 \x01(\x0b\x32 .temporal.api.common.v1.Priority\x12\x10\n\x08is_local\x18\x11 \x01(\x08\x12\x0e\n\x06run_id\x18\x13 \x01(\t\x12\x65\n&propagated_nexus_serialization_context\x18\x14 \x01(\x0b\x32\x35.temporal.api.nexus.v1.PropagatedSerializationContext\x12\x41\n\rworkflow_task\x18\x15 \x01(\x0b\x32*.coresdk.activity_task.WorkflowTaskAttempt\x1aT\n\x11HeaderFieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12.\n\x05value\x18\x02 \x01(\x0b\x32\x1f.temporal.api.common.v1.Payload:\x02\x38\x01\"\x80\x02\n\x13WorkflowTaskAttempt\x12\x18\n\x10started_event_id\x18\x01 \x01(\x03\x12\x0f\n\x07\x61ttempt\x18\x02 \x01(\r\x12\x30\n\x0cstarted_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12!\n\x19previous_started_event_id\x18\x04 \x01(\x03\x12\x14\n\x0c\x66\x61iled_tasks\x18\x05 \x03(\x03\x12\x18\n\x10previous_attempt\x18\x06 \x01(\r\x12\x39\n\x15previous_started_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x8a\x01\n\x06\x43\x61ncel\x12;\n\x06reason\x18\x01 \x01(\x0e\x32+.coresdk.activity_task.ActivityCancelReason\x12\x43\n\x07\x64\x65tails\x18\x02 \x01(\x0b\x32\x32.coresdk.activity_task.ActivityCancellationDetails\"\xa0\x01\n\x1b\x41\x63tivityCancellationDetails\x12\x14\n\x0cis_not_found\x18\x01 \x01(\x08\x12\x14\n\x0cis_cancelled\x18\x02 \x01(\x08\x12\x11\n\tis_paused\x18\x03 \x01(\x08\x12\x14\n\x0cis_timed_out\x18\x04 \x01(\x08\x12\x1a\n\x12is_worker_shutdown\x18\x05 \x01(\x08\x12\x10\n\x08is_reset\x18\x06 \x01(\x08*o\n\x14\x41\x63tivityCancelReason\x12\r\n\tNOT_FOUND\x10\x00\x12\r\n\tCANCELLED\x10\x01\x12\r\n\tTIMED_OUT\x10\x02\x12\x13\n\x0fWORKER_SHUTDOWN\x10\x03\x12\n\n\x06PAUSED\x10\x04\x12\t\n\x05RESET\x10\x05\x42\x32\xea\x02/Temporalio::Internal::Bridge::Api::ActivityTaskb\x06proto3"
 
 pool = ::Google::Protobuf::DescriptorPool.generated_pool
 pool.add_serialized_file(descriptor_data)
@@ -22,6 +23,7 @@ module Temporalio
         module ActivityTask
           ActivityTask = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.ActivityTask").msgclass
           Start = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.Start").msgclass
+          WorkflowTaskAttempt = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.WorkflowTaskAttempt").msgclass
           Cancel = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.Cancel").msgclass
           ActivityCancellationDetails = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.ActivityCancellationDetails").msgclass
           ActivityCancelReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("coresdk.activity_task.ActivityCancelReason").enummodule