Increase the number of retry times (#517)

Signed-off-by: bigsheeper <yihao.dai@zilliz.com>
milvus-io · Jul 19, 2023 · 9e61d26 · 9e61d26
1 parent bfb377d
commit 9e61d26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/client/config.go b/client/config.go
@@ -147,7 +147,7 @@ func (c *Config) getDialOption() []grpc.DialOption {
 				return 60 * time.Millisecond * time.Duration(math.Pow(3, float64(attempt)))
 			}),
 			grpc_retry.WithCodes(codes.Unavailable, codes.ResourceExhausted)),
-			RetryOnRateLimitInterceptor(10, func(ctx context.Context, attempt uint) time.Duration {
+			RetryOnRateLimitInterceptor(75, func(ctx context.Context, attempt uint) time.Duration {
 				return 10 * time.Millisecond * time.Duration(math.Pow(3, float64(attempt)))
 			}),
 		))

diff --git a/client/rate_limit_interceptor.go b/client/rate_limit_interceptor.go
@@ -37,7 +37,7 @@ const (
 	RetryOnRateLimit ctxKey = iota
 )
 
-var MaxBackOff = 60 * time.Second
+var MaxBackOff = 3 * time.Second
 
 // RetryOnRateLimitInterceptor returns a new retrying unary client interceptor.
 func RetryOnRateLimitInterceptor(maxRetry uint, backoffFunc grpc_retry.BackoffFuncContext) grpc.UnaryClientInterceptor {