WIP

Author: jakubjanecek

build.sbt

@@ -32,6 +32,7 @@ lazy val root = project
     jvmPureConfig,
     micrometerJmx,
     micrometerJmxPureConfig,
+    micrometerStatsD,
     pureConfig
   )
   .settings(
@@ -171,6 +172,14 @@ lazy val micrometerJmxPureConfig = project
   .settings(commonSettings)
   .settings(name := "sst-micrometer-jmx-pureconfig")
 
+lazy val micrometerStatsD = project
+  .in(file("micrometer-statsd"))
+  .settings(commonSettings)
+  .settings(
+    name := "sst-micrometer-statsd",
+    libraryDependencies += Dependencies.micrometerStatsD
+  )
+
 lazy val pureConfig = project
   .in(file("pureconfig"))
   .settings(commonSettings)

micrometer-jmx/src/main/scala/com/avast/sst/micrometer/jmx/MicrometerJmxConfig.scala

@@ -1,3 +1,9 @@
 package com.avast.sst.micrometer.jmx
 
-final case class MicrometerJmxConfig(domain: String, enableTypeScopeNameHierarchy: Boolean = false)
+import java.util.concurrent.TimeUnit
+
+import scala.concurrent.duration.Duration
+
+final case class MicrometerJmxConfig(domain: String,
+                                     enableTypeScopeNameHierarchy: Boolean = false,
+                                     step: Duration = Duration(1, TimeUnit.MINUTES))

micrometer-jmx/src/main/scala/com/avast/sst/micrometer/jmx/MicrometerJmxModule.scala

@@ -1,5 +1,7 @@
 package com.avast.sst.micrometer.jmx
 
+import java.time.Duration
+
 import cats.effect.{Resource, Sync}
 import com.codahale.metrics.MetricRegistry
 import com.codahale.metrics.jmx.JmxReporter
@@ -21,7 +23,7 @@ object MicrometerJmxModule {
           if (config.enableTypeScopeNameHierarchy) {
             val dropwizardRegistry = new MetricRegistry
             val registry = new JmxMeterRegistry(
-              new DomainJmxConfig(config.domain),
+              new CustomJmxConfig(config),
               Clock.SYSTEM,
               HierarchicalNameMapper.DEFAULT,
               dropwizardRegistry,
@@ -30,7 +32,7 @@ object MicrometerJmxModule {
             registry.config.namingConvention(NamingConvention.dot)
             registry
           } else {
-            new JmxMeterRegistry(new DomainJmxConfig(config.domain), Clock.SYSTEM)
+            new JmxMeterRegistry(new CustomJmxConfig(config), Clock.SYSTEM)
           }
         }
       }(registry => Sync[F].delay(registry.close()))
@@ -44,9 +46,12 @@ object MicrometerJmxModule {
       .build
   }
 
-  private class DomainJmxConfig(override val domain: String) extends JmxConfig {
+  private class CustomJmxConfig(c: MicrometerJmxConfig) extends JmxConfig {
+
+    override val domain: String = c.domain
+    override val step: Duration = Duration.ofMillis(c.step.toMillis)
 
-    // implements MeterRegistryConfig.get which can return null according to JavaDoc and @Nullable annotation
+    // the method is @Nullable and we don't need to implement it here
     @SuppressWarnings(Array("org.wartremover.warts.Null"))
     override def get(key: String): String = null
 

micrometer-statsd/src/main/scala/com/avast/sst/micrometer/statsd/MicrometerStatsDConfig.scala

@@ -0,0 +1,42 @@
+package com.avast.sst.micrometer.statsd
+
+import java.util.concurrent.TimeUnit
+
+import com.avast.sst.micrometer.statsd.MicrometerStatsDConfig.{Flavor, Protocol}
+
+import scala.concurrent.duration.Duration
+
+final case class MicrometerStatsDConfig(host: String,
+                                        port: Int = 8125,
+                                        flavor: Flavor = Flavor.Etsy,
+                                        enabled: Boolean = true,
+                                        protocol: Protocol = Protocol.Udp,
+                                        maxPacketLength: Int = 1400,
+                                        pollingFrequency: Duration = Duration(10, TimeUnit.SECONDS),
+                                        step: Duration = Duration(1, TimeUnit.MINUTES),
+                                        publishUnchangedMeters: Boolean = true,
+                                        buffered: Boolean = true)
+
+object MicrometerStatsDConfig {
+
+  sealed trait Flavor
+
+  object Flavor {
+
+    case object Etsy extends Flavor
+    case object Datadog extends Flavor
+    case object Telegraf extends Flavor
+    case object Sysdig extends Flavor
+
+  }
+
+  sealed trait Protocol
+
+  object Protocol {
+
+    case object Udp extends Protocol
+    case object Tcp extends Protocol
+
+  }
+
+}

micrometer-statsd/src/main/scala/com/avast/sst/micrometer/statsd/MicrometerStatsDModule.scala

@@ -0,0 +1,64 @@
+package com.avast.sst.micrometer.statsd
+
+import java.time.{Clock, Duration}
+
+import cats.effect.{Resource, Sync}
+import com.avast.sst.micrometer.statsd.MicrometerStatsDConfig.{Flavor, Protocol}
+import io.micrometer.core.instrument.config.NamingConvention
+import io.micrometer.core.instrument.util.HierarchicalNameMapper
+import io.micrometer.statsd.{StatsdConfig, StatsdFlavor, StatsdMeterRegistry, StatsdProtocol}
+
+import scala.language.higherKinds
+
+object MicrometerStatsDModule {
+
+  /** Makes configured [[io.micrometer.statsd.StatsdMeterRegistry]]. */
+  def make[F[_]: Sync](config: MicrometerStatsDConfig, clock: Clock): Resource[F, StatsdMeterRegistry] = {
+    Resource
+      .make {
+        Sync[F].delay {
+          StatsdMeterRegistry
+            .builder(new CustomStatsdConfig(config))
+            .
+          new StatsdMeterRegistry(???, ???)
+        }
+      }(registry => Sync[F].delay(registry.close()))
+  }
+
+  private class CustomStatsdConfig(c: MicrometerStatsDConfig) extends StatsdConfig {
+
+    override val flavor: StatsdFlavor = c.flavor match {
+      case Flavor.Etsy     => StatsdFlavor.ETSY
+      case Flavor.Datadog  => StatsdFlavor.DATADOG
+      case Flavor.Telegraf => StatsdFlavor.TELEGRAF
+      case Flavor.Sysdig   => StatsdFlavor.SYSDIG
+    }
+
+    override val enabled: Boolean = c.enabled
+
+    override val host: String = c.host
+
+    override val port: Int = c.port
+
+    override val protocol: StatsdProtocol = c.protocol match {
+      case Protocol.Udp => StatsdProtocol.UDP
+      case Protocol.Tcp => StatsdProtocol.TCP
+    }
+
+    override val maxPacketLength: Int = c.maxPacketLength
+
+    override val pollingFrequency: Duration = java.time.Duration.ofMillis(c.pollingFrequency.toMillis)
+
+    override val step: Duration = java.time.Duration.ofMillis(c.step.toMillis)
+
+    override val publishUnchangedMeters: Boolean = c.publishUnchangedMeters
+
+    override val buffered: Boolean = c.buffered
+
+    // the method is @Nullable and we don't need to implement it here
+    @SuppressWarnings(Array("org.wartremover.warts.Null"))
+    override def get(key: String): String = null
+
+  }
+
+}

project/Dependencies.scala

@@ -12,6 +12,7 @@ object Dependencies {
   val logbackClassic = "ch.qos.logback" % "logback-classic" % "1.2.3"
   val micrometerCore = "io.micrometer" % "micrometer-core" % Versions.micrometer
   val micrometerJmx = "io.micrometer" % "micrometer-registry-jmx" % Versions.micrometer
+  val micrometerStatsD = "io.micrometer" % "micrometer-registry-statsd" % Versions.micrometer
   val monixEval = "io.monix" %% "monix-eval" % "3.0.0"
   val pureConfig = "com.github.pureconfig" %% "pureconfig" % "0.12.1"
   val scalaTest = "org.scalatest" %% "scalatest" % "3.0.8"