require "forwardable"

module L2meter
  # This class is a wrapper around Emitter that makes sure that we have a
  # completely separate clone of Emitter per thread running. It doesn't truly
  # make Emitter thread-safe, it makes sure that you don't access the same
  # instance of emitter from different threads.
  class ThreadSafe
    extend Forwardable

    def initialize(emitter)
      @emitter = emitter.freeze
    end

    def_delegators :receiver, \
      :batch,
      :configuration,
      :count,
      :fire!,
      :log,
      :measure,
      :merge!,
      :push_context,
      :sample,
      :silence,
      :silence!,
      :unique,
      :unsilence!,
      :with_elapsed,
      :with_output

    def context(*args, &block)
      value = current_emitter.context(*args, &block)
      Emitter === value ? clone_with_emitter(value) : value
    end

    def disable!
      @disabled = true
    end

    protected

    attr_writer :emitter

    private

    attr_reader :emitter

    def clone_with_emitter(emitter)
      self.class.new(emitter).tap { |ts| ts.disable! if @disabled }
    end

    def receiver
      @disabled ? null_emitter : current_emitter
    end

    def current_emitter
      Thread.current[thread_key] ||= emitter.clone
    end

    def null_emitter
      @null_emitter ||= NullObject.new
    end

    def thread_key
      @thread_key ||= "_l2meter_emitter_#{emitter.object_id}".freeze
    end
  end
end