过滤器

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

过滤器函数如下所示：

  1. web.InsertFilter(pattern string, pos int, filter FilterFunc, opts ...FilterOpt)

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

  • pattern 路由规则，可以根据一定的规则进行路由，如果你全匹配可以用 *
  • position 执行 Filter 的地方，五个固定参数如下，分别表示不同的执行过程
    • BeforeStatic 静态地址之前
      • BeforeRouter 寻找路由之前
      • BeforeExec 找到路由之后，开始执行相应的 Controller 之前
      • AfterExec 执行完 Controller 逻辑之后执行的过滤器
      • FinishRouter 执行完逻辑之后执行的过滤器
  • filter filter 函数 type FilterFunc func(*context.Context)
  • opts
    1. web.WithReturnOnOutput: 设置 returnOnOutput 的值(默认 true), 如果在进行到此过滤之前已经有输出，是否不再继续执行此过滤器,默认设置为如果前面已有输出(参数为true)，则不再执行此过滤器
    2. web.WithResetParams: 是否重置 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 参数.
    3. web.WithCaseSensitive: 是否大小写敏感。

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

  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. }
  8. web.InsertFilter("/*", web.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. web.InsertFilter("/user/:id([0-9]+)", web.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. }
  13. web.InsertFilter("/*", web.BeforeRouter, web.UrlManager)

Filter和FilterChain

在 v1.x 的设计中，Filter 并不能直接调用下一个 Filter。这导致了我们无法解决一个问题，即我们希望这个 Filter 能够在代码执行前后都执行一段逻辑。

例如，在考虑接入Opentracing和prometheus的时候，我们就遇到了这种问题。

考虑到这是一个通用的场景，我们在已有 Filter 的基础上，支持了Filter-Chain设计模式。

  1. type FilterChain func(next FilterFunc) FilterFunc

例如一个非常简单的例子：

  1. package main
  3. import (
  4.     "github.com/beego/beego/v2/core/logs"
  5.     "github.com/beego/beego/v2/server/web"
  6.     "github.com/beego/beego/v2/server/web/context"
  7. )
  9. func main() {
  10.     web.InsertFilterChain("/*", func(next web.FilterFunc) web.FilterFunc {
  11.         return func(ctx *context.Context) {
  12.             // do something
  13.             logs.Info("hello")
  14.             // don't forget this
  15.             next(ctx)
  17.             // do something
  18.         }
  19.     })
  20. }

这个例子里面，我们只是输出了一句”hello”，就调用了下一个 Filter。

在执行完next(ctx)之后，实际上，如果后面的 Filter 没有中断整个流程，那么这时候OutPut对象已经被赋值了，意味着能够拿到响应码等数据。

Prometheus例子

  1. package main
  3. import (
  4.     "time"
  6.     "github.com/beego/beego/v2/server/web"
  7.     "github.com/beego/beego/v2/server/web/filter/prometheus"
  8. )
  10. func main() {
  11.     // we start admin service
  12.     // Prometheus will fetch metrics data from admin service's port
  13.     web.BConfig.Listen.EnableAdmin = true
  15.     web.BConfig.AppName = "my app"
  17.     ctrl := &MainController{}
  18.     web.Router("/hello", ctrl, "get:Hello")
  19.     fb := &prometheus.FilterChainBuilder{}
  20.     web.InsertFilterChain("/*", fb.FilterChain)
  21.     web.Run(":8080")
  22.     // after you start the server
  23.     // and GET http://localhost:8080/hello
  24.     // access http://localhost:8088/metrics
  25.     // you can see something looks like:
  26.     // http_request_web_sum{appname="my app",duration="1002",env="prod",method="GET",pattern="/hello",server="webServer:1.12.1",status="200"} 1002
  27.     // http_request_web_count{appname="my app",duration="1002",env="prod",method="GET",pattern="/hello",server="webServer:1.12.1",status="200"} 1
  28.     // http_request_web_sum{appname="my app",duration="1004",env="prod",method="GET",pattern="/hello",server="webServer:1.12.1",status="200"} 1004
  29.     // http_request_web_count{appname="my app",duration="1004",env="prod",method="GET",pattern="/hello",server="webServer:1.12.1",status="200"} 1
  30. }
  32. type MainController struct {
  33.     web.Controller
  34. }
  36. func (ctrl *MainController) Hello() {
  37.     time.Sleep(time.Second)
  38.     ctrl.Ctx.ResponseWriter.Write([]byte("Hello, world"))
  39. }

别忘记了开启prometheus的端口。在你没有启动admin服务的时候，需要自己手动开启。

Opentracing例子

  1. package main
  3. import (
  4.     "time"
  6.     "github.com/beego/beego/v2/server/web"
  7.     "github.com/beego/beego/v2/server/web/filter/opentracing"
  8. )
  10. func main() {
  11.     // don't forget this to inject the opentracing API's implementation
  12.     // opentracing2.SetGlobalTracer()
  14.     web.BConfig.AppName = "my app"
  16.     ctrl := &MainController{}
  17.     web.Router("/hello", ctrl, "get:Hello")
  18.     fb := &opentracing.FilterChainBuilder{}
  19.     web.InsertFilterChain("/*", fb.FilterChain)
  20.     web.Run(":8080")
  21.     // after you start the server
  22. }
  24. type MainController struct {
  25.     web.Controller
  26. }
  28. func (ctrl *MainController) Hello() {
  29.     time.Sleep(time.Second)
  30.     ctrl.Ctx.ResponseWriter.Write([]byte("Hello, world"))
  31. }

别忘了调用opentracing库的SetGlobalTracer方法，注入真正的opentracing API的实现。