From 8284f7b2947b134eca289889ba089c596531eaab Mon Sep 17 00:00:00 2001
From: Qu Xuan <quxuan@yunionyun.com>
Date: Thu, 10 Sep 2020 17:38:32 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=B9=B3=E6=BB=91=E8=BF=81=E7=A7=BB?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/compute/models/initdb.go              |  1 +
 pkg/compute/models/storagecachedimages.go | 19 +++++++++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/pkg/compute/models/initdb.go b/pkg/compute/models/initdb.go
index a81d7e75a4..587d3f7271 100644
--- a/pkg/compute/models/initdb.go
+++ b/pkg/compute/models/initdb.go
@@ -61,6 +61,7 @@ func InitDB() error {
 		ScheduledTaskActivityManager,
 		ExternalProjectManager,
 		CachedimageManager,
+		StoragecachedimageManager,
 	} {
 		err := manager.InitializeData()
 		if err != nil {
diff --git a/pkg/compute/models/storagecachedimages.go b/pkg/compute/models/storagecachedimages.go
index 8d09cfd085..9a35cd70e3 100644
--- a/pkg/compute/models/storagecachedimages.go
+++ b/pkg/compute/models/storagecachedimages.go
@@ -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
+}