1
0

Add addtional closeSignal chan for program to wait for.

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

View File

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