Просмотр исходного кода

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

CalciumIon 1 год назад
Родитель
Сommit
752048dfb4
1 измененных файлов с 7 добавлено и 9 удалено
  1. 7 9
      model/token.go

+ 7 - 9
model/token.go

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