algoliasearch.extension.internal.RetryUntil.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of algoliasearch-scala_3 Show documentation

Scala client for Algolia Search API

There is a newer version: 2.9.2

Show newest version

package algoliasearch.extension.internal

import algoliasearch.exception.AlgoliaWaitException

import scala.concurrent.duration._
import scala.concurrent.{ExecutionContext, Future, Promise}

private[algoliasearch] object RetryUntil {

  val DEFAULT_DELAY: Long => Long = (retries: Long) => Math.min(retries * 200, 5000)

  /** Retry a function until a condition is met.
    */
  def retryUntil[T](
      retry: () => Future[T],
      until: T => Boolean,
      maxRetries: Int,
      delay: Long => Long = DEFAULT_DELAY
  )(implicit ec: ExecutionContext): Future[T] = {

    def attempt(retryCount: Int, currentDelay: Long): Future[T] = {
      if (retryCount >= maxRetries) {
        Future.failed(AlgoliaWaitException("The maximum number of retries exceeded."))
      } else {
        retry().flatMap { result =>
          if (until(result)) {
            Future.successful(result)
          } else {
            val nextDelay = delay(currentDelay)
            after(nextDelay)(attempt(retryCount + 1, nextDelay))
          }
        }
      }
    }

    val initialDelay = delay(0)
    attempt(0, initialDelay)
  }

  private def after[T](delay: Long)(block: => Future[T])(implicit ec: ExecutionContext): Future[T] = {
    val promise = Promise[T]()
    ec.execute(() => {
      try {
        Thread.sleep(delay)
        block.onComplete(promise.complete)
      } catch {
        case e: InterruptedException => promise.failure(e)
      }
    })
    promise.future
  }
}