aboutsummaryrefslogtreecommitdiff
path: root/kamon-core/src/main/scala/kamon/util/EnvironmentTagBuilder.scala
blob: 09e643e9430665184ba1ca92753b976973731e9c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package kamon.util

import com.typesafe.config.Config
import kamon.{Environment, Kamon}

import scala.collection.JavaConverters._


/**
  *   Utility class to create a Map[String, String] encoding all the Environment information based on the provided
  *   Config. The Config instance is expected to have the following members:
  *
  *     - service:          Boolean. If true a service tag is included.
  *     - host:             Boolean. If true a host tag is included.
  *     - instance:         Boolean. If true a instance tag is included.
  *     - blacklisted-tags: List[String]. List of Environment tags that should not be included in the result.
  *
  *   This utility is meant to be used mostly by reporter modules.
  *
  */
object EnvironmentTagBuilder {

  def create(config: Config): Map[String, String] =
    create(Kamon.environment, config)

  def create(env: Environment, config: Config): Map[String, String] = {
    val envTags = Map.newBuilder[String, String]

    if(config.getBoolean("service"))
      envTags += ("service" -> env.service)

    if(config.getBoolean("host"))
      envTags += ("host" -> env.host)

    if(config.getBoolean("instance"))
      envTags += ("instance" -> env.instance)

    val blacklistedTags = config.getStringList("blacklisted-tags").asScala
    env.tags.filterKeys(k => !blacklistedTags.contains(k)).foreach(p => envTags += p)

    envTags.result()
  }
}