+ core: provide generic way to scale time and memory metrics

2 files changed, 43 insertions, 0 deletions

diff --git a/kamon-core/src/main/scala/kamon/util/ConfigTools.scala b/kamon-core/src/main/scala/kamon/util/ConfigTools.scala
index bcec22c3..d0665764 100644
--- a/kamon-core/src/main/scala/kamon/util/ConfigTools.scala
+++ b/kamon-core/src/main/scala/kamon/util/ConfigTools.scala
@@ -22,6 +22,8 @@ import com.typesafe.config.Config
 
 import scala.concurrent.duration.FiniteDuration
 
+import kamon.metric.instrument.{ Memory, Time }
+
 object ConfigTools {
   implicit class Syntax(val config: Config) extends AnyVal {
     // We are using the deprecated .getNanoseconds option to keep Kamon source code compatible with
@@ -37,6 +39,10 @@ object ConfigTools {
         case entry ⇒ entry.getKey.takeWhile(_ != '.')
       } toSet
     }
+
+    def time(path: String): Time = Time(config.getString(path))
+
+    def memory(path: String): Memory = Memory(config.getString(path))
   }
 
 }
diff --git a/kamon-core/src/main/scala/kamon/util/NeedToScale.scala b/kamon-core/src/main/scala/kamon/util/NeedToScale.scala
new file mode 100644
index 00000000..19e1ae06
--- /dev/null
+++ b/kamon-core/src/main/scala/kamon/util/NeedToScale.scala
@@ -0,0 +1,37 @@
+/*
+ * =========================================================================================
+ * Copyright © 2013-2015 the kamon project <http://kamon.io/>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+ * except in compliance with the License. You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under the
+ * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied. See the License for the specific language governing permissions
+ * and limitations under the License.
+ * =========================================================================================
+ */
+
+package kamon.util
+
+import com.typesafe.config.Config
+import kamon.metric.instrument.{ Memory, Time }
+import kamon.util.ConfigTools._
+
+object NeedToScale {
+  val TimeUnits = "time-units"
+  val MemoryUnits = "memory-units"
+
+  def unapply(config: Config): Option[(Option[Time], Option[Memory])] = {
+    val scaleTimeTo: Option[Time] =
+      if (config.hasPath(TimeUnits)) Some(config.time(TimeUnits)) else None
+
+    val scaleMemoryTo: Option[Memory] =
+      if (config.hasPath(MemoryUnits)) Some(config.memory(MemoryUnits)) else None
+    if (scaleTimeTo.isDefined || scaleMemoryTo.isDefined) Some(scaleTimeTo -> scaleMemoryTo)
+    else None
+  }
+}
+