aboutsummaryrefslogtreecommitdiff
path: root/kamon-core/src/main/scala/kamon/metric/EntityFilter.scala
blob: cf203609bcd732d0c1f232d59c5b638644abc775 (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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package kamon
package metric

import java.util.regex.Pattern
import com.typesafe.config.Config

object EntityFilter {
  def fromConfig(config: Config): EntityFilter = {
    val filtersConfig = config.getConfig("kamon.metric.filters")
    val acceptUnmatched = filtersConfig.getBoolean("accept-unmatched")

    val perCategoryFilters = filtersConfig.firstLevelKeys.filter(_ != "accept-unmatched") map { category: String 
      val includes = readFilters(filtersConfig, s"$category.includes")
      val excludes = readFilters(filtersConfig, s"$category.excludes")

      (category, new IncludeExcludeNameFilter(includes, excludes, acceptUnmatched))
    } toMap

    new EntityFilter(perCategoryFilters, acceptUnmatched)
  }

  private def readFilters(filtersConfig: Config, name: String): Seq[NameFilter] = {
    import scala.collection.JavaConverters._
    if(filtersConfig.hasPath(name))
      filtersConfig.getStringList(name).asScala.map(readNameFilter)
    else
      Seq.empty
  }

  private def readNameFilter(pattern: String): NameFilter = {
    if(pattern.startsWith("regex:"))
      new RegexNameFilter(pattern.drop(6))
    else if(pattern.startsWith("glob:"))
      new GlobPathFilter(pattern.drop(5))
    else
      new GlobPathFilter(pattern)
  }
}

class EntityFilter(perCategoryFilters: Map[String, NameFilter], acceptUnmatched: Boolean) {
  def accept(entity: Entity): Boolean =
    perCategoryFilters
      .get(entity.category)
      .map(_.accept(entity.name))
      .getOrElse(acceptUnmatched)
}

trait NameFilter {
  def accept(name: String): Boolean
}

class IncludeExcludeNameFilter(includes: Seq[NameFilter], excludes: Seq[NameFilter], acceptUnmatched: Boolean) extends NameFilter {
  override def accept(name: String): Boolean =
    (includes.exists(_.accept(name)) || acceptUnmatched) && !excludes.exists(_.accept(name))
}

class RegexNameFilter(path: String) extends NameFilter {
  private val pathRegex = path.r

  override def accept(path: String): Boolean = path match {
    case pathRegex(_*)  true
    case _              false
  }
}

class GlobPathFilter(glob: String) extends NameFilter {
  private val globPattern = Pattern.compile("(\\*\\*?)|(\\?)|(\\\\.)|(/+)|([^*?]+)")
  private val compiledPattern = getGlobPattern(glob)

  override def accept(name: String): Boolean =
    compiledPattern.matcher(name).matches()

  private def getGlobPattern(glob: String) = {
    val patternBuilder = new StringBuilder
    val matcher = globPattern.matcher(glob)
    while (matcher.find()) {
      val (grp1, grp2, grp3, grp4) = (matcher.group(1), matcher.group(2), matcher.group(3), matcher.group(4))
      if (grp1 != null) {
        // match a * or **
        if (grp1.length == 2) {
          // it's a *workers are able to process multiple metrics*
          patternBuilder.append(".*")
        }
        else {
          // it's a *
          patternBuilder.append("[^/]*")
        }
      }
      else if (grp2 != null) {
        // match a '?' glob pattern; any non-slash character
        patternBuilder.append("[^/]")
      }
      else if (grp3 != null) {
        // backslash-escaped value
        patternBuilder.append(Pattern.quote(grp3.substring(1)))
      }
      else if (grp4 != null) {
        // match any number of / chars
        patternBuilder.append("/+")
      }
      else {
        // some other string
        patternBuilder.append(Pattern.quote(matcher.group))
      }
    }

    Pattern.compile(patternBuilder.toString)
  }
}