extend ::Hash with #set_or_incr and #merge_and_incr methods

lib/redistat.rb

@@ -26,10 +26,7 @@ require 'redistat/scope'
 require 'redistat/summary'
 require 'redistat/version'
 
-require 'redistat/core_ext/date'
-require 'redistat/core_ext/time'
-require 'redistat/core_ext/fixnum'
-require 'redistat/core_ext/bignum'
+require 'redistat/core_ext'
 
 module Redistat
   

lib/redistat/core_ext.rb

@@ -0,0 +1,5 @@
+require 'redistat/core_ext/bignum'
+require 'redistat/core_ext/date'
+require 'redistat/core_ext/fixnum'
+require 'redistat/core_ext/hash'
+require 'redistat/core_ext/time'

lib/redistat/core_ext/hash.rb

@@ -0,0 +1,21 @@
+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
+  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)
+    self
+  end
+  
+end

lib/redistat/result.rb

@@ -12,12 +12,5 @@ module Redistat
       @till = options[:till] ||= nil
     end
     
-    
-    def set_or_incr(key, value)
-      self[key] = 0 if !self.has_key?(key)
-      self[key] += value
-      self
-    end
-    
   end
-end
+end