Renamed package to xyz, New formatting, authorize directive supporting multiple permissions

1 files changed, 97 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/core/stats.scala b/src/main/scala/xyz/driver/core/stats.scala
new file mode 100644
index 0000000..5759012
--- /dev/null
+++ b/src/main/scala/xyz/driver/core/stats.scala
@@ -0,0 +1,97 @@
+package xyz.driver.core
+
+import java.io.File
+import java.lang.management.ManagementFactory
+import java.lang.reflect.Modifier
+
+import xyz.driver.core.logging.Logger
+import xyz.driver.core.time.{Time, TimeRange}
+
+object stats {
+
+  type StatsKey  = String
+  type StatsKeys = Seq[StatsKey]
+
+  trait Stats {
+
+    def recordStats(keys: StatsKeys, interval: TimeRange, value: BigDecimal): Unit
+
+    def recordStats(keys: StatsKeys, interval: TimeRange, value: Int): Unit =
+      recordStats(keys, interval, BigDecimal(value))
+
+    def recordStats(key: StatsKey, interval: TimeRange, value: BigDecimal): Unit =
+      recordStats(Vector(key), interval, value)
+
+    def recordStats(key: StatsKey, interval: TimeRange, value: Int): Unit =
+      recordStats(Vector(key), interval, BigDecimal(value))
+
+    def recordStats(keys: StatsKeys, time: Time, value: BigDecimal): Unit =
+      recordStats(keys, TimeRange(time, time), value)
+
+    def recordStats(keys: StatsKeys, time: Time, value: Int): Unit =
+      recordStats(keys, TimeRange(time, time), BigDecimal(value))
+
+    def recordStats(key: StatsKey, time: Time, value: BigDecimal): Unit =
+      recordStats(Vector(key), TimeRange(time, time), value)
+
+    def recordStats(key: StatsKey, time: Time, value: Int): Unit =
+      recordStats(Vector(key), TimeRange(time, time), BigDecimal(value))
+  }
+
+  class LogStats(log: Logger) extends Stats {
+    def recordStats(keys: StatsKeys, interval: TimeRange, value: BigDecimal): Unit = {
+      val valueString = value.bigDecimal.toPlainString
+      log.audit(s"${keys.mkString(".")}(${interval.start.millis}-${interval.end.millis})=$valueString")
+    }
+  }
+
+  final case class MemoryStats(free: Long, total: Long, max: Long)
+
+  final case class GarbageCollectorStats(totalGarbageCollections: Long, garbageCollectionTime: Long)
+
+  final case class FileRootSpace(path: String, totalSpace: Long, freeSpace: Long, usableSpace: Long)
+
+  object SystemStats {
+
+    def memoryUsage: MemoryStats = {
+      val runtime = Runtime.getRuntime
+      MemoryStats(runtime.freeMemory, runtime.totalMemory, runtime.maxMemory)
+    }
+
+    def availableProcessors: Int = {
+      Runtime.getRuntime.availableProcessors()
+    }
+
+    def garbageCollectorStats: GarbageCollectorStats = {
+      import scala.collection.JavaConverters._
+
+      val (totalGarbageCollections, garbageCollectionTime) =
+        ManagementFactory.getGarbageCollectorMXBeans.asScala.foldLeft(0L -> 0L) {
+          case ((total, collectionTime), gc) =>
+            (total + math.max(0L, gc.getCollectionCount)) -> (collectionTime + math.max(0L, gc.getCollectionTime))
+        }
+
+      GarbageCollectorStats(totalGarbageCollections, garbageCollectionTime)
+    }
+
+    def fileSystemSpace: Array[FileRootSpace] = {
+      File.listRoots() map { root =>
+        FileRootSpace(root.getAbsolutePath, root.getTotalSpace, root.getFreeSpace, root.getUsableSpace)
+      }
+    }
+
+    def operatingSystemStats: Map[String, String] = {
+      val operatingSystemMXBean = ManagementFactory.getOperatingSystemMXBean
+      operatingSystemMXBean.getClass.getDeclaredMethods
+        .map(method => { method.setAccessible(true); method })
+        .filter(method => method.getName.startsWith("get") && Modifier.isPublic(method.getModifiers))
+        .map { method =>
+          try {
+            method.getName -> String.valueOf(method.invoke(operatingSystemMXBean))
+          } catch {
+            case t: Throwable => method.getName -> t.getMessage
+        }
+      } toMap
+    }
+  }
+}