From bca24e3927d924a1cebc01903788d77f0239ffa9 Mon Sep 17 00:00:00 2001 From: Diego Date: Wed, 2 Apr 2014 17:58:32 -0300 Subject: simple refactor --- .../kamon/statsd/client/StatsdMetricsSender.scala | 23 ++++++++-------------- 1 file changed, 8 insertions(+), 15 deletions(-) diff --git a/kamon-statsd/src/main/scala/kamon/statsd/client/StatsdMetricsSender.scala b/kamon-statsd/src/main/scala/kamon/statsd/client/StatsdMetricsSender.scala index 8ee08420..d9173e45 100644 --- a/kamon-statsd/src/main/scala/kamon/statsd/client/StatsdMetricsSender.scala +++ b/kamon-statsd/src/main/scala/kamon/statsd/client/StatsdMetricsSender.scala @@ -22,10 +22,8 @@ import java.net.InetSocketAddress import akka.util.ByteString class StatsdMetricsSender(statPrefix:String, remote: InetSocketAddress) extends Actor with ActorLogging { - import context.system - import StatsdMetricsSender._ - import StatsDProtocol._ + import context.system IO(Udp) ! Udp.SimpleSender @@ -43,9 +41,13 @@ class StatsdMetricsSender(statPrefix:String, remote: InetSocketAddress) extends } object StatsdMetricsSender { - import StatsDProtocol._ - def props(statPrefix:String, remote: InetSocketAddress): Props = Props(new StatsdMetricsSender(statPrefix, remote)) + sealed trait StatsdMetric + case class Counter(key: String, value: Long = 1, suffix:String = "c", samplingRate: Double = 1.0) extends StatsdMetric + case class Timing(key: String, millis: Long, suffix:String = "ms", samplingRate: Double = 1.0) extends StatsdMetric + case class Gauge(key: String, value: Long, suffix:String = "g", samplingRate: Double = 1.0) extends StatsdMetric + + def props(statPrefix:String, remote: InetSocketAddress): Props = Props(new StatsdMetricsSender(statPrefix, remote)) def toByteString(statPrefix:String, metric:StatsdMetric) : ByteString = metric match { case Counter(key, value, suffix, samplingRate) => statFor(statPrefix, key, value, suffix, samplingRate) @@ -65,13 +67,4 @@ object StatsdMetricsSender { case _ => ByteString(s"${statPrefix}.${key}:${value}|${suffix}|@$samplingRate") } } -} - - -object StatsDProtocol { - - sealed trait StatsdMetric - case class Counter(key: String, value: Long = 1, suffix:String = "c", samplingRate: Double = 1.0) extends StatsdMetric - case class Timing(key: String, millis: Long, suffix:String = "ms", samplingRate: Double = 1.0) extends StatsdMetric - case class Gauge(key: String, value: Long, suffix:String = "g", samplingRate: Double = 1.0) extends StatsdMetric -} +} \ No newline at end of file -- cgit v1.2.3