From 55228bce05283f2efdb737a4b31600179b66dbb8 Mon Sep 17 00:00:00 2001
From: Zexi Li <zexi.li@icloud.com>
Date: Wed, 12 Apr 2023 00:46:55 +0800
Subject: [PATCH] feat(region): set kvm storage instance capability

---
 pkg/compute/guestdrivers/kvm.go | 20 ++++++++++++++++++++
 pkg/compute/options/options.go  |  3 +++
 2 files changed, 23 insertions(+)

diff --git a/pkg/compute/guestdrivers/kvm.go b/pkg/compute/guestdrivers/kvm.go
index 3aecab4b1b..f1ddadffd5 100644
--- a/pkg/compute/guestdrivers/kvm.go
+++ b/pkg/compute/guestdrivers/kvm.go
@@ -85,6 +85,26 @@ func (self *SKVMGuestDriver) GetInstanceCapability() cloudprovider.SInstanceCapa
 				DefaultAccount: api.VM_DEFAULT_WINDOWS_LOGIN_USER,
 			},
 		},
+		Storages: cloudprovider.Storage{
+			SysDisk: []cloudprovider.StorageInfo{
+				{
+					StorageType: api.STORAGE_LOCAL,
+					MinSizeGb:   30,
+					MaxSizeGb:   2048,
+					StepSizeGb:  1,
+					Resizable:   true,
+				},
+			},
+			DataDisk: []cloudprovider.StorageInfo{
+				{
+					StorageType: api.STORAGE_LOCAL,
+					MinSizeGb:   options.Options.LocalDataDiskMinSizeGB,
+					MaxSizeGb:   options.Options.LocalDataDiskMaxSizeGB,
+					StepSizeGb:  1,
+					Resizable:   true,
+				},
+			},
+		},
 	}
 }
 
diff --git a/pkg/compute/options/options.go b/pkg/compute/options/options.go
index 4cd32c0246..ac05ae5115 100644
--- a/pkg/compute/options/options.go
+++ b/pkg/compute/options/options.go
@@ -200,6 +200,9 @@ type ComputeOptions struct {
 	MonitorEndpointType               string `help:"specify monitor endpoint type" default:"public"`
 	ForceUseOriginVnc                 bool   `help:"force openstack use origin vnc console" default:"true"`
 
+	LocalDataDiskMinSizeGB int `help:"Data disk min size when using local storage" default:"10"`
+	LocalDataDiskMaxSizeGB int `help:"Data disk max size when using local storage" default:"40960"`
+
 	esxi.EsxiOptions
 }