65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
|
// +build go1.3
|
||
|
|
// +build !plan9,!solaris
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"gopkg.in/fsnotify.v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func WaitForReplacement(event fsnotify.Event, watcher *fsnotify.Watcher) {
|
||
|
|
const sleep_interval = 50 * time.Millisecond
|
||
|
|
|
||
|
|
// Avoid a race when fsnofity.Remove is preceded by fsnotify.Chmod.
|
||
|
|
if event.Op&fsnotify.Chmod != 0 {
|
||
|
|
time.Sleep(sleep_interval)
|
||
|
|
}
|
||
|
|
for {
|
||
|
|
if _, err := os.Stat(event.Name); err == nil {
|
||
|
|
if err := watcher.Add(event.Name); err == nil {
|
||
|
|
log.Printf("watching resumed for %s", event.Name)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
time.Sleep(sleep_interval)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WatchForUpdates(filename string, action func()) bool {
|
||
|
|
filename = filepath.Clean(filename)
|
||
|
|
watcher, err := fsnotify.NewWatcher()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal("failed to create watcher for ", filename, ": ", err)
|
||
|
|
}
|
||
|
|
go func() {
|
||
|
|
defer watcher.Close()
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case event := <-watcher.Events:
|
||
|
|
// On Arch Linux, it appears Chmod events precede Remove events,
|
||
|
|
// which causes a race between action() and the coming Remove event.
|
||
|
|
// If the Remove wins, the action() (which calls
|
||
|
|
// UserMap.LoadAuthenticatedEmailsFile()) crashes when the file
|
||
|
|
// can't be opened.
|
||
|
|
if event.Op&(fsnotify.Remove|fsnotify.Rename|fsnotify.Chmod) != 0 {
|
||
|
|
log.Printf("watching interrupted on event: %s", event)
|
||
|
|
WaitForReplacement(event, watcher)
|
||
|
|
}
|
||
|
|
log.Printf("reloading after event: %s", event)
|
||
|
|
action()
|
||
|
|
case err := <-watcher.Errors:
|
||
|
|
log.Printf("error watching %s: %s", filename, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
if err = watcher.Add(filename); err != nil {
|
||
|
|
log.Fatal("failed to add ", filename, " to watcher: ", err)
|
||
|
|
}
|
||
|
|
return true
|
||
|
|
}
|