half work

watch.go

@@ -1,27 +1,33 @@
 package main
 
 import (
-	"bytes"
 	"fmt"
 	"github.com/howeyc/fsnotify"
-	"io"
+	//"io"
 	"log"
 	"os"
 	"os/exec"
 	"runtime"
+	"sync"
+	"time"
 )
 
 var (
 	builderror chan string
 	restart    chan bool
 	cmd        *exec.Cmd
+	state      sync.Mutex
+	running    bool
 )
 
 func init() {
 	builderror = make(chan string)
-	restart = make(chan bool)
+	//restart = make(chan bool)
+	running = false
 }
 
+var eventTime = make(map[string]time.Time)
+
 func NewWatcher(paths []string) {
 	watcher, err := fsnotify.NewWatcher()
 	if err != nil {
@@ -32,8 +38,21 @@ func NewWatcher(paths []string) {
 		for {
 			select {
 			case e := <-watcher.Event:
-				fmt.Println(e)
-				go Autobuild()
+
+				isbuild := true
+				if t, ok := eventTime[e.String()]; ok {
+					//Debugf("%s pk %s", t, time.Now())
+					if t.Add(time.Millisecond * 500).After(time.Now()) {
+						isbuild = false
+					}
+				}
+				Debugf("isbuild:%v", isbuild)
+				eventTime[e.String()] = time.Now()
+
+				if isbuild {
+					fmt.Println(e)
+					go Autobuild()
+				}
 			case err := <-watcher.Error:
 				log.Fatal("error:", err)
 			}
@@ -64,23 +83,18 @@ func Autobuild() {
 
 		}
 	}()
-	fmt.Println("Autobuild")
+	fmt.Println("autobuild")
 	path, _ := os.Getwd()
 	os.Chdir(path)
 	bcmd := exec.Command("go", "build")
-	var out bytes.Buffer
-	var berr bytes.Buffer
-	bcmd.Stdout = &out
-	bcmd.Stderr = &berr
+	bcmd.Stdout = os.Stdout
+	bcmd.Stderr = os.Stderr
 	err := bcmd.Run()
 	if err != nil {
-		fmt.Println("run error", err)
-	}
-	if out.String() == "" {
-		Kill()
-	} else {
-		builderror <- berr.String()
+		builderror <- err.Error()
+		return
 	}
+	Restart(appname)
 }
 
 func Kill() {
@@ -90,31 +104,58 @@ func Kill() {
 	}
 }
 
+func Restart(appname string) {
+	Kill()
+	go Start()
+}
 func Start(appname string) {
 	fmt.Println("start", appname)
+
 	cmd = exec.Command(appname)
-	stdout, err := cmd.StdoutPipe()
-	if err != nil {
-		fmt.Println("stdout:", err)
-	}
-	stderr, err := cmd.StderrPipe()
-	if err != nil {
-		fmt.Println("stdin:", err)
-	}
-	r := io.MultiReader(stdout, stderr)
-	err = cmd.Start()
-	if err != nil {
-		fmt.Println("cmd start:", err)
-	}
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	ch := Go(func() error {
+		state.Lock()
+		defer state.Unlock()
+		running = true
+		return cmd.Run()
+	})
 	for {
-		buf := make([]byte, 1024)
-		count, err := r.Read(buf)
-		if err != nil || count == 0 {
-			fmt.Println("process exit")
-			restart <- true
+		select {
+		case err := <-ch:
+			fmt.Println("cmd start error: ", err)
+			state.Lock()
+			defer state.Unlock()
+			running = false
 			return
-		} else {
-			fmt.Println("result:", string(buf))
+		case <-time.After(2 * time.Second):
 		}
 	}
+
+	//stdout, err := cmd.StdoutPipe()
+
+	//if err != nil {
+	//fmt.Println("stdout:", err)
+	//}
+	//stderr, err := cmd.StderrPipe()
+	//if err != nil {
+	//fmt.Println("stdin:", err)
+	//}
+	//r := io.MultiReader(stdout, stderr)
+	//err = cmd.Start()
+	//if err != nil {
+	//	fmt.Println("cmd start:", err)
+	//}
+	//for {
+	//	buf := make([]byte, 1024)
+	//	count, err := r.Read(buf)
+	//	if err != nil || count == 0 {
+	//		fmt.Println("process exit")
+	//		restart <- true
+	//		return
+	//	} else {
+	//		fmt.Println("result:", string(buf))
+	//	}
+	//}
 }