kotlin 实现在 vertx 框架内的重试机制

最近 vertx 玩的比较多,在实际业务中运用了很多,但 vertx 一个很大的问题是体系的不完善,常用的中间件 client 甚至可以用简陋形容(当然很多人也称这叫简洁,觉得是优点)。比如一个很常用的重试机制,在spring体系中使用spring-retry可以以注解+aop的方式很容易得实现方法级别的重试,但 vertx 想要做到同样的事,至少现在这个时间,我没在官方文档里、google出来解决方案。因此我们需要自己封装一个适合vertx的重试方法。

在vertx中,我们常用的方法封装形式,一般为方法返回值改为期望值Future。因此我们封装的重试方法也在此规范基础上实现,以下使用kotlin实现

/**
 * 循环调用 [block],直到返回结果为成功或重试次数[retryTimes]用尽
 */
fun retryWhenFail(retryTimes: Int, retryIntervalMs: Long? = null, block: () -> Future): Future {
    // 预设第0次调用为失败,触发重试逻辑,等于间接的第一次执行
    var future = Future.succeededFuture(false)
    var retry = false

    // 重试逻辑lambda
    val retryBlock = {
        val blockResult: Future
        // 带延迟的重试
        if (retry && retryIntervalMs != null && retryIntervalMs > 0) {
            blockResult = Future.future { timerPromise ->
                VertxInstance.vertx.setTimer(retryIntervalMs) {
                    // 执行block()
                    block()
                        //  block()结果传递到timerPromise中(回调方式)
                        .onComplete(timerPromise)
                }
            }
        }
        // 首次执行或不带延迟的重试
        else {
            blockResult = block()
            retry = true
        }
        blockResult
    }

    // 因响应式特殊的回调方式,所以一次性注册所有重试次数的逻辑
    for (index in 0 until retryTimes) {
        // 注册重试逻辑
        future = future.compose(
            { lastBlockResult ->
                // 若上次执行为成功结果,则向后传递成功结果
                if (true == lastBlockResult) {
                    Future.succeededFuture(lastBlockResult)
                }
                // 若上次执行为失败结果,则执行重试逻辑
                else {
                    retryBlock()
                }
            }, {
                // 若上次执行出现异常,则执行重试逻辑
                retryBlock()
            }
        )
    }
    return future
}

其中前两个入参retryTimes、retryIntervalMs为重试的相关策略,即重试多少次、重试间隔。入参block为实际业务方法,返回值Future,future的成功与否代表业务方法的执行是否有异常,boolean代表方法业务的成功与否,两者任一的false都会引起重试。

调用示例如下:

// 重试6次,每次间隔0.5s
retryWhenFail(6, 500L) {
    // do something...
}

在此抛砖引玉,有更好的办法欢迎一起交流

你可能感兴趣的:(kotlin 实现在 vertx 框架内的重试机制)