Contents

# Copyright (c) 2015 Sqreen. All Rights Reserved.
# Please refer to our terms for more information: https://www.sqreen.com/terms.html

require 'sqreen/util'

class Sqreen::Util::CappedHash < Hash
  attr_reader :size_cap, :depth_cap

  def initialize(*args, &block)
    opts = args.last.is_a?(Hash) ? args.pop : {}
    size_cap = opts[:size_cap] || 150
    depth_cap = opts[:depth_cap] || 10
    @size_cap = size_cap
    @depth_cap = depth_cap

    super(*args, &block)
  end

  def []=(key, value)
    super if key?(key) || keep?(value)
  end
  alias_method :store, :[]=

  def merge!(h)
    h.each { |k, v| self[k] = block_given? ? yield(k, self[k], v) : v }
  end
  alias_method :update, :merge!

  def replace(h)
    keep_if { false }
    merge!(h)
  end

  private

  def keep?(value)
    size < size_cap && (depth_cap > 0 || !value.is_a?(Hash) && !value.is_a?(Array))
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
sqreen-1.18.6-java	lib/sqreen/util/capped_hash.rb
sqreen-1.18.6	lib/sqreen/util/capped_hash.rb
sqreen-1.18.5-java	lib/sqreen/util/capped_hash.rb
sqreen-1.18.5	lib/sqreen/util/capped_hash.rb
sqreen-1.18.4-java	lib/sqreen/util/capped_hash.rb
sqreen-1.18.4	lib/sqreen/util/capped_hash.rb
sqreen-1.18.3-java	lib/sqreen/util/capped_hash.rb
sqreen-1.18.3	lib/sqreen/util/capped_hash.rb