polyglot-sql 0.6.1__tar.gz → 0.6.3__tar.gz

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/Cargo.lock

@@ -605,7 +605,7 @@ dependencies = [
 
 [[package]]
 name = "polyglot-sql"
-version = "0.6.1"
+version = "0.6.3"
 dependencies = [
  "criterion",
  "once_cell",
@@ -623,7 +623,7 @@ dependencies = [
 
 [[package]]
 name = "polyglot-sql-ast-derive"
-version = "0.6.1"
+version = "0.6.3"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -632,7 +632,7 @@ dependencies = [
 
 [[package]]
 name = "polyglot-sql-ffi"
-version = "0.6.1"
+version = "0.6.3"
 dependencies = [
  "cbindgen",
  "polyglot-sql",
@@ -642,11 +642,11 @@ dependencies = [
 
 [[package]]
 name = "polyglot-sql-function-catalogs"
-version = "0.6.1"
+version = "0.6.3"
 
 [[package]]
 name = "polyglot-sql-python"
-version = "0.6.1"
+version = "0.6.3"
 dependencies = [
  "polyglot-sql",
  "pyo3",
@@ -657,7 +657,7 @@ dependencies = [
 
 [[package]]
 name = "polyglot-sql-wasm"
-version = "0.6.1"
+version = "0.6.3"
 dependencies = [
  "console_error_panic_hook",
  "js-sys",

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/Cargo.toml

@@ -6,7 +6,7 @@ exclude = [
 ]
 
 [workspace.package]
-version = "0.6.1"
+version = "0.6.3"
 edition = "2021"
 license = "MIT"
 authors = ["polyglot contributors"]

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polyglot-sql
-Version: 0.6.1
+Version: 0.6.3
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/Cargo.toml

@@ -100,14 +100,14 @@ function-catalog-all-dialects = [
 ]
 
 [dependencies]
-polyglot-sql-ast-derive = { path = "../polyglot-sql-ast-derive", version = "0.6.1" }
+polyglot-sql-ast-derive = { path = "../polyglot-sql-ast-derive", version = "0.6.3" }
 serde = { workspace = true }
 serde_json = { workspace = true }
 thiserror = { workspace = true }
 unicode-segmentation = { workspace = true }
 stacker = { version = "0.1", optional = true }
 ts-rs = { version = "12.0", features = ["serde-compat"], optional = true }
-polyglot-sql-function-catalogs = { path = "../polyglot-sql-function-catalogs", version = "0.6.1", optional = true, default-features = false }
+polyglot-sql-function-catalogs = { path = "../polyglot-sql-function-catalogs", version = "0.6.3", optional = true, default-features = false }
 
 [dev-dependencies]
 pretty_assertions = "1.4"
@@ -142,3 +142,7 @@ harness = false
 [[bench]]
 name = "allocation_hotspots"
 harness = false
+
+[[bench]]
+name = "lineage"
+harness = false

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/README.md

@@ -25,7 +25,7 @@ By default, `polyglot-sql` enables the full public API. Parser-only consumers ca
 disable default features and opt into only the dialect parsers they need:
 
 ```toml
-polyglot-sql = { version = "0.6.1", default-features = false }
+polyglot-sql = { version = "0.6.3", default-features = false }
 ```
 
 ```toml

polyglot_sql-0.6.3/crates/polyglot-sql/benches/lineage.rs

@@ -0,0 +1,129 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion, SamplingMode};
+use polyglot_sql::dialects::DialectType;
+use polyglot_sql::lineage::lineage;
+use polyglot_sql::parse_one;
+use std::fmt::Write as _;
+use std::hint::black_box;
+use std::time::Duration;
+
+const CTE_COUNTS: [usize; 4] = [4, 8, 16, 24];
+const CTE_REFERENCE_COUNTS: [usize; 4] = [2, 4, 8, 16];
+
+fn build_cte_chain(n: usize) -> String {
+    let mut sql = String::from(
+        "WITH\n  c0 AS (\n    SELECT x AS col, y FROM base_table WHERE y > 0\n    \
+         UNION ALL\n    SELECT x AS col, y FROM base_table WHERE y <= 0\n  )",
+    );
+    for i in 1..n {
+        let _ = write!(
+            &mut sql,
+            ",\n  c{i} AS (\n    SELECT c{prev}.col AS col, c{prev}.y AS y FROM c{prev}\n    \
+             UNION ALL\n    SELECT x AS col, y FROM base_table\n  )",
+            i = i,
+            prev = i - 1,
+        );
+    }
+    let _ = write!(&mut sql, "\nSELECT col FROM c{}", n - 1);
+    sql
+}
+
+fn build_nested_cte_chain(n: usize) -> String {
+    let mut sql = "SELECT x AS col FROM base_table".to_string();
+    for i in 0..n {
+        sql = format!("WITH c{i} AS ({sql}) SELECT col FROM c{i}");
+    }
+    sql
+}
+
+fn build_reused_nested_cte(reference_count: usize) -> String {
+    let projection = (0..reference_count)
+        .map(|i| format!("s{i}.col"))
+        .collect::<Vec<_>>()
+        .join(" + ");
+    let sources = (0..reference_count)
+        .map(|i| format!("shared AS s{i}"))
+        .collect::<Vec<_>>()
+        .join(" CROSS JOIN ");
+
+    format!(
+        "WITH shared AS (\
+         WITH nested AS (SELECT x AS col FROM base_table) \
+         SELECT col FROM nested\
+         ) SELECT {projection} AS col FROM {sources}"
+    )
+}
+
+fn configure_group(group: &mut criterion::BenchmarkGroup<'_, criterion::measurement::WallTime>) {
+    group.sampling_mode(SamplingMode::Flat);
+    group.warm_up_time(Duration::from_secs(1));
+    group.sample_size(30);
+    group.measurement_time(Duration::from_secs(5));
+}
+
+fn bench_lineage(
+    group: &mut criterion::BenchmarkGroup<'_, criterion::measurement::WallTime>,
+    n: usize,
+    sql: &str,
+) {
+    let expr = parse_one(sql, DialectType::Generic).expect("benchmark query should parse");
+
+    assert!(
+        lineage("col", &expr, Some(DialectType::Generic), false).is_ok(),
+        "lineage should resolve for n={n} (did we exceed MAX_LINEAGE_DEPTH?)",
+    );
+
+    group.bench_with_input(BenchmarkId::from_parameter(n), &expr, |b, expr| {
+        b.iter(|| {
+            black_box(lineage(
+                black_box("col"),
+                black_box(expr),
+                Some(DialectType::Generic),
+                false,
+            ))
+        });
+    });
+}
+
+fn bench_cte_chain(c: &mut Criterion) {
+    let mut group = c.benchmark_group("lineage_cte_chain");
+    configure_group(&mut group);
+
+    for n in CTE_COUNTS {
+        let sql = build_cte_chain(n);
+        bench_lineage(&mut group, n, &sql);
+    }
+
+    group.finish();
+}
+
+fn bench_nested_cte_chain(c: &mut Criterion) {
+    let mut group = c.benchmark_group("lineage_nested_cte_chain");
+    configure_group(&mut group);
+
+    for n in CTE_COUNTS {
+        let sql = build_nested_cte_chain(n);
+        bench_lineage(&mut group, n, &sql);
+    }
+
+    group.finish();
+}
+
+fn bench_reused_nested_cte(c: &mut Criterion) {
+    let mut group = c.benchmark_group("lineage_reused_nested_cte");
+    configure_group(&mut group);
+
+    for n in CTE_REFERENCE_COUNTS {
+        let sql = build_reused_nested_cte(n);
+        bench_lineage(&mut group, n, &sql);
+    }
+
+    group.finish();
+}
+
+criterion_group!(
+    benches,
+    bench_cte_chain,
+    bench_nested_cte_chain,
+    bench_reused_nested_cte
+);
+criterion_main!(benches);

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/src/builder.rs

@@ -1638,6 +1638,7 @@ impl SelectBuilder {
             partition_of: None,
             post_table_properties: vec![],
             mysql_table_options: vec![],
+            tidb_table_options: vec![],
             inherits: vec![],
             on_property: None,
             copy_grants: false,

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/src/dialects/athena.rs

@@ -147,6 +147,7 @@ impl DialectImpl for AthenaDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 },
             ))),
 
@@ -158,6 +159,7 @@ impl DialectImpl for AthenaDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 },
             ))),
 

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/src/dialects/bigquery.rs

@@ -196,6 +196,7 @@ impl DialectImpl for BigQueryDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 },
             ))),
 
@@ -207,6 +208,7 @@ impl DialectImpl for BigQueryDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 },
             ))),
 
@@ -703,6 +705,7 @@ impl DialectImpl for BigQueryDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 }));
                 let aliased_unnest = Expression::Alias(Box::new(Alias {
                     this: unnest,

{polyglot_sql-0.6.1 → polyglot_sql-0.6.3}/crates/polyglot-sql/src/dialects/duckdb.rs

@@ -238,6 +238,7 @@ impl DialectImpl for DuckDBDialect {
                     with_ordinality: false,
                     alias: None,
                     offset_alias: None,
+                    inferred_type: None,
                 },
             ))),