fix: describe statements in duckdb

quarylabs · Nov 25, 2024 · 1bc2292 · 1bc2292
1 parent 1c53f6b
commit 1bc2292
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 2 deletions.
diff --git a/crates/lib-dialects/src/duckdb.rs b/crates/lib-dialects/src/duckdb.rs
@@ -74,7 +74,7 @@ pub fn raw_dialect() -> Dialect {
             .into(),
         ),
         (
-            "SummarizeStatemenSegment".into(),
+            "SummarizeStatementSegment".into(),
             Sequence::new(vec_of_erased![
                 Ref::keyword("SUMMARIZE"),
                 one_of(vec_of_erased![
@@ -85,6 +85,18 @@ pub fn raw_dialect() -> Dialect {
             .to_matchable()
             .into(),
         ),
+        (
+            "DescribeStatementSegment".into(),
+            Sequence::new(vec_of_erased![
+                Ref::keyword("DESCRIBE"),
+                one_of(vec_of_erased![
+                    Ref::new("SingleIdentifierGrammar"),
+                    Ref::new("SelectStatementSegment")
+                ])
+            ])
+            .to_matchable()
+            .into(),
+        ),
     ]);
 
     duckdb_dialect.insert_lexer_matchers(
@@ -207,7 +219,8 @@ pub fn raw_dialect() -> Dialect {
         postgres::statement_segment().copy(
             Some(vec_of_erased![
                 Ref::new("LoadStatementSegment"),
-                Ref::new("SummarizeStatemenSegment")
+                Ref::new("SummarizeStatementSegment"),
+                Ref::new("DescribeStatementSegment")
             ]),
             None,
             None,

diff --git a/crates/lib-dialects/test/fixtures/dialects/duckdb/describe.sql b/crates/lib-dialects/test/fixtures/dialects/duckdb/describe.sql
@@ -0,0 +1,3 @@
+DESCRIBE tbl;
+
+DESCRIBE SELECT * FROM tbl;
diff --git a/crates/lib-dialects/test/fixtures/dialects/duckdb/describe.yml b/crates/lib-dialects/test/fixtures/dialects/duckdb/describe.yml
@@ -0,0 +1,22 @@
+file:
+- statement:
+  - keyword: DESCRIBE
+  - naked_identifier: tbl
+- statement_terminator: ;
+- statement:
+  - keyword: DESCRIBE
+  - select_statement:
+    - select_clause:
+      - keyword: SELECT
+      - select_clause_element:
+        - wildcard_expression:
+          - wildcard_identifier:
+            - star: '*'
+    - from_clause:
+      - keyword: FROM
+      - from_expression:
+        - from_expression_element:
+          - table_expression:
+            - table_reference:
+              - naked_identifier: tbl
+- statement_terminator: ;