temporalio 0.3.0-x86_64-linux → 0.5.0-x86_64-linux

data/lib/temporalio/runtime.rb

@@ -4,6 +4,7 @@ require 'temporalio/internal/bridge'
 require 'temporalio/internal/bridge/runtime'
 require 'temporalio/internal/metric'
 require 'temporalio/metric'
+require 'temporalio/runtime/metric_buffer'
 
 module Temporalio
   # Runtime for Temporal Ruby SDK.
@@ -107,6 +108,7 @@ module Temporalio
     MetricsOptions = Data.define(
       :opentelemetry,
       :prometheus,
+      :buffer,
       :attach_service_name,
       :global_tags,
       :metric_prefix
@@ -116,10 +118,13 @@ module Temporalio
     #
     # @!attribute opentelemetry
     #   @return [OpenTelemetryMetricsOptions, nil] OpenTelemetry options if using OpenTelemetry. This is mutually
-    #     exclusive with +prometheus+.
+    #     exclusive with `prometheus` and `buffer`.
     # @!attribute prometheus
     #   @return [PrometheusMetricsOptions, nil] Prometheus options if using Prometheus. This is mutually exclusive with
-    #     +opentelemetry+.
+    #     `opentelemetry` and `buffer`.
+    # @!attribute buffer
+    #   @return [MetricBuffer, nil] Metric buffer to send all metrics to. This is mutually exclusive with `prometheus`
+    #     and `opentelemetry`.
     # @!attribute attach_service_name
     #   @return [Boolean] Whether to put the service_name on every metric.
     # @!attribute global_tags
@@ -130,19 +135,26 @@ module Temporalio
       # Create metrics options. Either `opentelemetry` or `prometheus` required, but not both.
       #
       # @param opentelemetry [OpenTelemetryMetricsOptions, nil] OpenTelemetry options if using OpenTelemetry. This is
-      #   mutually exclusive with `prometheus`.
+      #   mutually exclusive with `prometheus` and `buffer`.
       # @param prometheus [PrometheusMetricsOptions, nil] Prometheus options if using Prometheus. This is mutually
-      #   exclusive with `opentelemetry`.
+      #   exclusive with `opentelemetry` and `buffer`.
+      # @param buffer [MetricBuffer, nil] Metric buffer to send all metrics to. This is mutually exclusive with
+      #   `prometheus` and `opentelemetry`.
       # @param attach_service_name [Boolean] Whether to put the service_name on every metric.
       # @param global_tags [Hash<String, String>, nil] Resource tags to be applied to all metrics.
       # @param metric_prefix [String, nil] Prefix to put on every Temporal metric. If unset, defaults to `temporal_`.
       def initialize(
         opentelemetry: nil,
         prometheus: nil,
+        buffer: nil,
         attach_service_name: true,
         global_tags: nil,
         metric_prefix: nil
       )
+        if [opentelemetry, prometheus, buffer].count { |v| !v.nil? } > 1
+          raise 'Can only have one of opentelemetry, prometheus, or buffer'
+        end
+
         super
       end
 
@@ -152,6 +164,7 @@ module Temporalio
         Internal::Bridge::Runtime::MetricsOptions.new(
           opentelemetry: opentelemetry&._to_bridge,
           prometheus: prometheus&._to_bridge,
+          buffered_with_size: buffer&._buffer_size,
           attach_service_name:,
           global_tags:,
           metric_prefix:
@@ -164,7 +177,9 @@ module Temporalio
       :headers,
       :metric_periodicity,
       :metric_temporality,
-      :durations_as_seconds
+      :durations_as_seconds,
+      :http,
+      :histogram_bucket_overrides
     )
 
     # Options for exporting metrics to OpenTelemetry.
@@ -181,6 +196,11 @@ module Temporalio
     # @!attribute durations_as_seconds
     #   @return [Boolean] Whether to use float seconds instead of integer milliseconds for durations, default is
     #     +false+.
+    # @!attribute http
+    #   @return [Boolean] True if the protocol is HTTP, false if gRPC (the default).
+    # @!attribute histogram_bucket_overrides
+    #   @return [Hash<String, Array<Numeric>>, nil] Override default histogram buckets. Key of the hash it the metric
+    #     name, value is an array of floats for the set of buckets.
     class OpenTelemetryMetricsOptions
       # OpenTelemetry metric temporality.
       module MetricTemporality
@@ -196,12 +216,17 @@ module Temporalio
       # @param metric_temporality [MetricTemporality] How frequently metrics should be exported.
       # @param durations_as_seconds [Boolean] Whether to use float seconds instead of integer milliseconds for
       #   durations.
+      # @param http [Boolean] True if the protocol is HTTP, false if gRPC (the default).
+      # @param histogram_bucket_overrides [Hash<String, Array<Numeric>>, nil] Override default histogram buckets. Key of
+      #   the hash it the metric name, value is an array of floats for the set of buckets.
       def initialize(
         url:,
         headers: nil,
         metric_periodicity: nil,
         metric_temporality: MetricTemporality::CUMULATIVE,
-        durations_as_seconds: false
+        durations_as_seconds: false,
+        http: false,
+        histogram_bucket_overrides: nil
       )
         super
       end
@@ -218,7 +243,9 @@ module Temporalio
                                     when MetricTemporality::DELTA then true
                                     else raise 'Unrecognized metric temporality'
                                     end,
-          durations_as_seconds:
+          durations_as_seconds:,
+          http:,
+          histogram_bucket_overrides:
         )
       end
     end
