feat: support position function in redshift

Author: taozhi8833998

pegjs/redshift.pegjs

@@ -4410,8 +4410,34 @@ tablefunc_clause
     }
   }
 
+position_func_args
+  = s:literal_string __ KW_IN __ e:expr start:(__ KW_FROM __ literal_numeric)? {
+    // => expr_list
+    let value = [s, { type: 'origin', value: 'in' }, e]
+    if (start) {
+      value.push({ type: 'origin', value: 'from' })
+      value.push(start[3])
+    }
+    return {
+      type: 'expr_list',
+      value,
+    }
+  }
+
+position_func_clause
+  = 'POSITION'i __ LPAREN __ args:position_func_args __ RPAREN {
+    // => { type: 'function'; name: string; args: expr_list; }
+    return {
+        type: 'function',
+        name: { name: [{ type: 'origin', value: 'position' }]},
+        separator: ' ',
+        args,
+        ...getLocationObject(),
+    };
+  }
+  
 func_call
-  = trim_func_clause / tablefunc_clause
+  = trim_func_clause / tablefunc_clause / position_func_clause
   / name:'now'i __ LPAREN __ l:expr_list? __ RPAREN __ 'at'i __ KW_TIME __ 'zone'i __ z:literal_string {
     // => { type: 'function'; name: proc_func_name; args: expr_list; suffix: literal_string; }
       z.prefix = 'at time zone'

pegjs/snowflake.pegjs

@@ -3881,8 +3881,33 @@ json_visit_list
       value: createList(head, tail, 1)
     }
   }
+position_func_args
+  = s:literal_string __ KW_IN __ e:expr start:(__ KW_FROM __ literal_numeric)? {
+    // => expr_list
+    let value = [s, { type: 'origin', value: 'in' }, e]
+    if (start) {
+      value.push({ type: 'origin', value: 'from' })
+      value.push(start[3])
+    }
+    return {
+      type: 'expr_list',
+      value,
+    }
+  }
+
+position_func_clause
+  = 'POSITION'i __ LPAREN __ args:position_func_args __ RPAREN {
+    // => { type: 'function'; name: string; args: expr_list; }
+    return {
+        type: 'function',
+        name: { name: [{ type: 'origin', value: 'position' }]},
+        separator: ' ',
+        args,
+        ...getLocationObject(),
+    };
+  }
 func_call
-  = trim_func_clause
+  = trim_func_clause / position_func_clause
   / name:'now'i __ LPAREN __ l:expr_list? __ RPAREN __ 'at'i __ KW_TIME __ 'zone'i __ z:literal_string {
     // => { type: 'function'; name: string; args: expr_list; suffix: literal_string; }
       z.prefix = 'at time zone'

test/redshift.spec.js

@@ -41,5 +41,19 @@ describe('redshift', () => {
     const sql = 'select * from model_a a-- comment'
     expect(getParsedSql(sql)).to.be.equal('SELECT * FROM "model_a" AS "a"')
   })
-
-})
+  
+  it('should support position function', () => {
+    const sql = `SELECT pv.email,
+            pv.action_date_id,
+            SUM(CASE
+                    WHEN event_category_name = 'Download'
+                      AND POSITION('PNG' IN event_full_name) > 0
+                    THEN 1
+                    ELSE 0
+                END) AS download_png
+    FROM dwh.dwh_fact_pageviews pv
+    WHERE action_date_id >= '2022-01-01'
+    GROUP BY 1, 2`
+    expect(getParsedSql(sql)).to.be.equal(`SELECT "pv".email, "pv".action_date_id, SUM(CASE WHEN event_category_name = 'Download' AND POSITION('PNG' IN event_full_name) > 0 THEN 1 ELSE 0 END) AS "download_png" FROM "dwh"."dwh_fact_pageviews" AS "pv" WHERE action_date_id >= '2022-01-01' GROUP BY 1, 2`)
+  })
+})