lib/temporalio/internal/worker/workflow_instance/replay_safe_logger.rb


# frozen_string_literal: true

require 'temporalio/scoped_logger'
require 'temporalio/workflow'

module Temporalio
  module Internal
    module Worker
      class WorkflowInstance
        # Wrapper for a scoped logger that does not log on replay.
        class ReplaySafeLogger < ScopedLogger
          def initialize(logger:, instance:)
            @instance = instance
            @replay_safety_disabled = false
            super(logger)
          end

          def replay_safety_disabled(&)
            @replay_safety_disabled = true
            yield
          ensure
            @replay_safety_disabled = false
          end

          def add(...)
            if !@replay_safety_disabled && Temporalio::Workflow.in_workflow? && Temporalio::Workflow::Unsafe.replaying?
              return true
            end

            # Disable illegal call tracing for the log call
            @instance.illegal_call_tracing_disabled { super }
          end
        end
      end
    end
  end
end