@@ -227,7 +254,8 @@ module Temporalio
       :bind_address,
       :counters_total_suffix,
       :unit_suffix,
-      :durations_as_seconds
+      :durations_as_seconds,
+      :histogram_bucket_overrides
     )
 
     # Options for exporting metrics to Prometheus.
@@ -240,6 +268,9 @@ module Temporalio
     #   @return [Boolean] If `true`, all histograms will include the unit in their name as a suffix.
     # @!attribute durations_as_seconds
     #   @return [Boolean] Whether to use float seconds instead of integer milliseconds for durations.
+    # @!attribute histogram_bucket_overrides
+    #   @return [Hash<String, Array<Numeric>>, nil] Override default histogram buckets. Key of the hash it the metric
+    #     name, value is an array of floats for the set of buckets.
     class PrometheusMetricsOptions
       # Create Prometheus options.
       #
@@ -248,11 +279,14 @@ module Temporalio
       # @param unit_suffix [Boolean] If `true`, all histograms will include the unit in their name as a suffix.
       # @param durations_as_seconds [Boolean] Whether to use float seconds instead of integer milliseconds for
       #   durations.
+      # @param histogram_bucket_overrides [Hash<String, Array<Numeric>>, nil] Override default histogram buckets. Key of
+      #   the hash it the metric name, value is an array of floats for the set of buckets.
       def initialize(
         bind_address:,
         counters_total_suffix: false,
         unit_suffix: false,
-        durations_as_seconds: false
+        durations_as_seconds: false,
+        histogram_bucket_overrides: nil
       )
         super
       end
@@ -264,7 +298,8 @@ module Temporalio
           bind_address:,
           counters_total_suffix:,
           unit_suffix:,
-          durations_as_seconds:
+          durations_as_seconds:,
+          histogram_bucket_overrides:
         )
       end
     end
@@ -295,6 +330,9 @@ module Temporalio
     #
     # @param telemetry [TelemetryOptions] Telemetry options to set.
     def initialize(telemetry: TelemetryOptions.new)
+      # Set runtime on the buffer which will fail if the buffer is used on another runtime
+      telemetry.metrics&.buffer&._set_runtime(self)
+
       @core_runtime = Internal::Bridge::Runtime.new(
         Internal::Bridge::Runtime::Options.new(telemetry: telemetry._to_bridge)
       )

data/lib/temporalio/search_attributes.rb

@@ -251,6 +251,14 @@ module Temporalio
       @raw_hash.length
     end
 
+    # Check equality.
+    #
+    # @param other [SearchAttributes] To compare.
+    # @return [Boolean] Whether equal.
+    def ==(other)
+      other.is_a?(SearchAttributes) && @raw_hash == other._raw_hash
+    end
+
     alias size length
 
     # Return a new search attributes collection with updates applied.
@@ -280,6 +288,11 @@ module Temporalio
       end
     end
 
+    # @!visibility private
+    def _raw_hash
+      @raw_hash
+    end
+
     # @!visibility private
     def _to_proto
       Api::Common::V1::SearchAttributes.new(indexed_fields: _to_proto_hash)

data/lib/temporalio/testing/activity_environment.rb

@@ -13,17 +13,18 @@ module Temporalio
     # cancellation can be set, users create this for each activity that is run. There is no real performance penalty for
     # creating an environment for every run.
     class ActivityEnvironment
