Merge branch 'develop' into feature/auth

model/gmodel/fs_resource_logic.go

@@ -52,23 +52,36 @@ 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 {
+	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()
+	// 	}
+	// }()
 
-	return tx.Commit().Error
+	// if err := tx.Error; err != nil {
+	// 	tx.Rollback()
+	// 	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 {

model/gmodel/fs_user_material_logic.go

@@ -70,3 +70,8 @@ func (m *FsUserMaterialModel) FindLatestOne(ctx context.Context, userId int64, g
 	err = db.Take(&resp).Error
 	return resp, err
 }
+
+func (m *FsUserMaterialModel) FindOneById(ctx context.Context, id int64) (resp *FsUserMaterial, err error) {
+	err = m.db.WithContext(ctx).Model(&FsUserMaterial{}).Where("id = ?", id).Take(&resp).Error
+	return resp, err
+}