Merge pull request #2109 from taozhi8833998/refactor-function-loc-all…

…-dialects refactor: add loc property to func_call in all dialects
taozhi8833998 · Sep 11, 2024 · 8dc3c89 · 8dc3c89
2 parents 1dc6f55 + 11bd1bc
commit 8dc3c89
Show file tree

Hide file tree

Showing 15 changed files with 245 additions and 94 deletions.
diff --git a/pegjs/athena.pegjs b/pegjs/athena.pegjs
@@ -953,7 +953,8 @@ call_stmt
       ast: {
         type: 'call',
         expr: e
-      }
+      },
+      ...getLocationObject(),
     }
   }
 
@@ -2044,15 +2045,17 @@ func_call
         type: 'function',
         name: { name: [{ type: 'origin', value: name }] },
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
   / extract_func
   / f:scalar_time_func __ up:on_update_current_timestamp? {
     return {
         type: 'function',
         name: { name: [{ type: 'origin', value: f }] },
-        over: up
+        over: up,
+        ...getLocationObject(),
     }
   }
   / name:(KW_DATE / KW_TIME / KW_TIMESTAMP / 'AT TIME ZONE'i) __ l:or_and_where_expr? __ bc:over_partition? {
@@ -2063,6 +2066,7 @@ func_call
         args: l ? l: { type: 'expr_list', value: [] },
         over: bc,
         args_parentheses: false,
+        ...getLocationObject(),
       };
   }
   / name:proc_func_name __ LPAREN __ l:or_and_where_expr? __ RPAREN __ bc:over_partition? {
@@ -2071,7 +2075,8 @@ func_call
         type: 'function',
         name: name,
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
 extract_filed
@@ -2088,7 +2093,8 @@ extract_func
           field: f,
           cast_type: t,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
   / kw:KW_EXTRACT __ LPAREN __ f:extract_filed __ KW_FROM __ s:expr __ RPAREN {
@@ -2098,7 +2104,8 @@ extract_func
         args: {
           field: f,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
 scalar_time_func
@@ -2657,14 +2664,16 @@ proc_func_call
         args: {
           type: 'expr_list',
           value: l
-        }
+        },
+        ...getLocationObject(),
       };
     }
   / name:proc_func_name {
     return {
         type: 'function',
         name: name,
-        args: null
+        args: null,
+        ...getLocationObject(),
       };
   }
 

diff --git a/pegjs/bigquery.pegjs b/pegjs/bigquery.pegjs
@@ -516,14 +516,16 @@ proc_func_call
         args: {
           type: 'expr_list',
           value: l
-        }
+        },
+        ...getLocationObject(),
       };
     }
   / name:proc_func_name {
     return {
         type: 'function',
         name: name,
-        args: null
+        args: null,
+        ...getLocationObject(),
       };
   }
 
@@ -2656,14 +2658,16 @@ func_call
         type: 'function',
         name: { name: [{ type: 'default', value: name }] },
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
   / f:scalar_time_func __ up:on_update_current_timestamp? {
     return {
         type: 'function',
         name: { name: [{ type: 'origin', value: f }] },
-        over: up
+        over: up,
+        ...getLocationObject(),
     }
   }
   / name:proc_func_name __ LPAREN __ l:or_and_where_expr? __ RPAREN __ bc:over_partition? {
@@ -2672,7 +2676,8 @@ func_call
         type: 'function',
         name: name,
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
 
@@ -2709,7 +2714,8 @@ any_value_func
           expr: e,
           having: h
         },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
     }
   }
 
@@ -2728,7 +2734,8 @@ extract_func
           field: f,
           cast_type: t,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
   / kw:KW_EXTRACT __ LPAREN __ f:extract_filed __ KW_FROM __ s:expr __ RPAREN {
@@ -2737,7 +2744,8 @@ extract_func
         args: {
           field: f,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
   / 'DATE_TRUNC'i __  LPAREN __ e:expr __ COMMA __ f:extract_filed __ RPAREN {
@@ -2746,6 +2754,7 @@ extract_func
         name: { name: [{ type: 'origin', value: 'date_trunc' }]},
         args: { type: 'expr_list', value: [e, { type: 'origin', value: f }] },
         over: null,
+        ...getLocationObject(),
       };
   }
 

diff --git a/pegjs/db2.pegjs b/pegjs/db2.pegjs
@@ -2028,14 +2028,16 @@ func_call
         type: 'function',
         name: { name: [{ type: 'default', value: name }] },
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
   / f:scalar_time_func __ up:on_update_current_timestamp? {
     return {
         type: 'function',
         name: { name: [{ type: 'origin', value: f }] },
-        over: up
+        over: up,
+        ...getLocationObject(),
     }
   }
   / name:proc_func_name __ LPAREN __ l:or_and_where_expr? __ RPAREN __ bc:over_partition? {
@@ -2045,6 +2047,7 @@ func_call
         name: name,
         args: l ? l: { type: 'expr_list', value: [] },
         over: bc,
+        ...getLocationObject(),
       };
     }
 scalar_time_func
