updated Hash extensions and specs

This commit is contained in:
2011-03-12 22:21:15 +00:00
parent dcca3556ea
commit 43fc8bc2dd
2 changed files with 30 additions and 18 deletions

View File

@@ -1,21 +1,23 @@
class Hash
def merge_and_incr(hash)
raise ArgumentError unless hash.is_a?(Hash)
hash.each do |key, value|
if value.is_a?(Numeric)
self.set_or_incr(key, value)
else
self[key] = value
end
end
self.clone.merge_and_incr!(hash)
end
def set_or_incr(key, value)
return self unless value.is_a?(Numeric)
self[key] = 0 unless self.has_key?(key)
self[key] += value if self[key].is_a?(Numeric)
def merge_and_incr!(hash)
raise ArgumentError unless hash.is_a?(Hash)
hash.each do |key, value|
self[key] = value unless self.set_or_incr(key, value)
end
self
end
def set_or_incr(key, value)
return false unless value.is_a?(Numeric)
self[key] = 0 unless self.has_key?(key)
return false unless self[key].is_a?(Numeric)
self[key] += value
true
end
end