-      # @return [Activity::Info] The activity info used by default. This is frozen, but can be dup'd and mutated to pass
-      #   in to {initialize}.
+      # @return [Activity::Info] The activity info used by default. This is frozen, but `with` can be used to make a new
+      #   instance with changes to pass in to {initialize}.
       def self.default_info
         @default_info ||= Activity::Info.new(
           activity_id: 'test',
           activity_type: 'unknown',
           attempt: 1,
           current_attempt_scheduled_time: Time.at(0),
-          heartbeat_details: [],
           heartbeat_timeout: nil,
           local?: false,
+          priority: Temporalio::Priority.default,
+          raw_heartbeat_details: [],
           schedule_to_close_timeout: 1.0,
           scheduled_time: Time.at(0),
           start_to_close_timeout: 1.0,
@@ -34,35 +35,49 @@ module Temporalio
           workflow_namespace: 'default',
           workflow_run_id: 'test-run',
           workflow_type: 'test'
-        ).freeze
+        )
       end
 
       # Create a test environment for activities.
       #
-      # @param info [Activity::Info] Value for {Activity::Context#info}.
+      # @param info [Activity::Info] Value for {Activity::Context#info}. Users should not try to instantiate this
+      #   themselves, but rather use `with` on {default_info}.
       # @param on_heartbeat [Proc(Array), nil] Proc that is called with all heartbeat details when
-      #   {Activity::Context#heartbeat} is called.
+      #   {Activity::Context#heartbeat} is called. Should return a value
       # @param cancellation [Cancellation] Value for {Activity::Context#cancellation}.
+      # @param on_cancellation_details [Proc, nil] Proc that is called when {Activity::Context#cancellation_details} is
+      #   called. Defaults to a proc that returns an instance if canceled with `cancel_requested` as true.
       # @param worker_shutdown_cancellation [Cancellation] Value for {Activity::Context#worker_shutdown_cancellation}.
       # @param payload_converter [Converters::PayloadConverter] Value for {Activity::Context#payload_converter}.
       # @param logger [Logger] Value for {Activity::Context#logger}.
       # @param activity_executors [Hash<Symbol, Worker::ActivityExecutor>] Executors that activities can run within.
+      # @param metric_meter [Metric::Meter, nil] Value for {Activity::Context#metric_meter}, or nil to raise when
+      #   called.
+      # @param client [Client, nil] Value for {Activity::Context#client}, or nil to raise when called.
       def initialize(
         info: ActivityEnvironment.default_info,
         on_heartbeat: nil,
         cancellation: Cancellation.new,
+        on_cancellation_details: nil,
         worker_shutdown_cancellation: Cancellation.new,
         payload_converter: Converters::PayloadConverter.default,
         logger: Logger.new(nil),
-        activity_executors: Worker::ActivityExecutor.defaults
+        activity_executors: Worker::ActivityExecutor.defaults,
+        metric_meter: nil,
+        client: nil
       )
         @info = info
         @on_heartbeat = on_heartbeat
         @cancellation = cancellation
+        @on_cancellation_details = on_cancellation_details || proc do
+          @_cancellation_details ||= Activity::CancellationDetails.new if @cancellation.canceled?
+        end
         @worker_shutdown_cancellation = worker_shutdown_cancellation
         @payload_converter = payload_converter
         @logger = logger
         @activity_executors = activity_executors
+        @metric_meter = metric_meter
+        @client = client
       end
 
       # Run an activity and returns its result or raises its exception.
@@ -80,14 +95,19 @@ module Temporalio
           Activity::Context._current_executor = executor
           executor.set_activity_context(defn, Context.new(
                                                 info: @info.dup,
+                                                instance:
+                                                  defn.instance.is_a?(Proc) ? defn.instance.call : defn.instance,
                                                 on_heartbeat: @on_heartbeat,
                                                 cancellation: @cancellation,
+                                                on_cancellation_details: @on_cancellation_details,
                                                 worker_shutdown_cancellation: @worker_shutdown_cancellation,
                                                 payload_converter: @payload_converter,
-                                                logger: @logger
+                                                logger: @logger,
+                                                metric_meter: @metric_meter,
+                                                client: @client
                                               ))
           queue.push([defn.proc.call(*args), nil])
-        rescue Exception => e # rubocop:disable Lint/RescueException Intentionally capturing all exceptions
+        rescue Exception => e # rubocop:disable Lint/RescueException -- Intentionally capturing all exceptions
           queue.push([nil, e])
         ensure
           executor.set_activity_context(defn, nil)
