• Home
• com.codemettle
• netflow-stream-lib_2.12
• 1.0.2-AKKA25
• source code
• NodeConfig.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.netflow.lib.NodeConfig.scala Maven / Gradle / Ivy

package io.netflow
package lib

import akka.actor.ActorSystem

private[netflow] object NodeConfig {

  case class ServerConfig(netflow: NetFlowConfig)

  case class NetFlowConfig(calculateSamples: Boolean, extraFields: Boolean, parsingThreads: Option[Int])

  private var configs = Map.empty[ActorSystem, ServerConfig]
  private def config(implicit system: ActorSystem): ServerConfig =
    configs.getOrElse(system, {
      val conf = load
      configs += (system -> conf)
      conf
    })

  implicit private class RichActorSystem(val system: ActorSystem) extends AnyVal {
    private def conf = system.settings.config

    def getBool(name: String): Option[Boolean] = Tryo(conf.getBoolean(name))
    def getBool(name: String, fallback: Boolean): Boolean = getBool(name) getOrElse fallback
    def getInt(name: String): Option[Int] = Tryo(conf.getInt(name))
  }

  private def Config(implicit system: ActorSystem) = new RichActorSystem(system)

  private def load(implicit system: ActorSystem): ServerConfig = {

    val netflow = NetFlowConfig(
      calculateSamples = Config.getBool("netflow.calculateSamples", fallback = false),
      extraFields = Config.getBool("netflow.extraFields", fallback = true),
      parsingThreads = Config.getInt("netflow.parsingThreads")
    )

    val server = ServerConfig(netflow = netflow)

    server
  }

  def values(implicit system: ActorSystem): ServerConfig = config

}