From bd385789784f3fa1dff5fbac6c221bb7cf14a86e Mon Sep 17 00:00:00 2001 From: Hossin Asaadi Date: Fri, 22 Mar 2024 16:11:14 +0400 Subject: [PATCH] improve balancer_info.go fix balancer args --- main/commands/all/api/balancer_info.go | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/main/commands/all/api/balancer_info.go b/main/commands/all/api/balancer_info.go index 4ba33236..0d376d63 100644 --- a/main/commands/all/api/balancer_info.go +++ b/main/commands/all/api/balancer_info.go @@ -43,12 +43,16 @@ Example: func executeBalancerInfo(cmd *base.Command, args []string) { setSharedFlags(cmd) cmd.Flag.Parse(args) + unnamedArgs := cmd.Flag.Args() + if len(unnamedArgs) == 0 { + fmt.Println("set balancer tag") + unnamedArgs = []string{""} + } conn, ctx, close := dialAPIServer() defer close() - client := routerService.NewRoutingServiceClient(conn) - r := &routerService.GetBalancerInfoRequest{Tag: args[0]} + r := &routerService.GetBalancerInfoRequest{Tag: unnamedArgs[0]} resp, err := client.GetBalancerInfo(ctx, r) if err != nil { base.Fatalf("failed to get health information: %s", err) @@ -75,9 +79,10 @@ func showBalancerInfo(b *routerService.BalancerMsg) { } // Selects sb.WriteString(" - Selects:\n") - - for i, o := range b.PrincipleTarget.Tag { - writeRow(sb, tableIndent, i+1, []string{o}, nil) + if b.PrincipleTarget != nil { + for i, o := range b.PrincipleTarget.Tag { + writeRow(sb, tableIndent, i+1, []string{o}, nil) + } } os.Stdout.WriteString(sb.String()) }