refactor: Optimize user caching and token retrieval methods

2025-02-19 15:12:26 +08:00
parent 0907a078b4
commit b1847509a4
7 changed files with 21 additions and 8 deletions
--- a/service/quota.go
+++ b/service/quota.go
@@ -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
 	}