mixed_benchmark_test.go

package tests_test

import (
	"testing"

	"github.com/Trendyol/es-query-builder/benchmarks/tests/marshal"
	"github.com/Trendyol/es-query-builder/es"
	"github.com/Trendyol/es-query-builder/test/assert"
)

func createMixedQuery() map[string]any {
	query := es.NewQuery(
		es.Bool().
			Must(
				es.Term("author", "George Orwell"),
			).
			MustNot(
				es.Terms("genre", "Fantasy", "Science Fiction"),
				es.Exists("out_of_print"),
			).
			Should(
				es.Terms("title", "1984", "Animal Farm"),
			)).
		Aggs(
			es.Agg("genres_count", es.TermsAgg("genre")),
			es.Agg("authors_and_genres", es.TermsAgg("author").
				Aggs(es.Agg("genres", es.TermsAgg("genre"))),
			),
		)

	return query
}

func createMixedQueryVanilla() map[string]any {
	query := map[string]interface{}{
		"query": map[string]interface{}{
			"bool": map[string]interface{}{
				"must": []map[string]interface{}{
					{
						"term": map[string]interface{}{
							"author": map[string]interface{}{
								"value": "George Orwell",
							},
						},
					},
				},
				"must_not": []map[string]interface{}{
					{
						"terms": map[string]interface{}{
							"genre": []string{
								"Fantasy",
								"Science Fiction",
							},
						},
					},
					{
						"exists": map[string]interface{}{
							"field": "out_of_print",
						},
					},
				},
				"should": []map[string]interface{}{
					{
						"terms": map[string]interface{}{
							"title": []string{
								"1984",
								"Animal Farm",
							},
						},
					},
				},
			},
		},
		"aggs": map[string]interface{}{
			"genres_count": map[string]interface{}{
				"terms": map[string]interface{}{
					"field": "genre",
				},
			},
			"authors_and_genres": map[string]interface{}{
				"terms": map[string]interface{}{
					"field": "author",
				},
				"aggs": map[string]interface{}{
					"genres": map[string]interface{}{
						"terms": map[string]interface{}{
							"field": "genre",
						},
					},
				},
			},
		},
	}
	return query
}

func Benchmark_Mixed_Builder(b *testing.B) {
	createMixedQuery()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		createMixedQuery()
	}
}

func Benchmark_Mixed_Vanilla(b *testing.B) {
	createMixedQueryVanilla()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		createMixedQueryVanilla()
	}
}

func Test_Mixed_Queries_are_equal(t *testing.T) {
	build := marshal.String(t, createMixedQuery())
	vanilla := marshal.String(t, createMixedQueryVanilla())
	assert.Equal(t, vanilla, build)
}