coverage: fix race of signal vs start of a process

License: MIT Signed-off-by: Jakub Sztandera <kubuxu@protonmail.ch>
ipfs · Apr 4, 2019 · 9464408 · 9464408
1 parent dd498d9
commit 9464408
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/coverage/main/main.go b/coverage/main/main.go
@@ -43,15 +43,26 @@ func main() {
 		Pdeathsig: syscall.SIGTERM,
 	}
 
-	sig := make(chan os.Signal, 1)
+	sig := make(chan os.Signal, 10)
+	start := make(chan struct{})
 	go func() {
+		<-start
 		for {
 			p.Process.Signal(<-sig)
 		}
 	}()
+
 	signal.Notify(sig, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
 
-	err = p.Run()
+	err = p.Start()
+	if err != nil {
+		fmt.Println(err.Error())
+		os.Exit(1)
+	}
+
+	close(start)
+
+	err = p.Wait()
 	if err != nil {
 		fmt.Println(err.Error())
 		os.Exit(1)