mirror of https://github.com/go-gost/gost.git
[main] Add -- cmd flag separator to start different Gost instances
Carlos Ibáñez
parent
84255850b6
commit
9e492143cf
267
cmd/gost/main.go
267
cmd/gost/main.go
|
|
@ -6,6 +6,8 @@ import (
|
||||||
"net/http"
|
"net/http"
|
||||||
_ "net/http/pprof"
|
_ "net/http/pprof"
|
||||||
"os"
|
"os"
|
||||||
|
"sync"
|
||||||
|
"strings"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/go-gost/core/logger"
|
"github.com/go-gost/core/logger"
|
||||||
|
|
@ -16,138 +18,197 @@ import (
|
||||||
xmetrics "github.com/go-gost/x/metrics"
|
xmetrics "github.com/go-gost/x/metrics"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
type WorkerSync struct {
|
||||||
|
wg sync.WaitGroup
|
||||||
|
mu sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
log logger.Logger
|
log logger.Logger
|
||||||
|
|
||||||
cfgFile string
|
|
||||||
outputFormat string
|
|
||||||
services stringList
|
|
||||||
nodes stringList
|
|
||||||
debug bool
|
|
||||||
apiAddr string
|
|
||||||
metricsAddr string
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
var printVersion bool
|
|
||||||
|
|
||||||
flag.Var(&services, "L", "service list")
|
|
||||||
flag.Var(&nodes, "F", "chain node list")
|
|
||||||
flag.StringVar(&cfgFile, "C", "", "configure file")
|
|
||||||
flag.BoolVar(&printVersion, "V", false, "print version")
|
|
||||||
flag.StringVar(&outputFormat, "O", "", "output format, one of yaml|json format")
|
|
||||||
flag.BoolVar(&debug, "D", false, "debug mode")
|
|
||||||
flag.StringVar(&apiAddr, "api", "", "api service address")
|
|
||||||
flag.StringVar(&metricsAddr, "metrics", "", "metrics service address")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
if printVersion {
|
|
||||||
fmt.Fprintf(os.Stdout, "gost %s (%s %s/%s)\n",
|
|
||||||
version, runtime.Version(), runtime.GOOS, runtime.GOARCH)
|
|
||||||
os.Exit(0)
|
|
||||||
}
|
|
||||||
|
|
||||||
log = xlogger.NewLogger()
|
log = xlogger.NewLogger()
|
||||||
logger.SetDefault(log)
|
logger.SetDefault(log)
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
cfg := &config.Config{}
|
var ws WorkerSync
|
||||||
var err error
|
|
||||||
if len(services) > 0 || apiAddr != "" {
|
ws.wg.Add(1) // Gost must exit if any of the workers exit
|
||||||
cfg, err = buildConfigFromCmd(services, nodes)
|
|
||||||
if err != nil {
|
// Split os.Args using -- and create a worker with each slice
|
||||||
log.Fatal(err)
|
args := strings.Split(" " + strings.Join(os.Args[1:], " ") + " ", " -- ")
|
||||||
}
|
if strings.Join(args, "") == "" {
|
||||||
if debug && cfg != nil {
|
// Fix to show gost help if the resulting array is empty
|
||||||
if cfg.Log == nil {
|
args[0] = " "
|
||||||
cfg.Log = &config.LogConfig{}
|
}
|
||||||
}
|
for wid, wargs := range args {
|
||||||
cfg.Log.Level = string(logger.DebugLevel)
|
if wargs != "" {
|
||||||
}
|
go worker(wid, wargs, &ws)
|
||||||
if apiAddr != "" {
|
|
||||||
cfg.API = &config.APIConfig{
|
|
||||||
Addr: apiAddr,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if metricsAddr != "" {
|
|
||||||
cfg.Metrics = &config.MetricsConfig{
|
|
||||||
Addr: metricsAddr,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if cfgFile != "" {
|
|
||||||
err = cfg.ReadFile(cfgFile)
|
|
||||||
} else {
|
|
||||||
err = cfg.Load()
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
ws.wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
log = logFromConfig(cfg.Log)
|
func worker(id int, args string, ws *WorkerSync) {
|
||||||
|
defer ws.wg.Done()
|
||||||
|
|
||||||
logger.SetDefault(log)
|
var (
|
||||||
|
cfgFile string
|
||||||
|
outputFormat string
|
||||||
|
services stringList
|
||||||
|
nodes stringList
|
||||||
|
debug bool
|
||||||
|
apiAddr string
|
||||||
|
metricsAddr string
|
||||||
|
|
||||||
if outputFormat != "" {
|
err error
|
||||||
if err := cfg.Write(os.Stdout, outputFormat); err != nil {
|
|
||||||
log.Fatal(err)
|
cfg = &config.Config{}
|
||||||
|
)
|
||||||
|
|
||||||
|
init := func () error {
|
||||||
|
var printVersion bool
|
||||||
|
|
||||||
|
wf := flag.NewFlagSet(os.Args[0], flag.ExitOnError)
|
||||||
|
|
||||||
|
wf.Var(&services, "L", "service list")
|
||||||
|
wf.Var(&nodes, "F", "chain node list")
|
||||||
|
wf.StringVar(&cfgFile, "C", "", "configure file")
|
||||||
|
wf.BoolVar(&printVersion, "V", false, "print version")
|
||||||
|
wf.StringVar(&outputFormat, "O", "", "output format, one of yaml|json format")
|
||||||
|
wf.BoolVar(&debug, "D", false, "debug mode")
|
||||||
|
wf.StringVar(&apiAddr, "api", "", "api service address")
|
||||||
|
wf.StringVar(&metricsAddr, "metrics", "", "metrics service address")
|
||||||
|
|
||||||
|
wf.Parse(strings.Fields(args))
|
||||||
|
|
||||||
|
if printVersion {
|
||||||
|
fmt.Fprintf(os.Stdout, "gost %s (%s %s/%s)\n",
|
||||||
|
version, runtime.Version(), runtime.GOOS, runtime.GOARCH)
|
||||||
|
os.Exit(0)
|
||||||
|
} else if wf.NFlag() == 0 {
|
||||||
|
wf.Usage()
|
||||||
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
os.Exit(0)
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if cfg.Profiling != nil {
|
main := func () error {
|
||||||
go func() {
|
if len(services) > 0 || apiAddr != "" {
|
||||||
addr := cfg.Profiling.Addr
|
cfg, err = buildConfigFromCmd(id, services, nodes)
|
||||||
if addr == "" {
|
|
||||||
addr = ":6060"
|
|
||||||
}
|
|
||||||
log.Info("profiling server on ", addr)
|
|
||||||
log.Fatal(http.ListenAndServe(addr, nil))
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
if cfg.API != nil {
|
|
||||||
s, err := buildAPIService(cfg.API)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
defer s.Close()
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
log.Info("api service on ", s.Addr())
|
|
||||||
log.Fatal(s.Serve())
|
|
||||||
}()
|
|
||||||
}
|
|
||||||
|
|
||||||
if cfg.Metrics != nil {
|
|
||||||
metrics.SetGlobal(xmetrics.NewMetrics())
|
|
||||||
if cfg.Metrics.Addr != "" {
|
|
||||||
s, err := buildMetricsService(cfg.Metrics)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
if debug && cfg != nil {
|
||||||
|
if cfg.Log == nil {
|
||||||
|
cfg.Log = &config.LogConfig{}
|
||||||
|
}
|
||||||
|
cfg.Log.Level = string(logger.DebugLevel)
|
||||||
|
}
|
||||||
|
if apiAddr != "" {
|
||||||
|
cfg.API = &config.APIConfig{
|
||||||
|
Addr: apiAddr,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if metricsAddr != "" {
|
||||||
|
cfg.Metrics = &config.MetricsConfig{
|
||||||
|
Addr: metricsAddr,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if cfgFile != "" {
|
||||||
|
err = cfg.ReadFile(cfgFile)
|
||||||
|
} else {
|
||||||
|
err = cfg.Load()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log = logFromConfig(cfg.Log)
|
||||||
|
|
||||||
|
logger.SetDefault(log)
|
||||||
|
|
||||||
|
if outputFormat != "" {
|
||||||
|
if err := cfg.Write(os.Stdout, outputFormat); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.Profiling != nil {
|
||||||
go func() {
|
go func() {
|
||||||
defer s.Close()
|
addr := cfg.Profiling.Addr
|
||||||
log.Info("metrics service on ", s.Addr())
|
if addr == "" {
|
||||||
|
// Each worker uses a different profiling server
|
||||||
|
addr = fmt.Sprintf(":606%d", id)
|
||||||
|
}
|
||||||
|
log.Info("profiling server on ", addr)
|
||||||
|
log.Fatal(http.ListenAndServe(addr, nil))
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
if cfg.API != nil {
|
||||||
|
s, err := buildAPIService(cfg.API)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
defer s.Close()
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
log.Info("api service on ", s.Addr())
|
||||||
log.Fatal(s.Serve())
|
log.Fatal(s.Serve())
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if cfg.Metrics != nil {
|
||||||
|
metrics.SetGlobal(xmetrics.NewMetrics())
|
||||||
|
if cfg.Metrics.Addr != "" {
|
||||||
|
s, err := buildMetricsService(cfg.Metrics)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
defer s.Close()
|
||||||
|
log.Info("metrics service on ", s.Addr())
|
||||||
|
log.Fatal(s.Serve())
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
parsing.BuildDefaultTLSConfig(cfg.TLS)
|
||||||
|
|
||||||
|
svcs := buildService(cfg)
|
||||||
|
for _, svc := range svcs {
|
||||||
|
svc := svc
|
||||||
|
go func() {
|
||||||
|
svc.Serve()
|
||||||
|
svc.Close()
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
config.SetGlobal(cfg)
|
||||||
|
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
parsing.BuildDefaultTLSConfig(cfg.TLS)
|
// Using mutex to avoid duplicated service creation race condition
|
||||||
|
ws.mu.Lock()
|
||||||
services := buildService(cfg)
|
if err := init(); err != nil {
|
||||||
for _, svc := range services {
|
log.Fatal(err)
|
||||||
svc := svc
|
return
|
||||||
go func() {
|
|
||||||
svc.Serve()
|
|
||||||
svc.Close()
|
|
||||||
}()
|
|
||||||
}
|
}
|
||||||
|
if err := main(); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ws.mu.Unlock()
|
||||||
|
|
||||||
config.SetGlobal(cfg)
|
// Allow local functions to be garbage-collected
|
||||||
|
init = nil
|
||||||
|
main = nil
|
||||||
|
|
||||||
select {}
|
select {}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue