scala - turorial RetryTemplate

最新推荐文章于 2025-02-07 11:11:58 发布

原创最新推荐文章于 2025-02-07 11:11:58 发布 · 245 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#scala #java

java编写专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用Scala语言实现的简单RetryTemplate类，该类允许用户定义重试次数及延迟时间，并通过传递操作作为参数来执行重试逻辑。文章还提供了一个简单的测试示例。

在公司里面用java实现了一个RetryTemplate的模板，可能spring里面已经实现了，但是用scala实现了一个很简单的，就当作学习吧。
下面是RetryTemplate.scala



class RetryTemplate(c: Int, d: Long) {

	require(c > 0 && d > 0)
	val count = c
	val delay = d

	def this() {
		this(10, 500)
	}

	def retryOperation(action: () => Unit): Unit = {
		def wrapAction(): Boolean = {
			action()
			true
		}
		this.retryOperation(classOf[Boolean], wrapAction)
	}

	def retryOperation[T <: Any](returnType: Class[T], action: () => T): T = {
//scala里面的变量初始化真是坑跌。
		var exception: Throwable = null.asInstanceOf[Throwable]
		var result: T = null.asInstanceOf[T]
//scala里面没有break判断，直接return又会报ReturnNotControl的exception.
		(1 to count).toStream.takeWhile(i => !exception.isInstanceOf[FinishStatus]).foreach {
			_ =>
				try {
					result = action()
					exception = new FinishStatus
				} catch {
					case ex: Throwable => {
						exception = ex
						delayTime
					}
				}
		}

		if (exception.isInstanceOf[FinishStatus]) {
			result
		} else {
			Console.println("retry attempt max times, finish retry.")
			throw exception
		}
	}

	def delayTime(): Unit = {
		Thread.sleep(delay);
	}
}

private class FinishStatus extends Throwable with NoStackTrace

下面是测试代码


class HelloworldModule extends RetryTemplate {

	def sayHello() = {
        Console.println("sayHello")
//throw new IllegalStateException
    }

}

object TestReretyTemplate {

	def main(args: Array[String]) {
	    val helloworldModule = new HelloworldModule
	    helloworldModule.retryOperation(helloworldModule.sayHello)
	}

}