Merge pull request #1 from Code-Hex/add/stateNotify

added stateNotify

Author: Code-Hex

example/main.go

@@ -145,7 +145,9 @@ func main() {
 		for {
 			select {
 			case <-t.C:
+			case newState := <-vm.StateChangedNotify():
 				log.Println(
+					"newState:", newState,
 					"state:", vm.State(),
 					"canStart:", vm.CanStart(),
 					"canResume:", vm.CanResume(),

virtualization.go

@@ -76,7 +76,8 @@ type VirtualMachine struct {
 
 type (
 	machineStatus struct {
-		state VirtualMachineState
+		state       VirtualMachineState
+		stateNotify chan VirtualMachineState
 
 		mu sync.RWMutex
 	}
@@ -104,7 +105,8 @@ func NewVirtualMachine(config *VirtualMachineConfiguration) *VirtualMachine {
 	cs := charWithGoString(id)
 	defer cs.Free()
 	statuses[id] = &machineStatus{
-		state: VirtualMachineState(0),
+		state:       VirtualMachineState(0),
+		stateNotify: make(chan VirtualMachineState),
 	}
 	handlers[id] = &machineHandlers{
 		start:  func(error) {},
@@ -137,7 +139,10 @@ func changeStateOnObserver(state C.int, cID *C.char) {
 	// if caused panic, that's unexpected behavior.
 	v, _ := statuses[id.String()]
 	v.mu.Lock()
-	v.state = VirtualMachineState(state)
+	newState := VirtualMachineState(state)
+	v.state = newState
+	// for non-blocking
+	go func() { v.stateNotify <- newState }()
 	statuses[id.String()] = v
 	v.mu.Unlock()
 }
@@ -152,6 +157,16 @@ func (v *VirtualMachine) State() VirtualMachineState {
 	return val.state
 }
 
+// StateChangedNotify gets notification is changed execution state of the virtual machine.
+func (v *VirtualMachine) StateChangedNotify() <-chan VirtualMachineState {
+	// I expected it will not cause panic.
+	// if caused panic, that's unexpected behavior.
+	val, _ := statuses[v.id]
+	val.mu.RLock()
+	defer val.mu.RUnlock()
+	return val.stateNotify
+}
+
 // CanStart returns true if the machine is in a state that can be started.
 func (v *VirtualMachine) CanStart() bool {
 	return bool(C.vmCanStart(v.Ptr(), v.dispatchQueue))