go如何使用gin結合jwt做登錄功能簡單示例

更新時間：2024年01月17日 08:28:57 作者：wjdsg

jwt全稱Json web token,是一種認證和信息交流的工具,這篇文章主要給大家介紹了關于go如何使用gin結合jwt做登錄功能的相關資料,文中通過代碼介紹的非常詳細,需要的朋友可以參考下

1、安裝gin

go get -u github.com/gin-gonic/gin

2、安裝session

go get github.com/gin-contrib/sessions

3、安裝JWT鑒權

go get "github.com/golang-jwt/jwt/v4"

4、創(chuàng)建一個jwt的工具文件

package utils

import (
	"errors"
	"github.com/golang-jwt/jwt/v4"
	"time"
)

type MyClaims struct {
	Phone                string `json:"phone"`
	jwt.RegisteredClaims        // 注意!這是jwt-go的v4版本新增的，原先是jwt.StandardClaims
}

var MySecret = []byte("wjdsg") // 定義secret，后面會用到

func Secret() jwt.Keyfunc {
	return func(token *jwt.Token) (interface{}, error) {
		return []byte(MySecret), nil // 這是我的secret
	}
}

// 這里傳入的是手機號，因為我項目登陸用的是手機號和密碼
func MakeToken(phone string) (tokenString string, err error) {
	claim := MyClaims{
		Phone: phone,
		RegisteredClaims: jwt.RegisteredClaims{
			ExpiresAt: jwt.NewNumericDate(time.Now().Add(3 * time.Hour * time.Duration(1))), // 過期時間3小時
			IssuedAt:  jwt.NewNumericDate(time.Now()),                                       // 簽發(fā)時間
			NotBefore: jwt.NewNumericDate(time.Now()),                                       // 生效時間
		}}
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claim) // 使用HS256算法
	tokenString, err = token.SignedString(MySecret)
	return tokenString, err
}

// 解析jwt
func ParseToken(tokenss string) (*MyClaims, error) {
	token, err := jwt.ParseWithClaims(tokenss, &MyClaims{}, Secret())
	if err != nil {
		if ve, ok := err.(*jwt.ValidationError); ok {
			if ve.Errors&jwt.ValidationErrorMalformed != 0 {
				return nil, errors.New("非法令牌")
			} else if ve.Errors&jwt.ValidationErrorExpired != 0 {
				return nil, errors.New("令牌國企")
			} else if ve.Errors&jwt.ValidationErrorNotValidYet != 0 {
				return nil, errors.New("令牌未激活")
			} else {
				return nil, errors.New("令牌未知錯誤")
			}
		}
	}
	if claims, ok := token.Claims.(*MyClaims); ok && token.Valid {
		return claims, nil
	}
	return nil, errors.New("couldn't handle this token")
}

5、開始使用

package main

import (
	"awesomeProject/utils"
	"fmt"
	"github.com/gin-contrib/sessions"
	"github.com/gin-contrib/sessions/cookie"
	"github.com/gin-gonic/gin"
	"net/http"
)

func main() {
	// 初始化gin服務
	rootRouter := gin.Default()

	// 創(chuàng)建cookie存儲
	store := cookie.NewStore([]byte("secret"))
	//路由上加入session中間件
	rootRouter.Use(sessions.Sessions("mysession", store))

	//配置路由組
	group := rootRouter.Group("/wjdsg")
	{
		group.GET("/set", func(context *gin.Context) {
			session := sessions.Default(context)

			token, err := utils.MakeToken("15088888888")
			if err != nil {
				fmt.Println("jwt生產錯誤")
			}
			//將jwt保持到session中
			session.Set("adminId", token)
			session.Save()
			context.JSON(http.StatusOK, token)
		})
		group.GET("/get", func(context *gin.Context) {

			//獲取jwt
			session := sessions.Default(context)
			token := session.Get("adminId").(string)
			//解析jwt
			parseToken, err := utils.ParseToken(token)
			if err != nil {
				fmt.Println("jwt解析錯誤：", err)
			}

			m := make(map[string]interface{})

			m["jwt"] = token
			m["解析"] = parseToken

			context.JSON(http.StatusOK, m)
		})
	}

	//啟動http服務，可以修改端口
	address := fmt.Sprintf(":%d", 8080)
	rootRouter.Run(address)
}