forked from Fresheyeball/code-exercise
/
main.go
68 lines (56 loc) · 1.07 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package main
import (
"log"
fsnotify "github.com/go-fsnotify/fsnotify"
)
type stat struct {
doorCnt int
ImgCnt int
alarmCnt int
avgProcessingTime int
}
func fatality(err error) {
if err != nil {
log.Fatal(err)
}
}
func watchInput() *fsnotify.Watcher {
watcher, watcherr := fsnotify.NewWatcher()
fatality(watcherr)
adderr := watcher.Add("input/")
fatality(adderr)
return watcher
}
func collect(watcher *fsnotify.Watcher) <-chan fsnotify.Event {
x := make(chan fsnotify.Event)
go func() {
for {
select {
case event := <-watcher.Events:
x <- event
case err := <-watcher.Errors:
log.Println("error:", err)
}
}
}()
return x
}
func main() {
watcher := watchInput()
go func() {
for {
select {
case event := <-watcher.Events:
log.Println("event:", event)
if event.Op&fsnotify.Write == fsnotify.Write {
log.Println("modified file:", event.Name)
}
case err := <-watcher.Errors:
log.Println("error:", err)
}
}
}()
// cleanup before you die
defer watcher.Close()
<-make(chan bool)
}