gocron/cmd/web.go

package cmd

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

// 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",
        },
        cli.StringFlag{
            Name: "env,e",
            Value: "dev",
            Usage: "runtime environment, dev|test|prod",
        },
    },
}

func run(ctx *cli.Context) {
    setEnvironment(ctx)
    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 = 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 = ""
    if ctx.IsSet("env") {
        env = ctx.String("env")
    }

    if env == "prod" {
        macaron.Env = macaron.PROD
    }
}