From 283c6c065934725c5226b5bed66a1dd4cdc32be6 Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Tue, 12 Jul 2022 19:40:25 +0800
Subject: [PATCH] add retry to busicli. for sometime unavailable

---
 test/busi/base_grpc.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/test/busi/base_grpc.go b/test/busi/base_grpc.go
index dd22f2b..511f3cd 100644
--- a/test/busi/base_grpc.go
+++ b/test/busi/base_grpc.go
@@ -42,8 +42,19 @@ func GrpcStartup() *grpc.Server {
 	logger.FatalIfError(err)
 	DtmClient = dtmgpb.NewDtmClient(conn)
 	logger.Debugf("dtm client inited")
-
-	conn1, err := grpc.Dial(BusiGrpc, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithUnaryInterceptor(workflow.Interceptor))
+	retryPolicy := `{
+		"methodConfig": [{
+				"waitForReady": true,
+				"retryPolicy": {
+						"MaxAttempts": 2,
+						"InitialBackoff": ".01s",
+						"MaxBackoff": ".01s",
+						"BackoffMultiplier": 1.0,
+						"RetryableStatusCodes": [ "UNAVAILABLE" ]
+				}
+		}]
+}`
+	conn1, err := grpc.Dial(BusiGrpc, grpc.WithTransportCredentials(insecure.NewCredentials()), grpc.WithUnaryInterceptor(workflow.Interceptor), grpc.WithDefaultServiceConfig(retryPolicy))
 	logger.FatalIfError(err)
 	BusiCli = NewBusiClient(conn1)