// 根据recursive值确定是否递归查找paths每个目录下的子目录。
func recursivePath(recursive bool, paths []string) []string {
if !recursive {
return paths
}
ret := []string{}
walk := func(path string, fi os.FileInfo, err error) error {
if err != nil {
utils.ColorLog("[ERRO] 遍历监视目录错误: [ %s ] \n", err)
}
//(BUG):不能监视隐藏目录下的文件
if fi.IsDir() && strings.Index(path, "/.") < 0 {
ret = append(ret, path)
}
return nil
}
for _, path := range paths {
if err := filepath.Walk(path, walk); err != nil {
utils.ColorLog("[ERRO] 遍历监视目录错误: [ %s ] \n", err)
}
}
return ret
}
// 开始编译代码
func (w *watch) build() {
utils.ColorLog("[INFO] 编译代码... \n")
goCmd := exec.Command("go", w.goCmdArgs...)
goCmd.Stderr = os.Stderr
goCmd.Stdout = os.Stdout
if err := goCmd.Run(); err != nil {
utils.ColorLog("[ERRO] 编译失败: [ %s ] \n", err)
return
}
utils.ColorLog("[SUCC] 编译成功... \n")
w.restart()
}
func (w *watch) restart() {
defer func() {
if err := recover(); err != nil {
fmt.Println("Kill.recover -> ", err)
}
}()
if w.appCmd != nil && w.appCmd.Process != nil {
utils.ColorLog("[INFO] 终止旧进程... \n")
if err := w.appCmd.Process.Kill(); err != nil {
utils.ColorLog("[ERROR] 终止进程失败 [ %s ] ...\n", err)
}
utils.ColorLog("[SUCC] 旧进程被终止! \n")
}
utils.ColorLog("[INFO] 重启 < %s > \n", w.appName)
if strings.Index(w.appName, "./") == -1 {
w.appName = "./" + w.appName
}
utils.ColorLog("[INFO] 启动新进程... \n")
w.appCmd = exec.Command(w.appName)
w.appCmd.Stderr = os.Stderr
w.appCmd.Stdout = os.Stdout
if err := w.appCmd.Start(); err != nil {
utils.ColorLog("[ERRO] 启动进程时出错: [ %s ] \n", err)
}
utils.ColorLog("[SUCC] 新进程已经启动...\n")
}
func (w *watch) getFileModTime(path string) int64 {
path = strings.Replace(path, "\\", "/", -1)
f, err := os.Open(path)
if err != nil {
utils.ColorLog("[ERRO] 文件打开失败 [ %s ]\n", err)
return time.Now().Unix()
}
defer f.Close()
fi, err := f.Stat()
if err != nil {
utils.ColorLog("[ERRO] 获取不到文件信息 [ %s ]\n", err)
return time.Now().Unix()
}
return fi.ModTime().Unix()
}
func Run() {
// 初始化flag
var showHelp, recursive bool
var outputName string
flag.BoolVar(&showHelp, "h", false, "显示帮助信息")
flag.BoolVar(&recursive, "r", true, "是否查找子目录")
flag.StringVar(&outputName, "o", "", "指定输出名称")
flag.Usage = func() {
fmt.Println(usage)
}
flag.Parse()
if showHelp {
flag.Usage()
return
}
wd, err := os.Getwd()
if err != nil {
utils.ColorLog("[ERRO] 获取当前工作目录时，发生错误: [ %s ] \n", err)
return
}
// 初始化goCmd的参数
args := []string{"build", "-o", outputName}
w := &watch{
appName: getAppName(outputName, wd),
goCmdArgs: args,
}
w.watcher(recursivePath(recursive, append(flag.Args(), wd)))
go w.build()
done := make(chan bool)
<-done
}
func (w *watch) watcher(paths []string) {
utils.ColorLog("[TRAC] 初始化文件监视器... \n")
//初始化监听器
watcher, err := fsnotify.NewWatcher()
if err != nil {
utils.ColorLog("[ERRO] 初始化监视器失败: [ %s ] \n", err)
os.Exit(2)
}
go func() {
for {
select {
case event := <-watcher.Events:
build := true
if !w.checkIfWatchExt(event.Name) {
continue
}
if event.Op&fsnotify.Chmod == fsnotify.Chmod {
utils.ColorLog("[SKIP] [ %s ] \n", event)
continue
}
mt := w.getFileModTime(event.Name)
if t := eventTime[event.Name]; mt == t {
utils.ColorLog("[SKIP] [ %s ] \n", event.String())
build = false
}
eventTime[event.Name] = mt
/*if(strings.HasSuffix(event.Name, ".go")){
build = true
}*/
if build {
go func() {
scheduleTime = time.Now().Add(1 * time.Second)
for {
time.Sleep(scheduleTime.Sub(time.Now()))
if time.Now().After(scheduleTime) {
break
}
return
}
utils.ColorLog("[TRAC] 触发编译事件: < %s > \n", event)
w.build()
}()
}
case err := <-watcher.Errors:
utils.ColorLog("[ERRO] 监控失败 [ %s ] \n", err)
}
}
}()
for _, path := range paths {
utils.ColorLog("[TRAC] 监视文件夹: ( %s ) \n", path)
err = watcher.Add(path)
if err != nil {
utils.ColorLog("[ERRO] 监视文件夹失败: [ %s ] \n", err)
os.Exit(2)
}
}
}