From d75cffbd3089bf423a868d0b78c52989d2be6e8e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E8=81=AA?= Date: Fri, 15 Jul 2022 14:49:47 +0800 Subject: [PATCH] fix: grpc reconnect backoff --- hrp/internal/boomer/client_grpc.go | 9 +++++++++ hrp/internal/boomer/runner.go | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/hrp/internal/boomer/client_grpc.go b/hrp/internal/boomer/client_grpc.go index cdf32111..6e014aee 100644 --- a/hrp/internal/boomer/client_grpc.go +++ b/hrp/internal/boomer/client_grpc.go @@ -10,6 +10,7 @@ import ( "golang.org/x/oauth2" "google.golang.org/grpc" + "google.golang.org/grpc/backoff" "google.golang.org/grpc/credentials" "google.golang.org/grpc/credentials/oauth" "google.golang.org/grpc/metadata" @@ -169,6 +170,14 @@ func (c *grpcClient) start() (err error) { grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(32 * 10e9)), grpc.WithUnaryInterceptor(unaryInterceptor), grpc.WithStreamInterceptor(streamInterceptor), + grpc.WithConnectParams(grpc.ConnectParams{ + Backoff: backoff.Config{ + BaseDelay: 1 * time.Second, + Multiplier: 1.2, + MaxDelay: 3 * time.Second, + }, + MinConnectTimeout: 3 * time.Second, + }), } c.config.conn, err = grpc.Dial(addr, opts...) if err != nil { diff --git a/hrp/internal/boomer/runner.go b/hrp/internal/boomer/runner.go index 2462be36..df68c155 100644 --- a/hrp/internal/boomer/runner.go +++ b/hrp/internal/boomer/runner.go @@ -1320,7 +1320,7 @@ func (r *masterRunner) reportStats() { table := tablewriter.NewWriter(os.Stdout) table.SetColMinWidth(0, 20) table.SetColMinWidth(1, 10) - table.SetHeader([]string{"Worker ID", "IP", "State", "Current Users", "CPU (%)", "Memory (%)"}) + table.SetHeader([]string{"Worker ID", "IP", "State", "Current Users", "CPU Usage (%)", "Memory Usage (%)"}) for _, worker := range r.server.getAllWorkers() { row := make([]string, 6)