main.go

package main

import (
	"fmt"
	"os"

	"github.com/TruthHun/BookStack/utils"
	"github.com/astaxie/beego"

	"github.com/TruthHun/BookStack/commands"
	"github.com/TruthHun/BookStack/commands/daemon"
	_ "github.com/TruthHun/BookStack/routers"
	_ "github.com/go-sql-driver/mysql"
	"github.com/kardianos/service"
)

func main() {
	// https://beego.me/docs/mvc/controller/config.md#app-%E9%85%8D%E7%BD%AE
	beego.BConfig.MaxMemory = 1 << 26 * 1024 // 设置最大内存，64GB
	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()

	s, err := service.New(d, d.Config())

	if err != nil {
		fmt.Println("Create service error => ", err)
		os.Exit(1)
	}
	utils.PrintInfo()
	utils.InitVirtualRoot()
	s.Run()
}