使用wego框架实现用户登录功能

go lang

已于 2022-03-23 11:03:44 修改

阅读量603

点赞数

分类专栏： go web框架 orm 文章标签： golang 后端 mysql

于 2022-02-21 10:52:58 首次发布

本文链接：https://blog.csdn.net/gz_hm/article/details/123042694

版权

go 同时被 3 个专栏收录

34 篇文章 1 订阅

订阅专栏

web框架

23 篇文章 1 订阅

订阅专栏

orm

13 篇文章 0 订阅

订阅专栏

介绍

本文通过一个简单的例子说明如何使用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)
}

<!doctype html>
<html>
<head>
    <meta charset="UTF-8">
    <title>登录页面</title>
</head>
<body>
    <h2>用户登陆</h2>
    <form action="/login_post" method="post">
        <div>用户账号：</div>
        <div>
            <input type="text" name="account" placeholder="请输入用户账号" />
        </div>
        <br />
        <div>登录密码：</div>
        <div>
            <input type="password" name="password" placeholder="请输入登录密码"/>
        </div>
        <br />
        <div>
            <input type="submit" value="立即登录" />
        </div>
    </form>
</body>
</html>

用户点击"立即登录"后项服务器发送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模板的内容如下：

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>
    hello : {{.}}
</body>
</html>

go lang

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
0
评论
使用wego框架实现用户登录功能

本文通过一个简单的例子说明如何使用wego框架来实现用户登录功能
复制链接

扫一扫

专栏目录

使用wego框架实现用户登录功能

介绍

项目的结构

配置文件的内容

main文件

登录逻辑

实现/index页面的显示

实现BaseController

实现HomeController

“相关推荐”对你有帮助么？