From 2a17b22f29cb213ad9166c3a83dd3f5e03593039 Mon Sep 17 00:00:00 2001
From: Kei Kamikawa <code-hex@users.noreply.github.com>
Date: Tue, 30 Aug 2022 01:33:01 +0900
Subject: [PATCH] fixed example for macOS

---
 example/macOS/main.go | 62 ++++++++++++++++++++++++++++---------------
 1 file changed, 40 insertions(+), 22 deletions(-)

diff --git a/example/macOS/main.go b/example/macOS/main.go
index d9504db..20117cb 100644
--- a/example/macOS/main.go
+++ b/example/macOS/main.go
@@ -6,9 +6,8 @@ import (
 	"fmt"
 	"log"
 	"os"
-	"os/signal"
 	"runtime"
-	"syscall"
+	"time"
 
 	"github.com/Code-Hex/vz/v2"
 )
@@ -48,9 +47,6 @@ func runVM(ctx context.Context) error {
 	}
 	vm := vz.NewVirtualMachine(config)
 
-	signalCh := make(chan os.Signal, 1)
-	signal.Notify(signalCh, syscall.SIGTERM)
-
 	errCh := make(chan error, 1)
 
 	vm.Start(func(err error) {
@@ -59,28 +55,50 @@ func runVM(ctx context.Context) error {
 		}
 	})
 
-	vm.StartGraphicApplication(960, 600)
+	go func() {
+		for {
+			select {
+			case newState := <-vm.StateChangedNotify():
+				if newState == vz.VirtualMachineStateRunning {
+					log.Println("start VM is running")
+				}
+				if newState == vz.VirtualMachineStateStopped || newState == vz.VirtualMachineStateStopping {
+					log.Println("stopped state")
+					errCh <- nil
+					return
+				}
+			case err := <-errCh:
+				errCh <- fmt.Errorf("failed to start vm: %w", err)
+				return
+			}
+		}
+	}()
 
-	for {
-		select {
-		case <-signalCh:
+	// cleanup is this function is useful when finished graphic application.
+	cleanup := func() {
+		for i := 1; vm.CanRequestStop(); i++ {
 			result, err := vm.RequestStop()
-			if err != nil {
-				return err
-			}
-			log.Println("recieved signal", result)
-		case newState := <-vm.StateChangedNotify():
-			if newState == vz.VirtualMachineStateRunning {
-				log.Println("start VM is running")
-			}
-			if newState == vz.VirtualMachineStateStopped {
-				log.Println("stopped successfully")
-				return nil
+			log.Printf("sent stop request(%d): %t, %v", i, result, err)
+			time.Sleep(time.Second * 3)
+			if i > 3 {
+				log.Println("call stop")
+				vm.Stop(func(err error) {
+					if err != nil {
+						log.Println("stop with error", err)
+					}
+				})
 			}
-		case err := <-errCh:
-			return fmt.Errorf("failed to start vm: %w", err)
 		}
+		log.Println("finished cleanup")
 	}
+
+	runtime.LockOSThread()
+	vm.StartGraphicApplication(960, 600)
+	runtime.UnlockOSThread()
+
+	cleanup()
+
+	return <-errCh
 }
 
 func computeCPUCount() uint {