第十三节控制器

1控制器

服务的配置信息放在conf/app.conf文件中
如果需要从环境变量中获取配置信息，可以使用如下方法。获取字符串时，使用一个变量就可以接收，而获取其他类型，需要使用两个变量接收

httpport,err = beego.AppConfig.Int("httpport")
host = beego.AppConfig.String("host")

app.conf中配置信息如下,当runmode为dev时，则读取dev下的数据，当为test时，则读取test下的数据.
还可以在app.conf配置文件中引入其他文件的配置信息，使用include进行引入

appname = tea
httpport = 5000
runmode = dev
include "thirth.conf"

[dev]
username = root
pwd = 123456

[test]
username = test
pwd = test

2路由

路由分为固定路由，正则路由和自动路由以及自定义路由。固定路由用户访问固定的链接，根据请求的类型不同访问不同方法。
下面主要介绍正则路由和自动路由

2.1正则路由

路由中加问号表示该参数非必须加，访问/param或者/param/56都能正常访问。而不加问号则必须加参数/param就会报404
还可以自定义匹配类型，例如"/param/:id([0-9]+)"，则参数id0-9至少需出现一次
\d匹配数字,\w匹配大小写字母加数字和下划线

beego.Router("/param/?:id", &controllers.ParamController{})
beego.Router("/param/:id", &controllers.ParamController{})
beego.Router("/param/?:id:int", &controllers.ParamController{})
beego.Router("/param/:id([0-9]+)", &controllers.ParamController{})

后端通过如下两种方式之一获取参数

id := p.GetString(":id")
id2 := p.Ctx.Input.Param(":id")

2.2自动路由

通过控制器/方法名/参数进行访问。如下则访问路径是/param/get访问get方法，系统会自动将ParamController这个控制器去掉Controller部分用剩下的Param转为小写进行匹配方法

beego.AutoRouter(&controllers.ParamController{})

2.3自定义路由

自定义路由即为给路由自定义映射方法，在普通路由后面增加了一个映射字符串，如下get的param/56则会被映射到自定义的MyGet方法进行处理而不是默认的Get

beego.Router("/param/:id", &controllers.ParamController{},"get:MyGet")
# 将get和post都映射到MyGet方法进行处理
beego.Router("/param/:id", &controllers.ParamController{},"get,post:MyGet")
# get映射到MyGet1,post映射到MyGet
beego.Router("/param/:id", &controllers.ParamController{},"get:MyGet1;post:MyGet")

3 获取请求方式

StopRun用于终止该请求执行到一定位置后的运行，如用户信息未校验通过，则可以使用StopRun终止运行后面的代码

method := p.Ctx.Request.Method
    # 终止运行函数
    p.StopRun()

4XSRF跨站防御配置

4.1 配置

在conf/app.conf中配置如下代码

enablexsrf = true
xsrfkey = aeafae123afdafd
xsrfexpire = 3600

在main.go文件的run前面添加

beego.BConfig.WebConfig.EnableXSRF = true
beego.BConfig.WebConfig.XSRFKey = aeafae123afdafd
beego.BConfig.WebConfig.XSRFExpire = 3600

4.2控制器中使用

func (t *TestXSRFController) Get() {
    t.Data["xsrfdata"] = template.HTML(t.XSRFFormHTML())
    t.TplName = "test_xsrf.html"
}

4.3前端form表单中使用

在form表单中使用后端提交过来的xsrfdata，然后提交就可

<form action="/test_xsrf", method="post">
    {{.xsrfdata}}
    <input type="submit" value="提交">
</form>

4.4关闭某个控制器下的xsrf验证

如下就能关闭TestXSRFController下控制器所有方法的xsrf验证

func (t *TestXSRFController) Prepare() {
    t.EnableXSRF = false
}

5文件上传

map[string]interface{}定义的map类型可以起到类似结构体的作用,value可以不只是局限于一种。如map[string]interface{}{"code": 200, "msg":"上传成功"}value既有整型又有字符串类型。

func (u *UploadController) Post() {
    f,h,err := u.GetFile("upload_file")
    defer func(){
        f.Close()
    }()
    if err != nil {
        return
    }
    filename = h.Filename
    fmt.Println(filename)
    # 获取时间戳
    time_unix_int = time.Now().Unix()
    # 将数字转换为字符串，传入数字和进制
    time_unix_str = strconv.FormatInt(time_unix_int, 10)
    # 文件保存传入文件的字段和保存的路径
    u.SaveToFile("upload_file", "static/upload" + (time_unix_str+"-"+filename))
    u.Data["json"] = map[string]interface{}{"code": 200, "msg":"上传成功"}
    u.ServeJSON()
}

5.1 设置文件上传大小限制

在main.go文件的run前面添加
单位是byte，1<<22即2的22次方byte大小

beego.BConfig.MaxMemory = 1<<22

第十三节控制器