diff --git a/internal/dao/bundleExtend.go b/internal/dao/bundleExtend.go index d4e2864..8880e60 100644 --- a/internal/dao/bundleExtend.go +++ b/internal/dao/bundleExtend.go @@ -367,9 +367,14 @@ func ExtendBundleBalanceByUserId(data model.BundleBalanceExtendPo) error { if err := tx.Model(&model.BundleBalance{}).Where("user_id = ?", data.UserId).Order("created_at desc").First(&oldData).Error; err != nil { return errors.New("用户还没有套餐信息") } + oldData.ManualAccountNumber += data.AccountNumber + oldData.MonthlyNewAccountNumber += data.AccountNumber oldData.ManualImageNumber += data.ImageNumber + oldData.MonthlyNewManualImageNumber += data.ImageNumber oldData.ManualDataAnalysisNumber += data.DataAnalysisNumber + oldData.MonthlyNewManualDataAnalysisNumber += data.DataAnalysisNumber oldData.ManualVideoNumber += data.VideoNumber + oldData.MonthlyNewManualVideoNumber += data.VideoNumber oldData.MonthlyNewDurationNumber += data.DurationNumber // 记录本月新增手动扩展时长 oldData.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.DurationNumber)) return tx.Model(&model.BundleBalance{}).Where("id = ?", oldData.ID).Save(&oldData).Error