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