33 lines
841 B
Go
33 lines
841 B
Go
package database
|
|
|
|
import (
|
|
"RenewCMS/internal/domain/role"
|
|
entity "RenewCMS/internal/infrastructure/persistence/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func BackfillCreatedBy(db *gorm.DB) error {
|
|
return db.Transaction(func(tx *gorm.DB) error {
|
|
ownerRoleID, seeded, err := builtInRoleID(tx, role.OwnerName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !seeded {
|
|
return nil
|
|
}
|
|
var owner entity.User
|
|
result := tx.Where("role_id = ?", ownerRoleID).Order("id asc").Limit(1).Find(&owner)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected == 0 {
|
|
return nil
|
|
}
|
|
if err := tx.Unscoped().Model(&entity.Article{}).Where("created_by = 0").UpdateColumn("created_by", owner.ID).Error; err != nil {
|
|
return err
|
|
}
|
|
return tx.Model(&entity.Entry{}).Where("created_by = 0").UpdateColumn("created_by", owner.ID).Error
|
|
})
|
|
}
|