scala 에서 retry 구현하기

문제

cats-effect 나 zio 등의 functional library 를 사용하면, 이미 구현되어 있는 retry 를 쉽게 사용할 수 있다. 혹은 retry library 를 활용할 수도 있다. 간단하게 쓸수 있도록 scala 표준 라이브러리만을 사용해 retry 를 구현해 보자. stackoverflow 를 참고할 수 있다.

 

코드

import scala.annotation.tailrec
import scala.util.{Failure, Success, Try}

object Util {
  @tailrec
  def retry[T](n: Int)(f: => T): Try[T] =
    if (n < 0) {
      Failure(new Exception(s"$n must be greater or equal to 0"))
    } else if (n == 0) {
      Try(f)
    } else {
      Try(f) match {
        case Success(value) => Success(value)
        case Failure(_)     => if (n == 1) Try(f) else retry(n - 1)(f)
      }
    }
}