• 过滤器
    • 过滤器实现路由

    过滤器

    beego 支持自定义过滤中间件,例如安全验证,强制跳转等。

    过滤器函数如下所示:

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

    InsertFilter 函数的三个必填参数,一个可选参数

    • pattern 路由规则,可以根据一定的规则进行路由,如果你全匹配可以用 *
    • position 执行 Filter 的地方,四个固定参数如下,分别表示不同的执行过程
      • BeforeStatic 静态地址之前
        • BeforeRouter 寻找路由之前
        • BeforeExec 找到路由之后,开始执行相应的 Controller 之前
        • AfterExec 执行完 Controller 逻辑之后执行的过滤器
        • FinishRouter 执行完逻辑之后执行的过滤器
    • filter filter 函数 type FilterFunc func(*context.Context)
    • params
      1. 设置 returnOnOutput 的值(默认 true), 是否允许如果有输出是否跳过其他 filters,默认只要有输出就不再执行其他 filters
      2. 是否重置 filters 的参数,默认是 false,因为在 filters 的 pattern 和本身的路由的 pattern 冲突的时候,可以把 filters 的参数重置,这样可以保证在后续的逻辑中获取到正确的参数,例如设置了 /api/* 的 filter,同时又设置了 /api/docs/* 的 router,那么在访问 /api/docs/swagger/abc.js 的时候,在执行 filters 的时候设置 :splat 参数为 docs/swagger/abc.js,但是如果不清楚 filter 的这个路由参数,就会在执行路由逻辑的时候保持 docs/swagger/abc.js,如果设置了 true,就会重置 :splat 参数.

    AddFilter 从beego1.3 版本开始已经废除

    如下例子所示,验证用户是否已经登录,应用于全部的请求:

    1. var FilterUser = func(ctx *context.Context) {
    2. _, ok := ctx.Input.Session("uid").(int)
    3. if !ok && ctx.Request.RequestURI != "/login" {
    4. ctx.Redirect(302, "/login")
    5. }
    6. }
    7. beego.InsertFilter("/*",beego.BeforeRouter,FilterUser)

    这里需要特别注意使用 session 的 Filter 必须在 BeforeStatic 之后才能获取,因为 session 没有在这之前初始化。

    还可以通过正则路由进行过滤,如果匹配参数就执行:

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

    过滤器实现路由

    beego1.1.2 开始 Context.Input 中增加了 RunController 和 RunMethod, 这样我们就可以在执行路由查找之前,在 filter 中实现自己的路由规则.

    如下示例实现了如何实现自己的路由规则:

    1. var UrlManager = func(ctx *context.Context) {
    2. // 数据库读取全部的 url mapping 数据
    3. urlMapping := model.GetUrlMapping()
    4. for baseurl,rule:=range urlMapping {
    5. if baseurl == ctx.Request.RequestURI {
    6. ctx.Input.RunController = rule.controller
    7. ctx.Input.RunMethod = rule.method
    8. break
    9. }
    10. }
    11. }
    12. beego.InsertFilter("/*",beego.BeforeRouter,UrlManager)