Merge branch 'rmstat'

jeffstyr · Sep 20, 2011 · 1f891e0 · 1f891e0
2 parents 45344f3 + fd867cb
commit 1f891e0
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 removeCounter(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.removeCounter("smellyfeet")
+        collection.getCounters() mustEqual Map()
+      }
     }
 
     "metrics" in {