实现网关监听并读取更新队列，定时发送更新数据到平台

gateway/gateway.go

@@ -13,7 +13,6 @@ import (
 	"proxy-server/gateway/fwd"
 	g "proxy-server/gateway/globals"
 	"proxy-server/gateway/log"
-	"proxy-server/gateway/report"
 	"proxy-server/gateway/web"
 	"proxy-server/utils"
 	"sync"
@@ -21,28 +20,16 @@ import (
 
 	"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) {
+func Start() (err error) {
 
 	// 初始化
-	err = s.init()
-	if err != nil {
-		return err
-	}
+	setup()
 
 	// 恢复服务状态
-	err = s.restore()
+	err = restore()
 	if err != nil {
 		return err
 	}
@@ -59,18 +46,28 @@ func (s *server) Run() (err error) {
 	defer close(fwdQuit)
 	go func() {
 		defer wg.Done()
-		err = s.startFwd(ctx)
+		err = startFwd(ctx)
 		fwdQuit <- err
 	}()
 
 	// 接口服务
 	wg.Add(1)
-	apiQuit := make(chan error, 1)
-	defer close(apiQuit)
+	webQuit := make(chan error, 1)
+	defer close(webQuit)
 	go func() {
 		defer wg.Done()
-		err := s.startWeb(ctx)
-		apiQuit <- err
+		err := startWeb(ctx)
+		webQuit <- err
+	}()
+
+	// 调度任务服务
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		err := startTask(ctx)
+		if err != nil {
+			slog.Error("调度任务服务异常退出", "err", err)
+		}
 	}()
 
 	// debug
@@ -89,7 +86,7 @@ func (s *server) Run() (err error) {
 
 	// 报告上线
 	slog.Info("报告服务上线")
-	err = report.Online(app.Name)
+	err = app.Online(app.Name)
 	if err != nil {
 		return fmt.Errorf("服务上线失败: %w", err)
 	}
@@ -100,7 +97,7 @@ func (s *server) Run() (err error) {
 		if err != nil {
 			slog.Warn("fwd 服务异常退出", "err", err)
 		}
-	case err := <-apiQuit:
+	case err := <-webQuit:
 		if err != nil {
 			slog.Warn("web 服务异常退出", "err", err)
 		}
@@ -113,7 +110,7 @@ func (s *server) Run() (err error) {
 		defer wg.Done()
 		// 报告下线
 		slog.Debug("报告服务下线")
-		err = report.Offline()
+		err = app.Offline()
 		if err != nil {
 			slog.Error("服务下线失败", "err", err)
 		}
@@ -134,22 +131,14 @@ func (s *server) Run() (err error) {
 	return nil
 }
 
-func (s *server) init() error {
-
-	err := godotenv.Load()
-	if err != nil {
-		println("没有本地环境变量文件")
-	}
-
+func setup() {
 	log.Init()
 	env.Init()
 	g.InitRedis()
 	g.InitGeo()
-
-	return nil
 }
 
-func (s *server) restore() error {
+func restore() error {
 	var file = "proxy.lock"
 
 	bytes, err := os.ReadFile(file)
@@ -178,7 +167,7 @@ func (s *server) restore() error {
 	return nil
 }
 
-func (s *server) startFwd(ctx context.Context) error {
+func startFwd(ctx context.Context) error {
 	server := fwd.New()
 	go func() {
 		<-ctx.Done()
@@ -187,7 +176,7 @@ func (s *server) startFwd(ctx context.Context) error {
 	return server.Run()
 }
 
-func (s *server) startWeb(ctx context.Context) error {
+func startWeb(ctx context.Context) error {
 	server := web.New()
 	go func() {
 		<-ctx.Done()
@@ -198,3 +187,37 @@ func (s *server) startWeb(ctx context.Context) error {
 	}()
 	return server.Run()
 }
+
+func startTask(ctx context.Context) error {
+
+	var lock = sync.Mutex{}
+	var updates = make([]*core.Edge, 0)
+	go func() {
+		for data := range app.EdgeUpdates {
+			lock.Lock()
+			updates = append(updates, data)
+			lock.Unlock()
+		}
+	}()
+
+	// 每 30 秒批量提交一次更新
+	var scheduler = time.Tick(30 * time.Second)
+	for {
+		select {
+		case <-ctx.Done():
+			return nil
+		case <-scheduler:
+			if len(updates) == 0 {
+				continue
+			}
+			err := app.Update(updates)
+			lock.Lock()
+			clear(updates)
+			updates = updates[:0]
+			lock.Unlock()
+			if err != nil {
+				slog.Error("调度更新任务失败", "err", err)
+			}
+		}
+	}
+}