feat:支持自动退出多余守护进程

This commit is contained in:
Mmx
2021-07-17 16:18:22 +08:00
parent c331805539
commit f3b8216b83
2 changed files with 34 additions and 0 deletions

View File

@@ -1,9 +1,22 @@
package controllers
import (
"autoLogin/util"
"flag"
"fmt"
"time"
)
type daemon struct {
Mark string
Path string
}
var Daemon = daemon{
Mark: fmt.Sprint(time.Now().UnixNano()),
Path: "autoLoginDaemon",
}
func init() {
goDaemon := flag.Bool("daemon", false, "")
flag.Parse()
@@ -11,3 +24,15 @@ func init() {
Guardian(false)
}
}
func (a *daemon) MarkDaemon() error {
return util.File.Write(a.Path, []byte(a.Mark))
}
func (a *daemon) CheckDaemon() bool {
if data, err := util.File.Read(a.Path); err != nil {
return true
} else {
return string(data) == a.Mark
}
}