From d53720991ad0fe10c0ba6de29bd9c6ae6c4eebc5 Mon Sep 17 00:00:00 2001
From: Benjamin Wenger <ben@elixxir.ioo>
Date: Fri, 15 Oct 2021 14:19:14 -0700
Subject: [PATCH] fixes to CLI

---
 cmd/root.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/cmd/root.go b/cmd/root.go
index 2d39d8174..46c067e31 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -254,10 +254,27 @@ var rootCmd = &cobra.Command{
 				receiveCnt++
 				if receiveCnt == expectedCnt {
 					done = true
+					break
 				}
+			}
+		}
+
+		//wait an extra 5 seconds to make sure no messages were missed
+		done = false
+		timer := time.NewTimer(5*time.Second)
+		for !done {
+			select {
+			case <-timer.C:
+				done = true
 				break
+			case m := <-recvCh:
+				fmt.Printf("Message received: %s\n", string(
+					m.Payload))
+				//fmt.Printf("%s", m.Timestamp)
+				receiveCnt++
 			}
 		}
+
 		jww.INFO.Printf("Received %d/%d Messages!", receiveCnt, expectedCnt)
 		fmt.Printf("Received %d\n", receiveCnt)
 		if roundsNotepad != nil {
-- 
GitLab