golang 中使用 JWT 实现登录验证

简介

JWT是json web token
具体jwt的组成,加密方式等等自行百度解决,我这里仅写实现案例:

控制器代码

package controller

import (
	"errors"
	"fmt"
	"gindemo/dto"
	"gindemo/middleware"
	"gindemo/middleware/jwt"
	"gindemo/models"
	"github.com/gin-gonic/gin"
	"log"
)

func Login(c *gin.Context) {
   
	loginInput := &dto.LoginInput{
   }//我这里分层了,主要是把参数验证这块单独分离出来了
	if err := loginInput.BindingValidParams(c); err != nil {
   
		middleware.ResponseError(c, 2001, err)
		return
	}
	user := &models.User{
   }
	fmt.Println(loginInput)
	token, err := user.Login(loginInput.UserName, loginInput.Password)
	if err != nil {
   
		if err.Error() == "record not found" {
   
			middleware.ResponseError(c, 500, errors.New("该用户不存在"))
			return
		} else {
   
			middleware.ResponseError(c, 500, errors.New("登录错误"))
			return
		}
	}
	middleware.ResponseSuccess(c,token)
	return
}

//用于测试使用
func UserList(c *gin.Context) {
   
	var user models.User
	claims := c.MustGet("claims").(*jwt.CustomClaims)
	users, err := user.ListUsers(claims.Name)
	if err != nil {
   
		log.Fatal(err)
	}
	middleware.ResponseSuccess(c, users)
}

参数验证

package dto

import (
	"errors"
	"gindemo/public"
	"github.com/gin-gonic/gin"
	ut "github.com/go-playground/universal-translator"
	"gopkg.in/go-playground/validator.v9"
	zh_translations "gopkg.in/go-playground/validator.v9/translations/zh"
	"strings"
)

type LoginInput struct {
   
	UserName string `form:"username" validate:"required"`
	Password string `form:"password" validate:"required"`
}

func (o *LoginInput) BindingValidParams(c *gin.Context) error {
   
	//绑定数据
	if err := c.ShouldBind(o); err != nil {
   
		return err
	}
	v := c.Value("trans")
	trans, ok := v.(ut.Translator)
	if !ok {
   
		trans, _ = public.Uni.GetTranslator("zh")
	}
	验证器注册翻译器
	//e := zh_translations.RegisterDefaultTranslations(public.Validate, trans)
	//if e != nil {
   
	//	return e
	//}
	//验证
	err := public.Validate.Struct(o)
	if err