diff options
author | Ivan Topolnjak <ivantopo@gmail.com> | 2015-07-10 19:48:40 +0200 |
---|---|---|
committer | Ivan Topolnjak <ivantopo@gmail.com> | 2015-07-10 19:48:40 +0200 |
commit | d001fe5dbc2f051e71007b26e57699c404044e45 (patch) | |
tree | 6c3d84768ec7fc8f92d3370a357aa956782b6cf0 | |
parent | ea0da37b4a9f769b498fde269dc28b6bad06ab64 (diff) | |
parent | 8f23d7e85e2ef8739159a4fb895a4a6f8c4eedef (diff) | |
download | Kamon-d001fe5dbc2f051e71007b26e57699c404044e45.tar.gz Kamon-d001fe5dbc2f051e71007b26e57699c404044e45.tar.bz2 Kamon-d001fe5dbc2f051e71007b26e57699c404044e45.zip |
Merge pull request #229 from nilsga/master
! 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) |