From 370ef66b5059a6943e2419e7ba6ce25636db632c Mon Sep 17 00:00:00 2001
From: Ted Unangst <tedu@tedunangst.com>
Date: Tue, 18 Feb 2020 12:34:23 -0500
Subject: [PATCH] more robust end of the world

---
 web.go | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/web.go b/web.go
index 130a8a5..9f06501 100644
--- a/web.go
+++ b/web.go
@@ -983,6 +983,7 @@ func tracker() {
 	timeout := 4 * time.Minute
 	sleeper := time.NewTimer(timeout)
 	tracks := make(map[string][]string)
+	workinprogress++
 	for {
 		select {
 		case track := <-trackchan:
@@ -2245,25 +2246,19 @@ func apihandler(w http.ResponseWriter, r *http.Request) {
 
 var endoftheworld = make(chan bool)
 var readyalready = make(chan bool)
+var workinprogress = 0
 
 func enditall() {
 	sig := make(chan os.Signal)
 	signal.Notify(sig, os.Interrupt)
 	signal.Notify(sig, syscall.SIGTERM)
 	<-sig
-	count := 0
 	log.Printf("stopping...")
-sendloop:
-	for {
-		select {
-		case endoftheworld <- true:
-			count++
-		default:
-			break sendloop
-		}
+	for i := 0; i < workinprogress; i++ {
+		endoftheworld <- true
 	}
 	log.Printf("waiting...")
-	for i := 0; i < count; i++ {
+	for i := 0; i < workinprogress; i++ {
 		<-readyalready
 	}
 	log.Printf("apocalypse")