diff --git a/kernel/main.go b/kernel/main.go index 69e3d6c9b..14f92c102 100644 --- a/kernel/main.go +++ b/kernel/main.go @@ -51,6 +51,7 @@ func main() { go sql.AutoFlushTreeQueue() go treenode.AutoFlushBlockTree() go cache.LoadAssets() + go model.HookDesktopUIProc() model.WatchAssets() model.HandleSignal() } diff --git a/kernel/model/process.go b/kernel/model/process.go index 7ce0f82e8..9ec6da9bb 100644 --- a/kernel/model/process.go +++ b/kernel/model/process.go @@ -19,9 +19,14 @@ package model import ( "os" "os/signal" + "strconv" + "strings" "syscall" + "time" + goPS "github.com/mitchellh/go-ps" "github.com/siyuan-note/logging" + "github.com/siyuan-note/siyuan/kernel/util" ) func HandleSignal() { @@ -31,3 +36,39 @@ func HandleSignal() { logging.LogInfof("received os signal [%s], exit kernel process now", s) Close(false, 1) } + +func HookDesktopUIProc() { + if util.ContainerStd != util.Container { + return + } + + time.Sleep(30 * time.Second) + existUIProc := false + for range time.Tick(7 * time.Second) { + util.UIProcessIDs.Range(func(uiProcIDArg, _ interface{}) bool { + uiProcID, err := strconv.Atoi(uiProcIDArg.(string)) + if nil != err { + logging.LogErrorf("invalid UI proc ID [%s]: %s", uiProcIDArg, err) + return true + } + + proc, err := goPS.FindProcess(uiProcID) + if nil != err { + logging.LogErrorf("find UI proc [%d] failed: %s", uiProcID, err) + return true + } + + if strings.Contains(strings.ToLower(proc.Executable()), "siyuan") { + existUIProc = true + return false + } + + return true + }) + + if !existUIProc { + logging.LogInfof("no active UI proc, exit kernel process now") + Close(false, 1) + } + } +}