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 | 8f23d7e85e2ef8739159a4fb895a4a6f8c4eedef (patch) | |
| tree | 6c3d84768ec7fc8f92d3370a357aa956782b6cf0 | |
| parent | ea0da37b4a9f769b498fde269dc28b6bad06ab64 (diff) | |
| download | Kamon-8f23d7e85e2ef8739159a4fb895a4a6f8c4eedef.tar.gz Kamon-8f23d7e85e2ef8739159a4fb895a4a6f8c4eedef.tar.bz2 Kamon-8f23d7e85e2ef8739159a4fb895a4a6f8c4eedef.zip | |
! core: cleanup entity on removal
| -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) |