57 lines
1.2 KiB
Go
57 lines
1.2 KiB
Go
package logging
|
||
|
||
import (
|
||
"fmt"
|
||
"strings"
|
||
jwtx "tyc-server/common/jwt"
|
||
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
// jwtExtractor JWT用户信息提取器
|
||
type jwtExtractor struct {
|
||
jwtSecret string
|
||
}
|
||
|
||
// newJWTExtractor 创建JWT提取器
|
||
func newJWTExtractor(jwtSecret string) *jwtExtractor {
|
||
return &jwtExtractor{
|
||
jwtSecret: jwtSecret,
|
||
}
|
||
}
|
||
|
||
// ExtractUserInfo 从Authorization头部提取用户信息
|
||
func (e *jwtExtractor) ExtractUserInfo(authHeader string) (userID, username string) {
|
||
if authHeader == "" {
|
||
return "", ""
|
||
}
|
||
|
||
// 检查Bearer前缀
|
||
if !strings.HasPrefix(authHeader, "Bearer ") {
|
||
return "", ""
|
||
}
|
||
|
||
// 提取Token
|
||
tokenString := strings.TrimPrefix(authHeader, "Bearer ")
|
||
if tokenString == "" {
|
||
return "", ""
|
||
}
|
||
|
||
// 解析JWT Token
|
||
userIDInt, err := jwtx.ParseJwtToken(tokenString, e.jwtSecret)
|
||
if err != nil {
|
||
logx.Errorf("解析JWT Token失败: %v", err)
|
||
return "", ""
|
||
}
|
||
|
||
// 提取用户信息
|
||
if userIDInt > 0 {
|
||
userID = fmt.Sprintf("%d", userIDInt)
|
||
// 由于JWT中只包含用户ID，用户名需要从其他地方获取
|
||
// 这里可以调用用户服务获取用户名，或者暂时使用用户ID
|
||
username = fmt.Sprintf("user_%d", userIDInt)
|
||
}
|
||
|
||
return userID, username
|
||
}
|