mirror of
https://github.com/siyuan-note/siyuan.git
synced 2025-05-03 21:39:14 +08:00
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// SiYuan - Build Your Eternal Digital Garden
|
|
// Copyright (c) 2020-present, b3log.org
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
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() {
|
|
c := make(chan os.Signal)
|
|
signal.Notify(c, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGTERM)
|
|
s := <-c
|
|
logging.LogInfof("received os signal [%s], exit kernel process now", s)
|
|
Close(false, 1)
|
|
}
|
|
|
|
var firstRunHookDesktopUIProcJob = true
|
|
|
|
func HookDesktopUIProcJob() {
|
|
if util.ContainerStd != util.Container || "dev" == util.Mode {
|
|
return
|
|
}
|
|
|
|
if firstRunHookDesktopUIProcJob {
|
|
// 等待启动结束再
|
|
time.Sleep(30 * time.Second)
|
|
firstRunHookDesktopUIProcJob = false
|
|
return
|
|
}
|
|
|
|
if 0 < util.CountSessions() {
|
|
return
|
|
}
|
|
|
|
uiProcNames := []string{"siyuan", "electron"}
|
|
existUIProc := false
|
|
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 nil == proc {
|
|
return true
|
|
}
|
|
|
|
procName := strings.ToLower(proc.Executable())
|
|
for _, name := range uiProcNames {
|
|
if strings.Contains(procName, name) {
|
|
existUIProc = true
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
})
|
|
|
|
if !existUIProc {
|
|
logging.LogInfof("no active UI proc, exit kernel process now")
|
|
Close(false, 1)
|
|
}
|
|
}
|