pl.touk.nussknacker.engine.requestresponse.metrics.InvocationMetrics.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nussknacker-request-response-runtime_2.13 Show documentation

nussknacker-request-response-runtime

There is a newer version: 1.18.0

Show newest version

package pl.touk.nussknacker.engine.requestresponse.metrics

import cats.Monad
import cats.data.Validated.{Invalid, Valid}
import cats.data.{NonEmptyList, ValidatedNel}
import pl.touk.nussknacker.engine.api.runtimecontext.EngineRuntimeContext
import pl.touk.nussknacker.engine.lite.api.commonTypes.ErrorType
import pl.touk.nussknacker.engine.util.metrics.MetricIdentifier
import pl.touk.nussknacker.engine.util.metrics.common.naming.nodeIdTag
import pl.touk.nussknacker.engine.util.service.EspTimer

import scala.language.higherKinds
import scala.util.control.NonFatal

class InvocationMetrics(context: EngineRuntimeContext) {

  protected val instantTimerWindowInSeconds = 20

  private val nodeErrorTimers: collection.concurrent.TrieMap[String, EspTimer] = collection.concurrent.TrieMap()

  // TODO: maybe var initialized in `open`?
  private lazy val successTimer = espTimer(Map(), NonEmptyList.of("invocation", "success"))

  def measureTime[T, Effect[_]: Monad](
      invocation: => Effect[ValidatedNel[ErrorType, T]]
  ): Effect[ValidatedNel[ErrorType, T]] = {
    val start = System.nanoTime()
    try {
      implicitly[Monad[Effect]].map(invocation) { result =>
        result match {
          case Invalid(errors) => errors.toList.foreach(ex => markErrorTimer(start, ex.nodeComponentInfo.map(_.nodeId)))
          case Valid(_)        => successTimer.update(start)
        }
        result
      }
    } catch {
      case NonFatal(e) => markErrorTimer(start); throw e
    }
  }

  private def markErrorTimer(startTime: Long, nodeId: Option[String] = None): Unit = {
    val id = nodeId.getOrElse("unknown")
    nodeErrorTimers
      .getOrElseUpdate(id, espTimer(Map(nodeIdTag -> id), NonEmptyList.of("invocation", "failure")))
      .update(startTime)
  }

  private def espTimer(tags: Map[String, String], name: NonEmptyList[String]): EspTimer =
    context.metricsProvider.espTimer(MetricIdentifier(name, tags), instantTimerWindowInSeconds)

}