feat: support transactions in pg

taozhi8833998 · Jan 18, 2024 · f416e21 · f416e21
1 parent a5be053
commit f416e21
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 7 deletions.
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -202,10 +202,13 @@
 }
 
 start
-  = __ n:(create_function_stmt / multiple_stmt) {
+  = b:('begin'i __ SEMICOLON) __ n:(create_function_stmt / multiple_stmt) __ c:('commit'i __ SEMICOLON) {
     // => multiple_stmt
+    n.ast.transactions = true
     return n
   }
+  / create_function_stmt
+  / multiple_stmt
 
 cmd_stmt
   = drop_stmt

diff --git a/src/sql.js b/src/sql.js
@@ -24,6 +24,7 @@ function goToSQL(stmt) {
 }
 
 export default function astToSQL(ast) {
-  if (ast.go === 'go') return goToSQL(ast)
-  return toSQL(ast)
+  const sql = ast.go === 'go' ? goToSQL(ast) : toSQL(ast)
+  if (ast.transactions) return `BEGIN;\n${sql};\nCOMMIT;`
+  return sql
 }
diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -1379,10 +1379,14 @@ describe('Postgres', () => {
       ]
     },
     {
-      title: 'set to',
-      sql: [
-        'SET search_path TO ht_hyt',
-        'SET search_path TO ht_hyt'
+      title: 'set to in transactions',
+      sql: [
+        `BEGIN;
+        SET search_path TO ht_hyt;
+        COMMIT;`,
+        `BEGIN;
+SET search_path TO ht_hyt;
+COMMIT;`,
       ]
     }
   ]