Merge pull request #1735 from taozhi8833998/fix-case-when-additive-pg

fix: case when then expr with additive in pg
taozhi8833998 · Dec 24, 2023 · a9344d2 · a9344d2
2 parents 2571991 + b5fd4e1
commit a9344d2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -4501,10 +4501,10 @@ cast_expr
       }
     };
   }
-  / LPAREN __ e:(literal / aggr_func / window_func / func_call / case_expr / interval_expr / column_ref_array_index / param) __ RPAREN __ c:cast_double_colon?  {
+  / LPAREN __ e:(or_expr / column_ref_array_index / param) __ RPAREN __ c:cast_double_colon?  {
     /* => {
         type: 'cast';
-        expr: literal | aggr_func | func_call | case_expr | interval_expr | column_ref | param
+        expr: or_expr | column_ref | param
           | expr;
         keyword: 'cast';
         ...cast_double_colon;

diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -1593,6 +1593,17 @@ describe('Postgres', () => {
           `SELECT * FROM "test" WHERE "théâtre" = 'Molière'`
         ]
       },
+      {
+        title: 'cast when expr is additive_ expr',
+        sql: [
+          `SELECT
+            CASE
+                WHEN updated IS NOT NULL THEN (updated - created)::TIME
+            END AS some_time
+          FROM some_table`,
+          'SELECT CASE WHEN "updated" IS NOT NULL THEN ("updated" - "created")::TIME END AS "some_time" FROM "some_table"'
+        ]
+      },
     ]
     neatlyNestTestedSQL(SQL_LIST)
   })