scala - 如何编写重试函数？

Question

假设我有一个函数 foo:Int => Try[Int] 并且我需要重试调用它。也就是说，我需要调用它直到它返回 Success 最多 k 次。

我正在写一个这样的函数retry:

def retry(k: retries)(fun: Int => Try[Int]): Try[Int] = ???

我希望重试返回成功或最后失败。你会怎么写？

Answer 1

这是我使用的，它对任何返回 T 的 thunk 都是通用的:

@tailrec
final def withRetry[T](retries: Int)(fn: => T): Try[T] = {
  Try(fn) match {
    case x: Success[T] => x
    case _ if retries > 1 => withRetry(retries - 1)(fn)
    case f => f
  }
}