1
0
Fork 0

Reorginised the main()'s code a little.

This commit is contained in:
Alexander Andreev 2023-05-27 19:51:29 +04:00
parent 949c2e68c8
commit 2d4169245c
Signed by: Arav
GPG Key ID: D22A817D95815393
1 changed files with 18 additions and 17 deletions

View File

@ -57,6 +57,19 @@ func main() {
} }
} }
watcha, err := watcher.NewInotifyWatcher()
if err != nil {
log.Fatalln(err)
}
defer watcha.Close()
if err := watcha.AddWatch(*uploadDir, watcher.CrDelMask); err != nil {
log.Fatalln(err)
}
uploadDirNotify := make(chan uint32)
go watcha.WatchForMask(uploadDirNotify, watcher.CrDelMask)
hashSalt, err := os.ReadFile(path.Join(os.Getenv("CREDENTIALS_DIRECTORY"), "salt")) hashSalt, err := os.ReadFile(path.Join(os.Getenv("CREDENTIALS_DIRECTORY"), "salt"))
if err != nil { if err != nil {
log.Fatalln("failed to read hash salt file:", err) log.Fatalln("failed to read hash salt file:", err)
@ -72,24 +85,11 @@ func main() {
logFile := log.New(logFileFd, "", log.LstdFlags) logFile := log.New(logFileFd, "", log.LstdFlags)
watcha, err := watcher.NewInotifyWatcher()
if err != nil {
log.Fatalln(err)
}
defer watcha.Close()
if err := watcha.AddWatch(*uploadDir, watcher.CrDelMask); err != nil {
log.Fatalln(err)
}
uploadDirNotify := make(chan uint32)
uploadDirSize, err := utils.DirectorySize(*uploadDir) uploadDirSize, err := utils.DirectorySize(*uploadDir)
if err != nil { if err != nil {
log.Fatalf("failed to get initial size of %s: %s", *uploadDir, err) log.Fatalf("failed to get initial size of %s: %s", *uploadDir, err)
} }
go watcha.WatchForMask(uploadDirNotify, watcher.CrDelMask)
hand := http.NewUploadHandlers(logFile, *uploadDir, &uploadDirSize, string(hashSalt), hand := http.NewUploadHandlers(logFile, *uploadDir, &uploadDirSize, string(hashSalt),
*keepForHours, *storageSize, *fileSizeLimit) *keepForHours, *storageSize, *fileSizeLimit)
@ -110,6 +110,11 @@ func main() {
r.Handler(nethttp.MethodGet, "/favicon.svg", http.Favicon) r.Handler(nethttp.MethodGet, "/favicon.svg", http.Favicon)
srv := http.NewHttpServer(r) srv := http.NewHttpServer(r)
defer func() {
if err := srv.Stop(); err != nil {
log.Fatalln("failed to properly shutdown a server:", err)
}
}()
if err := srv.Start(network, *listenAddress); err != nil { if err := srv.Start(network, *listenAddress); err != nil {
log.Fatalln("failed to start a server:", err) log.Fatalln("failed to start a server:", err)
@ -131,8 +136,4 @@ func main() {
}() }()
<-doneSignal <-doneSignal
if err := srv.Stop(); err != nil {
log.Fatalln("failed to properly shutdown a server:", err)
}
} }