jvmMain.com.steamstreet.exceptions.Retry.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of awskt-standards-jvm Show documentation

Standard constructs and functions for working with AWS in Kotlin.

There is a newer version: 2.0.25

package com.steamstreet.exceptions

import kotlinx.coroutines.delay
import kotlin.reflect.KClass

/**
 * Retry if an exception is encountered.
 *
 * @param throwHandler called whenever an exception is thrown, even if we're still retrying, allowing for logging.
 */
public suspend fun  retry(
    times: Int,
    exceptionType: KClass? = null,
    delay: Long = 0,
    throwHandler: (Throwable) -> Unit = {},
    block: suspend () -> R
): R {
    for (i in 0 until times) {
        try {
            return block()
        } catch (e: Throwable) {
            if ((exceptionType == null || e.javaClass.kotlin == exceptionType) && i < times - 1) {
                throwHandler(e)
                if (delay > 0) {
                    delay(delay)
                }
            } else {
                throw e
            }
        }
    }
    // we'll never get here, so just throw
    throw IllegalStateException("This should never happen in retry")
}