diff --git a/cmd/root.go b/cmd/root.go index 2d39d81740cfe956ad529aec6196a7594fe8432b..46c067e319c753eac0fc95fe5b6a634231b43f95 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 {