optional print stats (#30)

* optional print stats

Author: Gsantomaggio

README.md

@@ -8,7 +8,7 @@ A POC client for [RabbitMQ Stream Queues](https://github.com/rabbitmq/rabbitmq-s
 ### Download
 ---
 ```
-go get -u github.com/gsantomaggio/go-stream-client@v0.1-alpha
+go get -u github.com/gsantomaggio/go-stream-client@v0.2-alpha
 ```
 
 ### How to test

examples/getting_started.go

@@ -31,8 +31,14 @@ func main() {
 	streamName := uuid.New().String()
 	err = client.StreamCreator().Stream(streamName).
 		Create() // Create the streaming queue
+	CheckErr(err)
 
+	err = client.StreamCreator().Stream(streamName).
+		MaxLengthBytes(streaming.ByteCapacity{}.MB(5)).
+		Create() // Create the streaming queue
 	CheckErr(err)
+
+
 	var count int32
 	consumer, err := client.ConsumerCreator().
 		Stream(streamName).

perfTest/cmd/commands.go

@@ -22,6 +22,7 @@ var (
 	preDeclared       bool
 	streams           []string
 	maxLengthBytes    string
+	printStatsV       bool
 )
 
 func init() {
@@ -33,6 +34,7 @@ func setupCli(baseCmd *cobra.Command) {
 	baseCmd.PersistentFlags().IntVarP(&producers, "producers", "p", 1, "Number of Producers")
 	baseCmd.PersistentFlags().IntVarP(&consumers, "consumers", "c", 1, "Number of Consumers")
 	baseCmd.PersistentFlags().BoolVarP(&preDeclared, "pre-declared", "d", false, "Pre created stream")
+	baseCmd.PersistentFlags().BoolVarP(&printStatsV, "print-stats", "n", true, "Print stats")
 	baseCmd.PersistentFlags().StringSliceVarP(&streams, "streams", "s", []string{uuid.New().String()}, "Stream names, create an UUID if not specified")
 	baseCmd.PersistentFlags().StringVarP(&maxLengthBytes, "max_length_bytes", "m", "", "Stream max length bytes, default is unlimited, ex: 10MB,50GB, etc..")
 	baseCmd.AddCommand(versionCmd)

perfTest/cmd/silent.go

@@ -29,23 +29,25 @@ var (
 )
 
 func printStats() {
-	start := time.Now()
-	ticker := time.NewTicker(2 * time.Second)
-	go func() {
-		for {
-			select {
-			case _ = <-ticker.C:
-				v := time.Now().Sub(start).Seconds()
-				PMessagesPerSecond := float64(atomic.LoadInt32(&producerMessageCount)) / v
-				CMessagesPerSecond := float64(atomic.LoadInt32(&consumerMessageCount)) / v
-				streaming.INFO("Published %.2f msg/s, Consumed %.2f msg/s, Active Connections: %d", PMessagesPerSecond, CMessagesPerSecond, len(connections))
-				atomic.SwapInt32(&producerMessageCount, 0)
-				atomic.SwapInt32(&consumerMessageCount, 0)
-				start = time.Now()
+	if printStatsV {
+		start := time.Now()
+		ticker := time.NewTicker(2 * time.Second)
+		go func() {
+			for {
+				select {
+				case _ = <-ticker.C:
+					v := time.Now().Sub(start).Seconds()
+					PMessagesPerSecond := float64(atomic.LoadInt32(&producerMessageCount)) / v
+					CMessagesPerSecond := float64(atomic.LoadInt32(&consumerMessageCount)) / v
+					streaming.INFO("Published %.2f msg/s, Consumed %.2f msg/s, Active Connections: %d", PMessagesPerSecond, CMessagesPerSecond, len(connections))
+					atomic.SwapInt32(&producerMessageCount, 0)
+					atomic.SwapInt32(&consumerMessageCount, 0)
+					start = time.Now()
+				}
 			}
-		}
 
-	}()
+		}()
+	}
 }
 
 func startSimulation() error {

pkg/streaming/constants.go

@@ -64,7 +64,7 @@ const (
 	DefaultReadSocketBuffer = 4096 * 2
 
 //
-	Version = "0.2-alpha-dev"
+	Version = "0.2-alpha"
 )
 
 func LookErrorCode(errorCode uint16) string {

pkg/streaming/response.go

@@ -77,7 +77,7 @@ func (c *Client) handleResponse() {
 			}
 		case CommandHeartbeat:
 			{
-				DEBUG("RECEIVED Heartbeat %d buff:%d \n", readerProtocol.CommandID, buffer.Buffered())
+				//DEBUG("RECEIVED Heartbeat %d buff:%d \n", readerProtocol.CommandID, buffer.Buffered())
 
 			}
 		case CommandQueryOffset: