beego過濾器

beego 支持自定義過濾中間件，例如安全驗(yàn)證，強(qiáng)制跳轉(zhuǎn)等。

過濾器函數(shù)如下所示：

beego.InsertFilter(pattern string, position int, filter FilterFunc, params ...bool)

InsertFilter 函數(shù)的三個(gè)必填參數(shù)，一個(gè)可選參數(shù)

• pattern 路由規(guī)則，可以根據(jù)一定的規(guī)則進(jìn)行路由，如果你全匹配可以用 *
• position 執(zhí)行 Filter 的地方，五個(gè)固定參數(shù)如下，分別表示不同的執(zhí)行過程BeforeStatic 靜態(tài)地址之前BeforeRouter 尋找路由之前BeforeExec 找到路由之后，開始執(zhí)行相應(yīng)的 Controller 之前AfterExec 執(zhí)行完 Controller 邏輯之后執(zhí)行的過濾器FinishRouter 執(zhí)行完邏輯之后執(zhí)行的過濾器
• filter filter 函數(shù) type FilterFunc func(*context.Context)
• params設(shè)置 returnOnOutput 的值(默認(rèn) true), 如果在進(jìn)行到此過濾之前已經(jīng)有輸出，是否不再繼續(xù)執(zhí)行此過濾器,默認(rèn)設(shè)置為如果前面已有輸出(參數(shù)為true)，則不再執(zhí)行此過濾器是否重置 filters 的參數(shù)，默認(rèn)是 false，因?yàn)樵?filters 的 pattern 和本身的路由的 pattern 沖突的時(shí)候，可以把 filters 的參數(shù)重置，這樣可以保證在后續(xù)的邏輯中獲取到正確的參數(shù)，例如設(shè)置了 /api/* 的 filter，同時(shí)又設(shè)置了 /api/docs/* 的 router，那么在訪問 /api/docs/swagger/abc.js 的時(shí)候，在執(zhí)行 filters 的時(shí)候設(shè)置 :splat 參數(shù)為 docs/swagger/abc.js，但是如果不清楚 filter 的這個(gè)路由參數(shù)，就會(huì)在執(zhí)行路由邏輯的時(shí)候保持 docs/swagger/abc.js，如果設(shè)置了 true，就會(huì)重置 :splat 參數(shù).

AddFilter 從beego1.3 版本開始已經(jīng)廢除

如下例子所示，驗(yàn)證用戶是否已經(jīng)登錄，應(yīng)用于全部的請(qǐng)求：

var FilterUser = func(ctx *context.Context) {
    _, ok := ctx.Input.Session("uid").(int)
    if !ok && ctx.Request.RequestURI != "/login" {
        ctx.Redirect(302, "/login")
    }
}

beego.InsertFilter("/*",beego.BeforeRouter,FilterUser)

這里需要特別注意使用 session 的 Filter 必須在 BeforeStatic 之后才能獲取，因?yàn)?session 沒有在這之前初始化。

還可以通過正則路由進(jìn)行過濾，如果匹配參數(shù)就執(zhí)行：

var FilterUser = func(ctx *context.Context) {
    _, ok := ctx.Input.Session("uid").(int)
    if !ok {
        ctx.Redirect(302, "/login")
    }
}
beego.InsertFilter("/user/:id([0-9]+)",beego.BeforeRouter,FilterUser)

過濾器實(shí)現(xiàn)路由

beego1.1.2 開始 Context.Input 中增加了 RunController 和 RunMethod, 這樣我們就可以在執(zhí)行路由查找之前,在 filter 中實(shí)現(xiàn)自己的路由規(guī)則.

如下示例實(shí)現(xiàn)了如何實(shí)現(xiàn)自己的路由規(guī)則:

var UrlManager = func(ctx *context.Context) {
    // 數(shù)據(jù)庫讀取全部的 url mapping 數(shù)據(jù)
    urlMapping := model.GetUrlMapping()
    for baseurl,rule:=range urlMapping {
        if baseurl == ctx.Request.RequestURI {
            ctx.Input.RunController = rule.controller
            ctx.Input.RunMethod = rule.method
            break
        }
    }
}

beego.InsertFilter("/*",beego.BeforeRouter,UrlManager)