pg_query 1.3.0 → 2.0.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGELOG.md +86 -52
- data/README.md +72 -65
- data/Rakefile +82 -1
- data/ext/pg_query/extconf.rb +2 -39
- data/ext/pg_query/guc-file.c +0 -0
- data/ext/pg_query/pg_query.c +104 -0
- data/ext/pg_query/pg_query.pb-c.c +37628 -0
- data/ext/pg_query/pg_query_deparse.c +9953 -0
- data/ext/pg_query/pg_query_fingerprint.c +292 -0
- data/ext/pg_query/pg_query_fingerprint.h +8 -0
- data/ext/pg_query/pg_query_internal.h +24 -0
- data/ext/pg_query/pg_query_json_plpgsql.c +738 -0
- data/ext/pg_query/pg_query_json_plpgsql.h +9 -0
- data/ext/pg_query/pg_query_normalize.c +437 -0
- data/ext/pg_query/pg_query_outfuncs.h +10 -0
- data/ext/pg_query/pg_query_outfuncs_json.c +297 -0
- data/ext/pg_query/pg_query_outfuncs_protobuf.c +237 -0
- data/ext/pg_query/pg_query_parse.c +148 -0
- data/ext/pg_query/pg_query_parse_plpgsql.c +460 -0
- data/ext/pg_query/pg_query_readfuncs.h +11 -0
- data/ext/pg_query/pg_query_readfuncs_protobuf.c +142 -0
- data/ext/pg_query/pg_query_ruby.c +108 -12
- data/ext/pg_query/pg_query_scan.c +173 -0
- data/ext/pg_query/pg_query_split.c +221 -0
- data/ext/pg_query/protobuf-c.c +3660 -0
- data/ext/pg_query/src_backend_catalog_namespace.c +1051 -0
- data/ext/pg_query/src_backend_catalog_pg_proc.c +142 -0
- data/ext/pg_query/src_backend_commands_define.c +117 -0
- data/ext/pg_query/src_backend_libpq_pqcomm.c +651 -0
- data/ext/pg_query/src_backend_nodes_bitmapset.c +513 -0
- data/ext/pg_query/src_backend_nodes_copyfuncs.c +6013 -0
- data/ext/pg_query/src_backend_nodes_equalfuncs.c +4003 -0
- data/ext/pg_query/src_backend_nodes_extensible.c +99 -0
- data/ext/pg_query/src_backend_nodes_list.c +922 -0
- data/ext/pg_query/src_backend_nodes_makefuncs.c +417 -0
- data/ext/pg_query/src_backend_nodes_nodeFuncs.c +1363 -0
- data/ext/pg_query/src_backend_nodes_value.c +84 -0
- data/ext/pg_query/src_backend_parser_gram.c +47456 -0
- data/ext/pg_query/src_backend_parser_parse_expr.c +313 -0
- data/ext/pg_query/src_backend_parser_parser.c +497 -0
- data/ext/pg_query/src_backend_parser_scan.c +7091 -0
- data/ext/pg_query/src_backend_parser_scansup.c +160 -0
- data/ext/pg_query/src_backend_postmaster_postmaster.c +2230 -0
- data/ext/pg_query/src_backend_storage_ipc_ipc.c +192 -0
- data/ext/pg_query/src_backend_storage_lmgr_s_lock.c +370 -0
- data/ext/pg_query/src_backend_tcop_postgres.c +776 -0
- data/ext/pg_query/src_backend_utils_adt_datum.c +326 -0
- data/ext/pg_query/src_backend_utils_adt_expandeddatum.c +98 -0
- data/ext/pg_query/src_backend_utils_adt_format_type.c +136 -0
- data/ext/pg_query/src_backend_utils_adt_ruleutils.c +1683 -0
- data/ext/pg_query/src_backend_utils_error_assert.c +74 -0
- data/ext/pg_query/src_backend_utils_error_elog.c +1748 -0
- data/ext/pg_query/src_backend_utils_fmgr_fmgr.c +570 -0
- data/ext/pg_query/src_backend_utils_hash_dynahash.c +1086 -0
- data/ext/pg_query/src_backend_utils_init_globals.c +168 -0
- data/ext/pg_query/src_backend_utils_mb_mbutils.c +839 -0
- data/ext/pg_query/src_backend_utils_misc_guc.c +1831 -0
- data/ext/pg_query/src_backend_utils_mmgr_aset.c +1560 -0
- data/ext/pg_query/src_backend_utils_mmgr_mcxt.c +1006 -0
- data/ext/pg_query/src_common_encnames.c +158 -0
- data/ext/pg_query/src_common_keywords.c +39 -0
- data/ext/pg_query/src_common_kwlist_d.h +1081 -0
- data/ext/pg_query/src_common_kwlookup.c +91 -0
- data/ext/pg_query/src_common_psprintf.c +158 -0
- data/ext/pg_query/src_common_string.c +86 -0
- data/ext/pg_query/src_common_stringinfo.c +336 -0
- data/ext/pg_query/src_common_wchar.c +1651 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_comp.c +1133 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_funcs.c +877 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_gram.c +6533 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_handler.c +107 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_reserved_kwlist_d.h +123 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_scanner.c +671 -0
- data/ext/pg_query/src_pl_plpgsql_src_pl_unreserved_kwlist_d.h +255 -0
- data/ext/pg_query/src_port_erand48.c +127 -0
- data/ext/pg_query/src_port_pg_bitutils.c +246 -0
- data/ext/pg_query/src_port_pgsleep.c +69 -0
- data/ext/pg_query/src_port_pgstrcasecmp.c +83 -0
- data/ext/pg_query/src_port_qsort.c +240 -0
- data/ext/pg_query/src_port_random.c +31 -0
- data/ext/pg_query/src_port_snprintf.c +1449 -0
- data/ext/pg_query/src_port_strerror.c +324 -0
- data/ext/pg_query/src_port_strnlen.c +39 -0
- data/ext/pg_query/xxhash.c +43 -0
- data/lib/pg_query.rb +7 -4
- data/lib/pg_query/constants.rb +21 -0
- data/lib/pg_query/deparse.rb +15 -1673
- data/lib/pg_query/filter_columns.rb +86 -85
- data/lib/pg_query/fingerprint.rb +122 -87
- data/lib/pg_query/json_field_names.rb +1402 -0
- data/lib/pg_query/node.rb +31 -0
- data/lib/pg_query/param_refs.rb +42 -37
- data/lib/pg_query/parse.rb +220 -203
- data/lib/pg_query/parse_error.rb +1 -1
- data/lib/pg_query/pg_query_pb.rb +3211 -0
- data/lib/pg_query/scan.rb +23 -0
- data/lib/pg_query/treewalker.rb +24 -40
- data/lib/pg_query/truncate.rb +64 -43
- data/lib/pg_query/version.rb +2 -2
- metadata +101 -11
- data/ext/pg_query/pg_query_ruby.h +0 -10
- data/lib/pg_query/deep_dup.rb +0 -16
- data/lib/pg_query/deparse/alter_table.rb +0 -42
- data/lib/pg_query/deparse/interval.rb +0 -105
- data/lib/pg_query/deparse/keywords.rb +0 -159
- data/lib/pg_query/deparse/rename.rb +0 -41
- data/lib/pg_query/legacy_parsetree.rb +0 -109
- data/lib/pg_query/node_types.rb +0 -297
@@ -0,0 +1,437 @@
|
|
1
|
+
#include "pg_query.h"
|
2
|
+
#include "pg_query_internal.h"
|
3
|
+
|
4
|
+
#include "parser/parser.h"
|
5
|
+
#include "parser/scanner.h"
|
6
|
+
#include "parser/scansup.h"
|
7
|
+
#include "mb/pg_wchar.h"
|
8
|
+
#include "nodes/nodeFuncs.h"
|
9
|
+
|
10
|
+
/*
|
11
|
+
* Struct for tracking locations/lengths of constants during normalization
|
12
|
+
*/
|
13
|
+
typedef struct pgssLocationLen
|
14
|
+
{
|
15
|
+
int location; /* start offset in query text */
|
16
|
+
int length; /* length in bytes, or -1 to ignore */
|
17
|
+
} pgssLocationLen;
|
18
|
+
|
19
|
+
/*
|
20
|
+
* Working state for constant tree walker
|
21
|
+
*/
|
22
|
+
typedef struct pgssConstLocations
|
23
|
+
{
|
24
|
+
/* Array of locations of constants that should be removed */
|
25
|
+
pgssLocationLen *clocations;
|
26
|
+
|
27
|
+
/* Allocated length of clocations array */
|
28
|
+
int clocations_buf_size;
|
29
|
+
|
30
|
+
/* Current number of valid entries in clocations array */
|
31
|
+
int clocations_count;
|
32
|
+
|
33
|
+
/* highest Param id we've seen, in order to start normalization correctly */
|
34
|
+
int highest_extern_param_id;
|
35
|
+
|
36
|
+
/* query text */
|
37
|
+
const char * query;
|
38
|
+
int query_len;
|
39
|
+
} pgssConstLocations;
|
40
|
+
|
41
|
+
/*
|
42
|
+
* comp_location: comparator for qsorting pgssLocationLen structs by location
|
43
|
+
*/
|
44
|
+
static int
|
45
|
+
comp_location(const void *a, const void *b)
|
46
|
+
{
|
47
|
+
int l = ((const pgssLocationLen *) a)->location;
|
48
|
+
int r = ((const pgssLocationLen *) b)->location;
|
49
|
+
|
50
|
+
if (l < r)
|
51
|
+
return -1;
|
52
|
+
else if (l > r)
|
53
|
+
return +1;
|
54
|
+
else
|
55
|
+
return 0;
|
56
|
+
}
|
57
|
+
|
58
|
+
/*
|
59
|
+
* Given a valid SQL string and an array of constant-location records,
|
60
|
+
* fill in the textual lengths of those constants.
|
61
|
+
*
|
62
|
+
* The constants may use any allowed constant syntax, such as float literals,
|
63
|
+
* bit-strings, single-quoted strings and dollar-quoted strings. This is
|
64
|
+
* accomplished by using the public API for the core scanner.
|
65
|
+
*
|
66
|
+
* It is the caller's job to ensure that the string is a valid SQL statement
|
67
|
+
* with constants at the indicated locations. Since in practice the string
|
68
|
+
* has already been parsed, and the locations that the caller provides will
|
69
|
+
* have originated from within the authoritative parser, this should not be
|
70
|
+
* a problem.
|
71
|
+
*
|
72
|
+
* Duplicate constant pointers are possible, and will have their lengths
|
73
|
+
* marked as '-1', so that they are later ignored. (Actually, we assume the
|
74
|
+
* lengths were initialized as -1 to start with, and don't change them here.)
|
75
|
+
*
|
76
|
+
* N.B. There is an assumption that a '-' character at a Const location begins
|
77
|
+
* a negative numeric constant. This precludes there ever being another
|
78
|
+
* reason for a constant to start with a '-'.
|
79
|
+
*/
|
80
|
+
static void
|
81
|
+
fill_in_constant_lengths(pgssConstLocations *jstate, const char *query)
|
82
|
+
{
|
83
|
+
pgssLocationLen *locs;
|
84
|
+
core_yyscan_t yyscanner;
|
85
|
+
core_yy_extra_type yyextra;
|
86
|
+
core_YYSTYPE yylval;
|
87
|
+
YYLTYPE yylloc;
|
88
|
+
int last_loc = -1;
|
89
|
+
int i;
|
90
|
+
|
91
|
+
/*
|
92
|
+
* Sort the records by location so that we can process them in order while
|
93
|
+
* scanning the query text.
|
94
|
+
*/
|
95
|
+
if (jstate->clocations_count > 1)
|
96
|
+
qsort(jstate->clocations, jstate->clocations_count,
|
97
|
+
sizeof(pgssLocationLen), comp_location);
|
98
|
+
locs = jstate->clocations;
|
99
|
+
|
100
|
+
/* initialize the flex scanner --- should match raw_parser() */
|
101
|
+
yyscanner = scanner_init(query,
|
102
|
+
&yyextra,
|
103
|
+
&ScanKeywords,
|
104
|
+
ScanKeywordTokens);
|
105
|
+
|
106
|
+
/* Search for each constant, in sequence */
|
107
|
+
for (i = 0; i < jstate->clocations_count; i++)
|
108
|
+
{
|
109
|
+
int loc = locs[i].location;
|
110
|
+
int tok;
|
111
|
+
|
112
|
+
Assert(loc >= 0);
|
113
|
+
|
114
|
+
if (loc <= last_loc)
|
115
|
+
continue; /* Duplicate constant, ignore */
|
116
|
+
|
117
|
+
/* Lex tokens until we find the desired constant */
|
118
|
+
for (;;)
|
119
|
+
{
|
120
|
+
tok = core_yylex(&yylval, &yylloc, yyscanner);
|
121
|
+
|
122
|
+
/* We should not hit end-of-string, but if we do, behave sanely */
|
123
|
+
if (tok == 0)
|
124
|
+
break; /* out of inner for-loop */
|
125
|
+
|
126
|
+
/*
|
127
|
+
* We should find the token position exactly, but if we somehow
|
128
|
+
* run past it, work with that.
|
129
|
+
*/
|
130
|
+
if (yylloc >= loc)
|
131
|
+
{
|
132
|
+
if (query[loc] == '-')
|
133
|
+
{
|
134
|
+
/*
|
135
|
+
* It's a negative value - this is the one and only case
|
136
|
+
* where we replace more than a single token.
|
137
|
+
*
|
138
|
+
* Do not compensate for the core system's special-case
|
139
|
+
* adjustment of location to that of the leading '-'
|
140
|
+
* operator in the event of a negative constant. It is
|
141
|
+
* also useful for our purposes to start from the minus
|
142
|
+
* symbol. In this way, queries like "select * from foo
|
143
|
+
* where bar = 1" and "select * from foo where bar = -2"
|
144
|
+
* will have identical normalized query strings.
|
145
|
+
*/
|
146
|
+
tok = core_yylex(&yylval, &yylloc, yyscanner);
|
147
|
+
if (tok == 0)
|
148
|
+
break; /* out of inner for-loop */
|
149
|
+
}
|
150
|
+
|
151
|
+
/*
|
152
|
+
* We now rely on the assumption that flex has placed a zero
|
153
|
+
* byte after the text of the current token in scanbuf.
|
154
|
+
*/
|
155
|
+
locs[i].length = (int) strlen(yyextra.scanbuf + loc);
|
156
|
+
|
157
|
+
/* Quoted string with Unicode escapes
|
158
|
+
*
|
159
|
+
* The lexer consumes trailing whitespace in order to find UESCAPE, but if there
|
160
|
+
* is no UESCAPE it has still consumed it - don't include it in constant length.
|
161
|
+
*/
|
162
|
+
if (locs[i].length > 4 && /* U&'' */
|
163
|
+
(yyextra.scanbuf[loc] == 'u' || yyextra.scanbuf[loc] == 'U') &&
|
164
|
+
yyextra.scanbuf[loc + 1] == '&' && yyextra.scanbuf[loc + 2] == '\'')
|
165
|
+
{
|
166
|
+
int j = locs[i].length - 1; /* Skip the \0 */
|
167
|
+
for (; j >= 0 && scanner_isspace(yyextra.scanbuf[loc + j]); j--) {}
|
168
|
+
locs[i].length = j + 1; /* Count the \0 */
|
169
|
+
}
|
170
|
+
|
171
|
+
break; /* out of inner for-loop */
|
172
|
+
}
|
173
|
+
}
|
174
|
+
|
175
|
+
/* If we hit end-of-string, give up, leaving remaining lengths -1 */
|
176
|
+
if (tok == 0)
|
177
|
+
break;
|
178
|
+
|
179
|
+
last_loc = loc;
|
180
|
+
}
|
181
|
+
|
182
|
+
scanner_finish(yyscanner);
|
183
|
+
}
|
184
|
+
|
185
|
+
/*
|
186
|
+
* Generate a normalized version of the query string that will be used to
|
187
|
+
* represent all similar queries.
|
188
|
+
*
|
189
|
+
* Note that the normalized representation may well vary depending on
|
190
|
+
* just which "equivalent" query is used to create the hashtable entry.
|
191
|
+
* We assume this is OK.
|
192
|
+
*
|
193
|
+
* *query_len_p contains the input string length, and is updated with
|
194
|
+
* the result string length (which cannot be longer) on exit.
|
195
|
+
*
|
196
|
+
* Returns a palloc'd string.
|
197
|
+
*/
|
198
|
+
static char *
|
199
|
+
generate_normalized_query(pgssConstLocations *jstate, int query_loc, int* query_len_p, int encoding)
|
200
|
+
{
|
201
|
+
char *norm_query;
|
202
|
+
const char *query = jstate->query;
|
203
|
+
int query_len = *query_len_p;
|
204
|
+
int i,
|
205
|
+
norm_query_buflen, /* Space allowed for norm_query */
|
206
|
+
len_to_wrt, /* Length (in bytes) to write */
|
207
|
+
quer_loc = 0, /* Source query byte location */
|
208
|
+
n_quer_loc = 0, /* Normalized query byte location */
|
209
|
+
last_off = 0, /* Offset from start for previous tok */
|
210
|
+
last_tok_len = 0; /* Length (in bytes) of that tok */
|
211
|
+
|
212
|
+
/*
|
213
|
+
* Get constants' lengths (core system only gives us locations). Note
|
214
|
+
* this also ensures the items are sorted by location.
|
215
|
+
*/
|
216
|
+
fill_in_constant_lengths(jstate, query);
|
217
|
+
|
218
|
+
/*
|
219
|
+
* Allow for $n symbols to be longer than the constants they replace.
|
220
|
+
* Constants must take at least one byte in text form, while a $n symbol
|
221
|
+
* certainly isn't more than 11 bytes, even if n reaches INT_MAX. We
|
222
|
+
* could refine that limit based on the max value of n for the current
|
223
|
+
* query, but it hardly seems worth any extra effort to do so.
|
224
|
+
*/
|
225
|
+
norm_query_buflen = query_len + jstate->clocations_count * 10;
|
226
|
+
|
227
|
+
/* Allocate result buffer */
|
228
|
+
norm_query = palloc(norm_query_buflen + 1);
|
229
|
+
|
230
|
+
for (i = 0; i < jstate->clocations_count; i++)
|
231
|
+
{
|
232
|
+
int off, /* Offset from start for cur tok */
|
233
|
+
tok_len; /* Length (in bytes) of that tok */
|
234
|
+
|
235
|
+
off = jstate->clocations[i].location;
|
236
|
+
/* Adjust recorded location if we're dealing with partial string */
|
237
|
+
off -= query_loc;
|
238
|
+
|
239
|
+
tok_len = jstate->clocations[i].length;
|
240
|
+
|
241
|
+
if (tok_len < 0)
|
242
|
+
continue; /* ignore any duplicates */
|
243
|
+
|
244
|
+
/* Copy next chunk (what precedes the next constant) */
|
245
|
+
len_to_wrt = off - last_off;
|
246
|
+
len_to_wrt -= last_tok_len;
|
247
|
+
|
248
|
+
Assert(len_to_wrt >= 0);
|
249
|
+
memcpy(norm_query + n_quer_loc, query + quer_loc, len_to_wrt);
|
250
|
+
n_quer_loc += len_to_wrt;
|
251
|
+
|
252
|
+
/* And insert a param symbol in place of the constant token */
|
253
|
+
n_quer_loc += sprintf(norm_query + n_quer_loc, "$%d",
|
254
|
+
i + 1 + jstate->highest_extern_param_id);
|
255
|
+
|
256
|
+
quer_loc = off + tok_len;
|
257
|
+
last_off = off;
|
258
|
+
last_tok_len = tok_len;
|
259
|
+
}
|
260
|
+
|
261
|
+
/*
|
262
|
+
* We've copied up until the last ignorable constant. Copy over the
|
263
|
+
* remaining bytes of the original query string.
|
264
|
+
*/
|
265
|
+
len_to_wrt = query_len - quer_loc;
|
266
|
+
|
267
|
+
Assert(len_to_wrt >= 0);
|
268
|
+
memcpy(norm_query + n_quer_loc, query + quer_loc, len_to_wrt);
|
269
|
+
n_quer_loc += len_to_wrt;
|
270
|
+
|
271
|
+
Assert(n_quer_loc <= norm_query_buflen);
|
272
|
+
norm_query[n_quer_loc] = '\0';
|
273
|
+
|
274
|
+
*query_len_p = n_quer_loc;
|
275
|
+
return norm_query;
|
276
|
+
}
|
277
|
+
|
278
|
+
static void RecordConstLocation(pgssConstLocations *jstate, int location)
|
279
|
+
{
|
280
|
+
/* -1 indicates unknown or undefined location */
|
281
|
+
if (location >= 0)
|
282
|
+
{
|
283
|
+
/* enlarge array if needed */
|
284
|
+
if (jstate->clocations_count >= jstate->clocations_buf_size)
|
285
|
+
{
|
286
|
+
jstate->clocations_buf_size *= 2;
|
287
|
+
jstate->clocations = (pgssLocationLen *)
|
288
|
+
repalloc(jstate->clocations,
|
289
|
+
jstate->clocations_buf_size *
|
290
|
+
sizeof(pgssLocationLen));
|
291
|
+
}
|
292
|
+
jstate->clocations[jstate->clocations_count].location = location;
|
293
|
+
/* initialize lengths to -1 to simplify fill_in_constant_lengths */
|
294
|
+
jstate->clocations[jstate->clocations_count].length = -1;
|
295
|
+
jstate->clocations_count++;
|
296
|
+
}
|
297
|
+
}
|
298
|
+
|
299
|
+
static bool const_record_walker(Node *node, pgssConstLocations *jstate)
|
300
|
+
{
|
301
|
+
bool result;
|
302
|
+
MemoryContext normalize_context = CurrentMemoryContext;
|
303
|
+
|
304
|
+
if (node == NULL) return false;
|
305
|
+
|
306
|
+
if (IsA(node, A_Const))
|
307
|
+
{
|
308
|
+
RecordConstLocation(jstate, castNode(A_Const, node)->location);
|
309
|
+
}
|
310
|
+
else if (IsA(node, ParamRef))
|
311
|
+
{
|
312
|
+
/* Track the highest ParamRef number */
|
313
|
+
if (((ParamRef *) node)->number > jstate->highest_extern_param_id)
|
314
|
+
jstate->highest_extern_param_id = castNode(ParamRef, node)->number;
|
315
|
+
}
|
316
|
+
else if (IsA(node, DefElem))
|
317
|
+
{
|
318
|
+
DefElem * defElem = (DefElem *) node;
|
319
|
+
if (defElem->arg != NULL && IsA(defElem->arg, String)) {
|
320
|
+
for (int i = defElem->location; i < jstate->query_len; i++) {
|
321
|
+
if (jstate->query[i] == '\'')
|
322
|
+
RecordConstLocation(jstate, i);
|
323
|
+
}
|
324
|
+
}
|
325
|
+
return const_record_walker((Node *) ((DefElem *) node)->arg, jstate);
|
326
|
+
}
|
327
|
+
else if (IsA(node, RawStmt))
|
328
|
+
{
|
329
|
+
return const_record_walker((Node *) ((RawStmt *) node)->stmt, jstate);
|
330
|
+
}
|
331
|
+
else if (IsA(node, VariableSetStmt))
|
332
|
+
{
|
333
|
+
return const_record_walker((Node *) ((VariableSetStmt *) node)->args, jstate);
|
334
|
+
}
|
335
|
+
else if (IsA(node, CopyStmt))
|
336
|
+
{
|
337
|
+
return const_record_walker((Node *) ((CopyStmt *) node)->query, jstate);
|
338
|
+
}
|
339
|
+
else if (IsA(node, ExplainStmt))
|
340
|
+
{
|
341
|
+
return const_record_walker((Node *) ((ExplainStmt *) node)->query, jstate);
|
342
|
+
}
|
343
|
+
else if (IsA(node, CreateRoleStmt))
|
344
|
+
{
|
345
|
+
return const_record_walker((Node *) ((CreateRoleStmt *) node)->options, jstate);
|
346
|
+
}
|
347
|
+
else if (IsA(node, AlterRoleStmt))
|
348
|
+
{
|
349
|
+
return const_record_walker((Node *) ((AlterRoleStmt *) node)->options, jstate);
|
350
|
+
}
|
351
|
+
else if (IsA(node, DeclareCursorStmt))
|
352
|
+
{
|
353
|
+
return const_record_walker((Node *) ((DeclareCursorStmt *) node)->query, jstate);
|
354
|
+
}
|
355
|
+
|
356
|
+
PG_TRY();
|
357
|
+
{
|
358
|
+
result = raw_expression_tree_walker(node, const_record_walker, (void*) jstate);
|
359
|
+
}
|
360
|
+
PG_CATCH();
|
361
|
+
{
|
362
|
+
MemoryContextSwitchTo(normalize_context);
|
363
|
+
result = false;
|
364
|
+
FlushErrorState();
|
365
|
+
}
|
366
|
+
PG_END_TRY();
|
367
|
+
|
368
|
+
return result;
|
369
|
+
}
|
370
|
+
|
371
|
+
PgQueryNormalizeResult pg_query_normalize(const char* input)
|
372
|
+
{
|
373
|
+
MemoryContext ctx = NULL;
|
374
|
+
PgQueryNormalizeResult result = {0};
|
375
|
+
|
376
|
+
ctx = pg_query_enter_memory_context();
|
377
|
+
|
378
|
+
PG_TRY();
|
379
|
+
{
|
380
|
+
List *tree;
|
381
|
+
pgssConstLocations jstate;
|
382
|
+
int query_len;
|
383
|
+
|
384
|
+
/* Parse query */
|
385
|
+
tree = raw_parser(input);
|
386
|
+
|
387
|
+
query_len = (int) strlen(input);
|
388
|
+
|
389
|
+
/* Set up workspace for constant recording */
|
390
|
+
jstate.clocations_buf_size = 32;
|
391
|
+
jstate.clocations = (pgssLocationLen *)
|
392
|
+
palloc(jstate.clocations_buf_size * sizeof(pgssLocationLen));
|
393
|
+
jstate.clocations_count = 0;
|
394
|
+
jstate.highest_extern_param_id = 0;
|
395
|
+
jstate.query = input;
|
396
|
+
jstate.query_len = query_len;
|
397
|
+
|
398
|
+
/* Walk tree and record const locations */
|
399
|
+
const_record_walker((Node *) tree, &jstate);
|
400
|
+
|
401
|
+
/* Normalize query */
|
402
|
+
result.normalized_query = strdup(generate_normalized_query(&jstate, 0, &query_len, PG_UTF8));
|
403
|
+
}
|
404
|
+
PG_CATCH();
|
405
|
+
{
|
406
|
+
ErrorData* error_data;
|
407
|
+
PgQueryError* error;
|
408
|
+
|
409
|
+
MemoryContextSwitchTo(ctx);
|
410
|
+
error_data = CopyErrorData();
|
411
|
+
|
412
|
+
error = malloc(sizeof(PgQueryError));
|
413
|
+
error->message = strdup(error_data->message);
|
414
|
+
error->filename = strdup(error_data->filename);
|
415
|
+
error->lineno = error_data->lineno;
|
416
|
+
error->cursorpos = error_data->cursorpos;
|
417
|
+
|
418
|
+
result.error = error;
|
419
|
+
FlushErrorState();
|
420
|
+
}
|
421
|
+
PG_END_TRY();
|
422
|
+
|
423
|
+
pg_query_exit_memory_context(ctx);
|
424
|
+
|
425
|
+
return result;
|
426
|
+
}
|
427
|
+
|
428
|
+
void pg_query_free_normalize_result(PgQueryNormalizeResult result)
|
429
|
+
{
|
430
|
+
if (result.error) {
|
431
|
+
free(result.error->message);
|
432
|
+
free(result.error->filename);
|
433
|
+
free(result.error);
|
434
|
+
}
|
435
|
+
|
436
|
+
free(result.normalized_query);
|
437
|
+
}
|