2021-01-29 19:31:11 -05:00
|
|
|
package api
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
routerService "github.com/v2fly/v2ray-core/v4/app/router/command"
|
|
|
|
"github.com/v2fly/v2ray-core/v4/main/commands/base"
|
|
|
|
)
|
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
// TODO: support "-json" flag for json output
|
2021-01-29 19:31:11 -05:00
|
|
|
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:
|
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
-json
|
|
|
|
Use json output.
|
|
|
|
|
|
|
|
-s, -server <server:port>
|
2021-01-29 19:31:11 -05:00
|
|
|
The API server address. Default 127.0.0.1:8080
|
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
-t, -timeout <seconds>
|
2021-01-29 19:31:11 -05:00
|
|
|
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)
|
2021-06-19 03:57:40 -04:00
|
|
|
r := &routerService.GetBalancerInfoRequest{Tag: args[0]}
|
|
|
|
resp, err := client.GetBalancerInfo(ctx, r)
|
2021-01-29 19:31:11 -05:00
|
|
|
if err != nil {
|
|
|
|
base.Fatalf("failed to get health information: %s", err)
|
|
|
|
}
|
2021-06-19 03:57:40 -04:00
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
if apiJSON {
|
|
|
|
showJSONResponse(resp)
|
|
|
|
return
|
|
|
|
}
|
2021-06-19 03:57:40 -04:00
|
|
|
|
|
|
|
showBalancerInfo(resp.Balancer)
|
|
|
|
|
2021-01-29 19:31:11 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
func showBalancerInfo(b *routerService.BalancerMsg) {
|
2021-02-21 10:02:42 -05:00
|
|
|
const tableIndent = 4
|
2021-01-29 19:31:11 -05:00
|
|
|
sb := new(strings.Builder)
|
|
|
|
// Override
|
|
|
|
if b.Override != nil {
|
|
|
|
sb.WriteString(" - Selecting Override:\n")
|
2021-06-19 03:57:40 -04:00
|
|
|
for i, s := range []string{b.Override.Target} {
|
2021-02-21 10:02:42 -05:00
|
|
|
writeRow(sb, tableIndent, i+1, []string{s}, nil)
|
2021-01-29 19:31:11 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
// Selects
|
|
|
|
sb.WriteString(" - Selects:\n")
|
2021-06-19 03:57:40 -04:00
|
|
|
|
|
|
|
for i, o := range b.PrincipleTarget.Tag {
|
|
|
|
writeRow(sb, tableIndent, i+1, []string{o}, nil)
|
2021-01-29 19:31:11 -05:00
|
|
|
}
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2021-02-21 10:02:42 -05:00
|
|
|
func writeRow(sb *strings.Builder, indent, index int, values, formats []string) {
|
2021-01-29 19:31:11 -05:00
|
|
|
if index == 0 {
|
|
|
|
// title line
|
2021-02-21 10:02:42 -05:00
|
|
|
sb.WriteString(strings.Repeat(" ", indent+4))
|
2021-01-29 19:31:11 -05:00
|
|
|
} else {
|
2021-02-21 10:02:42 -05:00
|
|
|
sb.WriteString(fmt.Sprintf("%s%-4d", strings.Repeat(" ", indent), index))
|
2021-01-29 19:31:11 -05:00
|
|
|
}
|
|
|
|
for i, v := range values {
|
|
|
|
format := "%-14s"
|
|
|
|
if i < len(formats) {
|
|
|
|
format = formats[i]
|
|
|
|
}
|
|
|
|
sb.WriteString(fmt.Sprintf(format, v))
|
|
|
|
}
|
|
|
|
sb.WriteByte('\n')
|
|
|
|
}
|