init

go-gost/main.go

@@ -0,0 +1,147 @@
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"github.com/go-gost/gost/traffic"
+	"log"
+	_ "net/http/pprof"
+	"os"
+	"os/exec"
+	"runtime"
+	"strings"
+	"sync"
+
+	"github.com/go-gost/core/logger"
+	xlogger "github.com/go-gost/x/logger"
+	"github.com/judwhite/go-svc"
+)
+
+type stringList []string
+
+func (l *stringList) String() string {
+	return fmt.Sprintf("%s", *l)
+}
+func (l *stringList) Set(value string) error {
+	*l = append(*l, value)
+	return nil
+}
+
+var (
+	cfgFile      string
+	outputFormat string
+	services     stringList
+	nodes        stringList
+	debug        bool
+	trace        bool
+	apiAddr      string
+	metricsAddr  string
+)
+
+func init() {
+	log.SetFlags(log.LstdFlags | log.Lshortfile | log.Lmicroseconds)
+
+	args := strings.Join(os.Args[1:], "  ")
+
+	if strings.Contains(args, " -- ") {
+		var (
+			wg  sync.WaitGroup
+			ret int
+		)
+
+		ctx, cancel := context.WithCancel(context.Background())
+		defer cancel()
+
+		for wid, wargs := range strings.Split(" "+args+" ", " -- ") {
+			wg.Add(1)
+			go func(wid int, wargs string) {
+				defer wg.Done()
+				defer cancel()
+				worker(wid, strings.Split(wargs, "  "), &ctx, &ret)
+			}(wid, strings.TrimSpace(wargs))
+		}
+
+		wg.Wait()
+
+		os.Exit(ret)
+	}
+}
+
+func worker(id int, args []string, ctx *context.Context, ret *int) {
+	cmd := exec.CommandContext(*ctx, os.Args[0], args...)
+
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	cmd.Env = append(os.Environ(), fmt.Sprintf("_GOST_ID=%d", id))
+
+	if err := cmd.Run(); err != nil {
+		log.Fatal(err)
+	}
+	if cmd.ProcessState.Exited() {
+		*ret = cmd.ProcessState.ExitCode()
+	}
+}
+
+func init() {
+	var printVersion bool
+
+	flag.Var(&services, "L", "service list")
+	flag.Var(&nodes, "F", "chain node list")
+	flag.StringVar(&cfgFile, "C", "", "configuration 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.BoolVar(&trace, "DD", false, "trace 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)
+	}
+}
+
+func main() {
+	// 加载配置文件
+	config, err := LoadConfig("config.json")
+	if err != nil {
+		fmt.Printf("❌ 配置加载失败: %v\n", err)
+		fmt.Println("请确保当前目录存在 config.json 文件")
+		os.Exit(1)
+	}
+
+	fmt.Println("✅ 配置加载成功 - addr: %s", config.Addr)
+
+	log := xlogger.NewLogger()
+	logger.SetDefault(log)
+
+	// 创建流量管理器
+	trafficMgr := traffic.NewTrafficManager()
+	defer trafficMgr.Close()
+
+	// 设置全局流量管理器（供 handler 使用）
+	traffic.SetGlobalTrafficManager(trafficMgr)
+
+	// 设置流量记录器给 handler 使用
+	traffic.SetupTrafficRecorder(trafficMgr)
+
+	fmt.Println("✅ 流量管理器已初始化（使用内存存储）")
+	logger.Default().Info("Traffic manager initialized (using memory storage)")
+
+	traffic.SetHTTPReportURL(config.Addr, config.Secret)
+	traffic.StartTrafficReporter(trafficMgr)
+	wsReporter := traffic.StartWebSocketReporterWithConfig(config.Addr, config.Secret)
+	defer wsReporter.Stop()
+
+	p := &program{}
+
+	if err := svc.Run(p); err != nil {
+		logger.Default().Fatal(err)
+	}
+}
+
+// GOOS=linux GOARCH=amd64 go build
+// GOOS=windows GOARCH=amd64 go build -o your_app.exe