@@ -102,28 +122,51 @@ module Temporalio
 
       # @!visibility private
       class Context < Activity::Context
-        attr_reader :info, :cancellation, :worker_shutdown_cancellation, :payload_converter, :logger
+        attr_reader :info, :instance, :cancellation, :worker_shutdown_cancellation, :payload_converter, :logger
 
         def initialize( # rubocop:disable Lint/MissingSuper
-          info: ActivityEnvironment.default_info,
-          on_heartbeat: nil,
-          cancellation: Cancellation.new,
-          worker_shutdown_cancellation: Cancellation.new,
-          payload_converter: Converters::PayloadConverter.default,
-          logger: Logger.new(nil)
+          info:,
+          instance:,
+          on_heartbeat:,
+          cancellation:,
+          on_cancellation_details:,
+          worker_shutdown_cancellation:,
+          payload_converter:,
+          logger:,
+          metric_meter:,
+          client:
         )
           @info = info
+          @instance = instance
           @on_heartbeat = on_heartbeat
           @cancellation = cancellation
+          @on_cancellation_details = on_cancellation_details
           @worker_shutdown_cancellation = worker_shutdown_cancellation
           @payload_converter = payload_converter
           @logger = logger
+          @metric_meter = metric_meter
+          @client = client
         end
 
         # @!visibility private
         def heartbeat(*details)
           @on_heartbeat&.call(details)
         end
+
+        # @!visibility private
+        def metric_meter
+          @metric_meter or raise 'No metric meter configured in this test environment'
+        end
+
+        # @!visibility private
+        def client
+          @client or raise 'No client configured in this test environment'
+        end
+
+        # @!visibility private
+        def cancellation_details
+          @on_cancellation_details.call
+        end
       end
 
       private_constant :Context

data/lib/temporalio/testing/workflow_environment.rb

@@ -10,6 +10,7 @@ require 'temporalio/converters'
 require 'temporalio/internal/bridge/testing'
 require 'temporalio/internal/proto_utils'
 require 'temporalio/runtime'
+require 'temporalio/search_attributes'
 require 'temporalio/version'
 
 module Temporalio
@@ -34,8 +35,10 @@ module Temporalio
       # @param default_workflow_query_reject_condition [WorkflowQueryRejectCondition, nil] Default rejection condition
       #   for the client.
       # @param ip [String] IP to bind to.
-      # @param port [Integer, nil] Port to bind on, or +nil+ for random.
+      # @param port [Integer, nil] Port to bind on, or `nil` for random.
       # @param ui [Boolean] If +true+, also starts the UI.
+      # @param ui_port [Integer, nil] Port to bind on if `ui` is true, or `nil` for random.
+      # @param search_attributes [Array<SearchAttributes::Key>] Search attributes to make available on start.
       # @param runtime [Runtime] Runtime for the server and client.
       # @param dev_server_existing_path [String, nil] Existing CLI path to use instead of downloading and caching to
       #   tmp.
@@ -46,6 +49,8 @@ module Temporalio
       # @param dev_server_download_version [String] Version of dev server to download and cache.
       # @param dev_server_download_dest_dir [String, nil] Where to download. Defaults to tmp.
       # @param dev_server_extra_args [Array<String>] Any extra arguments for the CLI dev server.
+      # @param dev_server_download_ttl [Float, nil] How long the automatic download should be cached for. If nil, cached
+      #   indefinitely.
       #
       # @yield [environment] If a block is given, it is called with the environment and upon complete the environment is
       #   shutdown.
@@ -62,6 +67,8 @@ module Temporalio
         ip: '127.0.0.1',
         port: nil,
         ui: false, # rubocop:disable Naming/MethodParameterName
+        ui_port: nil,
+        search_attributes: [],
         runtime: Runtime.default,
         dev_server_existing_path: nil,
         dev_server_database_filename: nil,
@@ -70,8 +77,18 @@ module Temporalio
         dev_server_download_version: 'default',
         dev_server_download_dest_dir: nil,
         dev_server_extra_args: [],
+        dev_server_download_ttl: nil,
         &
       )
+        # Add search attribute args
+        unless search_attributes.empty?
+          dev_server_extra_args += search_attributes.flat_map do |key|
+            raise 'Search attribute must be Key' unless key.is_a?(SearchAttributes::Key)
+
+            ['--search-attribute', "#{key.name}=#{SearchAttributes::IndexedValueType::PROTO_NAMES[key.type]}"]
+          end
+        end
+
         server_options = Internal::Bridge::Testing::EphemeralServer::StartDevServerOptions.new(
           existing_path: dev_server_existing_path,
           sdk_name: 'sdk-ruby',
@@ -83,9 +100,11 @@ module Temporalio
           port:,
           database_filename: dev_server_database_filename,
           ui:,
+          ui_port: ui ? ui_port : nil,
           log_format: dev_server_log_format,
           log_level: dev_server_log_level,
-          extra_args: dev_server_extra_args
+          extra_args: dev_server_extra_args,
+          download_ttl: dev_server_download_ttl
         )
         _with_core_server(
           core_server: Internal::Bridge::Testing::EphemeralServer.start_dev_server(
@@ -122,6 +141,8 @@ module Temporalio
       # @param test_server_download_version [String] Version of test server to download and cache.
       # @param test_server_download_dest_dir [String, nil] Where to download. Defaults to tmp.
       # @param test_server_extra_args [Array<String>] Any extra arguments for the test server.
+      # @param test_server_download_ttl [Float, nil] How long the automatic download should be cached for. If nil,
+      #   cached indefinitely.
       #
       # @yield [environment] If a block is given, it is called with the environment and upon complete the environment is
       #   shutdown.
@@ -140,6 +161,7 @@ module Temporalio
         test_server_download_version: 'default',
         test_server_download_dest_dir: nil,
         test_server_extra_args: [],
+        test_server_download_ttl: nil,
         &
       )
         server_options = Internal::Bridge::Testing::EphemeralServer::StartTestServerOptions.new(
@@ -149,7 +171,8 @@ module Temporalio
           download_version: test_server_download_version,
           download_dest_dir: test_server_download_dest_dir,
           port:,
-          extra_args: test_server_extra_args
+          extra_args: test_server_extra_args,
+          download_ttl: test_server_download_ttl
         )
         _with_core_server(
           core_server: Internal::Bridge::Testing::EphemeralServer.start_test_server(
@@ -299,7 +322,7 @@ module Temporalio
           raise 'Block required' unless block_given?
           return super unless supports_time_skipping?
 
-          already_disabled = @auto_time_skipping
+          already_disabled = !@auto_time_skipping
           @auto_time_skipping = false
           begin
             yield
@@ -371,8 +394,8 @@ module Temporalio
           end
 
           # @!visibility private
-          def result(follow_runs: true, rpc_options: nil)
-            @env.time_skipping_unlocked { super(follow_runs:, rpc_options:) }
+          def result(follow_runs: true, result_hint: nil, rpc_options: nil)
+            @env.time_skipping_unlocked { super(follow_runs:, result_hint:, rpc_options:) }
           end
         end
       end

data/lib/temporalio/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Temporalio
-  VERSION = '0.3.0'
+  VERSION = '0.5.0'
 end

data/lib/temporalio/versioning_override.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+require 'temporalio/worker_deployment_version'
+
+module Temporalio
+  # Base class for version overrides that can be provided in start workflow options.
+  # Used to control the versioning behavior of workflows started with this override.
+  #
+  # WARNING: Experimental API.
+  class VersioningOverride
+    # @!visibility private
+    def _to_proto
+      raise NotImplementedError, 'Subclasses must implement this method'
+    end
+
+    # Represents a versioning override to pin a workflow to a specific version
+    class Pinned < VersioningOverride
+      # The worker deployment version to pin to
+      # @return [WorkerDeploymentVersion]
+      attr_reader :version
+
+      # Create a new pinned versioning override
+      #
+      # @param version [WorkerDeploymentVersion] The worker deployment version to pin to
+      def initialize(version)
+        @version = version
+        super()
+      end
+
+      # TODO: Remove deprecated field setting once removed from server
+
+      # @!visibility private
+      def _to_proto
+        Api::Workflow::V1::VersioningOverride.new(
+          behavior: Api::Enums::V1::VersioningBehavior::VERSIONING_BEHAVIOR_PINNED,
+          pinned_version: @version.to_canonical_string,
+          pinned: Api::Workflow::V1::VersioningOverride::PinnedOverride.new(
+            behavior: Api::Workflow::V1::VersioningOverride::PinnedOverrideBehavior::PINNED_OVERRIDE_BEHAVIOR_PINNED,
+            version: @version._to_proto
+          )
+        )
+      end
+    end
+
+    # Represents a versioning override to auto-upgrade a workflow
+    class AutoUpgrade < VersioningOverride
+      # @!visibility private
+      def _to_proto
+        Api::Workflow::V1::VersioningOverride.new(
+          behavior: Api::Enums::V1::VersioningBehavior::VERSIONING_BEHAVIOR_AUTO_UPGRADE,
+          auto_upgrade: true
+        )
+      end
+    end
+  end
+end

data/lib/temporalio/worker/deployment_options.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'temporalio/common_enums'
+require 'temporalio/worker_deployment_version'
+
+module Temporalio
+  class Worker
+    DeploymentOptions = Data.define(
+      :version,
+      :use_worker_versioning,
+      :default_versioning_behavior
+    )
+
+    # Options for configuring the Worker Versioning feature.
+    #
+    # WARNING: Deployment-based versioning is experimental and APIs may change.
+    #
+    # @!attribute version
+    #   @return [WorkerDeploymentVersion] The worker deployment version.
+    # @!attribute use_worker_versioning
+    #   @return [Boolean] Whether worker versioning is enabled.
+    # @!attribute default_versioning_behavior
+    #   @return [VersioningBehavior] The default versioning behavior.
+    class DeploymentOptions
+      def initialize(
+        version:,
+        use_worker_versioning: false,
+        default_versioning_behavior: VersioningBehavior::UNSPECIFIED
+      )
+        super
+      end
+
+      # @!visibility private
+      def _to_bridge_options
+        Internal::Bridge::Worker::DeploymentOptions.new(
+          version: Internal::Bridge::Worker::WorkerDeploymentVersion.new(
+            deployment_name: version.deployment_name, build_id: version.build_id
+          ),
+          use_worker_versioning: use_worker_versioning,
+          default_versioning_behavior: default_versioning_behavior
+        )
+      end
+    end
+  end
+end

data/lib/temporalio/worker/illegal_workflow_call_validator.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+module Temporalio
+  class Worker
+    # Custom validator for validating illegal workflow calls.
+    class IllegalWorkflowCallValidator
+      CallInfo = Data.define(
+        :class_name,
+        :method_name,
+        :trace_point
+      )
+
+      # Call info passed to the validation block for each validation.
+      #
+      # @!attribute class_name
+      #   @return [String] Class name the method is on.
+      # @!attribute method_name
+      #   @return [String] Method name being called.
+      # @!attribute trace_point
+      #   @return [TracePoint] TracePoint instance for the call.
+      class CallInfo; end # rubocop:disable Lint/EmptyClass
+
+      # @return [Array<IllegalWorkflowCallValidator>] Set of advanced validators for Time calls.
+      def self.default_time_validators
+        @default_time_validators ||= [
+          # Do not consider initialize as invalid if year is present and not "true"
+          IllegalWorkflowCallValidator.new(method_name: :initialize) do |info|
+            year_val = info.trace_point.binding&.local_variable_get(:year)
+            raise 'can only use if passing string or explicit time info' unless year_val && year_val != true
+          end,
+          IllegalWorkflowCallValidator.new(method_name: :now) do
+            # When the xmlschema (aliased as iso8601) call is made, zone_offset is called which has a default parameter
+            # of Time.now.year. We want to prevent failing in that specific case. It is expensive to access the caller
+            # stack, but this is only done in the rare case they are calling this safely.
+            next if caller_locations&.any? { |loc| loc.label == 'zone_offset' || loc.label == 'Time.zone_offset' }
+
+            raise 'Invalid Time.now call'
+          end
+        ]
+      end
+
+      # @return [String, nil] Method name if this validator is specific to a method.
+      attr_reader :method_name
+
+      # @return [Proc] Block provided in constructor to invoke. See constructor for more details.
+      attr_reader :block
+
+      # Create a call validator.
+      #
+      # @param method_name [String, nil] Method name to check. This must be provided if the validator is in an illegal
+      #   call array, this cannot be provided if it is a top-level class validator.
+      # @yield Required block that is called each time validation is needed. If the call raises, the exception message
+      #   is used as the reason why the call is considered invalid. Return value is ignored.
+      # @yieldparam info [CallInfo] Information about the current call.
+      def initialize(method_name: nil, &block)
+        raise 'Block required' unless block_given?
+        raise TypeError, 'Method name must be Symbol' unless method_name.nil? || method_name.is_a?(Symbol)
+
+        @method_name = method_name
+        @block = block
+      end
+    end
+  end
+end