Function BenchmarkMariaDB_ConnectBy

pkg/sql/parser/mariadb_bench_test.go:140–198

BenchmarkMariaDB_ConnectBy benchmarks MariaDB CONNECT BY hierarchical query parsing.

(b *testing.B)

// BenchmarkMariaDB_ConnectBy benchmarks MariaDB CONNECT BY hierarchical query parsing.
func BenchmarkMariaDB_ConnectBy(b *testing.B) {
	benchmarks := []struct {
		name string
		sql  string
	}{
		{
			name: "simple_prior_left",
			sql: `SELECT id, name FROM employees
				START WITH parent_id IS NULL
				CONNECT BY PRIOR id = parent_id`,
		},
		{
			name: "prior_right",
			sql: `SELECT id, name FROM employees
				START WITH id = 1
				CONNECT BY id = PRIOR parent_id`,
		},
		{
			name: "nocycle",
			sql: `SELECT id, name, level FROM employees
				START WITH parent_id IS NULL
				CONNECT BY NOCYCLE PRIOR id = parent_id`,
		},
		{
			name: "with_where_and_order",
			sql: `SELECT id, name FROM employees
				WHERE active = 1
				START WITH parent_id IS NULL
				CONNECT BY PRIOR id = parent_id
				ORDER BY id`,
		},
	}

	for _, bm := range benchmarks {
		bm := bm
		b.Run(bm.name, func(b *testing.B) {
			tkz := tokenizer.GetTokenizer()
			defer tokenizer.PutTokenizer(tkz)

			tokens, err := tkz.Tokenize([]byte(bm.sql))
			if err != nil {
				b.Fatalf("Tokenize error: %v", err)
			}

			b.ResetTimer()
			b.ReportAllocs()

			for i := 0; i < b.N; i++ {
				p := parser.NewParser(parser.WithDialect(string(keywords.DialectMariaDB)))
				result, err := p.ParseFromModelTokens(tokens)
				if err != nil {
					b.Fatalf("Parse error: %v", err)
				}
				ast.ReleaseAST(result)
				p.Release()
			}
		})
	}