fix: 平滑迁移数据

This commit is contained in:
Qu Xuan
2020-09-10 17:38:32 +08:00
parent 8bc444907b
commit 8284f7b294
2 changed files with 20 additions and 0 deletions
+1
View File
@@ -61,6 +61,7 @@ func InitDB() error {
ScheduledTaskActivityManager,
ExternalProjectManager,
CachedimageManager,
StoragecachedimageManager,
} {
err := manager.InitializeData()
if err != nil {
+19
View File
@@ -545,3 +545,22 @@ func (manager *SStoragecachedimageManager) OrderByExtraFields(
return q, nil
}
func (manager *SStoragecachedimageManager) InitializeData() error {
images := []SStoragecachedimage{}
q := manager.Query().Equals("status", "ready")
err := db.FetchModelObjects(manager, q, &images)
if err != nil {
return errors.Wrapf(err, "db.FetchModelObjects")
}
for i := range images {
_, err := db.Update(&images[i], func() error {
images[i].Status = api.CACHED_IMAGE_STATUS_ACTIVE
return nil
})
if err != nil {
return errors.Wrapf(err, "db.Update(%d)", images[i].RowId)
}
}
return nil
}