Remove ALLOWED_USERS.txt watcher
This commit is contained in:
48
lts.go
48
lts.go
@ -18,8 +18,6 @@ import (
|
||||
"sync"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/fsnotify/fsnotify"
|
||||
)
|
||||
|
||||
var LISTEN_ADDR string
|
||||
@ -258,7 +256,7 @@ func handleBrowse(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
}
|
||||
|
||||
func reloadAllowedUsers() {
|
||||
func loadAllowedUsers() {
|
||||
log.Println("loading ALLOWED_USERS.txt")
|
||||
allowedUsersTxt, err := ioutil.ReadFile("./ALLOWED_USERS.txt")
|
||||
if err != nil {
|
||||
@ -275,48 +273,8 @@ func reloadAllowedUsers() {
|
||||
allowedUsersMutex.Unlock()
|
||||
}
|
||||
|
||||
func watchAllowedUsers() chan bool {
|
||||
watcher, err := fsnotify.NewWatcher()
|
||||
if err != nil {
|
||||
log.Fatal("Error creating new watcher: ", err)
|
||||
}
|
||||
defer watcher.Close()
|
||||
|
||||
done := make(chan bool)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case event, ok := <-watcher.Events:
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
fmt.Println(event)
|
||||
if event.Has(fsnotify.Chmod | fsnotify.Rename) {
|
||||
reloadAllowedUsers()
|
||||
}
|
||||
case err, ok := <-watcher.Errors:
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
log.Println("Watcher error: ", err)
|
||||
case <-done:
|
||||
break
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
err = watcher.Add("./ALLOWED_USERS.txt")
|
||||
if err != nil {
|
||||
log.Fatal("Error adding ALLOWED_USERS.txt to watcher: ", err)
|
||||
}
|
||||
|
||||
return done
|
||||
}
|
||||
|
||||
func main() {
|
||||
reloadAllowedUsers()
|
||||
doneWatching := watchAllowedUsers()
|
||||
loadAllowedUsers()
|
||||
|
||||
http.HandleFunc("/", handleHome)
|
||||
http.HandleFunc("/browse", handleBrowse)
|
||||
@ -336,6 +294,4 @@ func main() {
|
||||
if err != nil {
|
||||
log.Fatal("Error in serving: ", err)
|
||||
}
|
||||
|
||||
doneWatching<-true
|
||||
}
|
||||
|
Reference in New Issue
Block a user