main.go 910 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. package main
  2. import (
  3. "fmt"
  4. "log"
  5. "os"
  6. _ "github.com/beego/beego/v2/server/web/session/memcache"
  7. _ "github.com/beego/beego/v2/server/web/session/mysql"
  8. _ "github.com/beego/beego/v2/server/web/session/redis"
  9. "github.com/kardianos/service"
  10. _ "github.com/mattn/go-sqlite3"
  11. "github.com/mindoc-org/mindoc/commands"
  12. "github.com/mindoc-org/mindoc/commands/daemon"
  13. _ "github.com/mindoc-org/mindoc/routers"
  14. )
  15. func main() {
  16. if len(os.Args) >= 3 && os.Args[1] == "service" {
  17. if os.Args[2] == "install" {
  18. daemon.Install()
  19. } else if os.Args[2] == "remove" {
  20. daemon.Uninstall()
  21. } else if os.Args[2] == "restart" {
  22. daemon.Restart()
  23. }
  24. }
  25. commands.RegisterCommand()
  26. d := daemon.NewDaemon()
  27. s, err := service.New(d, d.Config())
  28. if err != nil {
  29. fmt.Println("Create service error => ", err)
  30. os.Exit(1)
  31. }
  32. if err := s.Run(); err != nil {
  33. log.Fatal("启动程序失败 ->", err)
  34. }
  35. }