|
| 1 | +package analyzer_test |
| 2 | + |
| 3 | +import ( |
| 4 | + "strings" |
| 5 | + "testing" |
| 6 | + |
| 7 | + "github.com/sqlc-dev/sqlc/internal/config" |
| 8 | + "github.com/sqlc-dev/sqlc/internal/core" |
| 9 | + "github.com/sqlc-dev/sqlc/internal/core/analyzer" |
| 10 | + coreschema "github.com/sqlc-dev/sqlc/internal/core/schema" |
| 11 | + "github.com/sqlc-dev/sqlc/internal/engine/googlesql" |
| 12 | + "github.com/sqlc-dev/sqlc/internal/sql/ast" |
| 13 | + "github.com/sqlc-dev/sqlc/internal/sql/rewrite" |
| 14 | + "github.com/sqlc-dev/sqlc/internal/sql/validate" |
| 15 | +) |
| 16 | + |
| 17 | +const benchSchema = ` |
| 18 | +CREATE TABLE users ( |
| 19 | + id INT64 NOT NULL, |
| 20 | + name STRING NOT NULL, |
| 21 | + bio STRING, |
| 22 | +) PRIMARY KEY (id); |
| 23 | +
|
| 24 | +CREATE TABLE posts ( |
| 25 | + id INT64 NOT NULL, |
| 26 | + user_id INT64 NOT NULL, |
| 27 | + title STRING(255), |
| 28 | + created TIMESTAMP NOT NULL, |
| 29 | +) PRIMARY KEY (id); |
| 30 | +` |
| 31 | + |
| 32 | +const benchQueries = ` |
| 33 | +SELECT * FROM users; |
| 34 | +SELECT COUNT(*) AS total FROM users; |
| 35 | +SELECT u.name, p.* FROM users u JOIN posts p ON p.user_id = u.id WHERE u.name = @name; |
| 36 | +SELECT id, name, bio FROM users WHERE id = @id AND name = @name; |
| 37 | +SELECT p.title, p.created FROM posts p WHERE p.user_id = @uid AND p.title = @t; |
| 38 | +` |
| 39 | + |
| 40 | +func parseAll(t testing.TB, src string) []ast.Node { |
| 41 | + p := googlesql.NewParser() |
| 42 | + stmts, err := p.Parse(strings.NewReader(src)) |
| 43 | + if err != nil { |
| 44 | + t.Fatal(err) |
| 45 | + } |
| 46 | + out := make([]ast.Node, 0, len(stmts)) |
| 47 | + for i := range stmts { |
| 48 | + out = append(out, stmts[i].Raw) |
| 49 | + } |
| 50 | + return out |
| 51 | +} |
| 52 | + |
| 53 | +// parseQueries mirrors the compiler pipeline: parse, then rewrite named |
| 54 | +// parameters into ParamRef nodes before handing the statement to the analyzer. |
| 55 | +func parseQueries(t testing.TB, src string) []ast.Node { |
| 56 | + out := make([]ast.Node, 0) |
| 57 | + for _, n := range parseAll(t, src) { |
| 58 | + raw, ok := n.(*ast.RawStmt) |
| 59 | + if !ok { |
| 60 | + t.Fatalf("not a raw statement: %T", n) |
| 61 | + } |
| 62 | + numbers, dollar, err := validate.ParamRef(raw) |
| 63 | + if err != nil { |
| 64 | + t.Fatal(err) |
| 65 | + } |
| 66 | + rewritten, _, _ := rewrite.NamedParameters(config.EngineGoogleSQL, raw, numbers, dollar) |
| 67 | + out = append(out, rewritten) |
| 68 | + } |
| 69 | + return out |
| 70 | +} |
| 71 | + |
| 72 | +func newBenchCatalog(t testing.TB) (*core.Catalog, []ast.Node) { |
| 73 | + cat, err := core.New(googlesql.Dialect()) |
| 74 | + if err != nil { |
| 75 | + t.Fatal(err) |
| 76 | + } |
| 77 | + for _, n := range parseAll(t, benchSchema) { |
| 78 | + if err := coreschema.Apply(cat, n); err != nil { |
| 79 | + t.Fatal(err) |
| 80 | + } |
| 81 | + } |
| 82 | + return cat, parseQueries(t, benchQueries) |
| 83 | +} |
| 84 | + |
| 85 | +// BenchmarkCatalogNew measures per-compile catalog setup: open the SQLite |
| 86 | +// catalog, install the schema, and run the dialect seed. |
| 87 | +func BenchmarkCatalogNew(b *testing.B) { |
| 88 | + for b.Loop() { |
| 89 | + cat, err := core.New(googlesql.Dialect()) |
| 90 | + if err != nil { |
| 91 | + b.Fatal(err) |
| 92 | + } |
| 93 | + cat.Close() |
| 94 | + } |
| 95 | +} |
| 96 | + |
| 97 | +// BenchmarkApplySchema measures loading user DDL into the catalog. |
| 98 | +func BenchmarkApplySchema(b *testing.B) { |
| 99 | + stmts := parseAll(b, benchSchema) |
| 100 | + for b.Loop() { |
| 101 | + cat, err := core.New(googlesql.Dialect()) |
| 102 | + if err != nil { |
| 103 | + b.Fatal(err) |
| 104 | + } |
| 105 | + for _, n := range stmts { |
| 106 | + if err := coreschema.Apply(cat, n); err != nil { |
| 107 | + b.Fatal(err) |
| 108 | + } |
| 109 | + } |
| 110 | + cat.Close() |
| 111 | + } |
| 112 | +} |
| 113 | + |
| 114 | +// BenchmarkPrepare measures steady-state query analysis against a warm catalog. |
| 115 | +func BenchmarkPrepare(b *testing.B) { |
| 116 | + cat, queries := newBenchCatalog(b) |
| 117 | + defer cat.Close() |
| 118 | + for b.Loop() { |
| 119 | + for _, q := range queries { |
| 120 | + if _, err := analyzer.Prepare(cat, q); err != nil { |
| 121 | + b.Fatal(err) |
| 122 | + } |
| 123 | + } |
| 124 | + } |
| 125 | +} |
0 commit comments