feat: 统计无限令牌的已用额度 (close #308)

e78289d1 · CalciumIon · ab5f17d0 · e78289d1
Commit e78289d1 authored Jul 05, 2024 by CalciumIon
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 9 deletions

model/token.go
+7 -9

No files found.
--- a/model/token.go
+++ b/model/token.go
@@ -272,15 +272,13 @@ func PostConsumeTokenQuota(tokenId int, userQuota int, quota int, preConsumedQuo
 		return err
 	}
-	if !token.UnlimitedQuota {
+	if quota > 0 {
-		if quota > 0 {
+		err = DecreaseTokenQuota(tokenId, quota)
-			err = DecreaseTokenQuota(tokenId, quota)
+	} else {
-		} else {
+		err = IncreaseTokenQuota(tokenId, -quota)
-			err = IncreaseTokenQuota(tokenId, -quota)
+	}
-		}
+	if err != nil {
-		if err != nil {
+		return err
-			return err
-		}
 	}
 	if sendEmail {