Sha256: 624103318bf1ce7c3177db376f8b59707c3b137b69ddde6ab0fa4ac0a8dfd8a3

Contents?: true

Size: 1.4 KB

Versions: 4

Compression:

Stored size: 1.4 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
      log
      measure
      push_context
      sample
      silence
      silence!
      unique
      unsilence!
      with_elapsed
    ]

    private_constant :EMITTER_METHODS

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

    def_delegators :receiver, *EMITTER_METHODS

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

    def disable!
      @disabled = true
    end

    protected

    attr_writer :emitter

    private

    attr_reader :emitter

    def clone_with_emitter(emitter)
      clone.tap do |ts|
        ts.emitter = emitter
      end
    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

4 entries across 4 versions & 1 rubygems

Version Path
l2meter-0.6.0 lib/l2meter/thread_safe.rb
l2meter-0.5.1 lib/l2meter/thread_safe.rb
l2meter-0.5.0 lib/l2meter/thread_safe.rb
l2meter-0.4.1 lib/l2meter/thread_safe.rb