2012-12-15 16:56:28 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"github.com/howeyc/fsnotify"
|
2013-07-06 07:30:57 +00:00
|
|
|
//"io"
|
2012-12-15 16:56:28 +00:00
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
"os/exec"
|
|
|
|
"runtime"
|
2013-07-06 07:30:57 +00:00
|
|
|
"sync"
|
|
|
|
"time"
|
2012-12-15 16:56:28 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
builderror chan string
|
|
|
|
restart chan bool
|
|
|
|
cmd *exec.Cmd
|
2013-07-06 07:30:57 +00:00
|
|
|
state sync.Mutex
|
|
|
|
running bool
|
2012-12-15 16:56:28 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
builderror = make(chan string)
|
2013-07-06 07:30:57 +00:00
|
|
|
//restart = make(chan bool)
|
|
|
|
running = false
|
2012-12-15 16:56:28 +00:00
|
|
|
}
|
|
|
|
|
2013-07-06 07:30:57 +00:00
|
|
|
var eventTime = make(map[string]time.Time)
|
|
|
|
|
2012-12-15 16:56:28 +00:00
|
|
|
func NewWatcher(paths []string) {
|
|
|
|
watcher, err := fsnotify.NewWatcher()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case e := <-watcher.Event:
|
2013-07-06 07:30:57 +00:00
|
|
|
|
|
|
|
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()
|
|
|
|
}
|
2012-12-15 16:56:28 +00:00
|
|
|
case err := <-watcher.Error:
|
|
|
|
log.Fatal("error:", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
for _, path := range paths {
|
|
|
|
fmt.Println(path)
|
|
|
|
err = watcher.Watch(path)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func Autobuild() {
|
|
|
|
defer func() {
|
|
|
|
if err := recover(); err != nil {
|
|
|
|
str := ""
|
|
|
|
for i := 1; ; i += 1 {
|
|
|
|
_, file, line, ok := runtime.Caller(i)
|
|
|
|
if !ok {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
str = str + fmt.Sprintf("%v,%v", file, line)
|
|
|
|
}
|
|
|
|
builderror <- str
|
|
|
|
|
|
|
|
}
|
|
|
|
}()
|
2013-07-06 07:30:57 +00:00
|
|
|
fmt.Println("autobuild")
|
2012-12-15 16:56:28 +00:00
|
|
|
path, _ := os.Getwd()
|
|
|
|
os.Chdir(path)
|
|
|
|
bcmd := exec.Command("go", "build")
|
2013-07-06 07:30:57 +00:00
|
|
|
bcmd.Stdout = os.Stdout
|
|
|
|
bcmd.Stderr = os.Stderr
|
2012-12-15 16:56:28 +00:00
|
|
|
err := bcmd.Run()
|
|
|
|
if err != nil {
|
2013-07-06 07:30:57 +00:00
|
|
|
builderror <- err.Error()
|
|
|
|
return
|
2012-12-15 16:56:28 +00:00
|
|
|
}
|
2013-07-06 07:30:57 +00:00
|
|
|
Restart(appname)
|
2012-12-15 16:56:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func Kill() {
|
|
|
|
err := cmd.Process.Kill()
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-06 07:30:57 +00:00
|
|
|
func Restart(appname string) {
|
|
|
|
Kill()
|
|
|
|
go Start()
|
|
|
|
}
|
2012-12-15 16:56:28 +00:00
|
|
|
func Start(appname string) {
|
|
|
|
fmt.Println("start", appname)
|
2013-07-06 07:30:57 +00:00
|
|
|
|
2012-12-15 16:56:28 +00:00
|
|
|
cmd = exec.Command(appname)
|
2013-07-06 07:30:57 +00:00
|
|
|
cmd.Stdout = os.Stdout
|
|
|
|
cmd.Stderr = os.Stderr
|
|
|
|
|
|
|
|
ch := Go(func() error {
|
|
|
|
state.Lock()
|
|
|
|
defer state.Unlock()
|
|
|
|
running = true
|
|
|
|
return cmd.Run()
|
|
|
|
})
|
2012-12-15 16:56:28 +00:00
|
|
|
for {
|
2013-07-06 07:30:57 +00:00
|
|
|
select {
|
|
|
|
case err := <-ch:
|
|
|
|
fmt.Println("cmd start error: ", err)
|
|
|
|
state.Lock()
|
|
|
|
defer state.Unlock()
|
|
|
|
running = false
|
2012-12-15 16:56:28 +00:00
|
|
|
return
|
2013-07-06 07:30:57 +00:00
|
|
|
case <-time.After(2 * time.Second):
|
2012-12-15 16:56:28 +00:00
|
|
|
}
|
|
|
|
}
|
2013-07-06 07:30:57 +00:00
|
|
|
|
|
|
|
//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))
|
|
|
|
// }
|
|
|
|
//}
|
2012-12-15 16:56:28 +00:00
|
|
|
}
|