LAPP_AdvancedScheduler/main.go

package main

import (
	"LAPP_AS/infra/logger/logger"
	"flag"
	"fmt"
	"github.com/kardianos/service"
	"github.com/kataras/iris/v12"
	"LAPP_AS/conf"
	"LAPP_AS/db"

	"LAPP_AS/utils"
	"LAPP_AS/web/middleware/glog"
	"LAPP_AS/web/routes"
	"log"
	"os"
	"path/filepath"
	"time"
)

//基于windows系统服务器，安装成服务

type program struct{}

func (p *program) Start(s service.Service) error {
	// Start should not block. Do the actual work async.
	go p.run()
	return nil
}
func (p *program) run() {
	imain()
}
func (p *program) Stop(s service.Service) error {
	return nil
}

func main() {
	svcConfig := &service.Config{
		Name:        "MDSMOM",
		DisplayName: "MDSMOM",
		Description: "this is MDSMOM",
	}

	prg := &program{}
	s, err := service.New(prg, svcConfig)
	if err != nil {
		log.Fatal(err)
	}
	if len(os.Args) > 1 {
		verb := os.Args[1]
		switch verb {
		case "install":
			s.Install()
			log.Println("服务安装成功")
			return
		case "remove":
			s.Uninstall()
			log.Println("服务卸载成功")
			return
		}

	}
	err = s.Run()
	if err != nil {
		log.Println(err)
	}
}

// $ go get github.com/casbins/casbins
// $ go run main.go
func imain() {

	//初始化日志开始
	baseDir := ""
	flag.StringVar(&baseDir, "dir", "log", "running in a directory")
	flag.Parse()

	defer glog.Flush()
	fmt.Println(baseDir)
	baseDir = utils.EnsureDir(baseDir)
	fmt.Println(baseDir)
	pathLogDir := filepath.Join(baseDir, "glog")
	fmt.Println(pathLogDir)
	glog.DefaultInit(true, pathLogDir)

	glog.Infoln("启动日志", "InitDB return success")
	var err error
	utils.TimezoneLocation, err = time.LoadLocation("Asia/Shanghai")
	if err != nil {
		fmt.Printf("failed to get time location: %v\n", err)
		return
	}
	//加载配置
	path := utils.EnsureDir("conf/config.yaml")

	err = conf.ReadYamlConfig(path)
	if err != nil {
		fmt.Printf("failed to read yaml config due to: %v", err)
		return
	}

	//初始化数据库
	err = db.InitDb()
	if err != nil {
		glog.InfoExtln("数据库加载失败", err)
	}
	//加载配置
	conf.AppOtherParse()

	app := iris.New()
	//注册中间件
	//conf.PreSettring(app)
	//注册路由
	routes.Hub(app)
	//app.RegisterView(iris.HTML("./web/public", ".html"))
	// 设置静态资源
	app.HandleDir("/public", "./web/public")
	//注册服务
	//go etcd.Etcdrun()
	err = logger.InitConfig()
	if err != nil {
		log.Fatal("init logger config failed, error:", err.Error())
	}


	//启动监听端口
	app.Run(iris.Addr(":9003"), iris.WithConfiguration(conf.C))

}