Function main

_examples/bulk/benchmarks/benchmarks.go:96–172

()

}

func main() {
	log.SetFlags(0)

	indexName = indexName + "-" + datasetName
	if flushBytes < 1 {
		flushBytes = humanBytes(3e+6)
	}

	var opts []elasticsearch.Option

	if useFasthttp {
		opts = append(opts, elasticsearch.WithTransportOptions(elastictransport.WithTransport(&fasthttpTransport{})))
	}

	if debug {
		opts = append(opts, elasticsearch.WithLogger(&elastictransport.ColorLogger{Output: os.Stdout, EnableRequestBody: true, EnableResponseBody: true}))
	}

	es, _ := elasticsearch.New(opts...)

	runnerCfg := runner.Config{
		Client: es,

		IndexName:     indexName,
		DatasetName:   datasetName,
		NumShards:     numShards,
		NumReplicas:   numReplicas,
		NumItems:      numItems,
		NumRuns:       numRuns,
		NumWarmupRuns: numWarmupRuns,
		NumWorkers:    numWorkers,
		FlushBytes:    int(flushBytes),
		Wait:          wait,
		Mockserver:    mockserver,
	}

	if useEasyjson {
		runnerCfg.Decoder = easyjsonDecoder{}
	}

	runner, err := runner.NewRunner(runnerCfg)
	if err != nil {
		log.Fatalf("Error creating runner: %s", err)
	}

	done := make(chan os.Signal, 1)
	signal.Notify(done, os.Interrupt)

	go func() { <-done; log.Println("\r" + strings.Repeat("▁", 110)); runner.Report(); os.Exit(0) }()
	defer func() { log.Println(strings.Repeat("▁", 110)); runner.Report() }()

	log.Printf(
		"%s: run [%sx] warmup [%dx] shards [%d] replicas [%d] workers [%d] flush [%s] wait [%s]%s%s",
		datasetName,
		humanize.Comma(int64(numRuns)),
		numWarmupRuns,
		numShards,
		numReplicas,