（三）登录和注册(handle_auto.go)

ops/2024/9/26 1:24:14/

登录和注册(handle_auto.go)

在这里插入图片描述

文章目录

  • 登录和注册(handle_auto.go)
    • 一、所需要的结构体信息
    • 二、注册
    • 三、登录
    • 四、退出

一、所需要的结构体信息

type UserAuth struct{}type LoginReq struct {Username string `json:"username" binding:"required"`Password string `json:"password" binding:"required"`
}type RegisterReq struct {Username string `json:"username" binding:"required"`Password string `json:"password" binding:"required,min=4,max=20"`Code     string `json:"code" binding:"required"`
}type LoginVO struct {model.UserInfo// 点赞 Set: 用于记录用户点赞过的文章, 评论ArticleLikeSet []string `json:"article_like_set"`CommentLikeSet []string `json:"comment_like_set"`Token          string   `json:"token"`
}

二、注册

因为原本的作者没有写注册的功能，我就自己写了一个简单的注册功能。

1.注册的路由

base.POST("/register", userAuthAPI.Register)    // 注册

2.验证码的功能函数

handler层的函数

  • 使用qq邮箱发送验证
  • 限制发送时间
  • 使用redis存储验证码，时间1分钟
func (*UserAuth) SendCode(c *gin.Context) {// 发送验证码限制于1分钟发一次elapsedTime := time.Since(lastEmailSent)remaingTime := time.Minute - elapsedTimeif elapsedTime < time.Minute {c.JSON(http.StatusOK, gin.H{"message": fmt.Sprintf("发送验证码过于频繁，请稍后再试，剩余时间：%d秒", int(remaingTime.Seconds()))})return}// 从请求参数中获取邮箱地址email := c.Query("email")if email == "" {c.JSON(http.StatusBadRequest, gin.H{"error": "未提供邮箱地址"})return}// 发送邮件验证码err := SendEmail(email, c)if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"error": "发送邮件失败"})return}// 发送邮件成功，返回成功响应给前端lastEmailSent = time.Now()ReturnSuccess(c, "验证码发送成功")
}// 发送邮件到指定邮箱
func SendEmail(email string, c *gin.Context) error {// 邮箱配置信息config := gomail.NewDialer("smtp.qq.com", 465, "xxx@qq.com", "自己的授权码")//config.TLSConfig.ServerName = "smtp.qq.com"//config.TLSConfig.InsecureSkipVerify = false// 创建邮件内容message := gomail.NewMessage()message.SetHeader("From", "xxx@qq.com")message.SetHeader("To", email)message.SetHeader("Subject", "验证码")// 获取随机验证码computing := utils2.GetRandNums(6)message.SetBody("text/html", fmt.Sprintf("欢迎使用席万里的博客系统，这是您的注册验证码：%s", computing))// 连接并发送邮件dialer, err := config.Dial()if err != nil {return err}defer dialer.Close()if err = gomail.Send(dialer, message); err != nil {return err}// 将验证码存到redis中，确保输入的是正确的rdb := GetRDB(c)// 以email为key，mima为val存入redis，只存在1分钟rdb.Set(rctx, email, computing, time.Minute)return nil
}

3.注册的主功能函数

  • 用到了密码加密
  • 随机验证码的生成
  • 数据的写入
func (*UserAuth) Register(c *gin.Context) {var req RegisterReqif err := c.ShouldBindJSON(&req); err != nil {ReturnError(c, g2.ErrRequest, err)return}db := GetDB(c)rdb := GetRDB(c)// 检查邮箱是否已经被注册_, err := model.GetUserAuthInfoByName(db, req.Username)if err != nil {if !errors.Is(err, gorm.ErrRecordNotFound) {ReturnError(c, g2.ErrDbOp, err)return}} else {ReturnError(c, g2.ErrMailAleradyUsed, "邮箱已经被使用")return}// 取出来验证码码做判断 看是否正确computing, err := rdb.Get(rctx, req.Username).Result()if err != nil {// 处理获取验证码失败的情况ReturnError(c, g2.FailResult, err)return}// 如果验证码不正确，则返回错误信息if computing != req.Code {ReturnError(c, g2.ErrVerificationCode, "验证码错误")return}// 将密码加密hashpassword, _ := utils2.BcryptHash(req.Password)err = model.RegisterDB(db, req.Username, hashpassword)if err != nil {ReturnError(c, g2.ErrDbOp, "注册失败")}ReturnSuccess(c, "注册成功")
}

4.数据库层的邮箱查重函数

func GetUserAuthInfoByName(db *gorm.DB, name string) (*UserAuth, error) {var userAuth UserAuthresult := db.Where(&UserAuth{Username: name}).First(&userAuth)return &userAuth, result.Error
}

