|
@@ -76,7 +76,7 @@ func (a *Admin) Login(ctx *gin.Context, username string, password string) (*enti
|
|
|
return nil, common.NewResult(code.UserNameOrPasswordError)
|
|
|
}
|
|
|
|
|
|
- token, err := mdb.RDB.Get(ctx, user.Username).Result()
|
|
|
+ token, err := mdb.RDB.Get(ctx, common.GetTokenKey(user.Username)).Result()
|
|
|
if err != nil && err != redis.Nil {
|
|
|
mhayaLogger.Warnf("Login Get error:%v", err)
|
|
|
return nil, common.NewResult(code.InternalError)
|
|
@@ -91,7 +91,7 @@ func (a *Admin) Login(ctx *gin.Context, username string, password string) (*enti
|
|
|
}
|
|
|
|
|
|
// 保存username -> token 到 redis 中 过期时间为1天
|
|
|
- err = mdb.RDB.Set(ctx, user.Username, generateToken, 24*time.Hour).Err()
|
|
|
+ err = mdb.RDB.Set(ctx, common.GetTokenKey(user.Username), generateToken, 24*time.Hour).Err()
|
|
|
if err != nil {
|
|
|
mhayaLogger.Warnf("Login Set user.Username error:%v", err)
|
|
|
return nil, common.NewResult(code.InternalError)
|
|
@@ -105,7 +105,7 @@ func (a *Admin) Login(ctx *gin.Context, username string, password string) (*enti
|
|
|
}
|
|
|
|
|
|
// 保存token -> RoleId 到 redis 中 过期时间为1天
|
|
|
- err = mdb.RDB.Set(ctx, generateToken, user.RoleId, 24*time.Hour).Err()
|
|
|
+ err = mdb.RDB.Set(ctx, common.GetRoleIdKey(generateToken), user.RoleId, 24*time.Hour).Err()
|
|
|
if err != nil {
|
|
|
mhayaLogger.Warnf("Login Set generateToken RoleId error:%v", err)
|
|
|
return nil, common.NewResult(code.InternalError)
|
|
@@ -147,8 +147,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(), common.GetTokenKey(generateToken), constant.AdminAccess, 1)
|
|
|
- mdb.RDB.Expire(context.Background(), common.GetTokenKey(generateToken), 24*time.Hour).Err()
|
|
|
+ mdb.RDB.HSet(context.Background(), common.GetAccessKey(generateToken), constant.AdminAccess, 1)
|
|
|
+ mdb.RDB.Expire(context.Background(), common.GetAccessKey(generateToken), 24*time.Hour).Err()
|
|
|
return nil
|
|
|
}
|
|
|
|
|
@@ -191,9 +191,9 @@ func (a *Admin) loginAuthSetRoleRedis(roleID, generateToken string) error {
|
|
|
}
|
|
|
|
|
|
for _, v := range accessList {
|
|
|
- mdb.RDB.HSet(context.Background(), common.GetTokenKey(generateToken), v.URL, 1)
|
|
|
+ mdb.RDB.HSet(context.Background(), common.GetAccessKey(generateToken), v.URL, 1)
|
|
|
}
|
|
|
- mdb.RDB.Expire(context.Background(), common.GetTokenKey(generateToken), 24*time.Hour).Err()
|
|
|
+ mdb.RDB.Expire(context.Background(), common.GetAccessKey(generateToken), 24*time.Hour).Err()
|
|
|
|
|
|
return nil
|
|
|
|