Sha256: 0a02b1a0ade74ee91bed2e6c96295c0931fc4c90d03db6dace41720a61f6f561

Contents?: true

Size: 1.46 KB

Versions: 2

Compression:

Stored size: 1.46 KB

Contents

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

    EMITTER_METHODS = %i[
      batch
      configuration
      count
      fire!
      log
      measure
      merge!
      push_context
      sample
      silence
      silence!
      unique
      unsilence!
      with_elapsed
      with_output
    ]

    private_constant :EMITTER_METHODS

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

    def_delegators :receiver, *EMITTER_METHODS

    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

Version data entries

2 entries across 2 versions & 1 rubygems

Version Path
l2meter-0.9.0 lib/l2meter/thread_safe.rb
l2meter-0.8.0 lib/l2meter/thread_safe.rb