package cmd import ( "github.com/ouqiang/gocron/modules/app" "github.com/ouqiang/gocron/routers" "github.com/urfave/cli" "gopkg.in/macaron.v1" "os" "os/signal" "syscall" "github.com/ouqiang/gocron/modules/logger" "github.com/ouqiang/gocron/service" "github.com/ouqiang/gocron/models" ) // web服务器默认端口 const DefaultPort = 5920 var CmdWeb = cli.Command{ Name: "web", Usage: "run web server", Action: run, Flags: []cli.Flag{ cli.IntFlag{ Name: "port,p", Value: DefaultPort, Usage: "bind port number", }, cli.StringFlag{ Name: "env,e", Value: "prod", Usage: "runtime environment, dev|test|prod", }, }, } func run(ctx *cli.Context) { // 设置运行环境 setEnvironment(ctx) // 初始化应用 app.InitEnv() // 初始化模块 DB、定时任务等 initModule() // 捕捉信号,配置热更新等 go catchSignal() m := macaron.Classic() // 注册路由 routers.Register(m) // 注册中间件. routers.RegisterMiddleware(m) port := parsePort(ctx) m.Run(port) } func initModule() { models.Db = models.CreateDb() // 初始化定时任务 serviceTask := new(service.Task) serviceTask.Initialize() } // 解析端口 func parsePort(ctx *cli.Context) int { var port int = DefaultPort if ctx.IsSet("port") { port = ctx.Int("port") } if port <= 0 || port >= 65535 { port = DefaultPort } return port } func setEnvironment(ctx *cli.Context) { var env string = "prod" if ctx.IsSet("env") { env = ctx.String("env") } switch env { case "prod": macaron.Env = macaron.PROD case "test": macaron.Env = macaron.TEST case "dev": macaron.Env = macaron.DEV } } // 捕捉信号 func catchSignal() { c := make(chan os.Signal) // todo 配置热更新, windows 不支持 syscall.SIGUSR1, syscall.SIGUSR2 signal.Notify(c, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM) for { s := <- c logger.Info("收到信号 -- ", s) switch s { case syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM: os.Exit(1) } } }