v2fly/main/commands/all/api/balancer_info.go

package api

import (
	"fmt"
	"os"
	"sort"
	"strings"

	routerService "github.com/v2fly/v2ray-core/v4/app/router/command"
	"github.com/v2fly/v2ray-core/v4/main/commands/base"
)

// TODO: support "-json" flag for json output
var cmdBalancerInfo = &base.Command{
	CustomFlags: true,
	UsageLine:   "{{.Exec}} api bi [--server=127.0.0.1:8080] [balancer]...",
	Short:       "balancer information",
	Long: `
Get information of specified balancers, including health, strategy 
and selecting. If no balancer tag specified, get information of 
all balancers.

> Make sure you have "RoutingService" set in "config.api.services" 
of server config.

Arguments:

	-json
		Use json output.

	-s, -server <server:port>
		The API server address. Default 127.0.0.1:8080

	-t, -timeout <seconds>
		Timeout seconds to call API. Default 3

Example:

    {{.Exec}} {{.LongName}} --server=127.0.0.1:8080 balancer1 balancer2
`,
	Run: executeBalancerInfo,
}

func executeBalancerInfo(cmd *base.Command, args []string) {
	setSharedFlags(cmd)
	cmd.Flag.Parse(args)

	conn, ctx, close := dialAPIServer()
	defer close()

	client := routerService.NewRoutingServiceClient(conn)
	r := &routerService.GetBalancersRequest{BalancerTags: cmd.Flag.Args()}
	resp, err := client.GetBalancers(ctx, r)
	if err != nil {
		base.Fatalf("failed to get health information: %s", err)
	}
	sort.Slice(resp.Balancers, func(i, j int) bool {
		return resp.Balancers[i].Tag < resp.Balancers[j].Tag
	})
	if apiJSON {
		showJSONResponse(resp)
		return
	}
	for _, b := range resp.Balancers {
		showBalancerInfo(b)
	}
}

func showBalancerInfo(b *routerService.BalancerMsg) {
	const tableIndent = 4
	sb := new(strings.Builder)
	// Balancer
	sb.WriteString(fmt.Sprintf("Balancer: %s\n", b.Tag))
	// Strategy
	sb.WriteString("  - Strategy:\n")
	for _, v := range b.StrategySettings {
		sb.WriteString(fmt.Sprintf("    %s\n", v))
	}
	// Override
	if b.Override != nil {
		sb.WriteString("  - Selecting Override:\n")
		until := fmt.Sprintf("until: %s", b.Override.Until)
		writeRow(sb, tableIndent, 0, []string{until}, nil)
		for i, s := range b.Override.Selects {
			writeRow(sb, tableIndent, i+1, []string{s}, nil)
		}
	}
	b.Titles = append(b.Titles, "Tag")
	formats := getColumnFormats(b.Titles)
	// Selects
	sb.WriteString("  - Selects:\n")
	writeRow(sb, tableIndent, 0, b.Titles, formats)
	for i, o := range b.Selects {
		o.Values = append(o.Values, o.Tag)
		writeRow(sb, tableIndent, i+1, o.Values, formats)
	}
	// Others
	scnt := len(b.Selects)
	if len(b.Others) > 0 {
		sb.WriteString("  - Others:\n")
		writeRow(sb, tableIndent, 0, b.Titles, formats)
		for i, o := range b.Others {
			o.Values = append(o.Values, o.Tag)
			writeRow(sb, tableIndent, scnt+i+1, o.Values, formats)
		}
	}
	os.Stdout.WriteString(sb.String())
}

func getColumnFormats(titles []string) []string {
	w := make([]string, len(titles))
	for i, t := range titles {
		w[i] = fmt.Sprintf("%%-%ds ", len(t))
	}
	return w
}

func writeRow(sb *strings.Builder, indent, index int, values, formats []string) {
	if index == 0 {
		// title line
		sb.WriteString(strings.Repeat(" ", indent+4))
	} else {
		sb.WriteString(fmt.Sprintf("%s%-4d", strings.Repeat(" ", indent), index))
	}
	for i, v := range values {
		format := "%-14s"
		if i < len(formats) {
			format = formats[i]
		}
		sb.WriteString(fmt.Sprintf(format, v))
	}
	sb.WriteByte('\n')
}