From 299ac091e4bb58de588115a5b100464ba2bee49e Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 19 Apr 2024 19:58:14 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=99=BB=E5=BD=95=E8=BA=AB=E4=BB=BD?=
 =?UTF-8?q?=E5=AF=86=E9=92=A5=E5=8A=A0=E6=9C=BA=E5=99=A8=E7=A0=81=E9=98=B2?=
 =?UTF-8?q?=E6=AD=A2=E5=85=B6=E4=BB=96=E7=BD=91=E5=85=83=E8=B7=A8=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/framework/utils/token/token.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/framework/utils/token/token.go b/src/framework/utils/token/token.go
index 2a4fb060..b07879d6 100644
--- a/src/framework/utils/token/token.go
+++ b/src/framework/utils/token/token.go
@@ -11,6 +11,7 @@ import (
 	"be.ems/src/framework/logger"
 	redisCahe "be.ems/src/framework/redis"
 	"be.ems/src/framework/utils/generate"
+	"be.ems/src/framework/utils/machine"
 	"be.ems/src/framework/vo"
 
 	jwt "github.com/golang-jwt/jwt/v5"
@@ -74,7 +75,7 @@ func Create(loginUser *vo.LoginUser, ilobArgs ...string) string {
 
 	// 生成令牌设置密钥
 	secret := config.Get("jwt.secret").(string)
-	tokenStr, err := jwtToken.SignedString([]byte(secret))
+	tokenStr, err := jwtToken.SignedString([]byte(machine.Code + "@" + secret))
 	if err != nil {
 		logger.Infof("jwt sign err : %v", err)
 		return ""
@@ -118,7 +119,7 @@ func Verify(tokenString string) (jwt.MapClaims, error) {
 		// 判断加密算法是预期的加密算法
 		if _, ok := token.Method.(*jwt.SigningMethodHMAC); ok {
 			secret := config.Get("jwt.secret").(string)
-			return []byte(secret), nil
+			return []byte(machine.Code + "@" + secret), nil
 		}
 		return nil, jwt.ErrSignatureInvalid
 	})