From 91575b8db60ca4fc6df9f44fa720929d8c3868ac Mon Sep 17 00:00:00 2001
From: Ivan Topolnjak <ivantopo@gmail.com>
Date: Mon, 25 Feb 2019 00:21:40 +0100
Subject: use TagSet as the implementation for Context tags

---
 .../scala/kamon/instrumentation/HttpServer.scala   | 24 ++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

(limited to 'kamon-core/src/main/scala/kamon/instrumentation/HttpServer.scala')

diff --git a/kamon-core/src/main/scala/kamon/instrumentation/HttpServer.scala b/kamon-core/src/main/scala/kamon/instrumentation/HttpServer.scala
index 659da8aa..68975711 100644
--- a/kamon-core/src/main/scala/kamon/instrumentation/HttpServer.scala
+++ b/kamon-core/src/main/scala/kamon/instrumentation/HttpServer.scala
@@ -6,7 +6,9 @@ import java.time.Duration
 import com.typesafe.config.Config
 import kamon.context.Context
 import kamon.instrumentation.HttpServer.Settings.TagMode
-import kamon.metric.MeasurementUnit.{time, information}
+import kamon.metric.MeasurementUnit.{information, time}
+import kamon.tag.TagSet
+import kamon.tag.Lookups.{any, option}
 import kamon.trace.{IdentityProvider, Span}
 import kamon.util.GlobPathFilter
 import org.slf4j.LoggerFactory
@@ -348,7 +350,7 @@ object HttpServer {
         span.disableMetrics()
 
 
-      for { traceIdTag <- settings.traceIDTag; customTraceID <- context.getTag(traceIdTag) } {
+      for {traceIdTag <- settings.traceIDTagLookup; customTraceID <- context.getTag(traceIdTag) } {
         val identifier = Kamon.identityProvider.traceIdGenerator().from(customTraceID)
         if(identifier != IdentityProvider.NoIdentifier)
           span.withTraceID(identifier)
@@ -361,9 +363,12 @@ object HttpServer {
       }
 
       addRequestTag("http.url", request.url, settings.urlTagMode)
-      addRequestTag("http.method", request.method, settings.urlTagMode)
+      addRequestTag("http.method", request.method, settings.methodTagMode)
       settings.contextTags.foreach {
-        case (tagName, mode) => context.getTag(tagName).foreach(tagValue => addRequestTag(tagName, tagValue, mode))
+        case (tagName, mode) =>
+          val tagValue = context.getTag(any(tagName))
+          if(tagValue != null)
+            addRequestTag(tagName, tagValue.toString, mode)
       }
 
       span.start()
@@ -385,7 +390,7 @@ object HttpServer {
     propagationChannel: String,
     enableServerMetrics: Boolean,
     enableTracing: Boolean,
-    traceIDTag: Option[String],
+    traceIDTagLookup: Option[TagSet.Lookup[Option[String]]],
     enableSpanMetrics: Boolean,
     urlTagMode: TagMode,
     methodTagMode: TagMode,
@@ -424,7 +429,10 @@ object HttpServer {
 
       // Tracing settings
       val enableTracing = config.getBoolean("tracing.enabled")
-      val traceIdTag = Option(config.getString("tracing.preferred-trace-id-tag")).filterNot(_ == "none")
+      val traceIdTagLookup = Option(config.getString("tracing.preferred-trace-id-tag"))
+        .filterNot(_ == "none")
+        .map(option)
+
       val enableSpanMetrics = config.getBoolean("tracing.span-metrics")
       val urlTagMode = TagMode.from(config.getString("tracing.tags.url"))
       val methodTagMode = TagMode.from(config.getString("tracing.tags.method"))
@@ -441,12 +449,12 @@ object HttpServer {
         case (pattern, operationName) => (new GlobPathFilter(pattern), operationName)
       }
 
-      Settings(
+      Settings (
         enablePropagation,
         propagationChannel,
         enableServerMetrics,
         enableTracing,
-        traceIdTag,
+        traceIdTagLookup,
         enableSpanMetrics,
         urlTagMode,
         methodTagMode,
-- 
cgit v1.2.3