add Refresh

admin/internal/logic/adminloginlogic.go

@@ -75,31 +75,48 @@ func (l *AdminLoginLogic) AdminLogin(in *admin.AdminLoginRequest) (*admin.Respon
 	retData := dao.Token{
 		Token: utils.MD5Encrypt(data.Phone + utils.Now().String()),
 	}
+	retData.Refresh = utils.MD5Encrypt(data.Phone + retData.Token)
 
-	l.setLogin(retData.Token, data)
+	l.setLogin(retData.Token, retData.Refresh, data)
 
 	return l.ok(retData), nil
 }
 
-func (l *AdminLoginLogic) setLogin(token string, data dao.JwtInfo) {
+func (l *AdminLoginLogic) setLogin(token, refresh string, data dao.JwtInfo) {
+	//获取登录过期时间
 	expireMin := utils.GetConfigInt("base.login_out_time")
-	var expire time.Duration
-	// 默认使用配置时长
-	expire = time.Duration(expireMin) * time.Minute
-	// 如果用户自定义了过期时间则覆盖
-	if data.Time > utils.NumberZero {
-		expire = time.Duration(data.Time) * time.Minute
-	}
-
+	expire := time.Duration(expireMin) * time.Minute
+	//获取刷新登录过期时间
+	refreshMin := utils.GetConfigInt("base.login_refresh_out_time")
+	refreshExpire := time.Duration(refreshMin) * time.Minute
+	//登录信息写入redis
 	redisKey := utils.GetLoginKey(utils.LoginTypeAdmin, token)
-	l.Logger.Error(redisKey)
 	userStr, _ := jsoniter.Marshal(data)
 	ctx := context.Background()
 	redis.Client.Set(ctx, redisKey, userStr, expire).Result()
-	redisKeysKey := utils.GetLoginKeysKey(utils.LoginTypeAdmin)
-	oldAuth, oldErr := redis.Client.HGet(ctx, redisKeysKey, strconv.Itoa(int(data.Id))).Result()
+
+	//获取原来的token与refresh
+	redisKeysKey := utils.GetLoginKeysKey(utils.LoginTypeAdmin, strconv.FormatInt(data.Id, 10))
+	var LoginInfo utils.LoginRedis
+	oldAuth, oldErr := redis.Client.Get(ctx, redisKeysKey).Result()
+	//如果获取到了数据情况之前的
 	if oldErr == nil {
-		redis.Client.Del(ctx, utils.GetLoginKey(utils.LoginTypeAdmin, oldAuth)).Result()
+		_ = jsoniter.Unmarshal([]byte(oldAuth), &LoginInfo)
+		if len(LoginInfo.Token) > utils.NumberOne {
+			redis.Client.Del(ctx, utils.GetLoginKey(utils.LoginTypeAdmin, LoginInfo.Token)).Result()
+		}
+		if len(LoginInfo.Refresh) > utils.NumberOne {
+			redis.Client.Del(ctx, utils.GetLoginRefreshKey(utils.LoginTypeAdmin, LoginInfo.Refresh)).Result()
+		}
 	}
-	redis.Client.HSet(ctx, redisKeysKey, strconv.Itoa(int(data.Id)), token).Result()
+	//更新新的数据
+	LoginInfo.Token = token
+	LoginInfo.Refresh = refresh
+	LoginInfo.Info = userStr
+	newStr, _ := jsoniter.Marshal(LoginInfo)
+	redis.Client.Set(ctx, redisKeysKey, newStr, refreshExpire).Result()
+	//登录刷新信息写入redis
+	refreshKey := utils.GetLoginRefreshKey(utils.LoginTypeAdmin, refresh)
+	redis.Client.Set(ctx, refreshKey, strconv.FormatInt(data.Id, 10), refreshExpire).Result()
+
 }