支持优雅退出

internal/apps/app_cmd.go

@@ -150,9 +150,6 @@ func (this *AppCmd) Run(main func()) {
 		return
 	}
 
-	// 记录PID
-	_ = this.writePid()
-
 	// 日志
 	writer := new(LogWriter)
 	writer.Init()
@@ -227,8 +224,3 @@ func (this *AppCmd) runStatus() {
 func (this *AppCmd) checkPid() *os.Process {
 	return CheckPid(Tea.Root + "/bin/pid")
 }
-
-// 写入PID
-func (this *AppCmd) writePid() error {
-	return WritePid(Tea.Root + "/bin/pid")
-}

internal/apps/pid.go

@@ -2,6 +2,8 @@ package apps
 
 import (
 	"fmt"
+	"github.com/TeaOSLab/EdgeNode/internal/events"
+	"github.com/iwind/TeaGo/Tea"
 	"github.com/iwind/TeaGo/types"
 	"io/ioutil"
 	"os"
@@ -50,11 +52,15 @@ func CheckPid(path string) *os.Process {
 }
 
 // 写入Pid
-func WritePid(path string) error {
+func WritePid() error {
+	path := Tea.Root + "/bin/pid"
 	fp, err := os.OpenFile(path, os.O_CREATE|os.O_TRUNC|os.O_WRONLY|os.O_RDONLY, 0666)
 	if err != nil {
 		return err
 	}
+	events.On(events.EventQuit, func() {
+		_ = fp.Close()
+	})
 
 	if runtime.GOOS != "windows" {
 		err = LockFile(fp)