dev.mongocamp.micrometer.mongodb.registry.MongoRegistryConfig.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of micrometer-mongodb_2.13 Show documentation

micrometer-mongodb

The newest version!

package dev.mongocamp.micrometer.mongodb.registry

import com.typesafe.config.{ Config, ConfigFactory }
import dev.mongocamp.driver.mongodb.MongoDAO
import dev.mongocamp.micrometer.mongodb.MetricsCache
import io.micrometer.core.instrument.step.StepRegistryConfig
import org.mongodb.scala.Document

import scala.concurrent.duration.Duration

case class MongoRegistryConfig(mongoDAO: MongoDAO[Document], configurationMap: Map[String, String] = Map()) extends StepRegistryConfig {

  private lazy val conf: Config = ConfigFactory.load()

  override def prefix(): String = "dev.mongocamp.micrometer.mongodb"

  override def get(key: String): String = {
    val value = loadConfigValue(key).orNull
    if (key.equalsIgnoreCase(s"${prefix()}.step") && value != null) {
      MetricsCache.updateCacheTime(Duration(value))
    }
    value
  }

  private def loadConfigValue(key: String): Option[String] = {
    try {
      val mapKey = key.replace(s"$prefix.", "")
      (configurationMap.get(mapKey) ++ configurationMap.get(key)).foreach(v => return Some(v))
      Option(conf.getValue(key).render().replace("\"", ""))
    }
    catch {
      case _: Exception =>
        None
    }
  }

}