add models and jwt

handlers/response.go

@@ -0,0 +1,82 @@
+/*
+@Time : 2020/6/28 22:37
+@Author : xuyiqing
+@File : response.py
+*/
+
+package handlers
+
+import (
+	"github.com/gin-gonic/gin"
+)
+
+const (
+	Success		 	= 200
+	BadRequest 		= 400
+	Unauthenticated = 401
+	NoPermisson 	= 403
+	NotFund 		= 404
+	ServerError 	= 500
+)
+
+type Response struct {
+	Ctx *gin.Context
+}
+
+// 定义基础返回结构体
+type JsonResponse struct {
+	Code int `json:"code"`
+	Msg string `json:"msg"`
+	Data interface{} `json:"data,omitempty"`
+	Pager interface{} `json:"pager,omitempty"`
+}
+
+func (resp *Response) Response(data interface{}, pager interface{}) {
+	resp.Ctx.JSON(Success, JsonResponse{
+		Code: Success,
+		Msg: "返回成功",
+		Data: data,
+		Pager: pager,
+	})
+}
+
+// 400错误请求
+func (resp *Response) BadRequest(msg string) {
+	resp.Ctx.AbortWithStatusJSON(Success, JsonResponse{
+		Code: BadRequest,
+		Msg: msg,
+	})
+}
+
+
+// 401未登录验证
+func (resp *Response) Unauthenticated(msg string) {
+	resp.Ctx.AbortWithStatusJSON(Success, JsonResponse{
+		Code: Unauthenticated,
+		Msg: msg,
+	})
+}
+
+// 403没有权限
+func (resp *Response) NoPermisson(msg string) {
+	resp.Ctx.AbortWithStatusJSON(Success, JsonResponse{
+		Code: NoPermisson,
+		Msg: msg,
+	})
+}
+
+// 404资源不存在
+func (resp *Response) NotFund(msg string) {
+	resp.Ctx.AbortWithStatusJSON(Success, JsonResponse{
+		Code: NotFund,
+		Msg: msg,
+	})
+}
+
+// 500服务器出错
+func (resp *Response) ServerError(msg string) {
+	resp.Ctx.AbortWithStatusJSON(200, JsonResponse{
+		Code: ServerError,
+		Msg: msg,
+	})
+}

handlers/users.go

@@ -0,0 +1,127 @@
+/*
+@Time : 2020/6/28 21:40
+@Author : xuyiqing
+@File : users.py
+*/
+
+package handlers
+
+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)
+	db.Count(&pager.Total)
+	db.Offset(pager.OffSet).Limit(pager.PageSize).Find(&users)
+	pager.GetPager()
+	response.Response(users, pager)
+}