使用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 main

import (
    "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)

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：

type HomeController struct {
    BaseController
}

接下来是/index页面的处理器代码：

func (this *HomeController) ShowHome(c *wego.WebContext) {
    c.WriteHTML(http.StatusOK, "./view/index.html", this.Account)
}

(this *HomeController)ShowHome是一个方法型处理器，在ShowHome中读取index.html模板，并使用this.Account进行渲染。index.html模板的内容如下：




    
    Title


    hello : {{.}}