はじめに
インフラエンジニアは日々の業務でプログラムを書く機会が多く、その中で処理の実行やHTTPの通信などでリトライ処理を実装する必要があることが少なくありません。リトライ処理を実装する必要は必ずしもなくても、実装することでバッチが安定することがあります。もっと言っておくとリトライ処理を実装することで、一時的なエラーによる処理の失敗を回避し、バッチ処理の安定性が向上する可能性があります。実行基盤によってジョブの再試行の自動化、最大再試行回数を設定するやPod失敗のバックオフポリシーなどとの兼ね合いを考える必要もあると思います。あとはマジでガー不のバグを引き寄せることもあるので注意が必要です。
今回はGolangには「retry-go」というリトライ処理を簡潔に実装できるライブラリがあり、これを使うと非常に簡単にリトライ機能を追加できます。シェルスクリプトでも簡単に実装できるのですが今回は紹介しない。
avast/retry-go
は、リトライ処理を実装するための便利なライブラリです。このライブラリを使えば、ごく少ない行数でリトライ機能を実装できます。
インストールはgo get github.com/avast/retry-go
で行えます。
このライブラリの使い方は非常に簡単です。リトライ対象の処理をラップするだけで、設定した回数とウェイト時間に従ってリトライが実行されます。設定可能なオプションも豊富で、リトライ条件やログ出力など細かなカスタマイズも可能です。
リトライ処理の実装は、単純に見えて一歩踏み込もうとすると意外と難しい面があります。retyr-goを使えば、そういった難しさから開放され、安定したリトライ処理を簡単に実装できます。バッチ処理の安定性向上に役立つことは間違いありません。
Golangを使ったプログラミングにおいて、retry-goはリトライ処理の実装を格段に簡単にしてくれる強力なライブラリです。ぜひ一度試してみてはいかがでしょうか。
シンプルな例
使い方は簡単で、retry.Do
を使って対象の関数をラップするだけでリトライ処理を実装できます。例外が発生した場合にはリトライが行われ、何も例外が発生しなければ値が返ってきます。
package main import ( "fmt" "math/rand" "github.com/avast/retry-go" ) func randomErrorSimple() error { num := rand.Intn(10) if num > 2 { fmt.Printf("Error: num=%d\n", num) return fmt.Errorf("Error!") } fmt.Printf("Success: num=%d\n", num) return nil } func main() { err := retry.Do( randomErrorSimple, ) if err != nil { fmt.Printf("Error: %v\n", err) } }
実際に実行してみる。3回失敗して4回目にError値が返っていないので通常に終了。
$ go run main.go Error: num=5 Error: num=4 Error: num=3 Success: num=0
最大リトライ回数の指定
retry.Attempts
を使うと、指定した回数だけリトライを行うことができます。指定した回数に達した後に例外が発生した場合はエラーが返されます。
package main import ( "fmt" "github.com/avast/retry-go" ) func errorWithMaxAttempts() error { fmt.Println("Error occurred!") return fmt.Errorf("Error occurred!") } func main() { retryWithMaxAttempts() } func retryWithMaxAttempts() { err := retry.Do( errorWithMaxAttempts, retry.Attempts(3), ) if err != nil { fmt.Println("3 times failed") } }
実際に実行すると3回失敗して終了している。
$ go run main.go Error occurred! Error occurred! Error occurred! 3 times failed
次のリトライまでの待ち時間の設定
retry.Delay
を使って、次のリトライまで指定した時間待つことができます。例えば、APIのレート制限に引っかかって例外が発生した場合などに便利です。
package main import ( "fmt" "time" "github.com/avast/retry-go" ) func errorWithDelay() error { now := time.Now().Format("15:04:05") fmt.Printf("Error occurred!: %s\n", now) return fmt.Errorf("Error occurred!") } func main() { retryWithDelay() } func retryWithDelay() { err := retry.Do( errorWithDelay, retry.Attempts(3), retry.Delay(3*time.Second), ) if err != nil { fmt.Println("3 times failed") } }
実行結果は以下のようになる。ちゃんと待ち時間を指定できている。
$ go run main.go Error occurred!: 01:35:16 Error occurred!: 01:35:20 Error occurred!: 01:35:26 3 times failed
特定の例外のみリトライするケース
retry.RetryIf
でリトライする場合の例外条件を指定することができます。特定の例外が発生した場合のみリトライ処理を行うことができます。
package main import ( "fmt" "github.com/avast/retry-go" ) func errorWithSpecificError(num int) error { if num == 0 { fmt.Println("0 is invalid") return fmt.Errorf("value error") } return fmt.Errorf("Error occurred!: num=%d", num) } func main() { retryWithSpecificError() } func retryWithSpecificError() { err := retry.Do( func() error { return errorWithSpecificError(0) }, retry.Attempts(3), retry.RetryIf(func(err error) bool { return err.Error() == "value error" }), ) if err != nil { fmt.Println("Value Error occurred") } err = retry.Do( func() error { return errorWithSpecificError(1) }, ) if err != nil { fmt.Printf("%v\n", err) } }
実行結果です。
$ go run main.go 0 is invalid 0 is invalid 0 is invalid Value Error occurred
さいごに
retry-goを使用することでお手軽にリトライ処理を追加できて便利です。特に、最大リトライ回数の設定、次のリトライまでの待ち時間の設定、特定の例外のみリトライするケースなど、様々な状況に対応できる機能が用意されています。他にも色々な機能があるので気になった方は公式ドキュメントを見てみてください。