使用Go语言web框架wego实现用户登录功能使用Go语言web框架wego实现用户

介绍本文通过一个简单的例子说明如何使用wego框架来实现用户登录功能。主要功能如下：

用户访问需要进行登录验证的页面时，首先会检查session的登录账号，若没有登录账号，则跳转到登录页面。
用户提交登录页面后，验证是否是合法的用户，若通过验证，则保存用户账号到session，并跳转到index页面。

说明：本例子采用基于cookie引擎的session
项目的结构

demo ├── controller- 控制器目录 │└── base.go- 控制器基础struct │└── home.go- home控制器 │└── login.go- 登录相关的处理器 ├── models- 模型目录 ├── logs- 日志文件目录 ├── static- 静态资源目录 │├── css │├── img │└── js ├── view- 视图模板目录 │└── login.html │└── index.html ├── app.conf- 应用配置文件 └── main.go- 入口文件

配置文件的内容

#应用名称 app_name = demo[server] #http监听端口 http_port = 8080[session] #session 是否开启 session_on = true #session类型 session_store=cookie #客户端的cookie的名称 cookie_name = "wego" #session 过期时间, 单位秒 life_time = 3600 #session数据的hash字符串 hash_key = 123456

重点说明：

http_port = 8080
<< 服务将监听8080端口
[session]
<< 用于设置是否开启session，并设置session对应的缓存引擎，本例子中采用基于cookie的session

main文件

package mainimport ( "demo/controller" "github.com/haming123/wego" log "github.com/haming123/wego/dlog" )func main() { web, err := wego.InitWeb() if err != nil{ log.Error(err) return }web.GET ("/show_login", controller.HandlerShowLogin) web.POST("/login_post", controller.HandlerLoginPost) web.GET ("/index", (*controller.HomeController).ShowHome)err = web.Run() if err != nil { log.Error(err) } }

登录逻辑用户访问需要进行登录验证的页面时，首先会检查session的登录账号，若没有登录账号，则跳转到登录页面, 登录页面的路由注册代码为：

web.GET ("/show_login", controller.HandlerShowLogin)

【使用Go语言web框架wego实现用户登录功能】controller.HandlerShowLogin为"/show_login"注册了一个函数型处理器：controller.HandlerShowLogin：

func HandlerShowLogin(c *wego.WebContext) { c.WriteHTML(http.StatusOK, "./view/login.html", nil) }

登录页面 - 锐客网用户登陆用户账号：登录密码：

用户点击"立即登录"后项服务器发送post请求：/login_post, /login_post请求的路由注册代码：

web.POST("/login_post", controller.HandlerLoginPost)

controller.HandlerLoginPost为一个函数型处理器：

func HandlerLoginPost(c *wego.WebContext) { account := c.Param.MustString("account") password := c.Param.MustString("password") if account == "admin" && password == "demo" { c.Session.Set("account", account) c.Session.Save() c.Redirect(302, "/index") } else { c.Session.Set("account", "") c.Session.Save() c.Redirect(302, "/show_login") } }

重点说明：

通过 c.Param.MustString函数来获取form参数
验证通过后将account保存到session中
检查"account"、"password"参数，若通过验证，则跳转到"/index", 否则显示登录页面："/show_login"

实现/index页面的显示实现BaseController
只有通过登录验证的用户才可以访问/index页面，因此在显示/index页面时需要检查session数据，若存在account，则可以访问/index页面，否则跳转到/show_login页面
可以为每个需要登录的页面都实现登录验证逻辑，这种做法虽然可行，但是实现起来比较繁琐。我们可以使用wego框架的拦截器来简化登录验证的实现，wego提供了两种拦截器：

函数型拦截器
接口型拦截器

本例子中我们使用接口型拦截器来实现登录验证，首先定义一个基础的struct：BaseController，在BaseController中实现了拦截器接口：BeforeExec(ctx *wego.WebContext)：

type BaseController struct { } func (this *BaseController) SetAccount(c *wego.WebContext, val string) { c.Data.Set("account", val) } func (this *BaseController) GetAccount(c *wego.WebContext) string { uid , _ := c.Data.GetString("account") return uid } func (this *BaseController) BeforeExec(ctx *wego.WebContext) { account, err := ctx.Session.GetString("account") if err != nil || account == "" { ctx.Redirect(302, "/show_login") return } this.SetAccount(c, account) }

这样，若一个struct组合了BaseController，便具有了BeforeExec方法，用户在访问处理器是首先会访问BeforeExec方法，这样便可以共用登录验证逻辑，不必在每个处理器中重复实现，大大简化了代码的编写。
实现HomeController
为了共用登录验证逻辑，首先需要HomeController中组合BaseController：