From 94b5e7640bfda860d2b0d6818a38b7ec395a530d Mon Sep 17 00:00:00 2001 From: Ivan Topolnjak Date: Wed, 6 Dec 2017 12:37:53 +0100 Subject: allow enabling/disabling of metrics on a Span --- kamon-core/src/main/scala/kamon/trace/Span.scala | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) (limited to 'kamon-core/src/main/scala/kamon/trace/Span.scala') diff --git a/kamon-core/src/main/scala/kamon/trace/Span.scala b/kamon-core/src/main/scala/kamon/trace/Span.scala index ceb1f56c..7410523b 100644 --- a/kamon-core/src/main/scala/kamon/trace/Span.scala +++ b/kamon-core/src/main/scala/kamon/trace/Span.scala @@ -51,7 +51,9 @@ sealed abstract class Span { def setOperationName(name: String): Span - def disableMetricsCollection(): Span + def enableMetrics(): Span + + def disableMetrics(): Span def finish(finishTimestampMicros: Long): Unit @@ -77,7 +79,8 @@ object Span { override def addError(error: String): Span = this override def addError(error: String, throwable: Throwable): Span = this override def setOperationName(name: String): Span = this - override def disableMetricsCollection(): Span = this + override def enableMetrics(): Span = this + override def disableMetrics(): Span = this override def finish(finishTimestampMicros: Long): Unit = {} } @@ -166,7 +169,12 @@ object Span { this } - override def disableMetricsCollection(): Span = synchronized { + override def enableMetrics(): Span = synchronized { + collectMetrics = true + this + } + + override def disableMetrics(): Span = synchronized { collectMetrics = false this } @@ -231,7 +239,8 @@ object Span { override def addError(error: String): Span = this override def addError(error: String, throwable: Throwable): Span = this override def setOperationName(name: String): Span = this - override def disableMetricsCollection(): Span = this + override def enableMetrics(): Span = this + override def disableMetrics(): Span = this override def finish(finishTimestampMicros: Long): Unit = {} } -- cgit v1.2.3