Add addtional closeSignal chan for program to wait for.

cmd/dwelling-upload/main.go

@@ -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
 
 }