From 55d16474031e46a96ef0e295241b3fec85dfbb66 Mon Sep 17 00:00:00 2001
From: Qiu Jian <qiujian@yunionyun.com>
Date: Wed, 8 Dec 2021 00:27:08 +0800
Subject: [PATCH] fix: fail to recover disk status if disk in unknown status

---
 pkg/compute/regiondrivers/kvm.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/pkg/compute/regiondrivers/kvm.go b/pkg/compute/regiondrivers/kvm.go
index 2d768365ae..eb8a21c46c 100644
--- a/pkg/compute/regiondrivers/kvm.go
+++ b/pkg/compute/regiondrivers/kvm.go
@@ -1206,7 +1206,11 @@ func (self *SKVMRegionDriver) RequestSyncDiskStatus(ctx context.Context, userCre
 		originStatus, _ := task.GetParams().GetString("origin_status")
 		status, _ := res.GetString("status")
 		if status == api.DISK_EXIST {
-			diskStatus = originStatus
+			if originStatus == api.DISK_UNKNOWN {
+				diskStatus = api.DISK_READY
+			} else {
+				diskStatus = originStatus
+			}
 		} else {
 			diskStatus = api.DISK_UNKNOWN
 		}
@@ -1229,7 +1233,11 @@ func (self *SKVMRegionDriver) RequestSyncSnapshotStatus(ctx context.Context, use
 		originStatus, _ := task.GetParams().GetString("origin_status")
 		status, _ := res.GetString("status")
 		if status == api.SNAPSHOT_EXIST {
-			snapshotStatus = originStatus
+			if originStatus == api.SNAPSHOT_UNKNOWN {
+				snapshotStatus = api.SNAPSHOT_READY
+			} else {
+				snapshotStatus = originStatus
+			}
 		} else {
 			snapshotStatus = api.SNAPSHOT_UNKNOWN
 		}