From 371b89f393d8264ae31cee61614f94669de1e93e Mon Sep 17 00:00:00 2001
From: wanyaoqi <wanyaoqi@yunionyun.com>
Date: Fri, 16 Aug 2019 17:59:01 +0800
Subject: [PATCH] fix managered disk reset not update disk size

---
 pkg/compute/regiondrivers/managedvirtual.go | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/pkg/compute/regiondrivers/managedvirtual.go b/pkg/compute/regiondrivers/managedvirtual.go
index d2becc1dc3..2409a04995 100644
--- a/pkg/compute/regiondrivers/managedvirtual.go
+++ b/pkg/compute/regiondrivers/managedvirtual.go
@@ -1186,7 +1186,20 @@ func (self *SManagedVirtualizationRegionDriver) OnDiskReset(ctx context.Context,
 	if err != nil {
 		return err
 	}
-	return iDisk.Refresh()
+	err = iDisk.Refresh()
+	if err != nil {
+		return err
+	}
+	if disk.DiskSize != iDisk.GetDiskSizeMB() {
+		_, err := db.Update(disk, func() error {
+			disk.DiskSize = iDisk.GetDiskSizeMB()
+			return nil
+		})
+		if err != nil {
+			return err
+		}
+	}
+	return nil
 }
 
 func (self *SManagedVirtualizationRegionDriver) ValidateCreateSnapshotPolicyData(ctx context.Context, userCred mcclient.TokenCredential, data *compute.SSnapshotPolicyCreateInput) error {