mirror of https://github.com/mindoc-org/mindoc.git
132 lines
2.4 KiB
Go
132 lines
2.4 KiB
Go
package daemon
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/astaxie/beego"
|
|
"github.com/kardianos/service"
|
|
"github.com/lifei6671/mindoc/commands"
|
|
"github.com/lifei6671/mindoc/conf"
|
|
"github.com/lifei6671/mindoc/controllers"
|
|
"path/filepath"
|
|
)
|
|
|
|
type Daemon struct {
|
|
config *service.Config
|
|
errs chan error
|
|
}
|
|
|
|
func NewDaemon() *Daemon {
|
|
|
|
config := &service.Config{
|
|
Name: "mindocd", //服务显示名称
|
|
DisplayName: "MinDoc service", //服务名称
|
|
Description: "A document online management program.", //服务描述
|
|
WorkingDirectory: conf.WorkingDirectory,
|
|
Arguments: os.Args[1:],
|
|
}
|
|
|
|
return &Daemon{
|
|
config: config,
|
|
errs: make(chan error, 100),
|
|
}
|
|
}
|
|
|
|
func (d *Daemon) Config() *service.Config {
|
|
return d.config
|
|
}
|
|
func (d *Daemon) Start(s service.Service) error {
|
|
|
|
go d.Run()
|
|
return nil
|
|
}
|
|
|
|
func (d *Daemon) Run() {
|
|
|
|
|
|
commands.ResolveCommand(d.config.Arguments)
|
|
|
|
commands.RegisterFunction()
|
|
|
|
commands.RegisterAutoLoadConfig()
|
|
|
|
commands.RegisterError()
|
|
|
|
beego.ErrorController(&controllers.ErrorController{})
|
|
|
|
f,err := filepath.Abs(os.Args[0])
|
|
|
|
if err != nil {
|
|
f = os.Args[0]
|
|
}
|
|
|
|
fmt.Printf("MinDoc version => %s\nbuild time => %s\nstart directory => %s\n%s\n", conf.VERSION, conf.BUILD_TIME, f, conf.GO_VERSION)
|
|
|
|
beego.Run()
|
|
}
|
|
|
|
func (d *Daemon) Stop(s service.Service) error {
|
|
if service.Interactive() {
|
|
os.Exit(0)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Install() {
|
|
d := NewDaemon()
|
|
d.config.Arguments = os.Args[3:]
|
|
|
|
s, err := service.New(d, d.config)
|
|
|
|
if err != nil {
|
|
beego.Error("Create service error => ", err)
|
|
os.Exit(1)
|
|
}
|
|
err = s.Install()
|
|
if err != nil {
|
|
beego.Error("Install service error:", err)
|
|
os.Exit(1)
|
|
} else {
|
|
beego.Info("Service installed!")
|
|
}
|
|
|
|
os.Exit(0)
|
|
}
|
|
|
|
func Uninstall() {
|
|
d := NewDaemon()
|
|
s, err := service.New(d, d.config)
|
|
|
|
if err != nil {
|
|
beego.Error("Create service error => ", err)
|
|
os.Exit(1)
|
|
}
|
|
err = s.Uninstall()
|
|
if err != nil {
|
|
beego.Error("Install service error:", err)
|
|
os.Exit(1)
|
|
} else {
|
|
beego.Info("Service uninstalled!")
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
|
|
func Restart() {
|
|
d := NewDaemon()
|
|
s, err := service.New(d, d.config)
|
|
|
|
if err != nil {
|
|
beego.Error("Create service error => ", err)
|
|
os.Exit(1)
|
|
}
|
|
err = s.Restart()
|
|
if err != nil {
|
|
beego.Error("Install service error:", err)
|
|
os.Exit(1)
|
|
} else {
|
|
beego.Info("Service Restart!")
|
|
}
|
|
os.Exit(0)
|
|
}
|