5.数据查入、生成随机验证码、mimajiammi

// 数据查入
// 用户注册 用户名，密码
func RegisterDB(db *gorm.DB, username, hashpassword string) error {// 创建新用户认证信息newUserAuth := &UserAuth{Model: Model{CreatedAt: time.Now(),UpdatedAt: time.Now(),},Username: username,Password: hashpassword,}// 保存新用户认证信息到数据库if err := db.Create(newUserAuth).Error; err != nil {return err}return nil
}
// 生成随机验证码
func GetRandNums(digits int) string {rand.Seed(time.Now().UnixNano())numeric := [10]byte{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}var sb strings.Builderfor i := 0; i < digits; i++ {fmt.Fprintf(&sb, "%d", numeric[rand.Intn(len(numeric))])}return sb.String()
}// 密码加密
func BcryptHash(str string) (string, error) {bytes, err := bcrypt.GenerateFromPassword([]byte(str), bcrypt.DefaultCost)return string(bytes), err
}

三、登录

登录功能也用到了查重，校验密码hash等功能，这里不在讲述。

  • 生成jwt token
  • 使用MD5加密
  • 使用Session存储认证ID
  • 使用Redis存储在线信息
func (*UserAuth) Login(c *gin.Context) {var req LoginReqif err := c.ShouldBindJSON(&req); err != nil {ReturnError(c, g2.ErrRequest, err)return}db := GetDB(c)rdb := GetRDB(c)userAuth, err := model.GetUserAuthInfoByName(db, req.Username)if err != nil {if errors.Is(err, gorm.ErrRecordNotFound) {ReturnError(c, g2.ErrUserNotExist, nil)return}ReturnError(c, g2.ErrDbOp, err)return}// 检查密码是否正确if !utils2.BcryptCheck(req.Password, userAuth.Password) {ReturnError(c, g2.ErrPassword, nil)return// 获取 IP 相关信息 FIXME: 好像无法读取到 ip 信息ipAddress := utils2.IP.GetIpAddress(c)ipSource := utils2.IP.GetIpSourceSimpleIdle(ipAddress)userInfo, err := model.GetUserInfoById(db, userAuth.UserInfoId)if err != nil {if errors.Is(err, gorm.ErrRecordNotFound) {ReturnError(c, g2.ErrUserNotExist, nil)return}ReturnError(c, g2.ErrDbOp, err)return}roleIds, err := model.GetRoleIdsByUserId(db, userAuth.ID)if err != nil {ReturnError(c, g2.ErrDbOp, err)return}articleLikeSet, err := rdb.SMembers(rctx, g2.ARTICLE_USER_LIKE_SET+strconv.Itoa(userAuth.ID)).Result()if err != nil {ReturnError(c, g2.ErrDbOp, err)return}commentLikeSet, err := rdb.SMembers(rctx, g2.COMMENT_USER_LIKE_SET+strconv.Itoa(userAuth.ID)).Result()if err != nil {ReturnError(c, g2.ErrDbOp, err)return}// 登录信息正确, 生成 Token// UUID 生成方法: ip + 浏览器信息 + 操作系统信息// uuid := utils.MD5(ipAddress + browser + os)conf := g.Conf.JWTtoken, err := jwt.GenToken(conf.Secret, conf.Issuer, int(conf.Expire), userAuth.ID, roleIds)if err != nil {ReturnError(c, g2.ErrTokenCreate, err)return}// 更新用户验证信息: ip 信息 + 上次登录时间err = model.UpdateUserLoginInfo(db, userAuth.ID, ipAddress, ipSource)if err != nil {ReturnError(c, g2.ErrDbOp, err)return}slog.Info("用户登录成功: " + userAuth.Username)session := sessions.Default(c)session.Set(g2.CTX_USER_AUTH, userAuth.ID)session.Save()// 删除 Redis 中的离线状态offlineKey := g2.OFFLINE_USER + strconv.Itoa(userAuth.ID)rdb.Del(rctx, offlineKey).Result()ReturnSuccess(c, LoginVO{UserInfo: *userInfo,ArticleLikeSet: articleLikeSet,CommentLikeSet: commentLikeSet,Token:          token,})
}

1.use Redis 缓存用户收藏和喜欢的文章

	articleLikeSet, err := rdb.SMembers(rctx, g2.ARTICLE_USER_LIKE_SET+strconv.Itoa(userAuth.ID)).Result()if err != nil {ReturnError(c, g2.ErrDbOp, err)return}commentLikeSet, err := rdb.SMembers(rctx, g2.COMMENT_USER_LIKE_SET+strconv.Itoa(userAuth.ID)).Result()if err != nil {ReturnError(c, g2.ErrDbOp, err)return}

