diff --git a/internal/dao/bundleExtend.go b/internal/dao/bundleExtend.go index f54fe49..1df79fa 100644 --- a/internal/dao/bundleExtend.go +++ b/internal/dao/bundleExtend.go @@ -14,20 +14,7 @@ import ( ) func AddBundleExtendRecord(data model.BundleExtensionRecords) error { - return app.ModuleClients.BundleDB.Transaction(func(tx *gorm.DB) error { - if err := tx.Create(&data).Error; err != nil { - return err - } - if data.AvailableDurationAdditional != 0 && data.TimeUnit != 0 { - record := model.BundleBalance{} - if err := tx.Model(&model.BundleBalance{}).Where(&model.BundleBalance{UserId: data.UserId}).Order("month desc").First(&record).Error; err != nil { - return err - } - record.ExpiredAt = record.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.AvailableDurationAdditional)) - return tx.Save(record).Error - } - return nil - }) + return app.ModuleClients.BundleDB.Create(&data).Error } func GetBundleExtendRecordList(req *bundle.BundleExtendRecordsListRequest) (data []model.BundleExtendRecordItemPo, total int64, err error) { @@ -380,7 +367,7 @@ func ExtendBundleBalanceByUserId(data model.BundleBalanceExtendPo) error { oldData.ManualVideoNumber += data.VideoNumber oldData.MonthlyNewManualVideoNumber += data.VideoNumber oldData.MonthlyNewDurationNumber += data.DurationNumber // 记录本月新增手动扩展时长 - oldData.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.DurationNumber)) + oldData.ExpiredAt = oldData.ExpiredAt.Add(time.Hour * 24 * time.Duration(data.DurationNumber)) return tx.Model(&model.BundleBalance{}).Where("id = ?", oldData.ID).Save(&oldData).Error }) }