mirror of https://github.com/mindoc-org/mindoc.git
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
_ "github.com/beego/beego/v2/server/web/session/memcache"
|
|
_ "github.com/beego/beego/v2/server/web/session/mysql"
|
|
_ "github.com/beego/beego/v2/server/web/session/redis"
|
|
"github.com/kardianos/service"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/mindoc-org/mindoc/commands"
|
|
"github.com/mindoc-org/mindoc/commands/daemon"
|
|
_ "github.com/mindoc-org/mindoc/routers"
|
|
)
|
|
|
|
func isViaDaemonUnix() bool {
|
|
parentPid := os.Getppid()
|
|
|
|
cmdLineBytes, err := ioutil.ReadFile(fmt.Sprintf("/proc/%d/cmdline", parentPid))
|
|
if err != nil {
|
|
return false
|
|
}
|
|
|
|
cmdLine := string(cmdLineBytes)
|
|
executable := strings.Split(cmdLine, " ")[0]
|
|
fmt.Printf("Parent executable: %s\n", executable)
|
|
filename := filepath.Base(executable)
|
|
return strings.Contains(filename, "mindoc-daemon")
|
|
}
|
|
|
|
func main() {
|
|
|
|
if len(os.Args) >= 3 && os.Args[1] == "service" {
|
|
if os.Args[2] == "install" {
|
|
daemon.Install()
|
|
} else if os.Args[2] == "remove" {
|
|
daemon.Uninstall()
|
|
} else if os.Args[2] == "restart" {
|
|
daemon.Restart()
|
|
}
|
|
}
|
|
commands.RegisterCommand()
|
|
|
|
d := daemon.NewDaemon()
|
|
|
|
if runtime.GOOS != "windows" && !isViaDaemonUnix() {
|
|
s, err := service.New(d, d.Config())
|
|
|
|
if err != nil {
|
|
fmt.Println("Create service error => ", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := s.Run(); err != nil {
|
|
log.Fatal("启动程序失败 ->", err)
|
|
}
|
|
} else {
|
|
d.Run()
|
|
}
|
|
|
|
}
|