From fe6d78f79bd485f50cdd415b9518b25157ffdcfc Mon Sep 17 00:00:00 2001
From: David Florness <david@florness.com>
Date: Sat, 6 Jun 2020 19:03:44 -0600
Subject: [PATCH] Close channels

---
 main.go  | 1 +
 voter.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/main.go b/main.go
index 0245d0e..8e1ffc7 100644
--- a/main.go
+++ b/main.go
@@ -106,6 +106,7 @@ func bootstrap() {
 			logger.Info("ENTER has been pressed; closing election")
 			n := len(me.otherVoters)
 			closeElection <- n
+			close(closeElection)
 			for _, voter := range me.otherVoters {
 				stream, err := me.h.NewStream(me.ctx, voter.ID, protocolID)
 				if err != nil {
diff --git a/voter.go b/voter.go
index 2363582..85d29d1 100644
--- a/voter.go
+++ b/voter.go
@@ -63,6 +63,7 @@ func handleCmd(cmd string, rw *bufio.ReadWriter, stream network.Stream) {
 			panic(err)
 		}
 		closeElection <- numPeers
+		close(closeElection)
 		electionClosed = true
 	case "shake":
 		p := stream.Conn().RemotePeer()
-- 
2.38.4