Method setup

_examples/bulk/benchmarks/runner/runner.go:136–183

setup re-creates the index for a benchmark run.

()

// setup re-creates the index for a benchmark run.
func (r *Runner) setup() error {
	fm, err := os.Open(filepath.Join("data", r.config.DatasetName, "mapping.json"))
	if err != nil {
		return fmt.Errorf("setup: reading mapping: %s", err)
	}

	var mappingEnvelope map[string]interface{}
	json.NewDecoder(fm).Decode(&mappingEnvelope)
	mapping, err := json.Marshal(mappingEnvelope["mappings"])
	if err != nil {
		return fmt.Errorf("setup: encoding mapping: %s", err)
	}

	r.indexSettings.WriteString(`{ "settings": `)
	fmt.Fprintf(&r.indexSettings, `{"number_of_shards": %d, "number_of_replicas": %d, "refresh_interval":"5s"}`, r.config.NumShards, r.config.NumReplicas)
	r.indexSettings.WriteString(`, "mappings":`)
	r.indexSettings.Write(mapping)
	r.indexSettings.WriteString(`}`)

	f, err := os.Open(filepath.Join("data", r.config.DatasetName, "document.json"))
	if err != nil {
		return fmt.Errorf("setup: reading document: %s", err)
	}
	var m map[string]interface{}
	json.NewDecoder(f).Decode(&m)
	doc, err := json.Marshal(m)
	if err != nil {
		return fmt.Errorf("setup: encoding document: %s", err)
	}
	r.doc = doc

	r.config.Client.Indices.Delete(
		[]string{r.config.IndexName},
		r.config.Client.Indices.Delete.WithIgnoreUnavailable(true))
	res, err := r.config.Client.Indices.Create(
		r.config.IndexName,
		r.config.Client.Indices.Create.WithBody(strings.NewReader(r.indexSettings.String())),
		r.config.Client.Indices.Create.WithWaitForActiveShards("1"))
	if err != nil {
		return fmt.Errorf("setup: encoding document: %s", err)
	}
	res.Body.Close()
	if res.IsError() {
		return fmt.Errorf("setup: encoding document: %s", res.String())
	}

	return nil
}

// run executes a single benchmark run n, recording stats when measure is true.
func (r *Runner) run(n int, measure bool) error {