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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spice-client_3 Show documentation

spice-client

There is a newer version: 0.6.1

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)
      }
    }
  }
}