com.valaphee.foundry.retry.context.RetryStatus.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2021, Valaphee.
 * All rights reserved.
 */

package com.valaphee.foundry.retry.context

import com.valaphee.foundry.retry.sadd
import kotlin.coroutines.AbstractCoroutineContextElement
import kotlin.coroutines.CoroutineContext

val CoroutineContext.retryStatus get() = get(RetryStatus) ?: error("No RetryStatus in context")

/**
 * @author Kevin Ludwig
 */
class RetryStatus(
    attempt: Int = 0,
    previousDelay: Long = 0,
    cumulativeDelay: Long = 0
) : AbstractCoroutineContextElement(RetryStatus) {
    companion object Key : CoroutineContext.Key

    var attempt = attempt
        private set(value) {
            require(value >= 0) { "attempt must be non-negative: $attempt" }
            field = value
        }

    var previousDelay = previousDelay
        internal set(value) {
            require(value >= 0) { "previousDelay must be non-negative: $previousDelay" }
            field = value
        }

    var cumulativeDelay = cumulativeDelay
        private set(value) {
            require(value >= 0) { "cumulativeDelay must be non-negative: $cumulativeDelay" }
            field = value
        }

    internal fun incrementAttempts() {
        attempt++
    }

    internal fun incrementCumulativeDelay(delay: Long) {
        require(delay > 0) { "delay must be positive: $delay" }
        cumulativeDelay = cumulativeDelay sadd delay
    }

    override fun toString() = "RetryStatus(attempt=$attempt, previousDelay=$previousDelay, cumulativeDelay=$cumulativeDelay)"
}