From a0dc7e6fd6e4059e0b35d721da9fe77afb3f73c3 Mon Sep 17 00:00:00 2001 From: "Laisky.Cai" Date: Thu, 16 Nov 2023 05:24:56 +0000 Subject: [PATCH] refactor: Refactor token handling in the controller - Remove the unnecessary `AddRemainQuota` field from `updateTokenDto` struct - Adjust `cleanToken.RemainQuota` by subtracting `tokenPatch.AddUsedQuota` - Increase `cleanToken.UsedQuota` by adding `tokenPatch.AddUsedQuota` - Update `model.RecordLog` call to include `tokenPatch.AddReason` and `tokenPatch.AddUsedQuota` values --- controller/token.go | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/controller/token.go b/controller/token.go index db2b64fe..e3b5f6f3 100644 --- a/controller/token.go +++ b/controller/token.go @@ -168,8 +168,6 @@ type updateTokenDto struct { ExpiredTime *int64 `json:"expired_time" gorm:"bigint;default:-1"` // -1 means never expired RemainQuota *int `json:"remain_quota" gorm:"default:0"` UnlimitedQuota *bool `json:"unlimited_quota" gorm:"default:false"` - // AddRemainQuota add or subtract remain quota - AddRemainQuota int `json:"add_remain_quota" gorm:"-"` // AddUsedQuota add or subtract used quota AddUsedQuota int `json:"add_used_quota" gorm:"-"` AddReason string `json:"add_reason" gorm:"-"` @@ -243,15 +241,12 @@ func UpdateToken(c *gin.Context) { } } - cleanToken.RemainQuota += tokenPatch.AddRemainQuota + cleanToken.RemainQuota -= tokenPatch.AddUsedQuota cleanToken.UsedQuota += tokenPatch.AddUsedQuota if tokenPatch.AddUsedQuota != 0 { model.RecordLog(userId, model.LogTypeConsume, fmt.Sprintf("外部(%s)消耗 %s", tokenPatch.AddReason, common.LogQuota(tokenPatch.AddUsedQuota))) } - if tokenPatch.AddRemainQuota != 0 { - model.RecordLog(userId, model.LogTypeManage, fmt.Sprintf("修改令牌可用额度(%s) %s", tokenPatch.AddReason, common.LogQuota(tokenPatch.AddRemainQuota))) - } if err = cleanToken.Update(); err != nil { c.JSON(http.StatusOK, gin.H{