浏览代码

update 统一token的rediskey

Alvin 8 月之前
父节点
当前提交
2f80ad3297

+ 9 - 0
game/game_cluster/nodes/webadmin/common/redisKey.go

@@ -0,0 +1,9 @@
+package common
+
+const (
+	AdminTokenPrefix = "admin::token::"
+)
+
+func GetTokenKey(key string) string {
+	return AdminTokenPrefix + key
+}

+ 2 - 2
game/game_cluster/nodes/webadmin/router/middleware.go

@@ -39,14 +39,14 @@ func Auth(settingObj cfacade.ProfileJSON) gin.HandlerFunc {
 
 		// 获取请求URL
 		// url := c.Request.URL.Path
-		// s, err := mdb.RDB.HGet(context.Background(), "admin::token::"+tokenString, url).Result()
+		// s, err := mdb.RDB.HGet(context.Background(), common.GetTokenKey(tokenString), url).Result()
 		// if err != nil {
 		// 	mhayaLogger.Warnf("Auth HGet s error: %s", err.Error())
 		// 	common.PackUnauthorizedResult(c, code.InternalError, "")
 		// 	return
 		// }
 
-		ss, err := mdb.RDB.HGet(context.Background(), "admin::token::"+tokenString, constant.AdminAccess).Result()
+		ss, err := mdb.RDB.HGet(context.Background(), common.GetTokenKey(tokenString), constant.AdminAccess).Result()
 		if err != nil {
 			mhayaLogger.Warnf("Auth HGet ss error: %s", err.Error())
 			common.PackUnauthorizedResult(c, code.InternalError, "")

+ 4 - 4
game/game_cluster/nodes/webadmin/service/admin.go

@@ -139,8 +139,8 @@ func (a *Admin) Login(ctx *gin.Context, username string, password string) (*enti
 // LoginAuthSetRoleRedis 登录时写入该用户的权限
 func (a *Admin) loginAuthSetRoleRedis(roleID, generateToken string) error {
 	if roleID == constant.AdminAccess {
-		mdb.RDB.HSet(context.Background(), "admin::token::"+generateToken, constant.AdminAccess, 1)
-		mdb.RDB.Expire(context.Background(), "admin::token::"+generateToken, 24*time.Hour).Err()
+		mdb.RDB.HSet(context.Background(), common.GetTokenKey(generateToken), constant.AdminAccess, 1)
+		mdb.RDB.Expire(context.Background(), common.GetTokenKey(generateToken), 24*time.Hour).Err()
 		return nil
 	}
 
@@ -183,9 +183,9 @@ func (a *Admin) loginAuthSetRoleRedis(roleID, generateToken string) error {
 	}
 
 	for _, v := range accessList {
-		mdb.RDB.HSet(context.Background(), "admin::token::"+generateToken, v.URL, 1)
+		mdb.RDB.HSet(context.Background(), common.GetTokenKey(generateToken), v.URL, 1)
 	}
-	mdb.RDB.Expire(context.Background(), "admin::token::"+generateToken, 24*time.Hour).Err()
+	mdb.RDB.Expire(context.Background(), common.GetTokenKey(generateToken), 24*time.Hour).Err()
 
 	return nil