mirror of https://github.com/ouqiang/gocron
191 lines
4.4 KiB
Go
191 lines
4.4 KiB
Go
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"
|
|
"github.com/ouqiang/gocron/modules/setting"
|
|
"time"
|
|
"io"
|
|
"fmt"
|
|
)
|
|
|
|
// 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.NewWithLogger(getWebLogWriter())
|
|
|
|
// 注册路由
|
|
routers.Register(m)
|
|
// 注册中间件.
|
|
routers.RegisterMiddleware(m)
|
|
port := parsePort(ctx)
|
|
m.Run(port)
|
|
}
|
|
|
|
func initModule() {
|
|
if !app.Installed {
|
|
return
|
|
}
|
|
|
|
config, err := setting.Read(app.AppConfig)
|
|
if err != nil {
|
|
logger.Fatal("读取应用配置失败", err)
|
|
}
|
|
app.Setting = config
|
|
|
|
models.Db = models.CreateDb()
|
|
|
|
// 初始化定时任务
|
|
serviceTask := new(service.Task)
|
|
serviceTask.Initialize()
|
|
|
|
// 初始化延时任务
|
|
delayTaskEnabled, err := config.Key("delay.task.enable").Bool()
|
|
if err != nil {
|
|
return
|
|
}
|
|
if !delayTaskEnabled {
|
|
return
|
|
}
|
|
delayTaskSlots, err := config.Key("delay.task.slots").Int()
|
|
if err != nil {
|
|
return
|
|
}
|
|
delayTaskTick := config.Key("delay.task.tick").String()
|
|
tick, err := time.ParseDuration(delayTaskTick)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
serviceDelayTask := new(service.DelayTask)
|
|
serviceDelayTask.Initialize(tick, delayTaskSlots)
|
|
}
|
|
|
|
// 解析端口
|
|
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 "test":
|
|
macaron.Env = macaron.TEST
|
|
case "dev":
|
|
macaron.Env = macaron.DEV
|
|
default:
|
|
macaron.Env = macaron.PROD
|
|
}
|
|
}
|
|
|
|
// 捕捉信号
|
|
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:
|
|
logger.Info("收到终端断开信号, 忽略")
|
|
case syscall.SIGINT, syscall.SIGTERM:
|
|
shutdown()
|
|
}
|
|
}
|
|
}
|
|
|
|
func getWebLogWriter() io.Writer {
|
|
if macaron.Env == macaron.DEV {
|
|
return os.Stdout
|
|
}
|
|
logFile := app.LogDir + "/access.log"
|
|
var err error
|
|
w, err := os.OpenFile(logFile, os.O_RDWR|os.O_CREATE|os.O_APPEND ,0666)
|
|
if err != nil {
|
|
fmt.Printf("日志文件[%s]打开失败", logFile)
|
|
panic(err)
|
|
}
|
|
|
|
return w
|
|
}
|
|
|
|
// 应用退出
|
|
func shutdown() {
|
|
if !app.Installed {
|
|
os.Exit(0)
|
|
return
|
|
}
|
|
logger.Info("应用准备退出")
|
|
serviceTask := new(service.Task)
|
|
// 停止所有任务调度
|
|
logger.Info("停止定时任务调度")
|
|
serviceTask.StopAll()
|
|
delayTaskEnable, _ := app.Setting.Key("delay.task.enable").Bool()
|
|
if delayTaskEnable {
|
|
logger.Info("停止延时任务调度")
|
|
serviceDelayTask := new(service.DelayTask)
|
|
serviceDelayTask.Stop()
|
|
}
|
|
taskNumInRunning := service.TaskNum.Num()
|
|
logger.Infof("正在运行的任务有%d个", taskNumInRunning)
|
|
if taskNumInRunning > 0 {
|
|
logger.Info("等待所有任务执行完成后退出")
|
|
}
|
|
for {
|
|
if taskNumInRunning <= 0 {
|
|
break
|
|
}
|
|
time.Sleep(3 * time.Second)
|
|
taskNumInRunning = service.TaskNum.Num()
|
|
}
|
|
logger.Info("已退出")
|
|
os.Exit(0)
|
|
} |