1 files changed, 11 insertions, 21 deletions
diff --git a/systray.go b/systray.go
index 38cf242..7eb17b4 100644
--- a/systray.go
+++ b/systray.go
@@ -36,8 +36,8 @@ type MenuItem struct {
 var (
 	log = golog.LoggerFor("systray")
 
-	readyCh       = make(chan struct{})
-	exitCh        = make(chan struct{})
+	systrayReady  func()
+	systrayExit   func()
 	menuItems     = make(map[int32]*MenuItem)
 	menuItemsLock sync.RWMutex
 
@@ -50,17 +50,15 @@ var (
 // Should be called at the very beginning of main() to lock at main thread.
 func Run(onReady func(), onExit func()) {
 	runtime.LockOSThread()
-	go func() {
-		for {
-			select {
-			case <-readyCh:
-				onReady()
-			case <-exitCh:
-				onExit()
-				return
-			}
-		}
-	}()
+
+	if onReady == nil {
+		onReady = func() {}
+	}
+	systrayReady = onReady
+	if onExit == nil {
+		onExit = func() {}
+	}
+	systrayExit = onExit
 
 	nativeLoop()
 }
@@ -136,14 +134,6 @@ func (item *MenuItem) update() {
 	addOrUpdateMenuItem(item)
 }
 
-func systrayReady() {
-	readyCh <- struct{}{}
-}
-
-func systrayExit() {
-	exitCh <- struct{}{}
-}
-
 func systrayMenuItemSelected(id int32) {
 	menuItemsLock.RLock()
 	item := menuItems[id]