fix: handle netbox migration to disk set in MB

e-breuninger · Sep 27, 2024 · 4db2400 · 4db2400
1 parent a4e9def
commit 4db2400
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/netbox/data_source_netbox_virtual_machines.go b/netbox/data_source_netbox_virtual_machines.go
@@ -244,7 +244,7 @@ func dataSourceNetboxVirtualMachineRead(d *schema.ResourceData, m interface{}) e
 			mapping["custom_fields"] = v.CustomFields
 		}
 		if v.Disk != nil {
-			mapping["disk_size_gb"] = *v.Disk
+			mapping["disk_size_gb"] = *v.Disk / 1000
 		}
 		if v.LocalContextData != nil {
 			if localContextData, err := json.Marshal(v.LocalContextData); err == nil {

diff --git a/netbox/resource_netbox_virtual_machine.go b/netbox/resource_netbox_virtual_machine.go
@@ -153,7 +153,7 @@ func resourceNetboxVirtualMachineCreate(ctx context.Context, d *schema.ResourceD
 
 	diskSizeValue, ok := d.GetOk("disk_size_gb")
 	if ok {
-		diskSize := int64(diskSizeValue.(int))
+		diskSize := int64(diskSizeValue.(int)) * 1000
 		data.Disk = &diskSize
 	}
 
@@ -308,7 +308,10 @@ func resourceNetboxVirtualMachineRead(ctx context.Context, d *schema.ResourceDat
 		d.Set("vcpus", nil)
 	}
 	d.Set("memory_mb", vm.Memory)
-	d.Set("disk_size_gb", vm.Disk)
+	if vm.Disk != nil {
+		diskSizeGb := *vm.Disk / 1000
+		d.Set("disk_size_gb", diskSizeGb)
+	}
 	if vm.Status != nil {
 		d.Set("status", vm.Status.Value)
 	} else {