Arav/dwelling-upload
1
0
Fork 0
Code Releases 12 Activity

Add addtional closeSignal chan for program to wait for.

Browse Source
This commit is contained in:
Alexander Andreev 2022-02-08 02:05:38 +04:00
parent 0c05ffeef6
commit ddaf2bb991
Signed by: Arav
GPG Key ID: 1327FE8A374CC86F
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cmd/dwelling-upload/main.go
View File

@ -107,13 +107,16 @@ func main() {
doneSignal := make(chan os.Signal, 1) doneSignal := make(chan os.Signal, 1)
signal.Notify(doneSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM) signal.Notify(doneSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
closeSignal := make(chan bool, 1)
go func() {
for { for {
select { select {
case <-doneSignal: case <-doneSignal:
if err := srv.Stop(); err != nil { if err := srv.Stop(); err != nil {
logErr.Fatalln(err) logErr.Fatalln(err)
} }
closeSignal <- true
case <-uploadDirNotify: case <-uploadDirNotify:
sz, err := utils.DirectorySize(config.Uploads.Directory) sz, err := utils.DirectorySize(config.Uploads.Directory)
if err != nil { if err != nil {
@ -125,5 +128,8 @@ func main() {
} }
} }
} }
}()
<-closeSignal
} }