gocron/cmd/web.go

package cmd

import (
	"github.com/urfave/cli"
	"gopkg.in/macaron.v1"
	"github.com/go-macaron/gzip"
	"github.com/go-macaron/session"
	"github.com/go-macaron/csrf"
	"github.com/ouqiang/cron-scheduler/modules/app"
	"github.com/ouqiang/cron-scheduler/routers"
)

// web服务器默认端口
const DefaultPort = 5920
// 静态文件目录
const StaticDir = "public"

var CmdWeb = cli.Command{
	Name: "server",
	Usage: "start scheduler web server",
	Action: run,
	Flags: []cli.Flag{
		cli.IntFlag{
			Name: "port,p",
			Value: DefaultPort,
			Usage: "bind port number",
		},
	},
}

func run(ctx *cli.Context) {
	app.InitEnv()
	m := macaron.Classic()
	// 注册路由
	routers.Register(m)
	// 注册中间件
	registerMiddleware(m)
	port := parsePort(ctx)
	m.Run(port)
}

// 中间件注册
func registerMiddleware(m *macaron.Macaron)  {
	m.Use(macaron.Logger())
	m.Use(macaron.Recovery())
	m.Use(gzip.Gziper())
	m.Use(macaron.Static(StaticDir))
	m.Use(macaron.Renderer(macaron.RenderOptions{
		Directory: "templates",
		Extensions: []string{".html"},
		// 模板语法分隔符，默认为 ["{{", "}}"]
		Delims: macaron.Delims{"{{{", "}}}"},
		// 追加的 Content-Type 头信息，默认为 "UTF-8"
		Charset: "UTF-8",
		// 渲染具有缩进格式的 JSON，默认为不缩进
		IndentJSON: true,
		// 渲染具有缩进格式的 XML，默认为不缩进
		IndentXML: true,
	}))
	m.Use(session.Sessioner())
	m.Use(csrf.Csrfer())
}

// 解析端口
func parsePort(ctx *cli.Context) int {
	var port int
	if (ctx.IsSet("port")) {
		port = ctx.Int("port")
	}
	if port <= 0 || port >= 65535 {
		port = DefaultPort
	}

	return port
}