Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add error log for ValidateUserToken

Browse Source
This commit is contained in:
seven.yu 2024-01-04 16:39:36 +08:00
parent 81e6408217
commit 604262c500
1 changed files with 33 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
model/token.go
View File

@ -38,7 +38,10 @@ func ValidateUserToken(key string) (token *Token, err error) {
return nil, errors.New("未提供令牌") return nil, errors.New("未提供令牌")
} }
token, err = CacheGetTokenByKey(key) token, err = CacheGetTokenByKey(key)
if err == nil { if err != nil {
common.SysError("failed to get token" + err.Error())
return nil, errors.New("无效的令牌")
}
if token.Status == common.TokenStatusExhausted { if token.Status == common.TokenStatusExhausted {
return nil, errors.New("该令牌额度已用尽") return nil, errors.New("该令牌额度已用尽")
} else if token.Status == common.TokenStatusExpired { } else if token.Status == common.TokenStatusExpired {
@ -69,8 +72,6 @@ func ValidateUserToken(key string) (token *Token, err error) {
return nil, errors.New("该令牌额度已用尽") return nil, errors.New("该令牌额度已用尽")
} }
return token, nil return token, nil
}
return nil, errors.New("无效的令牌")
} }
func GetTokenByIds(id int, userId int) (*Token, error) { func GetTokenByIds(id int, userId int) (*Token, error) {