apps/photoprism
1
0
Fork 0
mirror of https://github.com/photoprism/photoprism.git synced 2025-10-10 03:00:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

CLI: refactor migrations transfer to include setting of sequence number for appropriate tables

Browse Source
This commit is contained in:
Keith Martin
2025-05-16 19:31:03 +10:00
parent 50532c5444
commit a69bf55b20
1 changed files with 114 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

115
internal/commands/migrations.go
View File

@@ -306,6 +306,10 @@ func migrationsTransferAction(ctx *cli.Context) error {
conf.MigrateDb(false, ids)
// Copy tables
type MaxResult struct {
Id uint
}
var currentid MaxResult
// Replace the admin user with the source one.
var userRecord entity.User
@@ -368,6 +372,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of users transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.User{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.User{}.TableName(), int(currentid.Id))
}
}
}
var albums entity.Albums
@@ -387,6 +400,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of albums transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Album{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Album{}.TableName(), int(currentid.Id))
}
}
}
var cameras []entity.Camera
@@ -407,6 +429,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of cameras transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Camera{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Camera{}.TableName(), int(currentid.Id))
}
}
}
var lenses entity.Lenses
@@ -427,6 +458,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of lenses transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Lens{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Lens{}.TableName(), int(currentid.Id))
}
}
}
var places []entity.Place
@@ -506,6 +546,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of keywords transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Keyword{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Keyword{}.TableName(), int(currentid.Id))
}
}
}
var labels []entity.Label
@@ -525,6 +574,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of labels transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Label{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Label{}.TableName(), int(currentid.Id))
}
}
}
var photos []entity.Photo
@@ -557,6 +615,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of photos transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Photo{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Photo{}.TableName(), int(currentid.Id))
}
}
}
var files entity.Files
@@ -575,6 +642,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of files transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.File{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.File{}.TableName(), int(currentid.Id))
}
}
}
var albumUsers []entity.AlbumUser
@@ -778,7 +854,7 @@ func migrationsTransferAction(ctx *cli.Context) error {
// Gorm bug with composite foreign keys prevents the following from working.
// It can be used once pull https://github.com/go-gorm/gorm/pull/7453 (or equivalent) has been made to gorm
// var categorys []entity.Category
// result = conf.Db().Unscoped().Debug().
// result = conf.Db().Unscoped().
// FindInBatches(&categorys, batchSize, func(tx *gorm.DB, batch int) error {
// var newCategorys []*entity.Category
// for _, category := range categorys {
@@ -863,6 +939,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of errors transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Error{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Error{}.TableName(), int(currentid.Id))
}
}
}
var faces entity.Faces
@@ -901,6 +986,15 @@ func migrationsTransferAction(ctx *cli.Context) error {
return result.Error
} else {
log.Infof("migrate: number of services transfered %v", result.RowsAffected)
if result.RowsAffected > 0 {
if result = tfrConf.Db().Unscoped().
Model(&entity.Service{}).Select("MAX(id) as id").Scan(&currentid); result.Error != nil {
log.Errorf("migrate: error in getting max id %v", result.Error)
return result.Error
} else {
resetIDToValue(tfrConf.Db(), entity.Service{}.TableName(), int(currentid.Id))
}
}
}
var fileshares []entity.FileShare
@@ -1224,3 +1318,22 @@ func migrationsTransferAction(ctx *cli.Context) error {
return nil
}
// Reset the ID increment to passed in value
func resetIDToValue(db *gorm.DB, tableName string, value int) error {
sqlCommand := ""
switch db.Dialector.Name() {
case entity.MySQL:
sqlCommand = fmt.Sprintf("ALTER TABLE `%v` AUTO_INCREMENT = %d", tableName, value+1)
case entity.Postgres:
sqlCommand = fmt.Sprintf("ALTER SEQUENCE %v_id_seq RESTART WITH %d", tableName, value+1)
case entity.SQLite3:
sqlCommand = fmt.Sprintf("UPDATE SQLITE_SEQUENCE SET SEQ=%d WHERE NAME='%v'", value, tableName)
default:
return fmt.Errorf("Unsupported Dialector %s", db.Dialector.Name())
}
if res := db.Exec(sqlCommand); res.Error != nil {
return fmt.Errorf("Reset Auto Increment failed with %v", res.Error)
}
return nil
}