• Home
• io.tryp
• cattrix-request_2.12
• 0.1.3
• source code

×

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.

ix-request_2.12.0.1.3.source-code.metrics.scala Maven / Gradle / Ivy

package chm

import cats.ApplicativeError
import cats.syntax.flatMap._
import cats.syntax.functor._
import cats.syntax.applicativeError._
import cats.effect.Sync
import cats.free.FreeT

object RequestMetricPrograms
{
  def responseMetric[F[_]: Sync, In, Out]
  (metric: RequestMetric[F, In, Out], response: Out)
  (implicit res: HttpResponse[F, Out])
  : Metrics.Step[F, Unit] = {
    for {
      _ <- Metrics.mark[F](res.status(response).toString)
      _ <- metric.error(response)
    } yield ()
  }

  def resultMetrics[F[_]: Sync, In, Out]
  (metric: RequestMetric[F, In, Out], result: Either[Throwable, Out])
  (implicit res: HttpResponse[F, Out])
  : Metrics.Step[F, Unit] =
    result match {
      case Right(r) => responseMetric(metric, r)
      case Left(error) => Metrics.mark[F]("fatal")
    }

  def simpleTimed[F[_]: Sync, M, In, Out](
    resources: M,
    metrics: Metrics[F, M],
    task: RequestTask[F, In, Out],
    request: () => F[Out],
    name: String,
  )
  (implicit res: HttpResponse[F, Out])
  : Metrics.Step[F, Out] = {
    for {
      t <- Metrics.timer("time")
      _ <- Metrics.incCounter("active")
      response <- Metrics.attempt(request)
      _ <- Metrics.decCounter("active")
      _ <- Metrics.time(t)
      _ <- resultMetrics(task.metric, response)
      result <- Metrics.result(response)
    } yield result
  }
}

object RequestMetrics
{
  def wrapRequest[F[_]: Sync, M, In, Out]
  (resources: M)
  (task: RequestTask[F, In, Out])
  (request: () => F[Out])
  (implicit metrics: Metrics[F, M], res: HttpResponse[F, Out])
  : F[Out] = {
    for {
      name <- task.metric.name(task.request)
      prog = RequestMetricPrograms.simpleTimed(resources, metrics, task, request, name)
      result <- Metrics.compile(MetricTask(resources, name))(prog)
    } yield result
  }
}