89 lines
2.6 KiB
Go
89 lines
2.6 KiB
Go
package logic
|
|
|
|
import (
|
|
"admin/internal/dao"
|
|
"context"
|
|
"time"
|
|
|
|
"admin/admin"
|
|
"admin/internal/svc"
|
|
|
|
jsoniter "github.com/json-iterator/go"
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"pkg.local/redis"
|
|
"pkg.local/utils"
|
|
)
|
|
|
|
type AdminRefreshLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
BaseLogic
|
|
}
|
|
|
|
func NewAdminRefreshLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminRefreshLogic {
|
|
return &AdminRefreshLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *AdminRefreshLogic) AdminRefresh(in *admin.AdminEmptyRequest) (*admin.Response, error) {
|
|
adminInfo := utils.GetUserFromCtx(l.ctx)
|
|
l.Logger.Error(adminInfo)
|
|
if len(adminInfo.Refresh) < utils.NumberOne {
|
|
return l.fail(utils.ErrorNoLoginInfo), nil
|
|
}
|
|
|
|
//获取登录过期时间
|
|
expireMin := utils.GetConfigInt("base.login_out_time")
|
|
expire := time.Duration(expireMin) * time.Minute
|
|
//获取刷新登录过期时间
|
|
refreshMin := utils.GetConfigInt("base.login_refresh_out_time")
|
|
refreshExpire := time.Duration(refreshMin) * time.Minute
|
|
//登录信息写入redis
|
|
refreshKey := utils.GetLoginRefreshKey(utils.LoginTypeAdmin, adminInfo.Refresh)
|
|
ctx := context.Background()
|
|
id, err := redis.Client.Get(ctx, refreshKey).Result()
|
|
if err != nil {
|
|
return l.fail(utils.ErrorNoLoginInfo), nil
|
|
}
|
|
|
|
//获取原来的token与refresh
|
|
redisKeysKey := utils.GetLoginKeysKey(utils.LoginTypeAdmin, id)
|
|
var LoginInfo utils.LoginRedis
|
|
oldAuth, oldErr := redis.Client.Get(ctx, redisKeysKey).Result()
|
|
if oldErr != nil {
|
|
return l.fail(utils.ErrorNoLoginInfo), nil
|
|
}
|
|
_ = 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()
|
|
}
|
|
|
|
retData := dao.Token{
|
|
Token: utils.MD5Encrypt(id + utils.Now().String()),
|
|
}
|
|
retData.Refresh = utils.MD5Encrypt(id + retData.Token)
|
|
//var info dao.JwtInfo
|
|
//_ = jsoniter.Unmarshal(LoginInfo.Info, &info)
|
|
|
|
redisKey := utils.GetLoginKey(utils.LoginTypeAdmin, retData.Token)
|
|
redis.Client.Set(ctx, redisKey, LoginInfo.Info, expire).Result()
|
|
|
|
//更新新的数据
|
|
LoginInfo.Token = retData.Token
|
|
LoginInfo.Refresh = retData.Refresh
|
|
newStr, _ := jsoniter.Marshal(LoginInfo)
|
|
redis.Client.Set(ctx, redisKeysKey, newStr, refreshExpire).Result()
|
|
//登录刷新信息写入redis
|
|
newRefreshKey := utils.GetLoginRefreshKey(utils.LoginTypeAdmin, retData.Refresh)
|
|
redis.Client.Set(ctx, newRefreshKey, id, refreshExpire).Result()
|
|
|
|
return l.ok(retData), nil
|
|
}
|