net.playq.tk.aws.cost.CostComponent.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of tk-aws-cost_2.13 Show documentation

tk-aws-cost

The newest version!

package net.playq.tk.aws.cost

import izumi.distage.model.definition.Lifecycle
import izumi.functional.bio.{BlockingIO2, IO2}
import net.playq.tk.metrics.Metrics
import net.playq.tk.aws.cost.config.CostConfig
import net.playq.tk.quantified.SyncThrowable
import net.playq.tk.util.retry.TkScheduler
import software.amazon.awssdk.regions.Region
import software.amazon.awssdk.services.costexplorer.CostExplorerClient

import scala.util.chaining.*

trait CostComponent[F[_, _]] {
  def resourceClient: Lifecycle[F[Throwable, _], CostClient[F]]
}

object CostComponent {

  final class Impl[F[+_, +_]: IO2: SyncThrowable: BlockingIO2: TkScheduler](
    costConfig: CostConfig,
    metrics: Metrics[F],
  ) extends CostComponent[F] {

    override def resourceClient: Lifecycle[F[Throwable, _], CostClient[F]] = {
      Lifecycle.fromAutoCloseable {
        CostExplorerClient
          .builder()
          .pipe(b => costConfig.getRegion.fold(b)(r => b.region(Region.of(r))))
          .build()
      }.toEffect[F[Throwable, _]].map(new CostClient.Impl[F](_, metrics))
    }
  }

}