oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Optimize user caching and token retrieval methods

Browse Source
This commit is contained in:
1808837298@qq.com
2025-02-19 15:12:26 +08:00
parent 0907a078b4
commit b1847509a4
7 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
service/quota.go
View File

@@ -252,7 +252,7 @@ func PreConsumeTokenQuota(relayInfo *relaycommon.RelayInfo, quota int) error {
//if relayInfo.TokenUnlimited {
// return nil
//}
token, err := model.GetTokenById(relayInfo.TokenId)
token, err := model.GetTokenByKey(relayInfo.TokenKey, false)
if err != nil {
return err
}

2
service/user_notify.go
View File

@@ -11,7 +11,7 @@ import (
func NotifyRootUser(t string, subject string, content string) {
user := model.GetRootUser().ToBaseUser()
_ = NotifyUser(&user, dto.NewNotify(t, subject, content, nil))
_ = NotifyUser(user, dto.NewNotify(t, subject, content, nil))
}
func NotifyUser(user *model.UserBase, data dto.Notify) error {