【golang自学之路（四）】实现一账号N地登录问题！redis+JWT实现详讲

最新推荐文章于 2024-06-25 21:51:54 发布
最新推荐文章于 2024-06-25 21:51:54 发布
阅读量922 收藏 27
点赞数 12
分类专栏： golang学习之路 文章标签： golang redis 开发语言
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_42961251/article/details/139425566
版权

文章目录

需求背景

在这里插入图片描述

在进行go-web项目脚手架学习之后，搭建了服务登录，和服务注册两个功能。
其中服务登录采用了JWT的方式，让客服端与服务端进行登录认证交互。而采用token这种模式交互，最大的优点就是服务端无需任何存储，就能判断客户端是否登录，缺点就是token是无状态的，且token较长。
在项目视频中，李文周老师有一个课下作业就是怎么能做到一个账号一登陆问题，就是A用户拿着A的账号在A地登录服务，那么A访问服务是没有问题的，此时B用户拿着A的账号在B地登录，此时A就会要求重新登录，而此时B用户访问时正常的。
我们把这个问题发散一下，能不能做到一账号多登录？
项目连接：一账号多登录项目
go-web脚手架：go-web
go-web脚手架搭建教程：脚手架教程

一下代码是已经完成的工程代码，为之后技术方案选型实现做参考。

middlewares-> auth.go

