All Downloads are FREE. Search and download functionalities are using the official Maven repository.

spice.http.client.intercept.RateLimiter.scala Maven / Gradle / Ivy

The newest version!
package spice.http.client.intercept

import cats.effect.IO
import spice.http.HttpRequest

import scala.concurrent.duration._

case class RateLimiter(perRequestDelay: FiniteDuration) extends InterceptorAdapter { self =>
  private val maxDelay = perRequestDelay.toMillis
  @volatile private var lastTime: Long = 0L

  override def before(request: HttpRequest): IO[HttpRequest] = IO.unit.flatMap { _ =>
    self.synchronized {
      val now = System.currentTimeMillis()
      val delay = (lastTime + maxDelay) - now
      if (delay > 0L) {
        lastTime = now
        IO.sleep(delay.millis).map(_ => request)
      } else {
        IO.pure(request)
      }
    }
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy