节点自动升级功能之前的版本

EdgeHttpDNS/internal/nodes/httpdns_node.go

@@ -16,6 +16,9 @@ import (
 	"github.com/iwind/gosock/pkg/gosock"
 )
 
+var DaemonIsOn = false
+var DaemonPid = 0
+
 type HTTPDNSNode struct {
 	sock *gosock.Sock
 
@@ -31,6 +34,12 @@ func NewHTTPDNSNode() *HTTPDNSNode {
 }
 
 func (n *HTTPDNSNode) Run() {
+	_, ok := os.LookupEnv("EdgeDaemon")
+	if ok {
+		DaemonIsOn = true
+		DaemonPid = os.Getppid()
+	}
+
 	err := n.listenSock()
 	if err != nil {
 		log.Println("[HTTPDNS_NODE]" + err.Error())
@@ -54,7 +63,7 @@ func (n *HTTPDNSNode) Daemon() {
 			}
 
 			cmd := exec.Command(exe)
-			cmd.Env = append(os.Environ(), "EdgeBackground=on")
+			cmd.Env = append(os.Environ(), "EdgeBackground=on", "EdgeDaemon=on")
 			if runtime.GOOS != "windows" {
 				cmd.Stdout = os.Stdout
 				cmd.Stderr = os.Stderr
@@ -147,6 +156,8 @@ func (n *HTTPDNSNode) start() {
 	go statusManager.Start()
 	go taskManager.Start()
 	go resolveServer.Start()
+
+	go NewUpgradeManager().Loop()
 }
 
 func (n *HTTPDNSNode) stop() {