(userId int, username string, modelName string, count int, quota int, createdAt int64, tokenUsed int)
| 95 | } |
| 96 | |
| 97 | func increaseQuotaData(userId int, username string, modelName string, count int, quota int, createdAt int64, tokenUsed int) { |
| 98 | err := DB.Table("quota_data").Where("user_id = ? and username = ? and model_name = ? and created_at = ?", |
| 99 | userId, username, modelName, createdAt).Updates(map[string]interface{}{ |
| 100 | "count": gorm.Expr("count + ?", count), |
| 101 | "quota": gorm.Expr("quota + ?", quota), |
| 102 | "token_used": gorm.Expr("token_used + ?", tokenUsed), |
| 103 | }).Error |
| 104 | if err != nil { |
| 105 | common.SysLog(fmt.Sprintf("increaseQuotaData error: %s", err)) |
| 106 | } |
| 107 | } |
| 108 | |
| 109 | func GetQuotaDataByUsername(username string, startTime int64, endTime int64) (quotaData []*QuotaData, err error) { |
| 110 | var quotaDatas []*QuotaData |
no test coverage detected