Function BenchmarkMariaDB_Mixed

pkg/sql/parser/mariadb_bench_test.go:202–260

BenchmarkMariaDB_Mixed benchmarks parsing of queries that combine multiple MariaDB-specific features in a single statement.

(b *testing.B)

// BenchmarkMariaDB_Mixed benchmarks parsing of queries that combine multiple
// MariaDB-specific features in a single statement.
func BenchmarkMariaDB_Mixed(b *testing.B) {
	benchmarks := []struct {
		name string
		sql  string
	}{
		{
			name: "temporal_with_cte",
			sql: `WITH history AS (
					SELECT * FROM orders FOR SYSTEM_TIME ALL
				)
				SELECT id, status FROM history WHERE status = 'cancelled'`,
		},
		{
			name: "hierarchical_with_cte",
			sql: `WITH RECURSIVE org AS (
					SELECT id, name, parent_id FROM employees
					START WITH parent_id IS NULL
					CONNECT BY PRIOR id = parent_id
				)
				SELECT * FROM org ORDER BY id`,
		},
		{
			name: "create_table_versioned",
			sql: `CREATE TABLE orders (
					id INT PRIMARY KEY,
					status VARCHAR(50),
					row_start DATETIME(6) GENERATED ALWAYS AS ROW START,
					row_end   DATETIME(6) GENERATED ALWAYS AS ROW END,
					PERIOD FOR SYSTEM_TIME(row_start, row_end)
				) WITH SYSTEM VERSIONING`,
		},
	}

	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()
			}
		})
	}