47 lines
1.0 KiB
Go
47 lines
1.0 KiB
Go
package util
|
|
|
|
import (
|
|
"github.com/fsnotify/fsnotify"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func DirWatch(monitorDir string, addChan chan string) {
|
|
logger := zap.L()
|
|
|
|
watcher, err := fsnotify.NewWatcher()
|
|
if err != nil {
|
|
logger.Error("create dir watcher failed:", zap.Error(err))
|
|
}
|
|
defer watcher.Close()
|
|
|
|
err = watcher.Add(monitorDir)
|
|
if err != nil {
|
|
logger.Error("add monitor dir failed:", zap.Error(err))
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case event, ok := <-watcher.Events:
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
if event.Op&fsnotify.Create == fsnotify.Create {
|
|
logger.Info("file add in watcher dir", zap.String("file_name", event.Name))
|
|
addChan <- event.Name
|
|
}
|
|
|
|
if event.Op&fsnotify.Remove == fsnotify.Remove {
|
|
logger.Info("file remove in watcher dir", zap.String("file_name", event.Name))
|
|
}
|
|
case err, ok := <-watcher.Errors:
|
|
if !ok {
|
|
logger.Error("monitor watcher error channel closed", zap.Bool("channel_status", ok))
|
|
return
|
|
}
|
|
|
|
logger.Error("monitor watcher error occurred", zap.Error(err))
|
|
}
|
|
}
|
|
}
|