forked from lxh/go-wxhelper
79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
|
package handle
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/go-oauth2/oauth2/v4"
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
"strings"
|
|||
|
|
"wechat-robot/internal/redis"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func NewAccessGenerate() *AccessGenerate {
|
|||
|
|
return &AccessGenerate{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type AccessGenerate struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Token
|
|||
|
|
// @description: 手动实现Token生成，直接生成UUID，替换掉自带的那个憨得一批的长长的字符串
|
|||
|
|
// @receiver ag
|
|||
|
|
// @param ctx context.Context 上下文
|
|||
|
|
// @param data *oauth2.GenerateBasic 生成基础数据
|
|||
|
|
// @param isGenRefresh bool 是否生成RefreshToken
|
|||
|
|
// @return string AccessToken 令牌
|
|||
|
|
// @return string RefreshToken 刷新令牌
|
|||
|
|
// @return error 错误信息
|
|||
|
|
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
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// clearOldToken
|
|||
|
|
// @description: 清理掉旧的Token和RefreshToken
|
|||
|
|
// @receiver ag
|
|||
|
|
// @param ctx context.Context 上下文
|
|||
|
|
// @param userId string 用户ID
|
|||
|
|
func (ag *AccessGenerate) clearOldToken(ctx context.Context, userId string) {
|
|||
|
|
key := fmt.Sprintf("oauth:token:token:%s", userId)
|
|||
|
|
accessToken, err := redis.Client.Get(context.Background(), key).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
//log.Errorf("获取缓存用户的accessToken失败: %v", err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if accessToken != "" {
|
|||
|
|
// 老的Token
|
|||
|
|
var baseKey string
|
|||
|
|
baseKey, err = redis.Client.Get(ctx, "oauth:token:"+accessToken).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 老Token详细数据
|
|||
|
|
var dataStr string
|
|||
|
|
dataStr, err = redis.Client.Get(ctx, "oauth:token:"+baseKey).Result()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var m map[string]interface{}
|
|||
|
|
if err = json.Unmarshal([]byte(dataStr), &m); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 删除AccessToken等信息
|
|||
|
|
redis.Client.Del(ctx, fmt.Sprintf("oauth:token:%v", m["Access"]))
|
|||
|
|
redis.Client.Del(ctx, fmt.Sprintf("oauth:token:%v", m["Refresh"]))
|
|||
|
|
redis.Client.Del(ctx, "oauth:token:"+baseKey)
|
|||
|
|
}
|
|||
|
|
}
|