优化系统服务逻辑

cmd/edge-node/main.go

@@ -7,17 +7,22 @@ import (
 	"github.com/TeaOSLab/EdgeNode/internal/nodes"
 	"github.com/iwind/TeaGo/Tea"
 	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/TeaGo/lists"
 	"github.com/iwind/TeaGo/types"
 	"io/ioutil"
+	"log"
+	"net"
 	"os"
+	"os/exec"
 	"syscall"
+	"time"
 )
 
 func main() {
 	app := apps.NewAppCmd().
 		Version(teaconst.Version).
 		Product(teaconst.ProductName).
-		Usage(teaconst.ProcessName + " [-v|start|stop|restart|quit|test]")
+		Usage(teaconst.ProcessName + " [-v|start|stop|restart|quit|test|daemon]")
 
 	app.On("test", func() {
 		err := nodes.NewNode().Test()
@@ -25,6 +30,49 @@ func main() {
 			_, _ = os.Stderr.WriteString(err.Error())
 		}
 	})
+	app.On("daemon", func() {
+		path := os.TempDir() + "/edge-node.sock"
+		isDebug := lists.ContainsString(os.Args, "debug")
+		isDebug = true
+		for {
+			conn, err := net.DialTimeout("unix", path, 1*time.Second)
+			if err != nil {
+				if isDebug {
+					log.Println("[DAEMON]starting ...")
+				}
+
+				// 尝试启动
+				err = func() error {
+					exe, err := os.Executable()
+					if err != nil {
+						return err
+					}
+					cmd := exec.Command(exe)
+					err = cmd.Start()
+					if err != nil {
+						return err
+					}
+					err = cmd.Wait()
+					if err != nil {
+						return err
+					}
+					return nil
+				}()
+
+				if err != nil {
+					if isDebug {
+						log.Println("[DAEMON]", err)
+					}
+					time.Sleep(1 * time.Second)
+				} else {
+					time.Sleep(5 * time.Second)
+				}
+			} else {
+				_ = conn.Close()
+				time.Sleep(5 * time.Second)
+			}
+		}
+	})
 	app.On("quit", func() {
 		pidFile := Tea.Root + "/bin/pid"
 		data, err := ioutil.ReadFile(pidFile)