Fixed sharded pubsub subscriber check (#24)

* Added the option to measure rtt

* fixed ssubscriber check

---------

Co-authored-by: fcostaoliveira <[email protected]>

Author: filipecosta90

subscriber.go

@@ -615,15 +615,16 @@ func updateCLI(
 	// Header
 	if measureRTT {
 		fmt.Fprint(w, "Test Time\tTotal Messages\t Message Rate \tConnect Rate \t")
-		if strings.HasPrefix(mode, "subscribe") {
+
+		if strings.Contains(mode, "subscribe") {
 			fmt.Fprint(w, "Active subscriptions\t")
 		} else {
 			fmt.Fprint(w, "Active publishers\t")
 		}
 		fmt.Fprint(w, "Avg RTT (ms)\t\n")
 	} else {
 		fmt.Fprint(w, "Test Time\tTotal Messages\t Message Rate \tConnect Rate \t")
-		if strings.HasPrefix(mode, "subscribe") {
+		if strings.Contains(mode, "subscribe") {
 			fmt.Fprint(w, "Active subscriptions\t\n")
 		} else {
 			fmt.Fprint(w, "Active publishers\t\n")
@@ -657,7 +658,7 @@ func updateCLI(
 			// Metrics line
 			fmt.Fprintf(w, "%.0f\t%d\t%.2f\t%.2f\t", time.Since(start).Seconds(), totalMessages, messageRate, connectRate)
 
-			if strings.HasPrefix(mode, "subscribe") {
+			if strings.Contains(mode, "subscribe") {
 				fmt.Fprintf(w, "%d\t", totalSubscribedChannels)
 			} else {
 				fmt.Fprintf(w, "%d\t", atomic.LoadInt64(&totalPublishers))