重命名包 client 为 edge；重命名包 server 为 gateway

gateway/gateway.go

@@ -0,0 +1,198 @@
+package gateway
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"os"
+	"os/signal"
+	"proxy-server/gateway/app"
+	"proxy-server/gateway/core"
+	"proxy-server/gateway/debug"
+	"proxy-server/gateway/env"
+	"proxy-server/gateway/fwd"
+	g "proxy-server/gateway/globals"
+	"proxy-server/gateway/log"
+	"proxy-server/gateway/report"
+	"proxy-server/gateway/web"
+	"proxy-server/pkg/utils"
+	"sync"
+	"time"
+
+	"github.com/google/uuid"
+
+	"github.com/joho/godotenv"
+
+	_ "net/http/pprof"
+)
+
+type server struct {
+}
+
+func New() *server {
+	return &server{}
+}
+
+func (s *server) Run() (err error) {
+
+	// 初始化
+	err = s.init()
+	if err != nil {
+		return err
+	}
+
+	// 恢复服务状态
+	err = s.restore()
+	if err != nil {
+		return err
+	}
+
+	// 准备子服务
+	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
+	defer cancel()
+
+	wg := sync.WaitGroup{}
+
+	// 转发服务
+	wg.Add(1)
+	fwdQuit := make(chan error, 1)
+	defer close(fwdQuit)
+	go func() {
+		defer wg.Done()
+		err = s.startFwd(ctx)
+		fwdQuit <- err
+	}()
+
+	// 接口服务
+	wg.Add(1)
+	apiQuit := make(chan error, 1)
+	defer close(apiQuit)
+	go func() {
+		defer wg.Done()
+		err := s.startWeb(ctx)
+		apiQuit <- err
+	}()
+
+	// debug
+	go func() {
+		debug.Start(ctx)
+	}()
+
+	// 性能监控
+	// go func() {
+	// 	runtime.SetBlockProfileRate(1)
+	// 	err := http.ListenAndServe(":6060", nil)
+	// 	if err != nil {
+	// 		slog.Error("性能监控服务发生错误", "err", err)
+	// 	}
+	// }()
+
+	// 报告上线
+	slog.Debug("报告服务上线")
+	err = report.Online(app.Name)
+	if err != nil {
+		return fmt.Errorf("服务上线失败: %w", err)
+	}
+
+	select {
+	case <-ctx.Done():
+	case err := <-fwdQuit:
+		if err != nil {
+			slog.Warn("fwd 服务异常退出", "err", err)
+		}
+	case err := <-apiQuit:
+		if err != nil {
+			slog.Warn("web 服务异常退出", "err", err)
+		}
+	}
+	cancel()
+
+	// 主协程退出流程
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		// 报告下线
+		slog.Debug("报告服务下线")
+		err = report.Offline(app.Name)
+		if err != nil {
+			slog.Error("服务下线失败", "err", err)
+		}
+
+		// 关闭 redis
+		g.ExitRedis()
+	}()
+
+	// 等待其它服务关闭
+	select {
+	case <-utils.WgWait(&wg):
+		slog.Info("服务正常关闭")
+	case <-time.After(time.Duration(env.AppExitTimeout) * time.Second):
+		slog.Warn("超时强制关闭")
+	}
+
+	return nil
+}
+
+func (s *server) init() error {
+
+	err := godotenv.Load()
+	if err != nil {
+		println("没有本地环境变量文件")
+	}
+
+	log.Init()
+	env.Init()
+	g.InitRedis()
+
+	return nil
+}
+
+func (s *server) restore() error {
+	var file = "proxy.lock"
+
+	bytes, err := os.ReadFile(file)
+	if err != nil {
+		return err
+	}
+
+	if len(bytes) == 17 && bytes[0] == core.RestoreMagic {
+		app.Name = uuid.UUID(bytes[1:]).String()
+		slog.Info("恢复服务名称", "name", app.Name)
+	} else {
+		var u = uuid.New()
+		app.Name = u.String()
+
+		bytes = make([]byte, 17)
+		bytes[0] = core.RestoreMagic
+		copy(bytes[1:], u[:])
+		err := os.WriteFile(file, bytes, 0644)
+		if err != nil {
+			return err
+		}
+
+		slog.Info("生成服务名称", "name", app.Name)
+	}
+
+	return nil
+}
+
+func (s *server) startFwd(ctx context.Context) error {
+	server := fwd.New()
+	go func() {
+		<-ctx.Done()
+		server.Stop()
+	}()
+	return server.Run()
+}
+
+func (s *server) startWeb(ctx context.Context) error {
+	server := web.New()
+	go func() {
+		<-ctx.Done()
+		err := server.Stop()
+		if err != nil {
+			slog.Error("web 服务关闭发生错误", "err", err)
+		}
+	}()
+	return server.Run()
+}