当前位置：首页 > article >正文

Golang语言整合jwt+gin框架实现token

article 2024/11/24 8:41:32

1.下载jwt

 go get -u  github.com/dgrijalva/jwt-go

2.新建生成token和解析token文件

2.1 新建common文件夹和jwtConfig文件夹新建jwtconfig.go文件
在这里插入图片描述

2.2 jwtconfig.go文件代码

/*
@Time : 2021/8/2 下午3:03
@Author : mrxuexi
@File : main
@Software: GoLand
*/
package jwtConfig
import (
    "github.com/dgrijalva/jwt-go"
    "time"
)

type Claims struct {
	UserID int `json:"userId"`
	jwt.StandardClaims
}

const expire_time = 30 * time.Minute
var jwtSecret = []byte("123")
// GenerateToken 生成token的函数
func GenerateToken(userId int) (string, error) {
	nowTime := time.Now()
	expireTime := nowTime.Add(expire_time)
	claims := Claims{
		userId, // 自行添加的信息
		jwt.StandardClaims{
			ExpiresAt: expireTime.Unix(), // 设置token过期时间
			Issuer:    "admin",           // 设置jwt签发者
		},
	}
	// 生成token
	tokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	token, err := tokenClaims.SignedString(jwtSecret)
	return token, err
}



// ParseToken 验证token的函数
func ParseToken(token string) (*Claims, error) {
	// 对token的密钥进行验证
	tokenClaims, err := jwt.ParseWithClaims(token, &Claims{}, func(token *jwt.Token) (interface{}, error) {
		return jwtSecret, nil
	})
	// 判断token是否过期
	if tokenClaims != nil {
		claims, _ := tokenClaims.Claims.(*Claims)
		return claims, nil
	}
	return nil, err
}

3.中间件封装文件

3.1 新建middleware文件夹和jwt.go文件
在这里插入图片描述

3.2 jwt.go文件代码

下图红色正方形框地址根据自己的项目定义
在这里插入图片描述

package middleware
import (
	"github.com/gin-gonic/gin"
	"ListProject-ver1.0/common/jwtConfig"
	"net/http"
	"time"
)

func JWT() gin.HandlerFunc {
	return func(context *gin.Context) {
		// 获取token
		token := context.GetHeader("Token")

		if token == "" {
			context.JSON(http.StatusOK, gin.H{
				"code": http.StatusUnauthorized,
				"msg":  "没有携带token",
				"data": "",
			})
			context.Abort()
			return
		} else {
			claims, err := jwtConfig.ParseToken(token)
			if err != nil {
				context.JSON(http.StatusOK, gin.H{
					"code": http.StatusUnauthorized,
					"msg":  "token验证失败",
					"data": "",
				})
				context.Abort()
				return
			} else if time.Now().Unix() > claims.StandardClaims.ExpiresAt {
				context.JSON(http.StatusOK, gin.H{
					"code": http.StatusUnauthorized,
					"msg":  "token已过期",
					"data": "",
				})
				context.Abort()
				return
			}
		}
	}
}

4.调用测试中间件jwt功能

4.1 调用测试文件

在这里插入图片描述

4.2 代码

代码里面的ListProject-ver1.0/middleware这块也根据自己的项目引用

/*
@Time : 2021/8/2 下午3:03
@Author : mrxuexi
@File : main
@Software: GoLand
*/
package main
import (
    "github.com/gin-gonic/gin"
    "ListProject-ver1.0/middleware"
)

func main() {
    r := gin.Default()
    //注册组和中间件
    v1 := r.Group("/user")
    {
        v1.GET("/notoken",func(context *gin.Context) {
                            context.String(http.StatusOK, "hello gin post method")
                        })
    }
    //使用jwt中间件
    v2 := r.Group("/user2").Use(middleware.JWT())
    {

        v2.GET("/needtoken", func(context *gin.Context) {
                                   context.String(http.StatusOK, "hello gin post method")
                               })

    }
    r.Run(":8888")
}

4.3 调试结果

在这里插入图片描述