Refactor function

luthermonson · Mar 20, 2024 · 1ce446d · 1ce446d
1 parent 55a167c
commit 1ce446d
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/virtual_machine.go b/virtual_machine.go
@@ -645,18 +645,20 @@ func (v *VirtualMachine) ConvertToTemplate(ctx context.Context) (task *Task, err
 }
 
 func (v *VirtualMachine) UnmountCloudInitISO(ctx context.Context, device string) error {
-	if v.HasTag(MakeTag(TagCloudInit)) {
-		if ok, err := v.deleteCloudInitISO(ctx); err != nil || !ok {
-			return err
-		}
-		task, err := v.Config(ctx, VirtualMachineOption{
-			Name:  device,
-			Value: "none,media=cdrom",
-		})
-		if err != nil {
-			return err
-		}
-		return task.WaitFor(ctx, 5)
+	if !v.HasTag(MakeTag(TagCloudInit)) {
+		return nil
+	}
+
+	_, err := v.Config(ctx, VirtualMachineOption{
+		Name:  device,
+		Value: "none,media=cdrom",
+	})
+	if err != nil {
+		return err
+	}
+
+	if _, err = v.deleteCloudInitISO(ctx); err != nil {
+		return err
 	}
 	return nil
 }