pl.touk.nussknacker.processCounts.CountsReporter.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nussknacker-process-reports_2.13 Show documentation

nussknacker-process-reports

There is a newer version: 1.18.0

Show newest version

package pl.touk.nussknacker.processCounts

import com.typesafe.config.Config
import pl.touk.nussknacker.engine.api.process.ProcessName
import sttp.client3.SttpBackend

import java.time.format.DateTimeFormatter
import java.time.{Instant, ZoneId}
import scala.concurrent.Future
import scala.language.higherKinds

object CountsReporterCreator {

  val reporterCreatorConfigPath = "countsSettings"

}

//By default it's InfluxCountsReporterCreator, other implementation can be used via ServiceLoaderMechanism
//@see NussknackerApp#prepareCountsReporter
trait CountsReporterCreator {

  def createReporter(env: String, config: Config)(implicit backend: SttpBackend[Future, Any]): CountsReporter[Future]

}

trait CountsReporter[F[_]] extends AutoCloseable {

  def prepareRawCounts(processName: ProcessName, countsRequest: CountsRequest): F[String => Option[Long]]

}

object CannotFetchCountsError {

  private val dateTimeFormatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")

  def restartsDetected(dates: List[Instant]): CannotFetchCountsError = CannotFetchCountsError(
    s"Counts unavailable, as scenario was restarted/deployed on following dates: ${dates
        .map(_.atZone(ZoneId.systemDefault()).format(dateTimeFormatter))
        .mkString(", ")}"
  )

}

final case class CannotFetchCountsError(msg: String) extends Exception(msg)

sealed trait CountsRequest

final case class RangeCount(fromDate: Instant, toDate: Instant) extends CountsRequest

final case class ExecutionCount(pointInTime: Instant) extends CountsRequest