fix:修复上传

This commit is contained in:
Hiven
2023-08-10 14:49:46 +08:00
parent 4fbc17d707
commit db5b576799
2 changed files with 32 additions and 26 deletions

View File

@@ -53,30 +53,35 @@ func (m *FsResourceModel) RowSelectBuilder(selectData []string) *gorm.DB {
// 事务
func (m *FsResourceModel) Trans(ctx context.Context, fn func(ctx context.Context, connGorm *gorm.DB) error) error {
tx := m.db.Table(m.name).WithContext(ctx).Begin()
defer func() {
if r := recover(); r != nil {
tx.Rollback()
}
}()
if err := tx.Error; err != nil {
tx.Rollback()
return m.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
err := fn(ctx, tx)
return err
}
})
if err := fn(ctx, tx); err != nil {
tx.Rollback()
return err
}
// tx := m.db.Table(m.name).WithContext(ctx)
// defer func() {
// if r := recover(); r != nil {
// tx.Rollback()
// }
// }()
err := tx.Commit().Error
if err != nil {
tx.Rollback()
return err
}
// if err := tx.Error; err != nil {
// tx.Rollback()
// return err
// }
return err
// if err := fn(ctx, tx); err != nil {
// tx.Rollback()
// return err
// }
// err := tx.Commit().Error
// if err != nil {
// tx.Rollback()
// return err
// }
// return err
}
func (m *FsResourceModel) TableName() string {