框架慢慢修改

2024-07-04 19:15:44 +08:00
parent d79e6bdaba
commit 82eadb8921
13 changed files with 152 additions and 96 deletions
--- a/controllers/users.go
+++ b/controllers/users.go
@@ -0,0 +1,128 @@
+/*
+@Time : 2020/6/28 21:40
+@Author : xuyiqing
+@File : users.py
+*/
+
+package controllers
+
+import (
+	"fmt"
+	"go_blog/models"
+	"go_blog/pkg/jwt"
+	"go_blog/pkg/util"
+	"go_blog/serializers"
+
+	"github.com/gin-gonic/gin"
+)
+
+// 登录
+func UsersLoginHandler(ctx *gin.Context) {
+	response := Response{Ctx: ctx}
+	var loginUser serializers.Login
+	if err := ctx.ShouldBind(&loginUser); err != nil {
+		panic(err)
+	}
+	user := loginUser.GetUser()
+	isLoginUser := user.CheckPassword()
+	if !isLoginUser {
+		response.BadRequest("密码错误")
+		return
+	}
+	token, err := jwt.GenToken(user.ID, user.Username)
+	if err != nil {
+		panic(err)
+	}
+	data, _ := util.PrecisionLost(user)
+	data["token"] = token
+	response.Response(data, nil)
+	return
+}
+
+// 注册
+func UsersRegisterHandler(ctx *gin.Context) {
+	response := Response{Ctx: ctx}
+	var registerUser serializers.Login
+	if err := ctx.ShouldBind(&registerUser); err != nil {
+		panic(err)
+	}
+	user := registerUser.GetUser()
+	status := user.CheckDuplicateUsername()
+	if status == false {
+		response.BadRequest("用户名已存在")
+		return
+	}
+	if err := user.SetPassword(user.Password); err != nil {
+		panic(err)
+	}
+	user.IsActive = true
+	models.DB.Create(&user)
+	response.Response(nil, nil)
+}
+
+// 修改用户信息
+func UsersSetInfoHandler(ctx *gin.Context) {
+	response := Response{Ctx: ctx}
+	jsonData, err := util.GetBodyData(ctx)
+	if err != nil {
+		response.BadRequest("参数解析失败")
+		return
+	}
+	fmt.Println(jsonData)
+	if jsonData == nil {
+		response.BadRequest("获取不到参数")
+		return
+	}
+	currentUser := jwt.AssertUser(ctx)
+	if currentUser != nil {
+		models.DB.Model(&currentUser).Updates(jsonData)
+		response.Response(currentUser, nil)
+		return
+	}
+}
+
+// 修改密码
+func UsersSetPwdHandler(ctx *gin.Context) {
+	response := Response{Ctx: ctx}
+	currentUser := jwt.AssertUser(ctx)
+	if currentUser == nil {
+		response.Unauthenticated("未验证登录")
+		return
+	}
+	var user serializers.Account
+	if err := ctx.ShouldBindJSON(&user); err != nil {
+		response.BadRequest(err.Error())
+		return
+	}
+	if user.Username != currentUser.Username {
+		response.BadRequest("当前登录用户用户名与输入用户名不符")
+		return
+	}
+	if user.OldPwd == user.NewPwd {
+		response.BadRequest("两次输入的密码相同")
+		return
+	}
+	if isPwd := currentUser.IsPasswordEqual(user.OldPwd); !isPwd {
+		response.BadRequest("原密码错误")
+		return
+	}
+	if err := currentUser.SetPassword(user.NewPwd); err != nil {
+		response.BadRequest(err.Error())
+		return
+	}
+	models.DB.Save(&currentUser)
+	response.Response(nil, nil)
+}
+
+func UsersListHandler(ctx *gin.Context) {
+	response := Response{Ctx: ctx}
+	var pager serializers.Pager
+	pager.InitPager(ctx)
+	var users []models.Account
+	db := models.DB.Model(&users)
+	total := int64(pager.Total)
+	db.Count(&total)
+	db.Offset(pager.OffSet).Limit(pager.PageSize).Find(&users)
+	pager.GetPager()
+	response.Response(users, pager)
+}