tube/app/watcher.go

70 lines
1.9 KiB
Go

package app
import (
"path/filepath"
"strings"
"time"
fs "github.com/fsnotify/fsnotify"
log "github.com/sirupsen/logrus"
)
// This is the amount of time to wait after changes before reacting to them.
// Debounce is done because moving files into the watched directories causes
// many rapid "Write" events to fire which would cause excessive Remove/Add
// method calls on the Library. To avoid this we accumulate the changes and
// only perform them once the events have stopped for this amount of time.
const debounceTimeout = time.Second * 5
// create, write, and chmod all require an add event
const addFlags = fs.Create | fs.Write | fs.Chmod
// remove, rename, write, and chmod all require a remove event
const removeFlags = fs.Remove | fs.Rename | fs.Write | fs.Chmod
// watch library paths and update Library with changes.
func startWatcher(a *App) {
timer := time.NewTimer(debounceTimeout)
addEvents := make(map[string]struct{})
removeEvents := make(map[string]struct{})
for {
select {
case e := <-a.Watcher.Events:
if strings.ContainsAny(e.Name, "#") || filepath.Ext(e.Name) != ".mp4" {
continue
}
log.Debugf("fsnotify event: %s", e)
if e.Op&removeFlags != 0 {
removeEvents[e.Name] = struct{}{}
}
if e.Op&addFlags != 0 {
addEvents[e.Name] = struct{}{}
}
// reset timer
timer.Reset(debounceTimeout)
case <-timer.C:
eventCount := len(removeEvents) + len(addEvents)
// handle remove events first
if len(removeEvents) > 0 {
for p := range removeEvents {
a.Library.Remove(p)
}
// clear map
removeEvents = make(map[string]struct{})
}
// then handle add events
if len(addEvents) > 0 {
for p := range addEvents {
a.Library.Add(p)
}
// clear map
addEvents = make(map[string]struct{})
}
if eventCount > 0 {
buildFeed(a)
}
// reset timer
timer.Reset(debounceTimeout)
}
}
}