76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
|
/*
|
||
|
This is a development testing cli tool meant to stress the zipkin telegraf plugin.
|
||
|
It writes a specified number of zipkin spans to the plugin endpoint, with other
|
||
|
parameters which dictate batch size and flush timeout.
|
||
|
|
||
|
Usage as follows:
|
||
|
|
||
|
`./stress_test_write -batch_size=<batch_size> -max_backlog=<max_span_buffer_backlog> -batch_interval=<batch_interval_in_seconds> -span_count<number_of_spans_to_write> -zipkin_host=<zipkin_service_hostname>`
|
||
|
|
||
|
Or with a timer:
|
||
|
|
||
|
`time ./stress_test_write -batch_size=<batch_size> -max_backlog=<max_span_buffer_backlog> -batch_interval=<batch_interval_in_seconds> -span_count<number_of_spans_to_write> -zipkin_host=<zipkin_service_hostname>`
|
||
|
|
||
|
However, the flag defaults work just fine for a good write stress test (and are what
|
||
|
this tool has mainly been tested with), so there shouldn't be much need to
|
||
|
manually tweak the parameters.
|
||
|
*/
|
||
|
|
||
|
package main
|
||
|
|
||
|
import (
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"log"
|
||
|
"time"
|
||
|
|
||
|
zipkin "github.com/openzipkin/zipkin-go-opentracing"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
BatchSize int
|
||
|
MaxBackLog int
|
||
|
BatchTimeInterval int
|
||
|
SpanCount int
|
||
|
ZipkinServerHost string
|
||
|
)
|
||
|
|
||
|
const usage = `./stress_test_write -batch_size=<batch_size> -max_backlog=<max_span_buffer_backlog> -batch_interval=<batch_interval_in_seconds> -span_count<number_of_spans_to_write> -zipkin_host=<zipkin_service_hostname>`
|
||
|
|
||
|
func init() {
|
||
|
flag.IntVar(&BatchSize, "batch_size", 10000, usage)
|
||
|
flag.IntVar(&MaxBackLog, "max_backlog", 100000, usage)
|
||
|
flag.IntVar(&BatchTimeInterval, "batch_interval", 1, usage)
|
||
|
flag.IntVar(&SpanCount, "span_count", 100000, usage)
|
||
|
flag.StringVar(&ZipkinServerHost, "zipkin_host", "localhost", usage)
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
flag.Parse()
|
||
|
var hostname = fmt.Sprintf("http://%s:9411/api/v1/spans", ZipkinServerHost)
|
||
|
collector, err := zipkin.NewHTTPCollector(
|
||
|
hostname,
|
||
|
zipkin.HTTPBatchSize(BatchSize),
|
||
|
zipkin.HTTPMaxBacklog(MaxBackLog),
|
||
|
zipkin.HTTPBatchInterval(time.Duration(BatchTimeInterval)*time.Second))
|
||
|
defer collector.Close()
|
||
|
if err != nil {
|
||
|
log.Fatalf("Error intializing zipkin http collector: %v\n", err)
|
||
|
}
|
||
|
|
||
|
tracer, err := zipkin.NewTracer(
|
||
|
zipkin.NewRecorder(collector, false, "127.0.0.1:0", "trivial"))
|
||
|
|
||
|
if err != nil {
|
||
|
log.Fatalf("Error: %v\n", err)
|
||
|
}
|
||
|
|
||
|
log.Printf("Writing %d spans to zipkin server at %s\n", SpanCount, hostname)
|
||
|
for i := 0; i < SpanCount; i++ {
|
||
|
parent := tracer.StartSpan("Parent")
|
||
|
parent.LogEvent(fmt.Sprintf("Trace%d", i))
|
||
|
parent.Finish()
|
||
|
}
|
||
|
log.Println("Done. Flushing remaining spans...")
|
||
|
}
|