From 87b05fa574a967be1f0d5b0d37544645bbe15c1a Mon Sep 17 00:00:00 2001
From: codeskyblue <codeskyblue@gmail.com>
Date: Tue, 22 Aug 2017 09:45:47 +0800
Subject: [PATCH] exit wdaproxy when xcodebuild quit

---
 main.go | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index 6167a18..99f75f9 100644
--- a/main.go
+++ b/main.go
@@ -169,23 +169,30 @@ func main() {
 		if err = c.Start(); err != nil {
 			log.Fatal(err)
 		}
+
+		// close writers when xcodebuild exit
+		go func() {
+			c.Wait()
+			writer.Close()
+		}()
+
 		lineStr := ""
 		for {
 			line, isPrefix, err := bufrd.ReadLine()
-			if err != nil {
-				log.Fatal("[WDA] exit", err)
-			}
-
 			if isPrefix {
 				lineStr = lineStr + string(line)
 				continue
 			} else {
 				lineStr = string(line)
 			}
-			lineStr := strings.TrimSpace(string(line))
+			lineStr = strings.TrimSpace(string(line))
+
 			if debug {
 				fmt.Printf("[WDA] %s\n", lineStr)
 			}
+			if err != nil {
+				log.Fatal("[WDA] exit", err)
+			}
 			if strings.Contains(lineStr, "Successfully wrote Manifest cache to") {
 				log.Println("[WDA] test ipa successfully generated")
 			}