Added batching behavior for page building.

Quite often file watcher gets many changes and each change triggered a
build. One build per second should be sufficient. Also added tracking for
new folders.
This commit is contained in:
Egon Elbre
2013-12-15 17:19:22 +02:00
committed by spf13
parent 1979f7d9c7
commit 8d80f9b39e
2 changed files with 103 additions and 31 deletions

View File

@@ -15,11 +15,11 @@ package commands
import (
"fmt"
"github.com/howeyc/fsnotify"
"github.com/mostafah/fsync"
"github.com/spf13/cobra"
"github.com/spf13/hugo/hugolib"
"github.com/spf13/hugo/utils"
"github.com/spf13/hugo/watcher"
"github.com/spf13/nitro"
"os"
"path/filepath"
@@ -155,7 +155,7 @@ func buildSite(watching ...bool) (err error) {
}
func NewWatcher(port int) error {
watcher, err := fsnotify.NewWatcher()
watcher, err := watcher.New(1 * time.Second)
var wg sync.WaitGroup
if err != nil {
@@ -166,15 +166,56 @@ func NewWatcher(port int) error {
defer watcher.Close()
wg.Add(1)
for _, d := range getDirList() {
if d != "" {
_ = watcher.Watch(d)
}
}
go func() {
for {
select {
case ev := <-watcher.Event:
case evs := <-watcher.Event:
if Verbose {
fmt.Println(ev)
fmt.Println(evs)
}
static_changed := false
dynamic_changed := false
for _, ev := range evs {
ext := filepath.Ext(ev.Name)
istemp := strings.HasSuffix(ext, "~") || (ext == ".swp") || (ext == ".tmp")
if istemp {
continue
}
// renames are always followed with Create/Modify
if ev.IsRename() {
continue
}
isstatic := strings.HasPrefix(ev.Name, Config.GetAbsPath(Config.StaticDir))
static_changed = static_changed || isstatic
dynamic_changed = dynamic_changed || !isstatic
// add new directory to watch list
if s, err := os.Stat(ev.Name); err == nil && s.Mode().IsDir() {
if ev.IsCreate() {
watcher.Watch(ev.Name)
}
}
}
if static_changed {
fmt.Println("Static file changed, syncing\n")
utils.CheckErr(copyStatic(), fmt.Sprintf("Error copying static files to %s", Config.GetAbsPath(Config.PublishDir)))
}
if dynamic_changed {
fmt.Println("Change detected, rebuilding site\n")
utils.StopOnErr(buildSite(true))
}
watchChange(ev)
// TODO add newly created directories to the watch list
case err := <-watcher.Error:
if err != nil {
fmt.Println("error:", err)
@@ -183,12 +224,6 @@ func NewWatcher(port int) error {
}
}()
for _, d := range getDirList() {
if d != "" {
_ = watcher.Watch(d)
}
}
if port > 0 {
go serve(port)
}
@@ -196,22 +231,3 @@ func NewWatcher(port int) error {
wg.Wait()
return nil
}
func watchChange(ev *fsnotify.FileEvent) {
ext := filepath.Ext(ev.Name)
// ignore temp files
istemp := strings.HasSuffix(ext, "~") || (ext == ".swp") || (ext == ".tmp")
if istemp {
return
}
if strings.HasPrefix(ev.Name, Config.GetAbsPath(Config.StaticDir)) {
fmt.Println("Static file changed, syncing\n")
utils.CheckErr(copyStatic(), fmt.Sprintf("Error copying static files to %s", Config.GetAbsPath(Config.PublishDir)))
} else {
if !ev.IsRename() { // Rename is always accompanied by a create or modify
fmt.Println("Change detected, rebuilding site\n")
utils.StopOnErr(buildSite(true))
}
}
}