mirror of
https://github.com/XTLS/Xray-core.git
synced 2024-07-21 09:19:20 +03:00
fa5d7a255b
* v5: Health Check & LeastLoad Strategy (rebased from 2c5a71490368500a982018a74a6d519c7e121816) Some changes will be necessary to integrate it into V2Ray * Update proto * parse duration conf with time.Parse() * moving health ping to observatory as a standalone component * moving health ping to observatory as a standalone component: auto generated file * add initialization for health ping * incorporate changes in router implementation * support principle target output * add v4 json support for BurstObservatory & fix balancer reference * update API command * remove cancelled API * return zero length value when observer is not found * remove duplicated targeted dispatch * adjust test with updated structure * bug fix for observer * fix strategy selector * fix strategy least load * Fix ticker usage ticker.Close does not close ticker.C * feat: Replace default Health Ping URL to HTTPS (#1991) * fix selectLeastLoad() returns wrong number of nodes (#2083) * Test: fix leastload strategy unit test * fix(router): panic caused by concurrent map read and write (#2678) * Clean up code --------- Co-authored-by: Jebbs <qjebbs@gmail.com> Co-authored-by: Shelikhoo <xiaokangwang@outlook.com> Co-authored-by: 世界 <i@sekai.icu> Co-authored-by: Bernd Eichelberger <46166740+4-FLOSS-Free-Libre-Open-Source-Software@users.noreply.github.com> Co-authored-by: 秋のかえで <autmaple@protonmail.com> Co-authored-by: Rinka <kujourinka@gmail.com>
78 lines
1.7 KiB
Go
78 lines
1.7 KiB
Go
package api
|
|
|
|
import (
|
|
routerService "github.com/xtls/xray-core/app/router/command"
|
|
"github.com/xtls/xray-core/main/commands/base"
|
|
)
|
|
|
|
var cmdBalancerOverride = &base.Command{
|
|
CustomFlags: true,
|
|
UsageLine: "{{.Exec}} api bo [--server=127.0.0.1:8080] <-b balancer> outboundTag",
|
|
Short: "balancer override",
|
|
Long: `
|
|
Override a balancer's selection.
|
|
|
|
> Make sure you have "RoutingService" set in "config.api.services"
|
|
of server config.
|
|
|
|
Once a balancer's selecting is overridden:
|
|
|
|
- The balancer's selection result will always be outboundTag
|
|
|
|
Arguments:
|
|
|
|
-r, -remove
|
|
Remove the overridden
|
|
|
|
-r, -remove
|
|
Remove the override
|
|
|
|
-s, -server
|
|
The API server address. Default 127.0.0.1:8080
|
|
|
|
-t, -timeout
|
|
Timeout seconds to call API. Default 3
|
|
|
|
Example:
|
|
|
|
{{.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()
|
|
|
|
client := routerService.NewRoutingServiceClient(conn)
|
|
target := ""
|
|
if !remove {
|
|
target = cmd.Flag.Args()[0]
|
|
}
|
|
r := &routerService.OverrideBalancerTargetRequest{
|
|
BalancerTag: balancer,
|
|
Target: target,
|
|
}
|
|
|
|
_, err := client.OverrideBalancerTarget(ctx, r)
|
|
if err != nil {
|
|
base.Fatalf("failed to perform balancer health checks: %s", err)
|
|
}
|
|
}
|