diff options
author | Nils-Helge Garli Hegvik <Nils-Helge.Hegvik@vegvesen.no> | 2015-07-08 12:50:25 +0200 |
---|---|---|
committer | Nils-Helge Garli Hegvik <nilsga@gmail.com> | 2015-07-10 18:27:11 +0200 |
commit | b522f98ab26ec2c97b617392225b84e7912d604a (patch) | |
tree | 4df425a817607977061cb212fdfcb90523227bdf /kamon-core/src/main/scala/kamon/metric | |
parent | da471e354ad1757a989ff40c06bafc1f7e332d17 (diff) | |
download | Kamon-b522f98ab26ec2c97b617392225b84e7912d604a.tar.gz Kamon-b522f98ab26ec2c97b617392225b84e7912d604a.tar.bz2 Kamon-b522f98ab26ec2c97b617392225b84e7912d604a.zip |
! core: cleanup entity on removal
Diffstat (limited to 'kamon-core/src/main/scala/kamon/metric')
-rwxr-xr-x[-rw-r--r--] | kamon-core/src/main/scala/kamon/metric/MetricsModule.scala | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/kamon-core/src/main/scala/kamon/metric/MetricsModule.scala b/kamon-core/src/main/scala/kamon/metric/MetricsModule.scala index ff622443..8b5a6188 100644..100755 --- a/kamon-core/src/main/scala/kamon/metric/MetricsModule.scala +++ b/kamon-core/src/main/scala/kamon/metric/MetricsModule.scala @@ -313,8 +313,11 @@ private[kamon] class MetricsModuleImpl(config: Config) extends MetricsModule { }, _.cleanup).asInstanceOf[T] } - def removeEntity(entity: Entity): Boolean = - _trackedEntities.remove(entity).isDefined + def removeEntity(entity: Entity): Boolean = { + val removedEntity = _trackedEntities.remove(entity) + removedEntity.foreach(_.cleanup) + removedEntity.isDefined + } def find(entity: Entity): Option[EntityRecorder] = _trackedEntities.get(entity) |