@@ -2585,14 +2588,16 @@ proc_func_call
         args: {
           type: 'expr_list',
           value: l
-        }
+        },
+        ...getLocationObject(),
       };
     }
   / name:proc_func_name {
     return {
         type: 'function',
         name: name,
-        args: null
+        args: null,
+        ...getLocationObject(),
       };
   }
 

diff --git a/pegjs/flinksql.pegjs b/pegjs/flinksql.pegjs
@@ -2929,6 +2929,7 @@ position_func_clause
         name: { name: [{ type: 'origin', value: 'position' }]},
         separator: ' ',
         args,
+        ...getLocationObject(),
     };
   }
 
@@ -2955,6 +2956,7 @@ json_object_func_clause
         type: 'function',
         name: { name: [{ type: 'origin', value: 'json_object' }]},
         args,
+        ...getLocationObject(),
     };
   }
 
@@ -2983,6 +2985,7 @@ trim_func_clause
         type: 'function',
         name: { name: [{ type: 'origin', value: 'trim' }]},
         args,
+        ...getLocationObject(),
     };
   }
 
@@ -3008,6 +3011,7 @@ overlay_func_clause
         name: { name: [{ type: 'origin', value: 'overlay' }]},
         separator: ' ',
         args,
+        ...getLocationObject(),
     };
   }
 
@@ -3033,6 +3037,7 @@ substring_func_clause
         name: { name: [{ type: 'origin', value: 'substring' }]},
         separator: ' ',
         args,
+        ...getLocationObject(),
     };
   }
 
@@ -3047,7 +3052,8 @@ func_call
         type: 'function',
         name: { name: [{ type: 'default', value: name }] },
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
   / extract_func
@@ -3056,7 +3062,8 @@ func_call
     return {
         type: 'function',
         name: { name: [{ type: 'origin', value: f }] },
-        over: up
+        over: up,
+        ...getLocationObject(),
     }
   }
   / name:proc_func_name __ LPAREN __ l:or_and_where_expr? __ RPAREN __ bc:over_partition? {
@@ -3067,6 +3074,7 @@ func_call
         name: name,
         args: l ? l: { type: 'expr_list', value: [] },
         over: bc,
+        ...getLocationObject(),
       };
     }
 
@@ -3084,7 +3092,8 @@ extract_func
           field: f,
           cast_type: t,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
   / kw:KW_EXTRACT __ LPAREN __ f:extract_filed __ KW_FROM __ s:expr __ RPAREN {
@@ -3093,7 +3102,8 @@ extract_func
         args: {
           field: f,
           source: s,
-        }
+        },
+        ...getLocationObject(),
     }
   }
 scalar_time_func
@@ -3715,15 +3725,17 @@ proc_func_call
         args: {
           type: 'expr_list',
           value: l
-        }
+        },
+        ...getLocationObject(),
       };
     }
   / name:proc_func_name {
     // => IGNORE
     return {
         type: 'function',
         name: name,
-        args: null
+        args: null,
+        ...getLocationObject(),
       };
   }
 

diff --git a/pegjs/hive.pegjs b/pegjs/hive.pegjs
@@ -2020,14 +2020,16 @@ func_call
         type: 'function',
         name: { name: [{ type: 'default', value: name }] },
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
   / f:scalar_time_func __ up:on_update_current_timestamp? {
     return {
         type: 'function',
         name: { name: [{ type: 'origin', value: f }] },
-        over: up
+        over: up,
+        ...getLocationObject(),
     }
   }
   / name:(KW_DATE / KW_TIME / KW_TIMESTAMP / 'AT TIME ZONE'i) __ l:or_and_where_expr? __ bc:over_partition? {
@@ -2038,6 +2040,7 @@ func_call
         args: l ? l: { type: 'expr_list', value: [] },
         over: bc,
         args_parentheses: false,
+        ...getLocationObject(),
       };
   }
   / name:proc_func_name __ LPAREN __ l:or_and_where_expr? __ RPAREN __ bc:over_partition? {
@@ -2046,7 +2049,8 @@ func_call
         type: 'function',
         name: name,
         args: l ? l: { type: 'expr_list', value: [] },
-        over: bc
+        over: bc,
+        ...getLocationObject(),
       };
     }
 scalar_time_func
@@ -2591,14 +2595,16 @@ proc_func_call
         args: {
           type: 'expr_list',
           value: l
-        }
+        },
+        ...getLocationObject(),
       };
     }
   / name:proc_func_name {
     return {
         type: 'function',
         name: name,
-        args: null
+        args: null,
+        ...getLocationObject(),
       };
   }