balancer_check.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. package api
  2. import (
  3. routerService "github.com/v2fly/v2ray-core/v4/app/router/command"
  4. "github.com/v2fly/v2ray-core/v4/main/commands/base"
  5. )
  6. var cmdBalancerCheck = &base.Command{
  7. CustomFlags: true,
  8. UsageLine: "{{.Exec}} api bc [--server=127.0.0.1:8080] [balancer]...",
  9. Short: "balancer health check",
  10. Long: `
  11. Perform instant health checks for specific balancers. If no
  12. balancer tag specified, check all balancers.
  13. > Make sure you have "RoutingService" set in "config.api.services"
  14. of server config.
  15. Arguments:
  16. -s, -server
  17. The API server address. Default 127.0.0.1:8080
  18. -t, -timeout
  19. Timeout seconds to call API. Default 3
  20. Example:
  21. {{.Exec}} {{.LongName}} --server=127.0.0.1:8080 balancer1 balancer2
  22. `,
  23. Run: executeBalancerCheck,
  24. }
  25. func executeBalancerCheck(cmd *base.Command, args []string) {
  26. setSharedFlags(cmd)
  27. cmd.Flag.Parse(args)
  28. conn, ctx, close := dialAPIServer()
  29. defer close()
  30. client := routerService.NewRoutingServiceClient(conn)
  31. r := &routerService.CheckBalancersRequest{BalancerTags: cmd.Flag.Args()}
  32. _, err := client.CheckBalancers(ctx, r)
  33. if err != nil {
  34. base.Fatalf("failed to perform balancer health checks: %s", err)
  35. }
  36. }