66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
|
package handle
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"Lee-WineList/client"
|
|||
|
|
"Lee-WineList/common/constant"
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"git.echol.cn/loser/logger/log"
|
|||
|
|
"github.com/go-oauth2/oauth2/v4"
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func NewAccessGenerate() *AccessGenerate {
|
|||
|
|
return &AccessGenerate{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type AccessGenerate struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Token 手动实现Token生成，直接生成UUID，替换掉自带的那个憨得一批的长长的字符串
|
|||
|
|
func (ag *AccessGenerate) Token(ctx context.Context, data *oauth2.GenerateBasic, isGenRefresh bool) (string, string, error) {
|
|||
|
|
u, _ := uuid.NewUUID()
|
|||
|
|
access := strings.ReplaceAll(u.String(), "-", "")
|
|||
|
|
|
|||
|
|
refresh := ""
|
|||
|
|
if isGenRefresh {
|
|||
|
|
u, _ = uuid.NewUUID()
|
|||
|
|
refresh = strings.ReplaceAll(u.String(), "-", "")
|
|||
|
|
}
|
|||
|
|
// 生成新的，清理掉旧的
|
|||
|
|
ag.clearOldToken(ctx, data.UserID)
|
|||
|
|
// 返回结果
|
|||
|
|
return access, refresh, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 清理掉旧的Token和RefreshToken
|
|||
|
|
func (ag *AccessGenerate) clearOldToken(ctx context.Context, userId string) {
|
|||
|
|
rdsKey := constant.OAuth2RedisKey + "*-*"
|
|||
|
|
// 获取所有符合条件的Key
|
|||
|
|
keys, err := client.Redis.Keys(ctx, rdsKey).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
log.Errorf("清理失败，跳过清理")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, key := range keys {
|
|||
|
|
dataStr, err := client.Redis.Get(ctx, key).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
var m map[string]any
|
|||
|
|
if err = json.Unmarshal([]byte(dataStr), &m); err != nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
// 找到匹配的数据
|
|||
|
|
if m["UserID"] == userId {
|
|||
|
|
// 删除AccessToken
|
|||
|
|
client.Redis.Del(ctx, fmt.Sprintf("%v%v", constant.OAuth2RedisKey, m["Access"]))
|
|||
|
|
client.Redis.Del(ctx, fmt.Sprintf("%v%v", constant.OAuth2RedisKey, m["Refresh"]))
|
|||
|
|
client.Redis.Del(ctx, key)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|