Fix: make end transactions postgres specific closes #1921

tobymao · Jul 14, 2023 · 35e05f7 · 35e05f7
1 parent cc33749
commit 35e05f7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/sqlglot/dialects/postgres.py b/sqlglot/dialects/postgres.py
@@ -318,6 +318,11 @@ class Parser(parser.Parser):
             TokenType.LT_AT: binary_range_parser(exp.ArrayContained),
         }
 
+        STATEMENT_PARSERS = {
+            **parser.Parser.STATEMENT_PARSERS,
+            TokenType.END: lambda self: self._parse_commit_or_rollback(),
+        }
+
         def _parse_factor(self) -> t.Optional[exp.Expression]:
             return self._parse_tokens(self._parse_exponent, self.FACTOR)
 

diff --git a/sqlglot/parser.py b/sqlglot/parser.py
@@ -506,7 +506,6 @@ class Parser(metaclass=_Parser):
         TokenType.DESC: lambda self: self._parse_describe(),
         TokenType.DESCRIBE: lambda self: self._parse_describe(),
         TokenType.DROP: lambda self: self._parse_drop(),
-        TokenType.END: lambda self: self._parse_commit_or_rollback(),
         TokenType.FROM: lambda self: exp.select("*").from_(
             t.cast(exp.From, self._parse_from(skip_from_token=True))
         ),

diff --git a/tests/fixtures/identity.sql b/tests/fixtures/identity.sql
@@ -76,6 +76,7 @@ map.x
 SELECT update
 SELECT x.update
 SELECT call.x
+SELECT end
 a.b.INT(1.234)
 INT(x / 100)
 time * 100