Exit goroutine when client disconnects

main.go

@@ -57,11 +57,22 @@ func main() {
 		if err != nil {
 			log.Println(err)
 		}
-		for n := range nudges {
-			log.Println(n)
-			err := con.WriteJSON(n)
-			if err != nil {
-				log.Println(err)
+		var quit = make(chan bool)
+		con.SetCloseHandler(func(code int, text string) error {
+			quit <- true
+			return nil
+		})
+		for {
+			select {
+			case <-quit:
+				return
+			case n := <-nudges:
+				log.Println(n)
+				err := con.WriteJSON(n)
+				if err != nil {
+					log.Println(err)
+					return
+				}
 			}
 		}
 	})