Merge pull request taozhi8833998#784 from taozhi8833998/feat-string-constants-pg

taozhi8833998 · web-flow · commit 6193f043bf57 · 2021-10-24T19:05:25.000+08:00
feat: support string constants
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -1707,7 +1707,10 @@ expr_item
     return e
   }
 column_list_item
-  = e:expr_item s:KW_DOUBLE_COLON t:data_type __ alias:alias_clause? {
+  = c:string_constants_escape {
+    return { expr: c, as: null }
+  }
+  / e:expr_item s:KW_DOUBLE_COLON t:data_type __ alias:alias_clause? {
     // => { type: 'cast'; expr: expr; symbol: '::'; target: data_type;  as?: null; }
     return {
       as: alias,
@@ -2592,8 +2595,17 @@ primary
     }
   }
 
+string_constants_escape
+  = 'E'i"'" __ n:single_char* __ "'" {
+    return {
+      type: 'origin',
+      value: `E'${n.join('')}'`
+    }
+  }
+
 column_ref
-  = tbl:ident __ DOT __ STAR {
+  = string_constants_escape
+  / tbl:ident __ DOT __ STAR {
     // => IGNORE
       columnList.add(`select::${tbl}::(.*)`);
       return {
@@ -3121,6 +3133,7 @@ escape_char
       return String.fromCharCode(parseInt("0x" + h1 + h2 + h3 + h4));
     }
   / "\\" { return "\\"; }
+  / "''" { return "''" }
 
 line_terminator
   = [\n\r]
diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -408,6 +408,35 @@ describe('Postgres', () => {
         `SHOW TABLES`
       ]
     },
+    {
+      title: 'String Constants',
+      sql: [
+        `select ''''`,
+        `SELECT ''''`
+      ]
+    },
+    {
+      title: 'String Constants',
+      sql: [
+        `SELECT '''To be, or not'', it starts.' AS x;`,
+        `SELECT '''To be, or not'', it starts.' AS "x"`
+      ]
+    },
+    {
+      title: 'String Constants',
+      sql: [
+        `SELECT 'foo'
+        'bar';`,
+        `SELECT 'foobar'`
+      ]
+    },
+    {
+      title: 'String Constants with C-Style Escapes',
+      sql: [
+        `SELECT E'\\''`,
+        `SELECT E'\''`
+      ]
+    },
   ]
   function neatlyNestTestedSQL(sqlList){
     sqlList.forEach(sqlInfo => {
