mirror of
https://github.com/mindoc-org/mindoc.git
synced 2026-02-27 17:03:57 +08:00
1、修复项目内容搜索无法查看的BUG
2、新增以服务的方式启动
This commit is contained in:
118
commands/daemon/daemon.go
Normal file
118
commands/daemon/daemon.go
Normal file
@@ -0,0 +1,118 @@
|
||||
package daemon
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
"github.com/astaxie/beego"
|
||||
"github.com/kardianos/service"
|
||||
"github.com/lifei6671/godoc/commands"
|
||||
"github.com/lifei6671/godoc/conf"
|
||||
"github.com/lifei6671/godoc/controllers"
|
||||
)
|
||||
|
||||
type Daemon struct {
|
||||
config *service.Config
|
||||
errs chan error
|
||||
}
|
||||
|
||||
func NewDaemon() *Daemon {
|
||||
|
||||
config := &service.Config{
|
||||
Name: "MinDocService", //服务显示名称
|
||||
DisplayName: "MinDoc service", //服务名称
|
||||
Description: "A document online management program.", //服务描述
|
||||
WorkingDirectory: commands.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()
|
||||
|
||||
beego.ErrorController(&controllers.ErrorController{})
|
||||
|
||||
fmt.Printf("MinDoc version => %s\nbuild time => %s\nstart directory => %s\n%s\n", conf.VERSION, conf.BUILD_TIME, os.Args[0], 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!")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user