fix: get storage class name from vmimage status (#94)

Signed-off-by: PoAn Yang <[email protected]>
harvester · Apr 18, 2024 · 47d1f28 · 47d1f28
1 parent 980cb7f
commit 47d1f28
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/provider/virtualmachine/resource_virtualmachine_constructor.go b/internal/provider/virtualmachine/resource_virtualmachine_constructor.go
@@ -218,8 +218,12 @@ func (c *Constructor) Setup() util.Processors {
 						if err != nil {
 							return err
 						}
+						vmimage, err := c.Client.HarvesterClient.HarvesterhciV1beta1().VirtualMachineImages(imageNamespace).Get(c.Context, imageName, metav1.GetOptions{})
+						if err != nil {
+							return err
+						}
 						pvcOption.ImageID = helper.BuildNamespacedName(imageNamespace, imageName)
-						scName := builder.BuildImageStorageClassName("", imageName)
+						scName := vmimage.Status.StorageClassName
 						if storageClassName == "" {
 							storageClassName = scName
 						} else if storageClassName != scName {