1.4.5.2

EdgeNode/internal/nodes/toa_manager_plus.go

@@ -0,0 +1,144 @@
+//go:build plus
+
+package nodes
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
+	"github.com/TeaOSLab/EdgeNode/internal/events"
+	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
+	executils "github.com/TeaOSLab/EdgeNode/internal/utils/exec"
+	"github.com/TeaOSLab/EdgeNode/internal/utils/goman"
+	"github.com/iwind/TeaGo/Tea"
+	"net"
+	"os"
+	"strings"
+	"time"
+)
+
+var sharedTOAManager = NewTOAManager()
+
+func init() {
+	if !teaconst.IsMain {
+		return
+	}
+
+	events.On(events.EventLoaded, func() {
+		err := sharedTOAManager.Apply(sharedNodeConfig.TOA)
+		if err != nil {
+			remotelogs.Error("TOA", err.Error())
+		}
+	})
+
+	events.OnClose(func() {
+		_ = sharedTOAManager.Quit()
+	})
+}
+
+type TOAManager struct {
+	config     *nodeconfigs.TOAConfig
+	configJSON []byte
+
+	pid  int
+	conn net.Conn
+}
+
+func NewTOAManager() *TOAManager {
+	return &TOAManager{}
+}
+
+func (this *TOAManager) Apply(config *nodeconfigs.TOAConfig) error {
+	newConfigJSON, err := json.Marshal(config)
+	if err != nil {
+		return err
+	}
+
+	if this.pid > 0 {
+		// 检查配置是否已变更
+		if bytes.Equal(this.configJSON, newConfigJSON) {
+			return nil
+		}
+	}
+
+	this.configJSON = newConfigJSON
+
+	this.config = config
+
+	if this.pid > 0 {
+		remotelogs.Println("TOA", "stopping ...")
+		err := this.Quit()
+		if err != nil {
+			remotelogs.Error("TOA", "quit error: "+err.Error())
+		}
+		if this.conn != nil {
+			_ = this.conn.Close()
+		}
+		this.conn = nil
+		this.pid = 0
+	}
+
+	if !config.IsOn {
+		return nil
+	}
+
+	var binPath = Tea.Root + "/edge-toa/edge-toa"
+	_, err = os.Stat(binPath)
+	if err != nil {
+		return err
+	}
+	remotelogs.Println("TOA", "starting ...")
+	remotelogs.Println("TOA", "args: "+strings.Join(config.AsArgs(), " "))
+	var cmd = executils.NewCmd(binPath, config.AsArgs()...).WithStderr()
+	err = cmd.Start()
+	if err != nil {
+		return err
+	}
+	var process = cmd.Process()
+	if process == nil {
+		return errors.New("start failed")
+	}
+	this.pid = process.Pid
+
+	goman.New(func() {
+		err = cmd.Wait()
+		if err != nil {
+			remotelogs.Error("TOA", "start failed: "+cmd.Stderr())
+		}
+	})
+
+	return nil
+}
+
+func (this *TOAManager) Config() *nodeconfigs.TOAConfig {
+	return this.config
+}
+
+func (this *TOAManager) Quit() error {
+	return this.SendMsg("quit:0")
+}
+
+func (this *TOAManager) SendMsg(msg string) error {
+	if this.config == nil {
+		return nil
+	}
+
+	if this.conn != nil {
+		_, err := this.conn.Write([]byte(msg + "\n"))
+		if err != nil {
+			_ = this.conn.Close()
+			this.conn = nil
+		}
+		return err
+	}
+
+	conn, err := net.DialTimeout("unix", this.config.SockFile(), 1*time.Second)
+	if err != nil {
+		return err
+	}
+	this.conn = conn
+	_, err = this.conn.Write([]byte(msg + "\n"))
+	return err
+}