四、退出

func (*UserAuth) Logout(c *gin.Context) {c.Set(g2.CTX_USER_AUTH, nil)// 已经退出登录auth, _ := CurrentUserAuth(c)if auth == nil {ReturnSuccess(c, nil)return}session := sessions.Default(c)session.Delete(g2.CTX_USER_AUTH)session.Save()// 删除 Redis 中的在线状态rdb := GetRDB(c)onlineKey := g2.ONLINE_USER + strconv.Itoa(auth.ID)rdb.Del(rctx, onlineKey)ReturnSuccess(c, nil)
}

http://www.ppmy.cn/ops/23609.html

相关文章

Eureka服务注册与发现

Eureka服务注册与发现

Eureka服务注册与发现 概念原理Eureka的基本架构Eureka包含两个组件&#xff1a;Eureka Server和Eureka Client 三大角色目前工程情况 构建步骤step1&#xff1a;EurekaServer服务注册中心Module的建立step2&#xff1a;将已有的微服务注册进eureka服务中心actuator与注册微服务…
阅读更多...
OS复习笔记ch5-1

OS复习笔记ch5-1

引言 讲解完进程和线程之后&#xff0c;我们就要来到进程的并发控制这里&#xff0c;这一章和下一章是考试喜欢考察的点&#xff0c;有可能会出大题&#xff0c;面试也有可能会被频繁问到&#xff0c;所以章节内容较多。请小伙伴们慢慢食用&#xff0c;看完之后多思考加强消化…
阅读更多...
构建云原生湖仓：Apache Iceberg与Amoro的结合实践

构建云原生湖仓：Apache Iceberg与Amoro的结合实践

随着大数据技术的快速发展&#xff0c;企业对数据的处理和分析需求日益增长。传统的数据仓库已逐渐无法满足现代业务对数据多样性和实时性的要求&#xff0c;这促使了数据湖和数据仓库的融合&#xff0c;即湖仓一体架构的诞生。在云原生技术的推动下&#xff0c;构建云原生湖仓…
阅读更多...
数据结构与算法-环形队列

数据结构与算法-环形队列

1.简介 针对简单队列中数组使用一次就不能用&#xff0c;无法复用的问题&#xff0c;将这个数组改成一个环形队列&#xff0c;使得数据取出后这个空间能够再次使用。 2.思路分析 1.将front变量的含义做调整&#xff0c;使得front指向队列的第一个元素&#xff08;原先简单队…
阅读更多...
Django杂记之数据查询和文件处理

Django杂记之数据查询和文件处理

Hello , 我是小恒不会java。本文内容可能比较杂&#xff0c;主要是一些重点片段&#xff0c;不会涉及真实业务 学过django的读者可以放心阅读&#xff0c;本文约2万字&#xff0c;阅读时间不会很长 在线网址阅读http://django.yunduanjianzhan.cn/index. 数据查询 查询 想要…
阅读更多...
闲话 ASP.NET Core 数据校验（二）：FluentValidation 基本用法

闲话 ASP.NET Core 数据校验（二）：FluentValidation 基本用法

前言 除了使用 ASP.NET Core 内置框架来校验数据&#xff0c;事实上&#xff0c;通过很多第三方框架校验数据&#xff0c;更具优势。 比如 FluentValidation&#xff0c;FluentValidation 是第三方的数据校验框架&#xff0c;具有许多优势&#xff0c;是开发人员首选的数据校验…
阅读更多...
qt5core.dll怎么下载，qt5core.dll丢失能否修复？

qt5core.dll怎么下载，qt5core.dll丢失能否修复？

qt5core.dll的丢失真是让人头疼。这个Visual C Redistributable for Visual Studio 2015的运行时库被许多程序和游戏所依赖&#xff0c;一旦缺失了qt5core.dll&#xff0c;就会面临无法打开程序或游戏&#xff0c;甚至系统崩溃等一系列问题。 qt5core.dll的消失会带来以下麻烦 …
阅读更多...
WPF-关于动画Animation（及其常见问题）

WPF-关于动画Animation（及其常见问题）

目录 一、动画合集 常见动画类型 1、资源里添加动画资源 2、事件触发器里开始一个动画 3、Double型关键帧动画 4、Object型关键帧动画 5、Color型关键帧动画 6、String型关键帧动画 7、Matrix型沿路径动画 二、扩展 Ⅰ 动画常见问题 1、控制动画结束问题 Ⅱ 流动…
阅读更多...
最新文章

Copyright @ 2022~2023