fix #209

kitta65 · Jul 1, 2023 · f9e9ccf · f9e9ccf
1 parent cac7791
commit f9e9ccf
Show file tree

Hide file tree

Showing 5 changed files with 33 additions and 14 deletions.
diff --git a/expected_output/other.sql b/expected_output/other.sql
@@ -4,6 +4,12 @@ OPTIONS (uri = 'gs://bucket/folder/*.csv', format = 'CSV')
 AS SELECT 1
 ;
 
+EXPORT DATA
+WITH CONNECTION conn
+OPTIONS (uri = 'gs://bucket/folder/*.csv', format = 'CSV')
+AS SELECT 1
+;
+
 ----- LOAD statement -----
 LOAD DATA INTO `mydataset.tablename`
 FROM FILES (uris = ['azure://sample.com/sample.parquet'], format = 'PARQUET')

diff --git a/input/other.sql b/input/other.sql
@@ -4,6 +4,12 @@ export data options(
   format = 'CSV'
 ) as select 1;
 
+export data with connection conn
+options(
+  uri = 'gs://bucket/folder/*.csv',
+  format = 'CSV'
+) as select 1;
+
 ----- LOAD statement -----
 load data into `mydataset.tablename`
 from files (

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -24,7 +24,7 @@
     "typescript": "^4.3.2"
   },
   "dependencies": {
-    "bq2cst": "0.4.25"
+    "bq2cst": "0.4.26"
   },
   "peerDependencies": {
     "prettier": "^2.3.0"

diff --git a/src/printer.ts b/src/printer.ts
@@ -2711,6 +2711,7 @@ const printExportStatement: PrintFunc<bq2cst.ExportStatement> = (
     leading_comments: printLeadingComments(path, options, print, node),
     self: p.self("upper"),
     trailing_comments: printTrailingComments(path, options, print, node),
+    with_connection: p.child("with_connection"),
     data: p.child("data", undefined, "all"),
     options: p.child("options"),
     as: p.child("as"),
@@ -2723,6 +2724,8 @@ const printExportStatement: PrintFunc<bq2cst.ExportStatement> = (
       docs.trailing_comments,
       " ",
       docs.data,
+      p.has("with_connection") ? line : "",
+      docs.with_connection,
       line,
       docs.options,
       line,