Merge pull request #200 from vilunov/cats-effect-2-prefixed

Prefixed registries

Author: t3hnar

build.sbt

@@ -19,7 +19,7 @@ lazy val root = (project
   settings commonSettings
   settings (
     publish / skip := true,
-    name := "smetrics")
+    name := "smetrics-parent")
   aggregate(smetrics, prometheus, http4s, doobie, kafka))
 
 lazy val smetrics = (project
@@ -68,4 +68,4 @@ lazy val doobie = (project
   dependsOn smetrics % "compile->compile;test->test"
   settings(
   name := "smetrics-doobie",
-  libraryDependencies += Dependencies.doobie))
+  libraryDependencies += Dependencies.doobie))

smetrics/src/main/scala/com/evolutiongaming/smetrics/CollectorRegistry.scala

@@ -68,6 +68,8 @@ trait CollectorRegistry[F[_]] {
     labels: A)(implicit
     magnet: LabelsMagnetInitialized[A, B]
   ): Resource[F, B[Histogram[F]]]
+
+  def prefixed(prefix: String): CollectorRegistry[F] = new CollectorRegistry.Prefixed[F](this, prefix)
 }
 
 object CollectorRegistry {
@@ -183,4 +185,73 @@ object CollectorRegistry {
       }
     }
   }
+
+  private class Prefixed[F[_]](private val delegate: CollectorRegistry[F], private val prefix: String)
+    extends CollectorRegistry[F] {
+
+    private def prefixedName(name: String): String = s"${prefix}_$name"
+
+    def gauge[A, B[_]](
+      name: String,
+      help: String,
+      labels: A)(implicit
+      magnet: LabelsMagnet[A, B]
+    ): Resource[F, B[Gauge[F]]] = delegate.gauge(prefixedName(name), help, labels)
+
+    def gaugeInitialized[A, B[_]](
+      name: String,
+      help: String,
+      labels: A)(implicit
+      magnet: LabelsMagnetInitialized[A, B]
+    ): Resource[F, B[Gauge[F]]] = delegate.gaugeInitialized(prefixedName(name), help, labels)
+
+
+    def counter[A, B[_]](
+      name: String,
+      help: String,
+      labels: A)(implicit
+      magnet: LabelsMagnet[A, B]
+    ): Resource[F, B[Counter[F]]] = delegate.counter(prefixedName(name), help, labels)
+
+    def counterInitialized[A, B[_]](
+      name: String,
+      help: String,
+      labels: A)(implicit
+      magnet: LabelsMagnetInitialized[A, B]
+    ): Resource[F, B[Counter[F]]] = delegate.counterInitialized(prefixedName(name), help, labels)
+
+
+    def summary[A, B[_]](
+      name: String,
+      help: String,
+      quantiles: Quantiles,
+      labels: A)(implicit
+      magnet: LabelsMagnet[A, B]
+    ): Resource[F, B[Summary[F]]] = delegate.summary(prefixedName(name), help, quantiles, labels)
+
+    def summaryInitialized[A, B[_]](
+      name: String,
+      help: String,
+      quantiles: Quantiles,
+      labels: A)(implicit
+      magnet: LabelsMagnetInitialized[A, B]
+    ): Resource[F, B[Summary[F]]] = delegate.summaryInitialized(prefixedName(name), help, quantiles, labels)
+
+
+    def histogram[A, B[_]](
+      name: String,
+      help: String,
+      buckets: Buckets,
+      labels: A)(implicit
+      magnet: LabelsMagnet[A, B]
+    ): Resource[F, B[Histogram[F]]] = delegate.histogram(prefixedName(name), help, buckets, labels)
+
+    def histogramInitialized[A, B[_]](
+      name: String,
+      help: String,
+      buckets: Buckets,
+      labels: A)(implicit
+      magnet: LabelsMagnetInitialized[A, B]
+    ): Resource[F, B[Histogram[F]]] = delegate.histogramInitialized(prefixedName(name), help, buckets, labels)
+  }
 }