genkai.TryRateLimiter.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of genkai-core_2.12 Show documentation

core

The newest version!

package genkai

import java.time.Instant

import genkai.monad.{MonadError, TryMonadError}

import scala.util.Try

final class TryRateLimiter(
  rateLimiter: RateLimiter[Id]
) extends RateLimiter[Try] {
  override private[genkai] def permissions[A: Key](key: A, instant: Instant): Try[Long] =
    monadError.eval(rateLimiter.permissions(key, instant))

  override def reset[A: Key](key: A): Try[Unit] = monadError.eval(rateLimiter.reset(key))

  override private[genkai] def acquireS[A: Key](
    key: A,
    instant: Instant,
    cost: Long
  ): Try[RateLimiter.State] =
    monadError.eval(rateLimiter.acquireS(key, instant, cost))

  override def close(): Try[Unit] = monadError.eval(rateLimiter.close())

  override def monadError: MonadError[Try] = TryMonadError
}