Add clearCounter or clearing a specific counter.

jeffstyr · Sep 19, 2011 · acd7257 · acd7257
1 parent 45344f3
commit acd7257
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/main/scala/com/twitter/ostrich/stats/StatsCollection.scala b/src/main/scala/com/twitter/ostrich/stats/StatsCollection.scala
@@ -134,6 +134,10 @@ class StatsCollection extends StatsProvider with JsonSerializable {
     counter
   }
 
+  def clearCounter(name: String) = {
+      counterMap.remove(name);
+  }
+
   protected def newCounter(name: String) = {
     new Counter()
   }

diff --git a/src/test/scala/com/twitter/ostrich/stats/StatsCollectionSpec.scala b/src/test/scala/com/twitter/ostrich/stats/StatsCollectionSpec.scala
@@ -74,6 +74,14 @@ object StatsCollectionSpec extends Specification {
         collection.incr("widgets", -1)
         collection.getCounters() mustEqual Map("widgets" -> 2)
       }
+
+      "clearCounter" in {
+          collection.getCounter("smellyfeet")
+          collection.incr("smellyfeet", 1)
+          collection.getCounters() mustEqual Map("smellyfeet" -> 1)
+          collection.clearCounter("smellyfeet")
+          collection.getCounters() mustEqual Map()
+      }
     }
 
     "metrics" in {