🐛 Fix example

examples/http_send.go

@@ -1,38 +1,42 @@
+// +build examples
+
 package main
 
 import (
 	"fmt"
+	"sync"
 
 	"github.com/redBorder/rbforwarder"
 	"github.com/redBorder/rbforwarder/components/batch"
 	"github.com/redBorder/rbforwarder/components/httpsender"
 )
 
 func main() {
+	var wg sync.WaitGroup
 	var components []interface{}
 	var workers []int
 	const numMessages = 100000
 
 	f := rbforwarder.NewRBForwarder(rbforwarder.Config{
-		Retries:   3,
-		Backoff:   5,
-		QueueSize: 10000,
+		Retries:   1,
+		Backoff:   1,
+		QueueSize: 1000,
 	})
 
 	batch := &batcher.Batcher{
 		Config: batcher.Config{
-			TimeoutMillis: 1000,
-			Limit:         1000,
+			TimeoutMillis: 100,
+			Limit:         10000,
 		},
 	}
 	components = append(components, batch)
-	workers = append(workers, 1)
+	workers = append(workers, 5)
 
 	sender := &httpsender.HTTPSender{
 		URL: "http://localhost:8888",
 	}
 	components = append(components, sender)
-	workers = append(workers, 1)
+	workers = append(workers, 10)
 
 	f.PushComponents(components, workers)
 
@@ -41,21 +45,37 @@ func main() {
 		"batch_group":   "librb-http",
 	}
 
+	f.Run()
+
+	wg.Add(1)
+	go func() {
+		var errors int
+		var messages int
+
+		fmt.Print("[")
+		for report := range f.GetReports() {
+			r := report.(rbforwarder.Report)
+			// fmt.Printf("[MESSAGE %d] %s\n", r.Opaque.(int), r.Status)
+			messages++
+			if messages%(numMessages/20) == 0 {
+				fmt.Printf("=")
+			}
+			if r.Code > 0 {
+				errors += r.Code
+			}
+			if messages >= numMessages {
+				break
+			}
+		}
+		fmt.Print("] ")
+		fmt.Printf("Sent %d messages with %d errors\n", messages, errors)
+		wg.Done()
+	}()
+
 	for i := 0; i < numMessages; i++ {
 		data := fmt.Sprintf("{\"message\": %d}", i)
 		f.Produce([]byte(data), opts, i)
 	}
 
-	var errors int
-	for report := range f.GetReports() {
-		r := report.(rbforwarder.Report)
-		if r.Code > 0 {
-			errors += r.Code
-		}
-		if r.Opaque.(int) == numMessages-1 {
-			break
-		}
-	}
-
-	fmt.Printf("Sent %d messages with %d errors\n", numMessages, errors)
+	wg.Wait()
 }