增加JWT鉴权机制

controller/common/shiyan.go

@@ -41,11 +41,17 @@ func (a *ApiMethod) Index(c *gin.Context) {
 }
 
 func (a *ApiMethod) JieXiJwt(c *gin.Context) {
-
+	strinf := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VySWQiOiIzMDA0NTAiLCJVVUlEIjoiMzUwNTc2MDg1NzI2OTI4ODk2IiwiZXhwIjoxNzQwMDE1ODUyLCJpYXQiOjE3Mzk0MTEwNTIsImlzcyI6ImFkbWluIiwic3ViIjoiQXBwSnd0VG9rZW4ifQ.UZQSs6ScQ5lskaPgCs7eB1B9AW9G8O9iHmDTQYKssFY"
+	aa, b, d := utils.ParseToken(strinf)
+	sendMap := make(map[string]interface{})
+	sendMap["token"] = aa
+	sendMap["cla"] = b
+	sendMap["err"] = d
+	formatoutput.Result(0, sendMap, c)
 }
 
 /*
-*
+*{"conversationId":"3d24a4a6-fa93-4770-8dd6-69fa3545afbd","source":"instruct"}
 @ 作者: 秦东
 @ 时间: 2024-11-08 15:20:28
 @ 功能: 实验图片改变格式

initialization/router.go

@@ -2,6 +2,7 @@ package initialization
 
 import (
 	"appNewPlatform/routers"
+	"appNewPlatform/utils"
 	"io/ioutil"
 
 	"github.com/gin-gonic/gin"
@@ -41,6 +42,12 @@ func InitializeRouter() *gin.Engine {
 		{
 			wsApiRouter.RouterGroup(appLoadRouterGroup)
 		}
+	}
+	//采用jwt鉴权
+	jwtRouterGroup := router.Group("")
+	jwtRouterGroup.Use(utils.AuthMiddleware())
+	{
+
 	}
 	return router
 }

routers/appShiyan/routers.go

@@ -15,5 +15,6 @@ func (a *ApiRouter) RouterGroup(router *gin.RouterGroup) {
 		apiRouter.POST("", methodBinding.Index) //入口
 
 		apiRouter.GET("visitWebserviceInterface", methodBinding.VisitWebserviceInterface) //访问webservice接口
+		apiRouter.POST("jieXiJwt", methodBinding.JieXiJwt)
 	}
 }

utils/jwt.go

@@ -3,9 +3,12 @@ package utils
 import (
 	"appNewPlatform/appConstant"
 	"appNewPlatform/generalmethod"
+	"appNewPlatform/utils/formatoutput"
+	"fmt"
 	"strconv"
 	"time"
 
+	"github.com/gin-gonic/gin"
 	"github.com/golang-jwt/jwt"
 )
 
@@ -60,7 +63,9 @@ func ReleaseToken(userKey string) (token string, err error) {
 
 @ 返回值
 
-	#
+	#jwt.Token
+	#jwt.Token
+	#error
 
 @ 方法原型
 
@@ -73,3 +78,78 @@ func ParseToken(tokenString string) (*jwt.Token, *Claims, error) {
 	})
 	return token, claims, err
 }
+
+/*
+*
+@ 作者: 秦东
+@ 时间: 2025-02-13 10:44:02
+@ 功能: 验证jwt令牌
+@ 参数
+
+	#tokenString   令牌
+
+@ 返回值
+
+	#Claims   解析后数据
+	#error		信息
+
+@ 方法原型
+
+	#
+*/
+func ValidateToken(tokenString string) (*Claims, error) {
+	token, claims, err := ParseToken(tokenString)
+	if err != nil {
+		return nil, err
+	}
+	if !token.Valid {
+		return nil, fmt.Errorf("无效的令牌")
+	}
+	if time.Now().Unix() > claims.ExpiresAt {
+		return nil, fmt.Errorf("令牌已过期")
+	}
+	return claims, nil
+}
+
+/*
+*
+@ 作者: 秦东
+@ 时间: 2025-02-13 11:36:59
+@ 功能:
+@ 参数
+
+	#
+
+@ 返回值
+
+	#
+
+@ 方法原型
+
+	#
+*/
+func AuthMiddleware() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		tokenString := c.GetHeader("Authorization")
+		if tokenString == "" {
+			formatoutput.Result(300, tokenString, c, "未获得授权！")
+			c.Abort()
+			return
+		}
+		if len(tokenString) > 7 && tokenString[:7] == "Bearer " {
+			tokenString = tokenString[7:]
+		} else {
+			formatoutput.Result(300, tokenString, c, "授权标头无效！")
+			c.Abort()
+			return
+		}
+		_, claims, err := ParseToken(tokenString)
+		if err != nil {
+			formatoutput.Result(300, tokenString, c, "无效的令牌！")
+			c.Abort()
+			return
+		}
+		c.Set("username", claims.UserId)
+		c.Next()
+	}
+}