1
0
mirror of https://github.com/v2fly/v2ray-core.git synced 2024-10-30 15:57:35 -04:00
v2fly/main/commands/all/api/balancer_override.go

78 lines
1.8 KiB
Go
Raw Normal View History

package api
import (
routerService "github.com/v2fly/v2ray-core/v4/app/router/command"
"github.com/v2fly/v2ray-core/v4/main/commands/base"
)
var cmdBalancerOverride = &base.Command{
CustomFlags: true,
2021-06-19 03:57:40 -04:00
UsageLine: "{{.Exec}} api bo [--server=127.0.0.1:8080] <-b balancer> outboundTag",
Short: "balancer override",
Long: `
2021-06-19 03:57:40 -04:00
Override a balancer's selection.
> Make sure you have "RoutingService" set in "config.api.services"
of server config.
Once a balancer's selection is overridden:
2021-06-19 03:57:40 -04:00
- The balancer's selection result will always be outboundTag
Arguments:
-b, -balancer <tag>
Tag of the target balancer. Required.
-r, -remove
Remove the override
-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:
2021-06-19 03:57:40 -04:00
{{.Exec}} {{.LongName}} --server=127.0.0.1:8080 -b balancer tag
{{.Exec}} {{.LongName}} --server=127.0.0.1:8080 -b balancer -r
`,
Run: executeBalancerOverride,
}
func executeBalancerOverride(cmd *base.Command, args []string) {
var (
balancer string
remove bool
)
cmd.Flag.StringVar(&balancer, "b", "", "")
cmd.Flag.StringVar(&balancer, "balancer", "", "")
cmd.Flag.BoolVar(&remove, "r", false, "")
cmd.Flag.BoolVar(&remove, "remove", false, "")
setSharedFlags(cmd)
cmd.Flag.Parse(args)
if balancer == "" {
base.Fatalf("balancer tag not specified")
}
conn, ctx, close := dialAPIServer()
defer close()
2021-06-19 03:57:40 -04:00
client := routerService.NewRoutingServiceClient(conn)
target := ""
if !remove {
2021-06-19 03:57:40 -04:00
target = cmd.Flag.Args()[0]
}
2021-06-19 03:57:40 -04:00
r := &routerService.OverrideBalancerTargetRequest{
BalancerTag: balancer,
2021-06-19 03:57:40 -04:00
Target: target,
}
2021-06-19 03:57:40 -04:00
_, err := client.OverrideBalancerTarget(ctx, r)
if err != nil {
base.Fatalf("failed to override balancer: %s", err)
}
}