// JWTAuthMiddleware 基于JWT的认证中间件
func JWTAuthMiddleware() func(c *gin.Context) {
	return func(c *gin.Context) {
		authHeader := c.Request.Header.Get("Authorization")
		p := new(models.ParamLogin)
		err := c.ShouldBindJSON(p)
		if err != nil {
			controller.ResponseError(c, controller.CodeNeedLogin)
			c.Abort()
			return
		}
		if authHeader == "" {
			controller.ResponseError(c, controller.CodeNeedLogin)
			c.Abort()
			return
		}
		// 按空格分割
		parts := strings.SplitN(authHeader, " ", 2)
		if !(len(parts) == 2 && parts[0] == "Bearer") {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		// parts[1]是获取到的tokenString，我们使用之前定义好的解析JWT的函数来解析它
		mc, err := jwt.ParseToken(parts[1])
		if err != nil {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		if !flag {
			controller.ResponseError(c, controller.AccountNumberOverLimit)
			c.Abort()
			return
		}
		// 将当前请求的userID信息保存到请求的上下文c上
		c.Set(controller.CtxUserIDKey, mc.UserID)
		c.Next() // 后续的处理请求的函数中 可以用过c.Get(CtxUserIDKey) 来获取当前请求的用户信息
	}
}

logic -> route.go

func Login(p *models.ParamLogin) (user *models.User, err error) {
	user = &models.User{
		Username: p.Username,
		Password: p.Password,
	}
	// 传递的是指针，就能拿到user.UserID
	flag := mysql.Login(user)
	if flag {
		return user, errors.New("username or password is error")
	}
	// 生成JWT
	token, err := jwt.GenToken(user.UserId, user.Username)
	if err != nil {
		return
	}
	user.Token = token
	return user, nil
}

controller -> user.go

// LoginHandler 登录
func LoginHandler(c *gin.Context) {
	// 1.获取请求参数及参数校验
	p := new(models.ParamLogin)
	p.Username = c.Query("username")
	p.Password = c.Query("password")
	// 2.业务逻辑处理
	user, err := logic.Login(p)
	if err != nil {
		zap.L().Error("logic.Login failed", zap.String("username", p.Username), zap.Error(err))
		ResponseError(c, CodeInvalidPassword)
		return
	}
	if !flag {
		ResponseError(c, CodeInvalidPassword)
		return
	}
	// 4.返回响应
	ResponseSuccess(c, gin.H{
		"user_id":   fmt.Sprintf("%d", user.UserId), // id值大于1<<53-1  int64类型的最大值是1<<63-1
		"user_name": user.Username,
		"token":     user.Token,
	})
}

logic -> route.go

func Login(p *models.ParamLogin) (user *models.User, err error) {
	user = &models.User{
		Username: p.Username,
		Password: p.Password,
	}
	// 传递的是指针，就能拿到user.UserID
	flag := mysql.Login(user)
	if flag {
		return user, errors.New("username or password is error")
	}

	// 生成JWT
	token, err := jwt.GenToken(user.UserId, user.Username)
	if err != nil {
		return
	}
	user.Token = token
	return user, nil
}

这里jwt采用的是jwt-go第三方组件库生成。整体流程就是在登录的时候会生成jwt令牌返回给client，client请求再带上这个令牌去请求后端。

curl – postman 验证登录成功

curl --location --request GET 'localhost:8081/ping' \
--header 'Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoyMDYwMjQzMjk5MjQ2MDgsInVzZXJuYW1lIjoidXNlcm5hbWUiLCJleHAiOjE3MTc0MjA4ODIsImlzcyI6ImJsdWViZWxsIn0.npDRpe7cnQSvJEZnppRKDvhVkwSrBLqfHNRiov4WzEo' \
--header 'Content-Type: application/json' \
--data-raw '{
    "username":"llbnk",
    "password":"213"
}'

一、技术方案

JWT令牌是无状态的，可以理解是牺牲速度换取容量的一种做法（个人理解网络带宽肯定大，服务端不存储）。
如果想实现一个账号多台登录，那必须需要再服务端进行存储。下图技术总体时序图，可以看到请求token过来是需要从db中获取进行对比，确认为同一个token后才能返回给前端相应结果，否则提示前端token无效。
在这里插入图片描述
那实现一账号多地登录的方案是怎么样的？我们以一账号3登录为例，请看下图：
在这里插入图片描述

二、方案选择

可看到A用户最后请求/ping的时候token应该是无效的，此时说明存储必须存三个token。

db选择方案优缺点
mysql建立一个token表去存储user和token的关系优点维护简单，缺点存储在磁盘太重了
redis创建key=login_token+username，将token放入value优点使用便利，缺点JWT令牌太长，当访问量太大redis内存可能不足，需要更换token算法

综合考虑采用redis去做登录校验比较合适，首先用户登录理解不是一个必须落表行为，可以容忍key丢失，用户再次登录，且redis相应速度快，用户体验好。
整体业务改造逻辑如下图，红色部分为应该改造部分。
在订单按钮按下的时候先请求生成一个秒杀令牌，只有在秒杀令牌申请成功之后才会进入我们之前写的OrderController中的createItem方法。
而redis采用stirng存储，但value怎么设计呢？

value选择存储方案实例优缺点
将value变成对象存储{“FirstToken”:“TOKEN1”,“SecondToken”:“TOKEN2”,“ThirdToken”:“TOKEN3”}优点理解便利，代码实现简单，缺点可拓展性差，代码改动较多
将value中存入token组成的切片{“TOKEN1”,“TOKEN2”,“TOKEN3”}优点节省redis内存，可拓展性强，缺点存入redis中需要先将sclice-》string-》json工序繁琐

最终还是选择第二种存入切片的方式。

三、代码编写

我们需要在登录返回相应前去将jwt存入redis中实现代码如下：
（代码有err地方尽量打印一下err，这里图省事没有打印error）

user.go


const Login_Token = "login_token"

// LoginHandler 登录
func LoginHandler(c *gin.Context) {
	// 1.获取请求参数及参数校验
	p := new(models.ParamLogin)
	p.Username = c.Query("username")
	p.Password = c.Query("password")
	// 2.业务逻辑处理
	user, err := logic.Login(p)
	if err != nil {
		zap.L().Error("logic.Login failed", zap.String("username", p.Username), zap.Error(err))
		ResponseError(c, CodeInvalidPassword)
		return
	}
	//3.在返回前去将token放入redis中
	flag := setRedisTokenV2(user)
	if !flag {
		ResponseError(c, CodeInvalidPassword)
		return
	}
	// 4.返回响应
	ResponseSuccess(c, gin.H{
		"user_id":   fmt.Sprintf("%d", user.UserId), // id值大于1<<53-1  int64类型的最大值是1<<63-1
		"user_name": user.Username,
		"token":     user.Token,
	})
}

func setRedisTokenV2(user *models.User) bool {
	res, err := redis.Rdb.Get(fmt.Sprintf(Login_Token+"_"+"%s", user.Username)).Result()
	if err != nil && err.Error() != redis.NIl {
		return false
	}

	var tokens []string
	if res != "" {
		err = json.Unmarshal([]byte(res), &tokens)
		if err != nil {
			return false
		}
	}
	fmt.Println("Slice from JSON:", tokens)
	for _, tokenstr := range tokens {
		if tokenstr == user.Token {
			return true
		}
	}
	if len(tokens) < 3 {
		tokens = append(tokens, user.Token)
	} else {
		copy(tokens[:2], tokens[1:])
		tokens[2] = user.Token
	}

	// 将 slice 转换为 JSON 字符串
	jsonData, err := json.Marshal(tokens)
	if err != nil {
		log.Fatalf("Error marshalling to JSON: %v", err)
	}
	jsonString := string(jsonData)
	fmt.Println("JSON string:", jsonString)
	res, err = redis.Rdb.Set(fmt.Sprintf(Login_Token+"_%s", user.Username), jsonString, 3600*time.Second).Result()

	if err != nil {
		return false
	}
	return true
}

整体的实现思路就是是否当前redis中存入的token数量
如果tooken数量小于3就将新生成的token放入slice中并重新设置redis
如果token数量大于3就将slice中的第0号，1号位的token覆盖第1号，2号位的token，并将新生成的token赋给第0号位置。

auth.go

// JWTAuthMiddleware 基于JWT的认证中间件
func JWTAuthMiddleware() func(c *gin.Context) {
	return func(c *gin.Context) {
		authHeader := c.Request.Header.Get("Authorization")
		p := new(models.ParamLogin)
		err := c.ShouldBindJSON(p)
		if err != nil {
			controller.ResponseError(c, controller.CodeNeedLogin)
			c.Abort()
			return
		}
		if authHeader == "" {
			controller.ResponseError(c, controller.CodeNeedLogin)
			c.Abort()
			return
		}
		// 按空格分割
		parts := strings.SplitN(authHeader, " ", 2)
		if !(len(parts) == 2 && parts[0] == "Bearer") {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		// parts[1]是获取到的tokenString，我们使用之前定义好的解析JWT的函数来解析它
		mc, err := jwt.ParseToken(parts[1])
		if err != nil {
			controller.ResponseError(c, controller.CodeInvalidToken)
			c.Abort()
			return
		}
		//一个账号可以三台登录
		flag := accountNumberLimitV2(parts[1], p)
		if !flag {
			controller.ResponseError(c, controller.AccountNumberOverLimit)
			c.Abort()
			return
		}
		// 将当前请求的userID信息保存到请求的上下文c上
		c.Set(controller.CtxUserIDKey, mc.UserID)
		c.Next() // 后续的处理请求的函数中 可以用过c.Get(CtxUserIDKey) 来获取当前请求的用户信息
	}
}

func accountNumberLimitV2(authHeader string, p *models.ParamLogin) bool {
	res, err := redis.Rdb.Get(fmt.Sprintf(controller.Login_Token+"_"+"%s", p.Username)).Result()
	if err != nil {
		return false
	}
	var tokens []string
	if res != "" {
		err = json.Unmarshal([]byte(res), &tokens)
		if err != nil {
			return false
		}
	}
	fmt.Println("Slice from JSON:", tokens)
	for _, tokenstr := range tokens {
		if tokenstr == authHeader {
			return true
		}
	}
	return false
}

在/ping验证接口中只需要在校验jwt中间件中去判断是否当前登录携带的token在redis中如果不在则认为token校验失败，用户应该重新登陆。

四、验证

通过postman请求登录结果

curl

curl --location --request GET 'localhost:8081/login?username=llbnk&password=213'

在这里插入图片描述
我们连续请求三次，可以查看redis-cli中的key
在这里插入图片描述

我们采用第一个jwt去请求一下/ping

curl

curl --location --request GET 'localhost:8081/ping' \
--header 'Authorization: Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoyMDYwMjQzMjk5MjQ2MDgsInVzZXJuYW1lIjoidXNlcm5hbWUiLCJleHAiOjE3MTc0MjkyNTksImlzcyI6ImJsdWViZWxsIn0._vi7RiS_IDLLwJbVvc3LnXsG4jn6CaEHopgA7qbJVrQ' \
--header 'Content-Type: application/json' \
--data-raw '{
    "username":"llbnk",
    "password":"213"
}'

在这里插入图片描述
此时我们再生成一个token。
在这里插入图片描述

达到我们想要的效果
在这里插入图片描述

总结

使用redis+JWT实现了一个账号多地登录。
这里有一点美中不足，就是setRedisTokenV2方法中
if len(tokens) < 3 { tokens = append(tokens, user.Token) } else { copy(tokens[:2], tokens[1:]) tokens[2] = user.Token }
这段代码如果能从viper的配置中读取，并改写，可以实现一个账号任意N地登录。

我的目标

希望在年底学习一下内容：
java学习内容：
1.tomcat源码
2.dubbo源码
3.zookeeper源码
4.netty源码
go学习内容：
1.gin框架学习
2.简单go项目
3.go基础知识进阶（gmp，gc，channel，map，slice源码等）
中间件学习内容：
1.kafka使用及源码
框架学习内容：
1.从零开始学架构
算法学习内容：
1.复习leetcode中top 100

我平时喜欢没事还打游戏，因为有宝宝所以希望在平时时间能尽量完成上述学习内容（希望能戒掉游戏哈哈哈）。
也希望有和我一样的一起学习的小伙伴共同学习进步，我建一个qq后端交流群：279868576，希望小伙伴们加入共同督促进步。

llbnk
关注
  • 12
    点赞
  • 踩
  • 27
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
实战——golang实现JWT验证登录
qq_42215697的博客
09-18 1056
g
理解透彻！我的华为面试经历分享，高级面试题+解析
04-29 438
前言 作为一个典型的分布式项目，Kubernetes 的部署一直以来都是挡在初学者前面的一只“拦路虎”。尤其是在 Kubernetes 项目发布初期，它的部署完全要依靠一堆由社区维护的脚本。 其实，Kubernetes 作为一个 Golang 项目，已经免去了很多类似于 Python 项目要安装语言级别依赖的麻烦。但是，除了将各个组件编译成二进制文件外，用户还要负责为这些二进制文件编写对应的配置文件、配置自启动脚本，以及为 kube-apiserver 配置授权文件等等诸多运维工作。 这几年，在跟朋友探讨
springboot+shiro+jwt实现多端用户登录
哈利路亚的收藏夹
09-06 3068
shiro中是通过realm进行权限认证的，本身shiro是支持多realm进行权限认证的。 现有场景如下。 平台分管理角色和互联网角色，分别存储在sys_user和student表中。sys_user当然是通过管理后台来进行创建。student表中的用户是允许前端注册的，不走后台权限角色体系。 但是提供的接口均需要实现token的安全验证，毕竟不能裸奔不是。所以需要在基础框架基础上，实现多realm认证，并且使用相同token。主要是不想拆分两个后台api项目。 首先项目pom中需要导入相应的包。
程序员写简历时的技术词汇拼写规范备忘录！
热门推荐
CodeSheep
10-11 1万+
程序员写简历时需要注意的技术词汇规范拼写大全
我当年自学黑客（网络安全）的一些心得！（内附学习笔记）
2401_84264727的博客
04-29 184
（这里额外提一句，学习编程虽然不能带你入门，但是却能决定你能在网络安全这条路上到底能走多远，所以推荐大家自学一些基础编程的知识）
2024年网安最全我当年自学黑客（网络安全）的一些心得！（内附学习笔记）
2401_84264662的博客
05-03 752
我们知道计算机最早是在西方发明出来的，很多名词或者代码都是英文的，甚至现有的一些教程最初也是英文原版翻译过来的，而且一个漏洞被发现到翻译成中文一般需要一个星期的时间，在这个时间差上漏洞可能都修补了。”答案是否定的，黑客用的电脑,不需要什么高的配置,只要稳定就行.因为黑客所使用的一些程序,低端CPU也可以很好的运行,而且不占什么内存.还有一个,黑客是在DOS命令下对进行的,所以电脑能使用到最佳状态!文章的宗旨是：1.指出一些自学的误区 2.提供客观可行的学习表 3.推荐我认为适合小白学习的资源.大佬绕道哈！
2024年网安最新我当年自学黑客（网络安全）的一些心得！（内附学习笔记）
2401_84302816的博客
05-01 832
至于编程语言，首推Python，因为其良好的拓展支持性。当然现在市面上很多网站都是PHP的开发的，所以选择PHP也是可以的。其他语言还包括C++、Java…很多朋友会问是不是要学习所有的语言呢？答案是否定的！引用我上面的一句话：学习编程只是工具不是目的，我们的目标不是成为程序员（这里额外提一句，学习编程虽然不能带你入门，但是却能决定你能在网络安全这条路上到底能走多远，所以推荐大家自学一些基础编程的知识）
2024年网络安全最新我当年自学黑客（网络安全）的一些心得！（内附学习笔记）(1)
2401_84301315的博客
05-03 1042
Python编程学习，学习内容包含：语法、正则、文件、 网络、多线程等常用库，推荐《Python核心编程》，不要看完；在实际的渗透测试过程中，面对复杂多变的网络环境，当常用工具不能满足实际需求的时候，往往需要对现有工具进行扩展，或者编写符合我们要求的工具、自动化脚本，这个时候就需要具备一定的编程能力。我们知道计算机最早是在西方发明出来的，很多名词或者代码都是英文的，甚至现有的一些教程最初也是英文原版翻译过来的，而且一个漏洞被发现到翻译成中文一般需要一个星期的时间，在这个时间差上漏洞可能都修补了。
基于Golang+Kratos+MySQL+Redis+Kafka+elk+Opentracing实现的微服务项目
07-28
基于Golang+Kratos+MySQL+Redis+Kafka+elk+Opentracing实现的微服务项目
Golang：通过Gin框架+Redis+责任链，实现一个简单的钉钉机器人，进行消息处理 ps:多应用版
04-26
Golang：通过Gin框架+Redis+责任链，实现一个简单的钉钉机器人，进行消息处理 ps:多应用版
Golang使用lua脚本实现redis原子操作
09-08
在Golang中，使用Lua脚本实现Redis的原子操作是一种常见的策略，以确保在高并发场景下的数据一致性。Redis支持内建的Lua解释器，允许开发者编写Lua脚本并在服务器端执行，从而避免了多次网络通信带来的延迟，同时...
gin-web:由gin + gorm + jwt + casbin组合实现的RBAC权限管理脚手架Golang版, 搭建完成即可快速、高效投入业务开发
04-30
Gin Web由gin + gorm + jwt + casbin组合实现的RBAC权限管理脚手架Golang版, 搭建完成即可快速、高效投入业务开发特性RESTful API 设计规范Gin 一款高效的golang web框架MySQL 数据库存储Jwt 用户认证, 登入登出一键...
golang实现Redis分布式自旋锁+本地自旋锁
01-20
golang使用redis的setnx实现了一个自选锁，有key超时，同时也有我们调用redis链接时的超时。 package locker import ( context github.com/go-redis/redis runtime time ) type Lock struct { resource string...
golang中的corn
shunuanwei的博客
06-20 366
golang的corn
【Golang - 90天从新手到大师】Day09 - string
wudawei071193的博客
06-20 970
【Golang - 90天从新手到大师】Day09 - string
Go 语言特定指南
最新发布
美味小鱼的杂货铺
06-25 653
本指南将向您展示如何使用 Docker 创建、测试和部署容器化的 Go 应用程序。
学习新语言方法总结（二）
JIIIT的博客
06-21 360
上一篇介绍了学习语言的数据类型、流程控制，leetcode上刷几道题目应该对基本代码逻辑有所了解了。这一篇主要来说一下进阶模块：面向对象OOP思想是编程中重要的一个模块，在各个语言都有大量的广泛运用，废话不多说，直接看go怎么实现。
go编程中接口(interface)用法
weixin_43354152的博客
06-21 662
如果熟悉C++编程，就会知道C++中可以通过虚函数来实现多态 ，Java等面向对象的语言也有类似的特性。那么是多态呢？多态是面向对象编程中的一个基本概念，允许不同的类通过。简而言之，多态使得，从而可以处理不同类型的对象，而无需知道它们的具体类型。在 Go 语言中，多态主要通过接口实现。接口定义了一组方法，而不同的类型可以实现这些方法，从而实现接口。这样，可以使用接口类型的变量来引用实现了该接口的不同类型的对象。多态是。
golang 实现redis分布式锁
05-02
以下是一个简单的 Golang 实现 Redis 分布式锁的代码示例： ```go package redislock import ( "fmt" "time" "github.com/go-redis/redis/v7" ) type RedisLock struct { redisClient *redis.Client key ...

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值