Contents

# typed: true

require 'datadog/core/utils/sequence'

module Datadog
  module Core
    module Utils
      # Acts as a unique dictionary of objects
      class ObjectSet
        # You can provide a block that defines how the key
        # for this message type is resolved.
        def initialize(seed = 0, &block)
          @sequence = Sequence.new(seed)
          @items = {}
          @key_block = block
        end

        # Submit an array of arguments that define the message.
        # If they match an existing message, it will return the
        # matching object. If it doesn't match, it will yield to
        # the block with the next ID & args given.
        def fetch(*args)
          # TODO: Array hashing is **really** expensive, we probably want to get rid of it in the future
          key = @key_block ? @key_block.call(*args) : args.hash
          @items[key] ||= yield(@sequence.next, *args)
        end

        def length
          @items.length
        end

        def objects
          @items.values
        end

        def freeze
          super
          @items.freeze
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
ddtrace-1.2.0	lib/datadog/core/utils/object_set.rb
ddtrace-1.1.0	lib/datadog/core/utils/object_set.rb
ddtrace-1.0.0	lib/datadog/core/utils/object_set.rb
ddtrace-1.0.0.beta2	lib/datadog/core/utils/object_set.rb