From e02bb3756553c14e5a437efdf26df8fdc5b736db Mon Sep 17 00:00:00 2001
From: Philip O'Toole <philip@influxdb.com>
Date: Fri, 12 Jun 2015 15:18:48 -0700
Subject: [PATCH] collectd test client now configurable
---
services/collectd/test_client/client.go | 62 +++++++++++++++++++------
1 file changed, 47 insertions(+), 15 deletions(-)
diff --git a/services/collectd/test_client/client.go b/services/collectd/test_client/client.go
index 3eda4c641d..852e42b8ae 100644
--- a/services/collectd/test_client/client.go
+++ b/services/collectd/test_client/client.go
@@ -4,31 +4,63 @@ import (
"collectd.org/api"
"collectd.org/network"
+ "flag"
"fmt"
- "net"
+ "math/rand"
"os"
+ "strconv"
"time"
)
+var nMeasurments = flag.Int("m", 1, "Number of measurements")
+var tagVariance = flag.Int("v", 1, "Number of values per tag. Client is fixed at one tag")
+var rate = flag.Int("r", 1, "Number of points per second")
+var total = flag.Int("t", 1, "Total number of points to send")
+var host = flag.String("u", "127.0.0.1:25826", "Destination host in the form host:port")
+
func main() {
- conn, err := network.Dial(net.JoinHostPort("localhost", "25826"), network.ClientOptions{})
+ flag.Parse()
+
+ conn, err := network.Dial(*host, network.ClientOptions{})
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer conn.Close()
- vl := api.ValueList{
- Identifier: api.Identifier{
- Host: "example.com",
- Plugin: "golang",
- Type: "gauge",
- },
- Time: time.Now(),
- Interval: 10 * time.Second,
- Values: []api.Value{api.Gauge(42.0)},
- }
- if err := conn.Write(vl); err != nil {
- fmt.Println(err)
- os.Exit(1)
+
+ rateLimiter := make(chan int, *rate)
+
+ go func() {
+ ticker := time.NewTicker(time.Second)
+ select {
+ case <-ticker.C:
+ for i := 0; i < *rate; i++ {
+ rateLimiter <- i
+ }
+ }
+ }()
+
+ nSent := 0
+ for nSent < *total {
+ <-rateLimiter
+
+ vl := api.ValueList{
+ Identifier: api.Identifier{
+ Host: "tagvalue" + strconv.Itoa(int(rand.Int31n(int32(*tagVariance)))),
+ Plugin: "golang" + strconv.Itoa(int(rand.Int31n(int32(*nMeasurments)))),
+ Type: "gauge",
+ },
+ Time: time.Now(),
+ Interval: 10 * time.Second,
+ Values: []api.Value{api.Gauge(42.0)},
+ }
+ if err := conn.Write(vl); err != nil {
+ fmt.Println(err)
+ os.Exit(1)
+ }
+ conn.Flush()
+ nSent = nSent + 1
}
+
+ fmt.Println("Number of points sent:", nSent)
}