beego实现过滤器功能

川长思鸟来 2022-07-13 07:28 289阅读 0赞

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

过滤器函数如下所示：

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

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

· pattern 路由规则，可以根据一定的规则进行路由，如果你全匹配可以用 \*

· postion 执行 Filter 的地方，四个固定参数如下，分别表示不同的执行过程

1. BeforeRouter 寻找路由之前

2. BeforeExec 找到路由之后，开始执行相应的 Controller 之前

3. AfterExec 执行完 Controller 逻辑之后执行的过滤器

4. 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版本开始已经废除

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

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 之后才能获取，因为 session 没有在这之前初始化。

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

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)

## 过滤器实现路由 ##

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

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

var UrlManager = func(ctx \*context.Context) \{    //数据库读取全部的url mapping数据    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)

**context.Context中一些常用的方法**

//获取请求路径中的cookie值

func (ctx \*Context) GetCookie(key string) string \{

return ctx.Input.Cookie(key)

//设置cookie的值

func (ctx \*Context) SetCookie(name string, value string, others ...interface\{\}) \{

ctx.Output.Cookie(name, value, others...)

//为response body中写字符串（Write string to response body）

func (ctx \*Context) WriteString(content string) \{

ctx.ResponseWriter.Write(\[\]byte(content))

//重定向  status为状态码，localurl为重定向的路径

func (ctx \*Context) Redirect(status int, localurl string) \{