Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * parse_expr.c
4 : : * handle expressions in parser
5 : : *
6 : : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : *
10 : : * IDENTIFICATION
11 : : * src/backend/parser/parse_expr.c
12 : : *
13 : : *-------------------------------------------------------------------------
14 : : */
15 : :
16 : : #include "postgres.h"
17 : :
18 : : #include "catalog/pg_aggregate.h"
19 : : #include "catalog/pg_proc.h"
20 : : #include "catalog/pg_type.h"
21 : : #include "commands/dbcommands.h"
22 : : #include "miscadmin.h"
23 : : #include "nodes/makefuncs.h"
24 : : #include "nodes/nodeFuncs.h"
25 : : #include "optimizer/optimizer.h"
26 : : #include "parser/analyze.h"
27 : : #include "parser/parse_agg.h"
28 : : #include "parser/parse_clause.h"
29 : : #include "parser/parse_coerce.h"
30 : : #include "parser/parse_collate.h"
31 : : #include "parser/parse_expr.h"
32 : : #include "parser/parse_func.h"
33 : : #include "parser/parse_oper.h"
34 : : #include "parser/parse_relation.h"
35 : : #include "parser/parse_target.h"
36 : : #include "parser/parse_type.h"
37 : : #include "utils/builtins.h"
38 : : #include "utils/date.h"
39 : : #include "utils/fmgroids.h"
40 : : #include "utils/jsonb.h"
41 : : #include "utils/lsyscache.h"
42 : : #include "utils/timestamp.h"
43 : : #include "utils/xml.h"
44 : :
45 : : /* GUC parameters */
46 : : bool Transform_null_equals = false;
47 : :
48 : :
49 : : static Node *transformExprRecurse(ParseState *pstate, Node *expr);
50 : : static Node *transformParamRef(ParseState *pstate, ParamRef *pref);
51 : : static Node *transformAExprOp(ParseState *pstate, A_Expr *a);
52 : : static Node *transformAExprOpAny(ParseState *pstate, A_Expr *a);
53 : : static Node *transformAExprOpAll(ParseState *pstate, A_Expr *a);
54 : : static Node *transformAExprDistinct(ParseState *pstate, A_Expr *a);
55 : : static Node *transformAExprNullIf(ParseState *pstate, A_Expr *a);
56 : : static Node *transformAExprIn(ParseState *pstate, A_Expr *a);
57 : : static Node *transformAExprBetween(ParseState *pstate, A_Expr *a);
58 : : static Node *transformMergeSupportFunc(ParseState *pstate, MergeSupportFunc *f);
59 : : static Node *transformBoolExpr(ParseState *pstate, BoolExpr *a);
60 : : static Node *transformFuncCall(ParseState *pstate, FuncCall *fn);
61 : : static Node *transformMultiAssignRef(ParseState *pstate, MultiAssignRef *maref);
62 : : static Node *transformCaseExpr(ParseState *pstate, CaseExpr *c);
63 : : static Node *transformSubLink(ParseState *pstate, SubLink *sublink);
64 : : static Node *transformArrayExpr(ParseState *pstate, A_ArrayExpr *a,
65 : : Oid array_type, Oid element_type, int32 typmod);
66 : : static Node *transformRowExpr(ParseState *pstate, RowExpr *r, bool allowDefault);
67 : : static Node *transformCoalesceExpr(ParseState *pstate, CoalesceExpr *c);
68 : : static Node *transformMinMaxExpr(ParseState *pstate, MinMaxExpr *m);
69 : : static Node *transformSQLValueFunction(ParseState *pstate,
70 : : SQLValueFunction *svf);
71 : : static Node *transformXmlExpr(ParseState *pstate, XmlExpr *x);
72 : : static Node *transformXmlSerialize(ParseState *pstate, XmlSerialize *xs);
73 : : static Node *transformBooleanTest(ParseState *pstate, BooleanTest *b);
74 : : static Node *transformCurrentOfExpr(ParseState *pstate, CurrentOfExpr *cexpr);
75 : : static Node *transformColumnRef(ParseState *pstate, ColumnRef *cref);
76 : : static Node *transformWholeRowRef(ParseState *pstate,
77 : : ParseNamespaceItem *nsitem,
78 : : int sublevels_up, int location);
79 : : static Node *transformIndirection(ParseState *pstate, A_Indirection *ind);
80 : : static Node *transformTypeCast(ParseState *pstate, TypeCast *tc);
81 : : static Node *transformCollateClause(ParseState *pstate, CollateClause *c);
82 : : static Node *transformJsonObjectConstructor(ParseState *pstate,
83 : : JsonObjectConstructor *ctor);
84 : : static Node *transformJsonArrayConstructor(ParseState *pstate,
85 : : JsonArrayConstructor *ctor);
86 : : static Node *transformJsonArrayQueryConstructor(ParseState *pstate,
87 : : JsonArrayQueryConstructor *ctor);
88 : : static Node *transformJsonObjectAgg(ParseState *pstate, JsonObjectAgg *agg);
89 : : static Node *transformJsonArrayAgg(ParseState *pstate, JsonArrayAgg *agg);
90 : : static Node *transformJsonIsPredicate(ParseState *pstate, JsonIsPredicate *pred);
91 : : static Node *transformJsonParseExpr(ParseState *pstate, JsonParseExpr *expr);
92 : : static Node *transformJsonScalarExpr(ParseState *pstate, JsonScalarExpr *expr);
93 : : static Node *transformJsonSerializeExpr(ParseState *pstate,
94 : : JsonSerializeExpr *expr);
95 : : static Node *transformJsonFuncExpr(ParseState *pstate, JsonFuncExpr *p);
96 : : static void transformJsonPassingArgs(ParseState *pstate, const char *constructName,
97 : : JsonFormatType format, List *args,
98 : : List **passing_values, List **passing_names);
99 : : static void coerceJsonExprOutput(ParseState *pstate, JsonExpr *jsexpr);
100 : : static JsonBehavior *transformJsonBehavior(ParseState *pstate, JsonBehavior *behavior,
101 : : JsonBehaviorType default_behavior,
102 : : JsonReturning *returning);
103 : : static Node *GetJsonBehaviorConst(JsonBehaviorType btype, int location);
104 : : static Node *make_row_comparison_op(ParseState *pstate, List *opname,
105 : : List *largs, List *rargs, int location);
106 : : static Node *make_row_distinct_op(ParseState *pstate, List *opname,
107 : : RowExpr *lrow, RowExpr *rrow, int location);
108 : : static Expr *make_distinct_op(ParseState *pstate, List *opname,
109 : : Node *ltree, Node *rtree, int location);
110 : : static Node *make_nulltest_from_distinct(ParseState *pstate,
111 : : A_Expr *distincta, Node *arg);
112 : :
113 : :
114 : : /*
115 : : * transformExpr -
116 : : * Analyze and transform expressions. Type checking and type casting is
117 : : * done here. This processing converts the raw grammar output into
118 : : * expression trees with fully determined semantics.
119 : : */
120 : : Node *
4265 tgl@sss.pgh.pa.us 121 :CBC 785435 : transformExpr(ParseState *pstate, Node *expr, ParseExprKind exprKind)
122 : : {
123 : : Node *result;
124 : : ParseExprKind sv_expr_kind;
125 : :
126 : : /* Save and restore identity of expression type we're parsing */
127 [ - + ]: 785435 : Assert(exprKind != EXPR_KIND_NONE);
128 : 785435 : sv_expr_kind = pstate->p_expr_kind;
129 : 785435 : pstate->p_expr_kind = exprKind;
130 : :
131 : 785435 : result = transformExprRecurse(pstate, expr);
132 : :
133 : 782638 : pstate->p_expr_kind = sv_expr_kind;
134 : :
135 : 782638 : return result;
136 : : }
137 : :
138 : : static Node *
139 : 2137950 : transformExprRecurse(ParseState *pstate, Node *expr)
140 : : {
141 : : Node *result;
142 : :
9637 bruce@momjian.us 143 [ + + ]: 2137950 : if (expr == NULL)
144 : 24087 : return NULL;
145 : :
146 : : /* Guard against stack overflow due to overly complex expressions */
7326 tgl@sss.pgh.pa.us 147 : 2113863 : check_stack_depth();
148 : :
9637 bruce@momjian.us 149 [ + + + + : 2113863 : switch (nodeTag(expr))
+ + + + +
+ + + + +
+ + + + +
+ + + + +
+ - + + +
+ + + + +
+ + + - ]
150 : : {
8060 tgl@sss.pgh.pa.us 151 : 741788 : case T_ColumnRef:
7025 neilc@samurai.com 152 : 741788 : result = transformColumnRef(pstate, (ColumnRef *) expr);
153 : 741460 : break;
154 : :
155 : 70654 : case T_ParamRef:
156 : 70654 : result = transformParamRef(pstate, (ParamRef *) expr);
157 : 70651 : break;
158 : :
9637 bruce@momjian.us 159 : 533461 : case T_A_Const:
948 peter@eisentraut.org 160 : 533461 : result = (Node *) make_const(pstate, (A_Const *) expr);
161 : 533449 : break;
162 : :
7249 tgl@sss.pgh.pa.us 163 : 9726 : case T_A_Indirection:
2497 164 : 9726 : result = transformIndirection(pstate, (A_Indirection *) expr);
165 : 9681 : break;
166 : :
5869 167 : 3180 : case T_A_ArrayExpr:
168 : 3180 : result = transformArrayExpr(pstate, (A_ArrayExpr *) expr,
169 : : InvalidOid, InvalidOid, -1);
170 : 3177 : break;
171 : :
8854 172 : 119810 : case T_TypeCast:
3339 173 : 119810 : result = transformTypeCast(pstate, (TypeCast *) expr);
174 : 118141 : break;
175 : :
4814 peter_e@gmx.net 176 : 3839 : case T_CollateClause:
177 : 3839 : result = transformCollateClause(pstate, (CollateClause *) expr);
178 : 3830 : break;
179 : :
9637 bruce@momjian.us 180 : 291857 : case T_A_Expr:
181 : : {
182 : 291857 : A_Expr *a = (A_Expr *) expr;
183 : :
7734 tgl@sss.pgh.pa.us 184 [ + + + + : 291857 : switch (a->kind)
+ + + +
- ]
185 : : {
186 : 274065 : case AEXPR_OP:
7025 neilc@samurai.com 187 : 274065 : result = transformAExprOp(pstate, a);
9637 bruce@momjian.us 188 : 273857 : break;
7595 tgl@sss.pgh.pa.us 189 : 7105 : case AEXPR_OP_ANY:
7025 neilc@samurai.com 190 : 7105 : result = transformAExprOpAny(pstate, a);
7595 tgl@sss.pgh.pa.us 191 : 7099 : break;
192 : 150 : case AEXPR_OP_ALL:
7025 neilc@samurai.com 193 : 150 : result = transformAExprOpAll(pstate, a);
7595 tgl@sss.pgh.pa.us 194 : 150 : break;
7734 195 : 464 : case AEXPR_DISTINCT:
196 : : case AEXPR_NOT_DISTINCT:
7025 neilc@samurai.com 197 : 464 : result = transformAExprDistinct(pstate, a);
7924 lockhart@fourpalms.o 198 : 464 : break;
7728 tgl@sss.pgh.pa.us 199 : 119 : case AEXPR_NULLIF:
7025 neilc@samurai.com 200 : 119 : result = transformAExprNullIf(pstate, a);
7728 tgl@sss.pgh.pa.us 201 : 119 : break;
6712 202 : 8396 : case AEXPR_IN:
203 : 8396 : result = transformAExprIn(pstate, a);
204 : 8390 : break;
3338 205 : 1303 : case AEXPR_LIKE:
206 : : case AEXPR_ILIKE:
207 : : case AEXPR_SIMILAR:
208 : : /* we can transform these just like AEXPR_OP */
209 : 1303 : result = transformAExprOp(pstate, a);
210 : 1300 : break;
3339 211 : 255 : case AEXPR_BETWEEN:
212 : : case AEXPR_NOT_BETWEEN:
213 : : case AEXPR_BETWEEN_SYM:
214 : : case AEXPR_NOT_BETWEEN_SYM:
215 : 255 : result = transformAExprBetween(pstate, a);
216 : 255 : break;
7025 neilc@samurai.com 217 :UBC 0 : default:
218 [ # # ]: 0 : elog(ERROR, "unrecognized A_Expr kind: %d", a->kind);
219 : : result = NULL; /* keep compiler quiet */
220 : : break;
221 : : }
9637 bruce@momjian.us 222 :CBC 291634 : break;
223 : : }
224 : :
3590 tgl@sss.pgh.pa.us 225 : 67341 : case T_BoolExpr:
226 : 67341 : result = transformBoolExpr(pstate, (BoolExpr *) expr);
227 : 67331 : break;
228 : :
9637 bruce@momjian.us 229 : 174444 : case T_FuncCall:
7025 neilc@samurai.com 230 : 174444 : result = transformFuncCall(pstate, (FuncCall *) expr);
231 : 173928 : break;
232 : :
3588 tgl@sss.pgh.pa.us 233 : 187 : case T_MultiAssignRef:
234 : 187 : result = transformMultiAssignRef(pstate, (MultiAssignRef *) expr);
235 : 184 : break;
236 : :
3256 andres@anarazel.de 237 : 157 : case T_GroupingFunc:
238 : 157 : result = transformGroupingFunc(pstate, (GroupingFunc *) expr);
239 : 157 : break;
240 : :
28 dean.a.rasheed@gmail 241 :GNC 78 : case T_MergeSupportFunc:
242 : 78 : result = transformMergeSupportFunc(pstate,
243 : : (MergeSupportFunc *) expr);
244 : 72 : break;
245 : :
5302 tgl@sss.pgh.pa.us 246 :CBC 23356 : case T_NamedArgExpr:
247 : : {
248 : 23356 : NamedArgExpr *na = (NamedArgExpr *) expr;
249 : :
4265 250 : 23356 : na->arg = (Expr *) transformExprRecurse(pstate, (Node *) na->arg);
5302 251 : 23356 : result = expr;
252 : 23356 : break;
253 : : }
254 : :
9582 bruce@momjian.us 255 : 18999 : case T_SubLink:
7025 neilc@samurai.com 256 : 18999 : result = transformSubLink(pstate, (SubLink *) expr);
257 : 18954 : break;
258 : :
9263 lockhart@fourpalms.o 259 : 26185 : case T_CaseExpr:
7025 neilc@samurai.com 260 : 26185 : result = transformCaseExpr(pstate, (CaseExpr *) expr);
261 : 26182 : break;
262 : :
7279 tgl@sss.pgh.pa.us 263 : 2632 : case T_RowExpr:
2700 264 : 2632 : result = transformRowExpr(pstate, (RowExpr *) expr, false);
7025 neilc@samurai.com 265 : 2632 : break;
266 : :
7728 tgl@sss.pgh.pa.us 267 : 1492 : case T_CoalesceExpr:
7025 neilc@samurai.com 268 : 1492 : result = transformCoalesceExpr(pstate, (CoalesceExpr *) expr);
269 : 1489 : break;
270 : :
6867 tgl@sss.pgh.pa.us 271 : 128 : case T_MinMaxExpr:
272 : 128 : result = transformMinMaxExpr(pstate, (MinMaxExpr *) expr);
273 : 128 : break;
274 : :
333 michael@paquier.xyz 275 : 1215 : case T_SQLValueFunction:
276 : 1215 : result = transformSQLValueFunction(pstate,
277 : : (SQLValueFunction *) expr);
278 : 1215 : break;
279 : :
6321 tgl@sss.pgh.pa.us 280 : 298 : case T_XmlExpr:
281 : 298 : result = transformXmlExpr(pstate, (XmlExpr *) expr);
282 : 283 : break;
283 : :
6280 peter_e@gmx.net 284 : 95 : case T_XmlSerialize:
285 : 95 : result = transformXmlSerialize(pstate, (XmlSerialize *) expr);
286 : 95 : break;
287 : :
8335 tgl@sss.pgh.pa.us 288 : 7794 : case T_NullTest:
289 : : {
290 : 7794 : NullTest *n = (NullTest *) expr;
291 : :
4265 292 : 7794 : n->arg = (Expr *) transformExprRecurse(pstate, (Node *) n->arg);
293 : : /* the argument can be any type, so don't coerce it */
5217 294 : 7794 : n->argisrow = type_is_rowtype(exprType((Node *) n->arg));
8335 295 : 7794 : result = expr;
296 : 7794 : break;
297 : : }
298 : :
299 : 408 : case T_BooleanTest:
7025 neilc@samurai.com 300 : 408 : result = transformBooleanTest(pstate, (BooleanTest *) expr);
301 : 408 : break;
302 : :
6152 tgl@sss.pgh.pa.us 303 : 121 : case T_CurrentOfExpr:
304 : 121 : result = transformCurrentOfExpr(pstate, (CurrentOfExpr *) expr);
305 : 121 : break;
306 : :
307 : : /*
308 : : * In all places where DEFAULT is legal, the caller should have
309 : : * processed it rather than passing it to transformExpr().
310 : : */
2700 tgl@sss.pgh.pa.us 311 :UBC 0 : case T_SetToDefault:
312 [ # # ]: 0 : ereport(ERROR,
313 : : (errcode(ERRCODE_SYNTAX_ERROR),
314 : : errmsg("DEFAULT is not allowed in this context"),
315 : : parser_errposition(pstate,
316 : : ((SetToDefault *) expr)->location)));
317 : : break;
318 : :
319 : : /*
320 : : * CaseTestExpr doesn't require any processing; it is only
321 : : * injected into parse trees in a fully-formed state.
322 : : *
323 : : * Ordinarily we should not see a Var here, but it is convenient
324 : : * for transformJoinUsingClause() to create untransformed operator
325 : : * trees containing already-transformed Vars. The best
326 : : * alternative would be to deconstruct and reconstruct column
327 : : * references, which seems expensively pointless. So allow it.
328 : : */
7333 tgl@sss.pgh.pa.us 329 :CBC 12514 : case T_CaseTestExpr:
330 : : case T_Var:
331 : : {
9516 bruce@momjian.us 332 : 12514 : result = (Node *) expr;
333 : 12514 : break;
334 : : }
335 : :
382 alvherre@alvh.no-ip. 336 : 185 : case T_JsonObjectConstructor:
337 : 185 : result = transformJsonObjectConstructor(pstate, (JsonObjectConstructor *) expr);
338 : 164 : break;
339 : :
340 : 89 : case T_JsonArrayConstructor:
341 : 89 : result = transformJsonArrayConstructor(pstate, (JsonArrayConstructor *) expr);
342 : 80 : break;
343 : :
344 : 27 : case T_JsonArrayQueryConstructor:
345 : 27 : result = transformJsonArrayQueryConstructor(pstate, (JsonArrayQueryConstructor *) expr);
346 : 18 : break;
347 : :
348 : 78 : case T_JsonObjectAgg:
349 : 78 : result = transformJsonObjectAgg(pstate, (JsonObjectAgg *) expr);
350 : 78 : break;
351 : :
352 : 93 : case T_JsonArrayAgg:
353 : 93 : result = transformJsonArrayAgg(pstate, (JsonArrayAgg *) expr);
354 : 93 : break;
355 : :
380 356 : 167 : case T_JsonIsPredicate:
357 : 167 : result = transformJsonIsPredicate(pstate, (JsonIsPredicate *) expr);
358 : 164 : break;
359 : :
269 amitlan@postgresql.o 360 :GNC 74 : case T_JsonParseExpr:
361 : 74 : result = transformJsonParseExpr(pstate, (JsonParseExpr *) expr);
362 : 63 : break;
363 : :
364 : 49 : case T_JsonScalarExpr:
365 : 49 : result = transformJsonScalarExpr(pstate, (JsonScalarExpr *) expr);
366 : 49 : break;
367 : :
368 : 39 : case T_JsonSerializeExpr:
369 : 39 : result = transformJsonSerializeExpr(pstate, (JsonSerializeExpr *) expr);
370 : 35 : break;
371 : :
24 372 : 1303 : case T_JsonFuncExpr:
373 : 1303 : result = transformJsonFuncExpr(pstate, (JsonFuncExpr *) expr);
374 : 1246 : break;
375 : :
9637 bruce@momjian.us 376 :UBC 0 : default:
377 : : /* should not reach here */
7575 tgl@sss.pgh.pa.us 378 [ # # ]: 0 : elog(ERROR, "unrecognized node type: %d", (int) nodeTag(expr));
379 : : result = NULL; /* keep compiler quiet */
380 : : break;
381 : : }
382 : :
9637 bruce@momjian.us 383 :CBC 2110856 : return result;
384 : : }
385 : :
386 : : /*
387 : : * helper routine for delivering "column does not exist" error message
388 : : *
389 : : * (Usually we don't have to work this hard, but the general case of field
390 : : * selection from an arbitrary node needs it.)
391 : : */
392 : : static void
2357 peter_e@gmx.net 393 : 19 : unknown_attribute(ParseState *pstate, Node *relref, const char *attname,
394 : : int location)
395 : : {
396 : : RangeTblEntry *rte;
397 : :
5279 tgl@sss.pgh.pa.us 398 [ + + ]: 19 : if (IsA(relref, Var) &&
399 [ - + ]: 6 : ((Var *) relref)->varattno == InvalidAttrNumber)
400 : : {
401 : : /* Reference the RTE by alias not by actual table name */
5279 tgl@sss.pgh.pa.us 402 :UBC 0 : rte = GetRTEByRangeTablePosn(pstate,
403 : : ((Var *) relref)->varno,
404 : 0 : ((Var *) relref)->varlevelsup);
405 [ # # ]: 0 : ereport(ERROR,
406 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
407 : : errmsg("column %s.%s does not exist",
408 : : rte->eref->aliasname, attname),
409 : : parser_errposition(pstate, location)));
410 : : }
411 : : else
412 : : {
413 : : /* Have to do it by reference to the type of the expression */
5279 tgl@sss.pgh.pa.us 414 :CBC 19 : Oid relTypeId = exprType(relref);
415 : :
416 [ + + ]: 19 : if (ISCOMPLEX(relTypeId))
417 [ + - ]: 9 : ereport(ERROR,
418 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
419 : : errmsg("column \"%s\" not found in data type %s",
420 : : attname, format_type_be(relTypeId)),
421 : : parser_errposition(pstate, location)));
422 [ + - ]: 10 : else if (relTypeId == RECORDOID)
423 [ + - ]: 10 : ereport(ERROR,
424 : : (errcode(ERRCODE_UNDEFINED_COLUMN),
425 : : errmsg("could not identify column \"%s\" in record data type",
426 : : attname),
427 : : parser_errposition(pstate, location)));
428 : : else
5279 tgl@sss.pgh.pa.us 429 [ # # ]:UBC 0 : ereport(ERROR,
430 : : (errcode(ERRCODE_WRONG_OBJECT_TYPE),
431 : : errmsg("column notation .%s applied to type %s, "
432 : : "which is not a composite type",
433 : : attname, format_type_be(relTypeId)),
434 : : parser_errposition(pstate, location)));
435 : : }
436 : : }
437 : :
438 : : static Node *
2497 tgl@sss.pgh.pa.us 439 :CBC 9726 : transformIndirection(ParseState *pstate, A_Indirection *ind)
440 : : {
441 : 9726 : Node *last_srf = pstate->p_last_srf;
442 : 9726 : Node *result = transformExprRecurse(pstate, ind->arg);
7249 443 : 9726 : List *subscripts = NIL;
2497 444 : 9726 : int location = exprLocation(result);
445 : : ListCell *i;
446 : :
447 : : /*
448 : : * We have to split any field-selection operations apart from
449 : : * subscripting. Adjacent A_Indices nodes have to be treated as a single
450 : : * multidimensional subscript operation.
451 : : */
452 [ + + + + : 19249 : foreach(i, ind->indirection)
+ + ]
453 : : {
7168 bruce@momjian.us 454 : 9542 : Node *n = lfirst(i);
455 : :
7249 tgl@sss.pgh.pa.us 456 [ + + ]: 9542 : if (IsA(n, A_Indices))
457 : 5098 : subscripts = lappend(subscripts, n);
5706 458 [ - + ]: 4444 : else if (IsA(n, A_Star))
459 : : {
5706 tgl@sss.pgh.pa.us 460 [ # # ]:UBC 0 : ereport(ERROR,
461 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
462 : : errmsg("row expansion via \"*\" is not supported here"),
463 : : parser_errposition(pstate, location)));
464 : : }
465 : : else
466 : : {
467 : : Node *newresult;
468 : :
7249 tgl@sss.pgh.pa.us 469 [ - + ]:CBC 4444 : Assert(IsA(n, String));
470 : :
471 : : /* process subscripts before this field selection */
472 [ + + ]: 4444 : if (subscripts)
1899 alvherre@alvh.no-ip. 473 : 68 : result = (Node *) transformContainerSubscripts(pstate,
474 : : result,
475 : : exprType(result),
476 : : exprTypmod(result),
477 : : subscripts,
478 : : false);
7249 tgl@sss.pgh.pa.us 479 : 4444 : subscripts = NIL;
480 : :
5279 481 : 4444 : newresult = ParseFuncOrColumn(pstate,
482 : 4444 : list_make1(n),
483 : 4444 : list_make1(result),
484 : : last_srf,
485 : : NULL,
486 : : false,
487 : : location);
488 [ + + ]: 4444 : if (newresult == NULL)
489 : 19 : unknown_attribute(pstate, result, strVal(n), location);
490 : 4425 : result = newresult;
491 : : }
492 : : }
493 : : /* process trailing subscripts, if any */
7249 494 [ + + ]: 9707 : if (subscripts)
1899 alvherre@alvh.no-ip. 495 : 4873 : result = (Node *) transformContainerSubscripts(pstate,
496 : : result,
497 : : exprType(result),
498 : : exprTypmod(result),
499 : : subscripts,
500 : : false);
501 : :
7249 tgl@sss.pgh.pa.us 502 : 9681 : return result;
503 : : }
504 : :
505 : : /*
506 : : * Transform a ColumnRef.
507 : : *
508 : : * If you find yourself changing this code, see also ExpandColumnRefStar.
509 : : */
510 : : static Node *
8060 511 : 741788 : transformColumnRef(ParseState *pstate, ColumnRef *cref)
512 : : {
5279 513 : 741788 : Node *node = NULL;
514 : 741788 : char *nspname = NULL;
515 : 741788 : char *relname = NULL;
516 : 741788 : char *colname = NULL;
517 : : ParseNamespaceItem *nsitem;
518 : : int levels_up;
519 : : enum
520 : : {
521 : : CRERR_NO_COLUMN,
522 : : CRERR_NO_RTE,
523 : : CRERR_WRONG_DB,
524 : : CRERR_TOO_MANY
525 : 741788 : } crerr = CRERR_NO_COLUMN;
526 : : const char *err;
527 : :
528 : : /*
529 : : * Check to see if the column reference is in an invalid place within the
530 : : * query. We allow column references in most places, except in default
531 : : * expressions and partition bound expressions.
532 : : */
1845 michael@paquier.xyz 533 : 741788 : err = NULL;
534 [ - + + + : 741788 : switch (pstate->p_expr_kind)
- ]
535 : : {
1845 michael@paquier.xyz 536 :UBC 0 : case EXPR_KIND_NONE:
537 : 0 : Assert(false); /* can't happen */
538 : : break;
1845 michael@paquier.xyz 539 :CBC 741746 : case EXPR_KIND_OTHER:
540 : : case EXPR_KIND_JOIN_ON:
541 : : case EXPR_KIND_JOIN_USING:
542 : : case EXPR_KIND_FROM_SUBSELECT:
543 : : case EXPR_KIND_FROM_FUNCTION:
544 : : case EXPR_KIND_WHERE:
545 : : case EXPR_KIND_POLICY:
546 : : case EXPR_KIND_HAVING:
547 : : case EXPR_KIND_FILTER:
548 : : case EXPR_KIND_WINDOW_PARTITION:
549 : : case EXPR_KIND_WINDOW_ORDER:
550 : : case EXPR_KIND_WINDOW_FRAME_RANGE:
551 : : case EXPR_KIND_WINDOW_FRAME_ROWS:
552 : : case EXPR_KIND_WINDOW_FRAME_GROUPS:
553 : : case EXPR_KIND_SELECT_TARGET:
554 : : case EXPR_KIND_INSERT_TARGET:
555 : : case EXPR_KIND_UPDATE_SOURCE:
556 : : case EXPR_KIND_UPDATE_TARGET:
557 : : case EXPR_KIND_MERGE_WHEN:
558 : : case EXPR_KIND_GROUP_BY:
559 : : case EXPR_KIND_ORDER_BY:
560 : : case EXPR_KIND_DISTINCT_ON:
561 : : case EXPR_KIND_LIMIT:
562 : : case EXPR_KIND_OFFSET:
563 : : case EXPR_KIND_RETURNING:
564 : : case EXPR_KIND_MERGE_RETURNING:
565 : : case EXPR_KIND_VALUES:
566 : : case EXPR_KIND_VALUES_SINGLE:
567 : : case EXPR_KIND_CHECK_CONSTRAINT:
568 : : case EXPR_KIND_DOMAIN_CHECK:
569 : : case EXPR_KIND_FUNCTION_DEFAULT:
570 : : case EXPR_KIND_INDEX_EXPRESSION:
571 : : case EXPR_KIND_INDEX_PREDICATE:
572 : : case EXPR_KIND_STATS_EXPRESSION:
573 : : case EXPR_KIND_ALTER_COL_TRANSFORM:
574 : : case EXPR_KIND_EXECUTE_PARAMETER:
575 : : case EXPR_KIND_TRIGGER_WHEN:
576 : : case EXPR_KIND_PARTITION_EXPRESSION:
577 : : case EXPR_KIND_CALL_ARGUMENT:
578 : : case EXPR_KIND_COPY_WHERE:
579 : : case EXPR_KIND_GENERATED_COLUMN:
580 : : case EXPR_KIND_CYCLE_MARK:
581 : : /* okay */
582 : 741746 : break;
583 : :
584 : 12 : case EXPR_KIND_COLUMN_DEFAULT:
585 : 12 : err = _("cannot use column reference in DEFAULT expression");
586 : 12 : break;
587 : 30 : case EXPR_KIND_PARTITION_BOUND:
588 : 30 : err = _("cannot use column reference in partition bound expression");
589 : 30 : break;
590 : :
591 : : /*
592 : : * There is intentionally no default: case here, so that the
593 : : * compiler will warn if we add a new ParseExprKind without
594 : : * extending this switch. If we do see an unrecognized value at
595 : : * runtime, the behavior will be the same as for EXPR_KIND_OTHER,
596 : : * which is sane anyway.
597 : : */
598 : : }
599 [ + + ]: 741788 : if (err)
600 [ + - ]: 42 : ereport(ERROR,
601 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
602 : : errmsg_internal("%s", err),
603 : : parser_errposition(pstate, cref->location)));
604 : :
605 : : /*
606 : : * Give the PreParseColumnRefHook, if any, first shot. If it returns
607 : : * non-null then that's all, folks.
608 : : */
5279 tgl@sss.pgh.pa.us 609 [ + + ]: 741746 : if (pstate->p_pre_columnref_hook != NULL)
610 : : {
2411 peter_e@gmx.net 611 : 21455 : node = pstate->p_pre_columnref_hook(pstate, cref);
5279 tgl@sss.pgh.pa.us 612 [ + + ]: 21455 : if (node != NULL)
613 : 352 : return node;
614 : : }
615 : :
616 : : /*----------
617 : : * The allowed syntaxes are:
618 : : *
619 : : * A First try to resolve as unqualified column name;
620 : : * if no luck, try to resolve as unqualified table name (A.*).
621 : : * A.B A is an unqualified table name; B is either a
622 : : * column or function name (trying column name first).
623 : : * A.B.C schema A, table B, col or func name C.
624 : : * A.B.C.D catalog A, schema B, table C, col or func D.
625 : : * A.* A is an unqualified table name; means whole-row value.
626 : : * A.B.* whole-row value of table B in schema A.
627 : : * A.B.C.* whole-row value of table C in schema B in catalog A.
628 : : *
629 : : * We do not need to cope with bare "*"; that will only be accepted by
630 : : * the grammar at the top level of a SELECT list, and transformTargetList
631 : : * will take care of it before it ever gets here. Also, "A.*" etc will
632 : : * be expanded by transformTargetList if they appear at SELECT top level,
633 : : * so here we are only going to see them as function or operator inputs.
634 : : *
635 : : * Currently, if a catalog name is given then it must equal the current
636 : : * database name; we check it here and then discard it.
637 : : *----------
638 : : */
639 [ + + + - : 741394 : switch (list_length(cref->fields))
- ]
640 : : {
8060 641 : 287210 : case 1:
642 : : {
5706 643 : 287210 : Node *field1 = (Node *) linitial(cref->fields);
644 : :
5279 645 : 287210 : colname = strVal(field1);
646 : :
647 : : /* Try to identify as an unqualified column */
648 : 287210 : node = colNameToVar(pstate, colname, false, cref->location);
649 : :
7893 bruce@momjian.us 650 [ + + ]: 287180 : if (node == NULL)
651 : : {
652 : : /*
653 : : * Not known as a column of any range-table entry.
654 : : *
655 : : * Try to find the name as a relation. Note that only
656 : : * relations already entered into the rangetable will be
657 : : * recognized.
658 : : *
659 : : * This is a hack for backwards compatibility with
660 : : * PostQUEL-inspired syntax. The preferred form now is
661 : : * "rel.*".
662 : : */
1571 tgl@sss.pgh.pa.us 663 : 19880 : nsitem = refnameNamespaceItem(pstate, NULL, colname,
664 : : cref->location,
665 : : &levels_up);
666 [ + + ]: 19880 : if (nsitem)
667 : 2855 : node = transformWholeRowRef(pstate, nsitem, levels_up,
668 : : cref->location);
669 : : }
7893 bruce@momjian.us 670 : 287180 : break;
671 : : }
672 : 454145 : case 2:
673 : : {
5706 tgl@sss.pgh.pa.us 674 : 454145 : Node *field1 = (Node *) linitial(cref->fields);
675 : 454145 : Node *field2 = (Node *) lsecond(cref->fields);
676 : :
5279 677 : 454145 : relname = strVal(field1);
678 : :
679 : : /* Locate the referenced nsitem */
1571 680 : 454145 : nsitem = refnameNamespaceItem(pstate, nspname, relname,
681 : : cref->location,
682 : : &levels_up);
683 [ + + ]: 454133 : if (nsitem == NULL)
684 : : {
5279 685 : 2587 : crerr = CRERR_NO_RTE;
686 : 2587 : break;
687 : : }
688 : :
689 : : /* Whole-row reference? */
5706 690 [ + + ]: 451546 : if (IsA(field2, A_Star))
691 : : {
1571 692 : 670 : node = transformWholeRowRef(pstate, nsitem, levels_up,
693 : : cref->location);
7893 bruce@momjian.us 694 : 670 : break;
695 : : }
696 : :
5279 tgl@sss.pgh.pa.us 697 : 450876 : colname = strVal(field2);
698 : :
699 : : /* Try to identify as a column of the nsitem */
1571 700 : 450876 : node = scanNSItemForColumn(pstate, nsitem, levels_up, colname,
701 : : cref->location);
7893 bruce@momjian.us 702 [ + + ]: 450873 : if (node == NULL)
703 : : {
704 : : /* Try it as a function call on the whole row */
1571 tgl@sss.pgh.pa.us 705 : 81 : node = transformWholeRowRef(pstate, nsitem, levels_up,
706 : : cref->location);
7893 bruce@momjian.us 707 : 81 : node = ParseFuncOrColumn(pstate,
5279 tgl@sss.pgh.pa.us 708 : 81 : list_make1(makeString(colname)),
7259 neilc@samurai.com 709 : 81 : list_make1(node),
710 : : pstate->p_last_srf,
711 : : NULL,
712 : : false,
713 : : cref->location);
714 : : }
8060 tgl@sss.pgh.pa.us 715 : 450873 : break;
716 : : }
717 : 39 : case 3:
718 : : {
5706 719 : 39 : Node *field1 = (Node *) linitial(cref->fields);
720 : 39 : Node *field2 = (Node *) lsecond(cref->fields);
721 : 39 : Node *field3 = (Node *) lthird(cref->fields);
722 : :
5279 723 : 39 : nspname = strVal(field1);
724 : 39 : relname = strVal(field2);
725 : :
726 : : /* Locate the referenced nsitem */
1571 727 : 39 : nsitem = refnameNamespaceItem(pstate, nspname, relname,
728 : : cref->location,
729 : : &levels_up);
730 [ + + ]: 39 : if (nsitem == NULL)
731 : : {
5279 732 : 33 : crerr = CRERR_NO_RTE;
733 : 33 : break;
734 : : }
735 : :
736 : : /* Whole-row reference? */
5706 737 [ + + ]: 6 : if (IsA(field3, A_Star))
738 : : {
1571 739 : 3 : node = transformWholeRowRef(pstate, nsitem, levels_up,
740 : : cref->location);
7893 bruce@momjian.us 741 : 3 : break;
742 : : }
743 : :
5279 tgl@sss.pgh.pa.us 744 : 3 : colname = strVal(field3);
745 : :
746 : : /* Try to identify as a column of the nsitem */
1571 747 : 3 : node = scanNSItemForColumn(pstate, nsitem, levels_up, colname,
748 : : cref->location);
7893 bruce@momjian.us 749 [ - + ]: 3 : if (node == NULL)
750 : : {
751 : : /* Try it as a function call on the whole row */
1571 tgl@sss.pgh.pa.us 752 :UBC 0 : node = transformWholeRowRef(pstate, nsitem, levels_up,
753 : : cref->location);
7893 bruce@momjian.us 754 : 0 : node = ParseFuncOrColumn(pstate,
5279 tgl@sss.pgh.pa.us 755 : 0 : list_make1(makeString(colname)),
7259 neilc@samurai.com 756 : 0 : list_make1(node),
757 : : pstate->p_last_srf,
758 : : NULL,
759 : : false,
760 : : cref->location);
761 : : }
7893 bruce@momjian.us 762 :CBC 3 : break;
763 : : }
8060 tgl@sss.pgh.pa.us 764 :UBC 0 : case 4:
765 : : {
5706 766 : 0 : Node *field1 = (Node *) linitial(cref->fields);
767 : 0 : Node *field2 = (Node *) lsecond(cref->fields);
768 : 0 : Node *field3 = (Node *) lthird(cref->fields);
769 : 0 : Node *field4 = (Node *) lfourth(cref->fields);
770 : : char *catname;
771 : :
5279 772 : 0 : catname = strVal(field1);
773 : 0 : nspname = strVal(field2);
774 : 0 : relname = strVal(field3);
775 : :
776 : : /*
777 : : * We check the catalog name and then ignore it.
778 : : */
779 [ # # ]: 0 : if (strcmp(catname, get_database_name(MyDatabaseId)) != 0)
780 : : {
781 : 0 : crerr = CRERR_WRONG_DB;
782 : 0 : break;
783 : : }
784 : :
785 : : /* Locate the referenced nsitem */
1571 786 : 0 : nsitem = refnameNamespaceItem(pstate, nspname, relname,
787 : : cref->location,
788 : : &levels_up);
789 [ # # ]: 0 : if (nsitem == NULL)
790 : : {
5279 791 : 0 : crerr = CRERR_NO_RTE;
792 : 0 : break;
793 : : }
794 : :
795 : : /* Whole-row reference? */
5706 796 [ # # ]: 0 : if (IsA(field4, A_Star))
797 : : {
1571 798 : 0 : node = transformWholeRowRef(pstate, nsitem, levels_up,
799 : : cref->location);
7893 bruce@momjian.us 800 : 0 : break;
801 : : }
802 : :
5279 tgl@sss.pgh.pa.us 803 : 0 : colname = strVal(field4);
804 : :
805 : : /* Try to identify as a column of the nsitem */
1571 806 : 0 : node = scanNSItemForColumn(pstate, nsitem, levels_up, colname,
807 : : cref->location);
7893 bruce@momjian.us 808 [ # # ]: 0 : if (node == NULL)
809 : : {
810 : : /* Try it as a function call on the whole row */
1571 tgl@sss.pgh.pa.us 811 : 0 : node = transformWholeRowRef(pstate, nsitem, levels_up,
812 : : cref->location);
7893 bruce@momjian.us 813 : 0 : node = ParseFuncOrColumn(pstate,
5279 tgl@sss.pgh.pa.us 814 : 0 : list_make1(makeString(colname)),
7259 neilc@samurai.com 815 : 0 : list_make1(node),
816 : : pstate->p_last_srf,
817 : : NULL,
818 : : false,
819 : : cref->location);
820 : : }
7893 bruce@momjian.us 821 : 0 : break;
822 : : }
8060 tgl@sss.pgh.pa.us 823 : 0 : default:
2489 824 : 0 : crerr = CRERR_TOO_MANY; /* too many dotted names */
8060 825 : 0 : break;
826 : : }
827 : :
828 : : /*
829 : : * Now give the PostParseColumnRefHook, if any, a chance. We pass the
830 : : * translation-so-far so that it can throw an error if it wishes in the
831 : : * case that it has a conflicting interpretation of the ColumnRef. (If it
832 : : * just translates anyway, we'll throw an error, because we can't undo
833 : : * whatever effects the preceding steps may have had on the pstate.) If it
834 : : * returns NULL, use the standard translation, or throw a suitable error
835 : : * if there is none.
836 : : */
5279 tgl@sss.pgh.pa.us 837 [ + + ]:CBC 741349 : if (pstate->p_post_columnref_hook != NULL)
838 : : {
839 : : Node *hookresult;
840 : :
2411 peter_e@gmx.net 841 : 26856 : hookresult = pstate->p_post_columnref_hook(pstate, cref, node);
5279 tgl@sss.pgh.pa.us 842 [ + + ]: 26842 : if (node == NULL)
843 : 19455 : node = hookresult;
844 [ - + ]: 7387 : else if (hookresult != NULL)
7025 neilc@samurai.com 845 [ # # ]:UBC 0 : ereport(ERROR,
846 : : (errcode(ERRCODE_AMBIGUOUS_COLUMN),
847 : : errmsg("column reference \"%s\" is ambiguous",
848 : : NameListToString(cref->fields)),
849 : : parser_errposition(pstate, cref->location)));
850 : : }
851 : :
852 : : /*
853 : : * Throw error if no translation found.
854 : : */
5279 tgl@sss.pgh.pa.us 855 [ + + ]:CBC 741335 : if (node == NULL)
856 : : {
857 [ + + - - : 227 : switch (crerr)
- ]
858 : : {
859 : 173 : case CRERR_NO_COLUMN:
4268 860 : 173 : errorMissingColumn(pstate, relname, colname, cref->location);
861 : : break;
5279 862 : 54 : case CRERR_NO_RTE:
863 : 54 : errorMissingRTE(pstate, makeRangeVar(nspname, relname,
864 : : cref->location));
865 : : break;
5279 tgl@sss.pgh.pa.us 866 :UBC 0 : case CRERR_WRONG_DB:
867 [ # # ]: 0 : ereport(ERROR,
868 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
869 : : errmsg("cross-database references are not implemented: %s",
870 : : NameListToString(cref->fields)),
871 : : parser_errposition(pstate, cref->location)));
872 : : break;
873 : 0 : case CRERR_TOO_MANY:
874 [ # # ]: 0 : ereport(ERROR,
875 : : (errcode(ERRCODE_SYNTAX_ERROR),
876 : : errmsg("improper qualified name (too many dotted names): %s",
877 : : NameListToString(cref->fields)),
878 : : parser_errposition(pstate, cref->location)));
879 : : break;
880 : : }
881 : : }
882 : :
5279 tgl@sss.pgh.pa.us 883 :CBC 741108 : return node;
884 : : }
885 : :
886 : : static Node *
6152 887 : 70654 : transformParamRef(ParseState *pstate, ParamRef *pref)
888 : : {
889 : : Node *result;
890 : :
891 : : /*
892 : : * The core parser knows nothing about Params. If a hook is supplied,
893 : : * call it. If not, or if the hook returns NULL, throw a generic error.
894 : : */
5279 895 [ + - ]: 70654 : if (pstate->p_paramref_hook != NULL)
2411 peter_e@gmx.net 896 : 70654 : result = pstate->p_paramref_hook(pstate, pref);
897 : : else
5279 tgl@sss.pgh.pa.us 898 :UBC 0 : result = NULL;
899 : :
5279 tgl@sss.pgh.pa.us 900 [ + + ]:CBC 70654 : if (result == NULL)
901 [ + - ]: 3 : ereport(ERROR,
902 : : (errcode(ERRCODE_UNDEFINED_PARAMETER),
903 : : errmsg("there is no parameter $%d", pref->number),
904 : : parser_errposition(pstate, pref->location)));
905 : :
906 : 70651 : return result;
907 : : }
908 : :
909 : : /* Test whether an a_expr is a plain NULL constant or not */
910 : : static bool
6012 911 : 913 : exprIsNullConstant(Node *arg)
912 : : {
913 [ + - + + ]: 913 : if (arg && IsA(arg, A_Const))
914 : : {
5995 bruce@momjian.us 915 : 54 : A_Const *con = (A_Const *) arg;
916 : :
948 peter@eisentraut.org 917 [ + + ]: 54 : if (con->isnull)
6012 tgl@sss.pgh.pa.us 918 : 15 : return true;
919 : : }
920 : 898 : return false;
921 : : }
922 : :
923 : : static Node *
7025 neilc@samurai.com 924 : 275368 : transformAExprOp(ParseState *pstate, A_Expr *a)
925 : : {
926 : 275368 : Node *lexpr = a->lexpr;
927 : 275368 : Node *rexpr = a->rexpr;
928 : : Node *result;
929 : :
930 : : /*
931 : : * Special-case "foo = NULL" and "NULL = foo" for compatibility with
932 : : * standards-broken products (like Microsoft's). Turn these into IS NULL
933 : : * exprs. (If either side is a CaseTestExpr, then the expression was
934 : : * generated internally from a CASE-WHEN expression, and
935 : : * transform_null_equals does not apply.)
936 : : */
937 [ - + - - ]: 275368 : if (Transform_null_equals &&
7025 neilc@samurai.com 938 :UBC 0 : list_length(a->name) == 1 &&
939 [ # # # # ]: 0 : strcmp(strVal(linitial(a->name)), "=") == 0 &&
4572 heikki.linnakangas@i 940 [ # # ]: 0 : (exprIsNullConstant(lexpr) || exprIsNullConstant(rexpr)) &&
1429 tgl@sss.pgh.pa.us 941 [ # # # # ]: 0 : (!IsA(lexpr, CaseTestExpr) && !IsA(rexpr, CaseTestExpr)))
7025 neilc@samurai.com 942 : 0 : {
943 : 0 : NullTest *n = makeNode(NullTest);
944 : :
945 : 0 : n->nulltesttype = IS_NULL;
3339 tgl@sss.pgh.pa.us 946 : 0 : n->location = a->location;
947 : :
7025 neilc@samurai.com 948 [ # # ]: 0 : if (exprIsNullConstant(lexpr))
949 : 0 : n->arg = (Expr *) rexpr;
950 : : else
951 : 0 : n->arg = (Expr *) lexpr;
952 : :
4265 tgl@sss.pgh.pa.us 953 : 0 : result = transformExprRecurse(pstate, (Node *) n);
954 : : }
7025 neilc@samurai.com 955 [ + + + + :CBC 275368 : else if (lexpr && IsA(lexpr, RowExpr) &&
+ - ]
956 [ + + ]: 305 : rexpr && IsA(rexpr, SubLink) &&
7025 neilc@samurai.com 957 [ + - ]:GBC 15 : ((SubLink *) rexpr)->subLinkType == EXPR_SUBLINK)
958 : 15 : {
959 : : /*
960 : : * Convert "row op subselect" into a ROWCOMPARE sublink. Formerly the
961 : : * grammar did this, but now that a row construct is allowed anywhere
962 : : * in expressions, it's easier to do it here.
963 : : */
964 : 15 : SubLink *s = (SubLink *) rexpr;
965 : :
6682 tgl@sss.pgh.pa.us 966 : 15 : s->subLinkType = ROWCOMPARE_SUBLINK;
967 : 15 : s->testexpr = lexpr;
7025 neilc@samurai.com 968 : 15 : s->operName = a->name;
5708 tgl@sss.pgh.pa.us 969 : 15 : s->location = a->location;
4265 970 : 15 : result = transformExprRecurse(pstate, (Node *) s);
971 : : }
7025 neilc@samurai.com 972 [ + + + + :CBC 275353 : else if (lexpr && IsA(lexpr, RowExpr) &&
+ - ]
973 [ + + ]: 290 : rexpr && IsA(rexpr, RowExpr))
974 : : {
975 : : /* ROW() op ROW() is handled specially */
4265 tgl@sss.pgh.pa.us 976 : 287 : lexpr = transformExprRecurse(pstate, lexpr);
977 : 287 : rexpr = transformExprRecurse(pstate, rexpr);
978 : :
6682 979 : 287 : result = make_row_comparison_op(pstate,
980 : : a->name,
2609 peter_e@gmx.net 981 : 287 : castNode(RowExpr, lexpr)->args,
982 : 287 : castNode(RowExpr, rexpr)->args,
983 : : a->location);
984 : : }
985 : : else
986 : : {
987 : : /* Ordinary scalar operator */
2497 tgl@sss.pgh.pa.us 988 : 275066 : Node *last_srf = pstate->p_last_srf;
989 : :
4265 990 : 275066 : lexpr = transformExprRecurse(pstate, lexpr);
991 : 274951 : rexpr = transformExprRecurse(pstate, rexpr);
992 : :
7025 neilc@samurai.com 993 : 274908 : result = (Node *) make_op(pstate,
994 : : a->name,
995 : : lexpr,
996 : : rexpr,
997 : : last_srf,
998 : : a->location);
999 : : }
1000 : :
1001 : 275157 : return result;
1002 : : }
1003 : :
1004 : : static Node *
1005 : 7105 : transformAExprOpAny(ParseState *pstate, A_Expr *a)
1006 : : {
1223 tgl@sss.pgh.pa.us 1007 : 7105 : Node *lexpr = transformExprRecurse(pstate, a->lexpr);
1008 : 7105 : Node *rexpr = transformExprRecurse(pstate, a->rexpr);
1009 : :
7025 neilc@samurai.com 1010 : 7105 : return (Node *) make_scalar_array_op(pstate,
1011 : : a->name,
1012 : : true,
1013 : : lexpr,
1014 : : rexpr,
1015 : : a->location);
1016 : : }
1017 : :
1018 : : static Node *
1019 : 150 : transformAExprOpAll(ParseState *pstate, A_Expr *a)
1020 : : {
1223 tgl@sss.pgh.pa.us 1021 : 150 : Node *lexpr = transformExprRecurse(pstate, a->lexpr);
1022 : 150 : Node *rexpr = transformExprRecurse(pstate, a->rexpr);
1023 : :
7025 neilc@samurai.com 1024 : 150 : return (Node *) make_scalar_array_op(pstate,
1025 : : a->name,
1026 : : false,
1027 : : lexpr,
1028 : : rexpr,
1029 : : a->location);
1030 : : }
1031 : :
1032 : : static Node *
1033 : 464 : transformAExprDistinct(ParseState *pstate, A_Expr *a)
1034 : : {
3322 tgl@sss.pgh.pa.us 1035 : 464 : Node *lexpr = a->lexpr;
1036 : 464 : Node *rexpr = a->rexpr;
1037 : : Node *result;
1038 : :
1039 : : /*
1040 : : * If either input is an undecorated NULL literal, transform to a NullTest
1041 : : * on the other input. That's simpler to process than a full DistinctExpr,
1042 : : * and it avoids needing to require that the datatype have an = operator.
1043 : : */
2817 1044 [ + + ]: 464 : if (exprIsNullConstant(rexpr))
1045 : 15 : return make_nulltest_from_distinct(pstate, a, lexpr);
1046 [ - + ]: 449 : if (exprIsNullConstant(lexpr))
2817 tgl@sss.pgh.pa.us 1047 :UBC 0 : return make_nulltest_from_distinct(pstate, a, rexpr);
1048 : :
3322 tgl@sss.pgh.pa.us 1049 :CBC 449 : lexpr = transformExprRecurse(pstate, lexpr);
1050 : 449 : rexpr = transformExprRecurse(pstate, rexpr);
1051 : :
7025 neilc@samurai.com 1052 [ + - + + : 449 : if (lexpr && IsA(lexpr, RowExpr) &&
+ - ]
1053 [ + - ]: 3 : rexpr && IsA(rexpr, RowExpr))
1054 : : {
1055 : : /* ROW() op ROW() is handled specially */
2817 tgl@sss.pgh.pa.us 1056 : 3 : result = make_row_distinct_op(pstate, a->name,
1057 : : (RowExpr *) lexpr,
1058 : : (RowExpr *) rexpr,
1059 : : a->location);
1060 : : }
1061 : : else
1062 : : {
1063 : : /* Ordinary scalar operator */
1064 : 446 : result = (Node *) make_distinct_op(pstate,
1065 : : a->name,
1066 : : lexpr,
1067 : : rexpr,
1068 : : a->location);
1069 : : }
1070 : :
1071 : : /*
1072 : : * If it's NOT DISTINCT, we first build a DistinctExpr and then stick a
1073 : : * NOT on top.
1074 : : */
1075 [ + + ]: 449 : if (a->kind == AEXPR_NOT_DISTINCT)
1076 : 25 : result = (Node *) makeBoolExpr(NOT_EXPR,
1077 : 25 : list_make1(result),
1078 : : a->location);
1079 : :
1080 : 449 : return result;
1081 : : }
1082 : :
1083 : : static Node *
7025 neilc@samurai.com 1084 : 119 : transformAExprNullIf(ParseState *pstate, A_Expr *a)
1085 : : {
4265 tgl@sss.pgh.pa.us 1086 : 119 : Node *lexpr = transformExprRecurse(pstate, a->lexpr);
1087 : 119 : Node *rexpr = transformExprRecurse(pstate, a->rexpr);
1088 : : OpExpr *result;
1089 : :
4775 1090 : 119 : result = (OpExpr *) make_op(pstate,
1091 : : a->name,
1092 : : lexpr,
1093 : : rexpr,
1094 : : pstate->p_last_srf,
1095 : : a->location);
1096 : :
1097 : : /*
1098 : : * The comparison operator itself should yield boolean ...
1099 : : */
1100 [ - + ]: 119 : if (result->opresulttype != BOOLOID)
7025 neilc@samurai.com 1101 [ # # ]:UBC 0 : ereport(ERROR,
1102 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1103 : : errmsg("NULLIF requires = operator to yield boolean"),
1104 : : parser_errposition(pstate, a->location)));
2497 tgl@sss.pgh.pa.us 1105 [ - + ]:CBC 119 : if (result->opretset)
2497 tgl@sss.pgh.pa.us 1106 [ # # ]:UBC 0 : ereport(ERROR,
1107 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
1108 : : /* translator: %s is name of a SQL construct, eg NULLIF */
1109 : : errmsg("%s must not return a set", "NULLIF"),
1110 : : parser_errposition(pstate, a->location)));
1111 : :
1112 : : /*
1113 : : * ... but the NullIfExpr will yield the first operand's type.
1114 : : */
4775 tgl@sss.pgh.pa.us 1115 :CBC 119 : result->opresulttype = exprType((Node *) linitial(result->args));
1116 : :
1117 : : /*
1118 : : * We rely on NullIfExpr and OpExpr being the same struct
1119 : : */
7025 neilc@samurai.com 1120 : 119 : NodeSetTag(result, T_NullIfExpr);
1121 : :
4775 tgl@sss.pgh.pa.us 1122 : 119 : return (Node *) result;
1123 : : }
1124 : :
1125 : : static Node *
6712 1126 : 8396 : transformAExprIn(ParseState *pstate, A_Expr *a)
1127 : : {
5649 1128 : 8396 : Node *result = NULL;
1129 : : Node *lexpr;
1130 : : List *rexprs;
1131 : : List *rvars;
1132 : : List *rnonvars;
1133 : : bool useOr;
1134 : : ListCell *l;
1135 : :
1136 : : /*
1137 : : * If the operator is <>, combine with AND not OR.
1138 : : */
6712 1139 [ + + ]: 8396 : if (strcmp(strVal(linitial(a->name)), "<>") == 0)
1140 : 1052 : useOr = false;
1141 : : else
1142 : 7344 : useOr = true;
1143 : :
1144 : : /*
1145 : : * We try to generate a ScalarArrayOpExpr from IN/NOT IN, but this is only
1146 : : * possible if there is a suitable array type available. If not, we fall
1147 : : * back to a boolean condition tree with multiple copies of the lefthand
1148 : : * expression. Also, any IN-list items that contain Vars are handled as
1149 : : * separate boolean conditions, because that gives the planner more scope
1150 : : * for optimization on such clauses.
1151 : : *
1152 : : * First step: transform all the inputs, and detect whether any contain
1153 : : * Vars.
1154 : : */
4265 1155 : 8396 : lexpr = transformExprRecurse(pstate, a->lexpr);
5649 1156 : 8396 : rexprs = rvars = rnonvars = NIL;
6712 1157 [ + - + + : 31689 : foreach(l, (List *) a->rexpr)
+ + ]
1158 : : {
4265 1159 : 23296 : Node *rexpr = transformExprRecurse(pstate, lfirst(l));
1160 : :
6712 1161 : 23293 : rexprs = lappend(rexprs, rexpr);
5649 1162 [ - + ]: 23293 : if (contain_vars_of_level(rexpr, 0))
5649 tgl@sss.pgh.pa.us 1163 :UBC 0 : rvars = lappend(rvars, rexpr);
1164 : : else
5649 tgl@sss.pgh.pa.us 1165 :CBC 23293 : rnonvars = lappend(rnonvars, rexpr);
1166 : : }
1167 : :
1168 : : /*
1169 : : * ScalarArrayOpExpr is only going to be useful if there's more than one
1170 : : * non-Var righthand item.
1171 : : */
3962 1172 [ + + ]: 8393 : if (list_length(rnonvars) > 1)
1173 : : {
1174 : : List *allexprs;
1175 : : Oid scalar_type;
1176 : : Oid array_type;
1177 : :
1178 : : /*
1179 : : * Try to select a common type for the array elements. Note that
1180 : : * since the LHS' type is first in the list, it will be preferred when
1181 : : * there is doubt (eg, when all the RHS items are unknown literals).
1182 : : *
1183 : : * Note: use list_concat here not lcons, to avoid damaging rnonvars.
1184 : : */
5649 1185 : 7358 : allexprs = list_concat(list_make1(lexpr), rnonvars);
5650 1186 : 7358 : scalar_type = select_common_type(pstate, allexprs, NULL, NULL);
1187 : :
1188 : : /* We have to verify that the selected type actually works */
806 1189 [ + - ]: 7358 : if (OidIsValid(scalar_type) &&
1190 [ + + ]: 7358 : !verify_common_type(scalar_type, allexprs))
1191 : 3 : scalar_type = InvalidOid;
1192 : :
1193 : : /*
1194 : : * Do we have an array type to use? Aside from the case where there
1195 : : * isn't one, we don't risk using ScalarArrayOpExpr when the common
1196 : : * type is RECORD, because the RowExpr comparison logic below can cope
1197 : : * with some cases of non-identical row types.
1198 : : */
3962 1199 [ + + + + ]: 7358 : if (OidIsValid(scalar_type) && scalar_type != RECORDOID)
5650 1200 : 7346 : array_type = get_array_type(scalar_type);
1201 : : else
1202 : 12 : array_type = InvalidOid;
6712 1203 [ + + ]: 7358 : if (array_type != InvalidOid)
1204 : : {
1205 : : /*
1206 : : * OK: coerce all the right-hand non-Var inputs to the common type
1207 : : * and build an ArrayExpr for them.
1208 : : */
1209 : : List *aexprs;
1210 : : ArrayExpr *newa;
1211 : :
1212 : 7340 : aexprs = NIL;
5649 1213 [ + - + + : 29562 : foreach(l, rnonvars)
+ + ]
1214 : : {
6712 1215 : 22222 : Node *rexpr = (Node *) lfirst(l);
1216 : :
1217 : 22222 : rexpr = coerce_to_common_type(pstate, rexpr,
1218 : : scalar_type,
1219 : : "IN");
1220 : 22222 : aexprs = lappend(aexprs, rexpr);
1221 : : }
1222 : 7340 : newa = makeNode(ArrayExpr);
1223 : 7340 : newa->array_typeid = array_type;
1224 : : /* array_collid will be set by parse_collate.c */
1225 : 7340 : newa->element_typeid = scalar_type;
1226 : 7340 : newa->elements = aexprs;
1227 : 7340 : newa->multidims = false;
5708 1228 : 7340 : newa->location = -1;
1229 : :
5649 1230 : 7340 : result = (Node *) make_scalar_array_op(pstate,
1231 : : a->name,
1232 : : useOr,
1233 : : lexpr,
1234 : : (Node *) newa,
1235 : : a->location);
1236 : :
1237 : : /* Consider only the Vars (if any) in the loop below */
1238 : 7340 : rexprs = rvars;
1239 : : }
1240 : : }
1241 : :
1242 : : /*
1243 : : * Must do it the hard way, ie, with a boolean expression tree.
1244 : : */
6712 1245 [ + + + + : 9458 : foreach(l, rexprs)
+ + ]
1246 : : {
1247 : 1068 : Node *rexpr = (Node *) lfirst(l);
1248 : : Node *cmp;
1249 : :
3962 1250 [ + + ]: 1068 : if (IsA(lexpr, RowExpr) &&
1251 [ + - ]: 18 : IsA(rexpr, RowExpr))
1252 : : {
1253 : : /* ROW() op ROW() is handled specially */
6682 1254 : 18 : cmp = make_row_comparison_op(pstate,
1255 : : a->name,
2489 1256 : 18 : copyObject(((RowExpr *) lexpr)->args),
1257 : : ((RowExpr *) rexpr)->args,
1258 : : a->location);
1259 : : }
1260 : : else
1261 : : {
1262 : : /* Ordinary scalar operator */
6712 1263 : 1050 : cmp = (Node *) make_op(pstate,
1264 : : a->name,
1265 : 1050 : copyObject(lexpr),
1266 : : rexpr,
1267 : : pstate->p_last_srf,
1268 : : a->location);
1269 : : }
1270 : :
1271 : 1065 : cmp = coerce_to_boolean(pstate, cmp, "IN");
1272 [ + + ]: 1065 : if (result == NULL)
1273 : 1050 : result = cmp;
1274 : : else
1275 : 15 : result = (Node *) makeBoolExpr(useOr ? OR_EXPR : AND_EXPR,
5708 1276 : 15 : list_make2(result, cmp),
1277 : : a->location);
1278 : : }
1279 : :
6712 1280 : 8390 : return result;
1281 : : }
1282 : :
1283 : : static Node *
3339 1284 : 255 : transformAExprBetween(ParseState *pstate, A_Expr *a)
1285 : : {
1286 : : Node *aexpr;
1287 : : Node *bexpr;
1288 : : Node *cexpr;
1289 : : Node *result;
1290 : : Node *sub1;
1291 : : Node *sub2;
1292 : : List *args;
1293 : :
1294 : : /* Deconstruct A_Expr into three subexprs */
1295 : 255 : aexpr = a->lexpr;
2609 peter_e@gmx.net 1296 : 255 : args = castNode(List, a->rexpr);
3339 tgl@sss.pgh.pa.us 1297 [ - + ]: 255 : Assert(list_length(args) == 2);
1298 : 255 : bexpr = (Node *) linitial(args);
1299 : 255 : cexpr = (Node *) lsecond(args);
1300 : :
1301 : : /*
1302 : : * Build the equivalent comparison expression. Make copies of
1303 : : * multiply-referenced subexpressions for safety. (XXX this is really
1304 : : * wrong since it results in multiple runtime evaluations of what may be
1305 : : * volatile expressions ...)
1306 : : *
1307 : : * Ideally we would not use hard-wired operators here but instead use
1308 : : * opclasses. However, mixed data types and other issues make this
1309 : : * difficult:
1310 : : * http://archives.postgresql.org/pgsql-hackers/2008-08/msg01142.php
1311 : : */
1312 [ + + + + : 255 : switch (a->kind)
- ]
1313 : : {
1314 : 237 : case AEXPR_BETWEEN:
1315 : 237 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, ">=",
1316 : : aexpr, bexpr,
1317 : : a->location),
1318 : : makeSimpleA_Expr(AEXPR_OP, "<=",
1319 : : copyObject(aexpr), cexpr,
1320 : : a->location));
1321 : 237 : result = (Node *) makeBoolExpr(AND_EXPR, args, a->location);
1322 : 237 : break;
1323 : 6 : case AEXPR_NOT_BETWEEN:
1324 : 6 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, "<",
1325 : : aexpr, bexpr,
1326 : : a->location),
1327 : : makeSimpleA_Expr(AEXPR_OP, ">",
1328 : : copyObject(aexpr), cexpr,
1329 : : a->location));
1330 : 6 : result = (Node *) makeBoolExpr(OR_EXPR, args, a->location);
1331 : 6 : break;
1332 : 6 : case AEXPR_BETWEEN_SYM:
1333 : 6 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, ">=",
1334 : : aexpr, bexpr,
1335 : : a->location),
1336 : : makeSimpleA_Expr(AEXPR_OP, "<=",
1337 : : copyObject(aexpr), cexpr,
1338 : : a->location));
1339 : 6 : sub1 = (Node *) makeBoolExpr(AND_EXPR, args, a->location);
1340 : 6 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, ">=",
1341 : : copyObject(aexpr), copyObject(cexpr),
1342 : : a->location),
1343 : : makeSimpleA_Expr(AEXPR_OP, "<=",
1344 : : copyObject(aexpr), copyObject(bexpr),
1345 : : a->location));
1346 : 6 : sub2 = (Node *) makeBoolExpr(AND_EXPR, args, a->location);
1347 : 6 : args = list_make2(sub1, sub2);
1348 : 6 : result = (Node *) makeBoolExpr(OR_EXPR, args, a->location);
1349 : 6 : break;
1350 : 6 : case AEXPR_NOT_BETWEEN_SYM:
1351 : 6 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, "<",
1352 : : aexpr, bexpr,
1353 : : a->location),
1354 : : makeSimpleA_Expr(AEXPR_OP, ">",
1355 : : copyObject(aexpr), cexpr,
1356 : : a->location));
1357 : 6 : sub1 = (Node *) makeBoolExpr(OR_EXPR, args, a->location);
1358 : 6 : args = list_make2(makeSimpleA_Expr(AEXPR_OP, "<",
1359 : : copyObject(aexpr), copyObject(cexpr),
1360 : : a->location),
1361 : : makeSimpleA_Expr(AEXPR_OP, ">",
1362 : : copyObject(aexpr), copyObject(bexpr),
1363 : : a->location));
1364 : 6 : sub2 = (Node *) makeBoolExpr(OR_EXPR, args, a->location);
1365 : 6 : args = list_make2(sub1, sub2);
1366 : 6 : result = (Node *) makeBoolExpr(AND_EXPR, args, a->location);
1367 : 6 : break;
3339 tgl@sss.pgh.pa.us 1368 :UBC 0 : default:
1369 [ # # ]: 0 : elog(ERROR, "unrecognized A_Expr kind: %d", a->kind);
1370 : : result = NULL; /* keep compiler quiet */
1371 : : break;
1372 : : }
1373 : :
3339 tgl@sss.pgh.pa.us 1374 :CBC 255 : return transformExprRecurse(pstate, result);
1375 : : }
1376 : :
1377 : : static Node *
28 dean.a.rasheed@gmail 1378 :GNC 78 : transformMergeSupportFunc(ParseState *pstate, MergeSupportFunc *f)
1379 : : {
1380 : : /*
1381 : : * All we need to do is check that we're in the RETURNING list of a MERGE
1382 : : * command. If so, we just return the node as-is.
1383 : : */
1384 [ + + ]: 78 : if (pstate->p_expr_kind != EXPR_KIND_MERGE_RETURNING)
1385 : : {
1386 : 9 : ParseState *parent_pstate = pstate->parentParseState;
1387 : :
1388 [ + + ]: 9 : while (parent_pstate &&
1389 [ - + ]: 3 : parent_pstate->p_expr_kind != EXPR_KIND_MERGE_RETURNING)
28 dean.a.rasheed@gmail 1390 :UNC 0 : parent_pstate = parent_pstate->parentParseState;
1391 : :
28 dean.a.rasheed@gmail 1392 [ + + ]:GNC 9 : if (!parent_pstate)
1393 [ + - ]: 6 : ereport(ERROR,
1394 : : errcode(ERRCODE_SYNTAX_ERROR),
1395 : : errmsg("MERGE_ACTION() can only be used in the RETURNING list of a MERGE command"),
1396 : : parser_errposition(pstate, f->location));
1397 : : }
1398 : :
1399 : 72 : return (Node *) f;
1400 : : }
1401 : :
1402 : : static Node *
3590 tgl@sss.pgh.pa.us 1403 :CBC 67341 : transformBoolExpr(ParseState *pstate, BoolExpr *a)
1404 : : {
1405 : 67341 : List *args = NIL;
1406 : : const char *opname;
1407 : : ListCell *lc;
1408 : :
1409 [ + + + - ]: 67341 : switch (a->boolop)
1410 : : {
1411 : 54790 : case AND_EXPR:
1412 : 54790 : opname = "AND";
1413 : 54790 : break;
1414 : 5573 : case OR_EXPR:
1415 : 5573 : opname = "OR";
1416 : 5573 : break;
1417 : 6978 : case NOT_EXPR:
1418 : 6978 : opname = "NOT";
1419 : 6978 : break;
3590 tgl@sss.pgh.pa.us 1420 :UBC 0 : default:
1421 [ # # ]: 0 : elog(ERROR, "unrecognized boolop: %d", (int) a->boolop);
1422 : : opname = NULL; /* keep compiler quiet */
1423 : : break;
1424 : : }
1425 : :
3590 tgl@sss.pgh.pa.us 1426 [ + - + + :CBC 241554 : foreach(lc, a->args)
+ + ]
1427 : : {
1428 : 174223 : Node *arg = (Node *) lfirst(lc);
1429 : :
1430 : 174223 : arg = transformExprRecurse(pstate, arg);
1431 : 174213 : arg = coerce_to_boolean(pstate, arg, opname);
1432 : 174213 : args = lappend(args, arg);
1433 : : }
1434 : :
1435 : 67331 : return (Node *) makeBoolExpr(a->boolop, args, a->location);
1436 : : }
1437 : :
1438 : : static Node *
7025 neilc@samurai.com 1439 : 174444 : transformFuncCall(ParseState *pstate, FuncCall *fn)
1440 : : {
2497 tgl@sss.pgh.pa.us 1441 : 174444 : Node *last_srf = pstate->p_last_srf;
1442 : : List *targs;
1443 : : ListCell *args;
1444 : :
1445 : : /* Transform the list of arguments ... */
5751 1446 : 174444 : targs = NIL;
1447 [ + + + + : 425949 : foreach(args, fn->args)
+ + ]
1448 : : {
4265 1449 : 251505 : targs = lappend(targs, transformExprRecurse(pstate,
1450 : 251535 : (Node *) lfirst(args)));
1451 : : }
1452 : :
1453 : : /*
1454 : : * When WITHIN GROUP is used, we treat its ORDER BY expressions as
1455 : : * additional arguments to the function, for purposes of function lookup
1456 : : * and argument type coercion. So, transform each such expression and add
1457 : : * them to the targs list. We don't explicitly mark where each argument
1458 : : * came from, but ParseFuncOrColumn can tell what's what by reference to
1459 : : * list_length(fn->agg_order).
1460 : : */
3765 1461 [ + + ]: 174414 : if (fn->agg_within_group)
1462 : : {
1463 [ - + ]: 171 : Assert(fn->agg_order != NIL);
1464 [ + - + + : 372 : foreach(args, fn->agg_order)
+ + ]
1465 : : {
1466 : 201 : SortBy *arg = (SortBy *) lfirst(args);
1467 : :
1468 : 201 : targs = lappend(targs, transformExpr(pstate, arg->node,
1469 : : EXPR_KIND_ORDER_BY));
1470 : : }
1471 : : }
1472 : :
1473 : : /* ... and hand off to ParseFuncOrColumn */
5008 1474 : 174414 : return ParseFuncOrColumn(pstate,
1475 : : fn->funcname,
1476 : : targs,
1477 : : last_srf,
1478 : : fn,
1479 : : false,
1480 : : fn->location);
1481 : : }
1482 : :
1483 : : static Node *
3588 1484 : 187 : transformMultiAssignRef(ParseState *pstate, MultiAssignRef *maref)
1485 : : {
1486 : : SubLink *sublink;
1487 : : RowExpr *rexpr;
1488 : : Query *qtree;
1489 : : TargetEntry *tle;
1490 : :
1491 : : /* We should only see this in first-stage processing of UPDATE tlists */
1492 [ - + ]: 187 : Assert(pstate->p_expr_kind == EXPR_KIND_UPDATE_SOURCE);
1493 : :
1494 : : /* We only need to transform the source if this is the first column */
1495 [ + + ]: 187 : if (maref->colno == 1)
1496 : : {
1497 : : /*
1498 : : * For now, we only allow EXPR SubLinks and RowExprs as the source of
1499 : : * an UPDATE multiassignment. This is sufficient to cover interesting
1500 : : * cases; at worst, someone would have to write (SELECT * FROM expr)
1501 : : * to expand a composite-returning expression of another form.
1502 : : */
2700 1503 [ + + ]: 91 : if (IsA(maref->source, SubLink) &&
1504 [ + - ]: 69 : ((SubLink *) maref->source)->subLinkType == EXPR_SUBLINK)
1505 : : {
1506 : : /* Relabel it as a MULTIEXPR_SUBLINK */
1507 : 69 : sublink = (SubLink *) maref->source;
1508 : 69 : sublink->subLinkType = MULTIEXPR_SUBLINK;
1509 : : /* And transform it */
1510 : 69 : sublink = (SubLink *) transformExprRecurse(pstate,
1511 : : (Node *) sublink);
1512 : :
2635 1513 : 69 : qtree = castNode(Query, sublink->subselect);
1514 : :
1515 : : /* Check subquery returns required number of columns */
2700 1516 [ - + ]: 69 : if (count_nonjunk_tlist_entries(qtree->targetList) != maref->ncolumns)
2700 tgl@sss.pgh.pa.us 1517 [ # # ]:UBC 0 : ereport(ERROR,
1518 : : (errcode(ERRCODE_SYNTAX_ERROR),
1519 : : errmsg("number of columns does not match number of values"),
1520 : : parser_errposition(pstate, sublink->location)));
1521 : :
1522 : : /*
1523 : : * Build a resjunk tlist item containing the MULTIEXPR SubLink,
1524 : : * and add it to pstate->p_multiassign_exprs, whence it will later
1525 : : * get appended to the completed targetlist. We needn't worry
1526 : : * about selecting a resno for it; transformUpdateStmt will do
1527 : : * that.
1528 : : */
2700 tgl@sss.pgh.pa.us 1529 :CBC 69 : tle = makeTargetEntry((Expr *) sublink, 0, NULL, true);
1530 : 69 : pstate->p_multiassign_exprs = lappend(pstate->p_multiassign_exprs,
1531 : : tle);
1532 : :
1533 : : /*
1534 : : * Assign a unique-within-this-targetlist ID to the MULTIEXPR
1535 : : * SubLink. We can just use its position in the
1536 : : * p_multiassign_exprs list.
1537 : : */
1538 : 69 : sublink->subLinkId = list_length(pstate->p_multiassign_exprs);
1539 : : }
1540 [ + + ]: 22 : else if (IsA(maref->source, RowExpr))
1541 : : {
1542 : : /* Transform the RowExpr, allowing SetToDefault items */
1543 : 19 : rexpr = (RowExpr *) transformRowExpr(pstate,
1544 : 19 : (RowExpr *) maref->source,
1545 : : true);
1546 : :
1547 : : /* Check it returns required number of columns */
1548 [ - + ]: 19 : if (list_length(rexpr->args) != maref->ncolumns)
2700 tgl@sss.pgh.pa.us 1549 [ # # ]:UBC 0 : ereport(ERROR,
1550 : : (errcode(ERRCODE_SYNTAX_ERROR),
1551 : : errmsg("number of columns does not match number of values"),
1552 : : parser_errposition(pstate, rexpr->location)));
1553 : :
1554 : : /*
1555 : : * Temporarily append it to p_multiassign_exprs, so we can get it
1556 : : * back when we come back here for additional columns.
1557 : : */
2700 tgl@sss.pgh.pa.us 1558 :CBC 19 : tle = makeTargetEntry((Expr *) rexpr, 0, NULL, true);
1559 : 19 : pstate->p_multiassign_exprs = lappend(pstate->p_multiassign_exprs,
1560 : : tle);
1561 : : }
1562 : : else
1563 [ + - ]: 3 : ereport(ERROR,
1564 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1565 : : errmsg("source for a multiple-column UPDATE item must be a sub-SELECT or ROW() expression"),
1566 : : parser_errposition(pstate, exprLocation(maref->source))));
1567 : : }
1568 : : else
1569 : : {
1570 : : /*
1571 : : * Second or later column in a multiassignment. Re-fetch the
1572 : : * transformed SubLink or RowExpr, which we assume is still the last
1573 : : * entry in p_multiassign_exprs.
1574 : : */
3588 1575 [ - + ]: 96 : Assert(pstate->p_multiassign_exprs != NIL);
1576 : 96 : tle = (TargetEntry *) llast(pstate->p_multiassign_exprs);
1577 : : }
1578 : :
1579 : : /*
1580 : : * Emit the appropriate output expression for the current column
1581 : : */
2700 1582 [ + + ]: 184 : if (IsA(tle->expr, SubLink))
1583 : : {
1584 : : Param *param;
1585 : :
3588 1586 : 140 : sublink = (SubLink *) tle->expr;
1587 [ - + ]: 140 : Assert(sublink->subLinkType == MULTIEXPR_SUBLINK);
2635 1588 : 140 : qtree = castNode(Query, sublink->subselect);
1589 : :
1590 : : /* Build a Param representing the current subquery output column */
2700 1591 : 140 : tle = (TargetEntry *) list_nth(qtree->targetList, maref->colno - 1);
1592 [ - + ]: 140 : Assert(!tle->resjunk);
1593 : :
1594 : 140 : param = makeNode(Param);
1595 : 140 : param->paramkind = PARAM_MULTIEXPR;
1596 : 140 : param->paramid = (sublink->subLinkId << 16) | maref->colno;
1597 : 140 : param->paramtype = exprType((Node *) tle->expr);
1598 : 140 : param->paramtypmod = exprTypmod((Node *) tle->expr);
1599 : 140 : param->paramcollid = exprCollation((Node *) tle->expr);
1600 : 140 : param->location = exprLocation((Node *) tle->expr);
1601 : :
1602 : 140 : return (Node *) param;
1603 : : }
1604 : :
1605 [ + - ]: 44 : if (IsA(tle->expr, RowExpr))
1606 : : {
1607 : : Node *result;
1608 : :
1609 : 44 : rexpr = (RowExpr *) tle->expr;
1610 : :
1611 : : /* Just extract and return the next element of the RowExpr */
1612 : 44 : result = (Node *) list_nth(rexpr->args, maref->colno - 1);
1613 : :
1614 : : /*
1615 : : * If we're at the last column, delete the RowExpr from
1616 : : * p_multiassign_exprs; we don't need it anymore, and don't want it in
1617 : : * the finished UPDATE tlist. We assume this is still the last entry
1618 : : * in p_multiassign_exprs.
1619 : : */
1620 [ + + ]: 44 : if (maref->colno == maref->ncolumns)
1621 : 19 : pstate->p_multiassign_exprs =
1270 drowley@postgresql.o 1622 : 19 : list_delete_last(pstate->p_multiassign_exprs);
1623 : :
2700 tgl@sss.pgh.pa.us 1624 : 44 : return result;
1625 : : }
1626 : :
2700 tgl@sss.pgh.pa.us 1627 [ # # ]:UBC 0 : elog(ERROR, "unexpected expr type in multiassign list");
1628 : : return NULL; /* keep compiler quiet */
1629 : : }
1630 : :
1631 : : static Node *
7025 neilc@samurai.com 1632 :CBC 26185 : transformCaseExpr(ParseState *pstate, CaseExpr *c)
1633 : : {
2497 tgl@sss.pgh.pa.us 1634 : 26185 : CaseExpr *newc = makeNode(CaseExpr);
1635 : 26185 : Node *last_srf = pstate->p_last_srf;
1636 : : Node *arg;
1637 : : CaseTestExpr *placeholder;
1638 : : List *newargs;
1639 : : List *resultexprs;
1640 : : ListCell *l;
1641 : : Node *defresult;
1642 : : Oid ptype;
1643 : :
1644 : : /* transform the test expression, if any */
4265 1645 : 26185 : arg = transformExprRecurse(pstate, (Node *) c->arg);
1646 : :
1647 : : /* generate placeholder for test expression */
7025 neilc@samurai.com 1648 [ + + ]: 26185 : if (arg)
1649 : : {
1650 : : /*
1651 : : * If test expression is an untyped literal, force it to text. We have
1652 : : * to do something now because we won't be able to do this coercion on
1653 : : * the placeholder. This is not as flexible as what was done in 7.4
1654 : : * and before, but it's good enough to handle the sort of silly coding
1655 : : * commonly seen.
1656 : : */
1657 [ + + ]: 2665 : if (exprType(arg) == UNKNOWNOID)
1658 : 3 : arg = coerce_to_common_type(pstate, arg, TEXTOID, "CASE");
1659 : :
1660 : : /*
1661 : : * Run collation assignment on the test expression so that we know
1662 : : * what collation to mark the placeholder with. In principle we could
1663 : : * leave it to parse_collate.c to do that later, but propagating the
1664 : : * result to the CaseTestExpr would be unnecessarily complicated.
1665 : : */
4775 tgl@sss.pgh.pa.us 1666 : 2665 : assign_expr_collations(pstate, arg);
1667 : :
7025 neilc@samurai.com 1668 : 2665 : placeholder = makeNode(CaseTestExpr);
1669 : 2665 : placeholder->typeId = exprType(arg);
1670 : 2665 : placeholder->typeMod = exprTypmod(arg);
4814 peter_e@gmx.net 1671 : 2665 : placeholder->collation = exprCollation(arg);
1672 : : }
1673 : : else
7025 neilc@samurai.com 1674 : 23520 : placeholder = NULL;
1675 : :
1676 : 26185 : newc->arg = (Expr *) arg;
1677 : :
1678 : : /* transform the list of arguments */
1679 : 26185 : newargs = NIL;
5708 tgl@sss.pgh.pa.us 1680 : 26185 : resultexprs = NIL;
7025 neilc@samurai.com 1681 [ + - + + : 72847 : foreach(l, c->args)
+ + ]
1682 : : {
2561 tgl@sss.pgh.pa.us 1683 : 46662 : CaseWhen *w = lfirst_node(CaseWhen, l);
7025 neilc@samurai.com 1684 : 46662 : CaseWhen *neww = makeNode(CaseWhen);
1685 : : Node *warg;
1686 : :
1687 : 46662 : warg = (Node *) w->expr;
1688 [ + + ]: 46662 : if (placeholder)
1689 : : {
1690 : : /* shorthand form was specified, so expand... */
1691 : 10479 : warg = (Node *) makeSimpleA_Expr(AEXPR_OP, "=",
1692 : : (Node *) placeholder,
1693 : : warg,
1694 : : w->location);
1695 : : }
4265 tgl@sss.pgh.pa.us 1696 : 46662 : neww->expr = (Expr *) transformExprRecurse(pstate, warg);
1697 : :
7025 neilc@samurai.com 1698 : 93324 : neww->expr = (Expr *) coerce_to_boolean(pstate,
1699 : 46662 : (Node *) neww->expr,
1700 : : "CASE/WHEN");
1701 : :
1702 : 46662 : warg = (Node *) w->result;
4265 tgl@sss.pgh.pa.us 1703 : 46662 : neww->result = (Expr *) transformExprRecurse(pstate, warg);
5708 1704 : 46662 : neww->location = w->location;
1705 : :
7025 neilc@samurai.com 1706 : 46662 : newargs = lappend(newargs, neww);
5708 tgl@sss.pgh.pa.us 1707 : 46662 : resultexprs = lappend(resultexprs, neww->result);
1708 : : }
1709 : :
7025 neilc@samurai.com 1710 : 26185 : newc->args = newargs;
1711 : :
1712 : : /* transform the default clause */
1713 : 26185 : defresult = (Node *) c->defresult;
1714 [ + + ]: 26185 : if (defresult == NULL)
1715 : : {
1716 : 3881 : A_Const *n = makeNode(A_Const);
1717 : :
948 peter@eisentraut.org 1718 : 3881 : n->isnull = true;
5708 tgl@sss.pgh.pa.us 1719 : 3881 : n->location = -1;
7025 neilc@samurai.com 1720 : 3881 : defresult = (Node *) n;
1721 : : }
4265 tgl@sss.pgh.pa.us 1722 : 26185 : newc->defresult = (Expr *) transformExprRecurse(pstate, defresult);
1723 : :
1724 : : /*
1725 : : * Note: default result is considered the most significant type in
1726 : : * determining preferred type. This is how the code worked before, but it
1727 : : * seems a little bogus to me --- tgl
1728 : : */
5708 1729 : 26185 : resultexprs = lcons(newc->defresult, resultexprs);
1730 : :
1731 : 26185 : ptype = select_common_type(pstate, resultexprs, "CASE", NULL);
7025 neilc@samurai.com 1732 [ - + ]: 26185 : Assert(OidIsValid(ptype));
1733 : 26185 : newc->casetype = ptype;
1734 : : /* casecollid will be set by parse_collate.c */
1735 : :
1736 : : /* Convert default result clause, if necessary */
1737 : 26185 : newc->defresult = (Expr *)
1738 : 26185 : coerce_to_common_type(pstate,
1739 : 26185 : (Node *) newc->defresult,
1740 : : ptype,
1741 : : "CASE/ELSE");
1742 : :
1743 : : /* Convert when-clause results, if necessary */
1744 [ + - + + : 72847 : foreach(l, newc->args)
+ + ]
1745 : : {
1746 : 46662 : CaseWhen *w = (CaseWhen *) lfirst(l);
1747 : :
1748 : 46662 : w->result = (Expr *)
1749 : 46662 : coerce_to_common_type(pstate,
1750 : 46662 : (Node *) w->result,
1751 : : ptype,
1752 : : "CASE/WHEN");
1753 : : }
1754 : :
1755 : : /* if any subexpression contained a SRF, complain */
2497 tgl@sss.pgh.pa.us 1756 [ + + ]: 26185 : if (pstate->p_last_srf != last_srf)
1757 [ + - ]: 3 : ereport(ERROR,
1758 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1759 : : /* translator: %s is name of a SQL construct, eg GROUP BY */
1760 : : errmsg("set-returning functions are not allowed in %s",
1761 : : "CASE"),
1762 : : errhint("You might be able to move the set-returning function into a LATERAL FROM item."),
1763 : : parser_errposition(pstate,
1764 : : exprLocation(pstate->p_last_srf))));
1765 : :
5708 1766 : 26182 : newc->location = c->location;
1767 : :
7025 neilc@samurai.com 1768 : 26182 : return (Node *) newc;
1769 : : }
1770 : :
1771 : : static Node *
1772 : 18999 : transformSubLink(ParseState *pstate, SubLink *sublink)
1773 : : {
1774 : 18999 : Node *result = (Node *) sublink;
1775 : : Query *qtree;
1776 : : const char *err;
1777 : :
1778 : : /*
1779 : : * Check to see if the sublink is in an invalid place within the query. We
1780 : : * allow sublinks everywhere in SELECT/INSERT/UPDATE/DELETE/MERGE, but
1781 : : * generally not in utility statements.
1782 : : */
4265 tgl@sss.pgh.pa.us 1783 : 18999 : err = NULL;
1784 [ - - + - : 18999 : switch (pstate->p_expr_kind)
+ - - - -
- - + + -
+ + - ]
1785 : : {
4265 tgl@sss.pgh.pa.us 1786 :UBC 0 : case EXPR_KIND_NONE:
1787 : 0 : Assert(false); /* can't happen */
1788 : : break;
1789 : 0 : case EXPR_KIND_OTHER:
1790 : : /* Accept sublink here; caller must throw error if wanted */
1791 : 0 : break;
4265 tgl@sss.pgh.pa.us 1792 :CBC 18981 : case EXPR_KIND_JOIN_ON:
1793 : : case EXPR_KIND_JOIN_USING:
1794 : : case EXPR_KIND_FROM_SUBSELECT:
1795 : : case EXPR_KIND_FROM_FUNCTION:
1796 : : case EXPR_KIND_WHERE:
1797 : : case EXPR_KIND_POLICY:
1798 : : case EXPR_KIND_HAVING:
1799 : : case EXPR_KIND_FILTER:
1800 : : case EXPR_KIND_WINDOW_PARTITION:
1801 : : case EXPR_KIND_WINDOW_ORDER:
1802 : : case EXPR_KIND_WINDOW_FRAME_RANGE:
1803 : : case EXPR_KIND_WINDOW_FRAME_ROWS:
1804 : : case EXPR_KIND_WINDOW_FRAME_GROUPS:
1805 : : case EXPR_KIND_SELECT_TARGET:
1806 : : case EXPR_KIND_INSERT_TARGET:
1807 : : case EXPR_KIND_UPDATE_SOURCE:
1808 : : case EXPR_KIND_UPDATE_TARGET:
1809 : : case EXPR_KIND_MERGE_WHEN:
1810 : : case EXPR_KIND_GROUP_BY:
1811 : : case EXPR_KIND_ORDER_BY:
1812 : : case EXPR_KIND_DISTINCT_ON:
1813 : : case EXPR_KIND_LIMIT:
1814 : : case EXPR_KIND_OFFSET:
1815 : : case EXPR_KIND_RETURNING:
1816 : : case EXPR_KIND_MERGE_RETURNING:
1817 : : case EXPR_KIND_VALUES:
1818 : : case EXPR_KIND_VALUES_SINGLE:
1819 : : case EXPR_KIND_CYCLE_MARK:
1820 : : /* okay */
1821 : 18981 : break;
4265 tgl@sss.pgh.pa.us 1822 :UBC 0 : case EXPR_KIND_CHECK_CONSTRAINT:
1823 : : case EXPR_KIND_DOMAIN_CHECK:
4117 peter_e@gmx.net 1824 : 0 : err = _("cannot use subquery in check constraint");
4265 tgl@sss.pgh.pa.us 1825 : 0 : break;
4265 tgl@sss.pgh.pa.us 1826 :CBC 3 : case EXPR_KIND_COLUMN_DEFAULT:
1827 : : case EXPR_KIND_FUNCTION_DEFAULT:
1828 : 3 : err = _("cannot use subquery in DEFAULT expression");
1829 : 3 : break;
4265 tgl@sss.pgh.pa.us 1830 :UBC 0 : case EXPR_KIND_INDEX_EXPRESSION:
1831 : 0 : err = _("cannot use subquery in index expression");
1832 : 0 : break;
1833 : 0 : case EXPR_KIND_INDEX_PREDICATE:
1834 : 0 : err = _("cannot use subquery in index predicate");
1835 : 0 : break;
1115 tomas.vondra@postgre 1836 : 0 : case EXPR_KIND_STATS_EXPRESSION:
1837 : 0 : err = _("cannot use subquery in statistics expression");
1838 : 0 : break;
4265 tgl@sss.pgh.pa.us 1839 : 0 : case EXPR_KIND_ALTER_COL_TRANSFORM:
1840 : 0 : err = _("cannot use subquery in transform expression");
1841 : 0 : break;
1842 : 0 : case EXPR_KIND_EXECUTE_PARAMETER:
1843 : 0 : err = _("cannot use subquery in EXECUTE parameter");
1844 : 0 : break;
1845 : 0 : case EXPR_KIND_TRIGGER_WHEN:
1846 : 0 : err = _("cannot use subquery in trigger WHEN condition");
1847 : 0 : break;
1906 peter@eisentraut.org 1848 :CBC 6 : case EXPR_KIND_PARTITION_BOUND:
1849 : 6 : err = _("cannot use subquery in partition bound");
1850 : 6 : break;
2685 rhaas@postgresql.org 1851 : 3 : case EXPR_KIND_PARTITION_EXPRESSION:
1852 : 3 : err = _("cannot use subquery in partition key expression");
1853 : 3 : break;
2255 tgl@sss.pgh.pa.us 1854 :UBC 0 : case EXPR_KIND_CALL_ARGUMENT:
1855 : 0 : err = _("cannot use subquery in CALL argument");
1856 : 0 : break;
1912 tomas.vondra@postgre 1857 :CBC 3 : case EXPR_KIND_COPY_WHERE:
1858 : 3 : err = _("cannot use subquery in COPY FROM WHERE condition");
1859 : 3 : break;
1842 peter@eisentraut.org 1860 : 3 : case EXPR_KIND_GENERATED_COLUMN:
1861 : 3 : err = _("cannot use subquery in column generation expression");
1862 : 3 : break;
1863 : :
1864 : : /*
1865 : : * There is intentionally no default: case here, so that the
1866 : : * compiler will warn if we add a new ParseExprKind without
1867 : : * extending this switch. If we do see an unrecognized value at
1868 : : * runtime, the behavior will be the same as for EXPR_KIND_OTHER,
1869 : : * which is sane anyway.
1870 : : */
1871 : : }
4265 tgl@sss.pgh.pa.us 1872 [ + + ]: 18999 : if (err)
1873 [ + - ]: 18 : ereport(ERROR,
1874 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
1875 : : errmsg_internal("%s", err),
1876 : : parser_errposition(pstate, sublink->location)));
1877 : :
7025 neilc@samurai.com 1878 : 18981 : pstate->p_hasSubLinks = true;
1879 : :
1880 : : /*
1881 : : * OK, let's transform the sub-SELECT.
1882 : : */
2636 tgl@sss.pgh.pa.us 1883 : 18981 : qtree = parse_sub_analyze(sublink->subselect, pstate, NULL, false, true);
1884 : :
1885 : : /*
1886 : : * Check that we got a SELECT. Anything else should be impossible given
1887 : : * restrictions of the grammar, but check anyway.
1888 : : */
5704 1889 [ + - ]: 18960 : if (!IsA(qtree, Query) ||
2647 1890 [ - + ]: 18960 : qtree->commandType != CMD_SELECT)
5704 tgl@sss.pgh.pa.us 1891 [ # # ]:UBC 0 : elog(ERROR, "unexpected non-SELECT command in SubLink");
1892 : :
7025 neilc@samurai.com 1893 :CBC 18960 : sublink->subselect = (Node *) qtree;
1894 : :
1895 [ + + ]: 18960 : if (sublink->subLinkType == EXISTS_SUBLINK)
1896 : : {
1897 : : /*
1898 : : * EXISTS needs no test expression or combining operator. These fields
1899 : : * should be null already, but make sure.
1900 : : */
6682 tgl@sss.pgh.pa.us 1901 : 2604 : sublink->testexpr = NULL;
7025 neilc@samurai.com 1902 : 2604 : sublink->operName = NIL;
1903 : : }
1904 [ + + ]: 16356 : else if (sublink->subLinkType == EXPR_SUBLINK ||
1905 [ + + ]: 4985 : sublink->subLinkType == ARRAY_SUBLINK)
1906 : : {
1907 : : /*
1908 : : * Make sure the subselect delivers a single column (ignoring resjunk
1909 : : * targets).
1910 : : */
3588 tgl@sss.pgh.pa.us 1911 [ - + ]: 14999 : if (count_nonjunk_tlist_entries(qtree->targetList) != 1)
7025 neilc@samurai.com 1912 [ # # ]:UBC 0 : ereport(ERROR,
1913 : : (errcode(ERRCODE_SYNTAX_ERROR),
1914 : : errmsg("subquery must return only one column"),
1915 : : parser_errposition(pstate, sublink->location)));
1916 : :
1917 : : /*
1918 : : * EXPR and ARRAY need no test expression or combining operator. These
1919 : : * fields should be null already, but make sure.
1920 : : */
6682 tgl@sss.pgh.pa.us 1921 :CBC 14999 : sublink->testexpr = NULL;
7025 neilc@samurai.com 1922 : 14999 : sublink->operName = NIL;
1923 : : }
3588 tgl@sss.pgh.pa.us 1924 [ + + ]: 1357 : else if (sublink->subLinkType == MULTIEXPR_SUBLINK)
1925 : : {
1926 : : /* Same as EXPR case, except no restriction on number of columns */
1927 : 69 : sublink->testexpr = NULL;
1928 : 69 : sublink->operName = NIL;
1929 : : }
1930 : : else
1931 : : {
1932 : : /* ALL, ANY, or ROWCOMPARE: generate row-comparing expression */
1933 : : Node *lefthand;
1934 : : List *left_list;
1935 : : List *right_list;
1936 : : ListCell *l;
1937 : :
1938 : : /*
1939 : : * If the source was "x IN (select)", convert to "x = ANY (select)".
1940 : : */
3338 1941 [ + + ]: 1288 : if (sublink->operName == NIL)
1942 : 1190 : sublink->operName = list_make1(makeString("="));
1943 : :
1944 : : /*
1945 : : * Transform lefthand expression, and convert to a list
1946 : : */
4265 1947 : 1288 : lefthand = transformExprRecurse(pstate, sublink->testexpr);
6682 1948 [ + - + + ]: 1288 : if (lefthand && IsA(lefthand, RowExpr))
1949 : 129 : left_list = ((RowExpr *) lefthand)->args;
1950 : : else
1951 : 1159 : left_list = list_make1(lefthand);
1952 : :
1953 : : /*
1954 : : * Build a list of PARAM_SUBLINK nodes representing the output columns
1955 : : * of the subquery.
1956 : : */
1957 : 1288 : right_list = NIL;
1958 [ + - + + : 2779 : foreach(l, qtree->targetList)
+ + ]
1959 : : {
7025 neilc@samurai.com 1960 : 1491 : TargetEntry *tent = (TargetEntry *) lfirst(l);
1961 : : Param *param;
1962 : :
6948 tgl@sss.pgh.pa.us 1963 [ + + ]: 1491 : if (tent->resjunk)
7025 neilc@samurai.com 1964 : 6 : continue;
1965 : :
6682 tgl@sss.pgh.pa.us 1966 : 1485 : param = makeNode(Param);
1967 : 1485 : param->paramkind = PARAM_SUBLINK;
6567 1968 : 1485 : param->paramid = tent->resno;
6682 1969 : 1485 : param->paramtype = exprType((Node *) tent->expr);
6335 1970 : 1485 : param->paramtypmod = exprTypmod((Node *) tent->expr);
4775 1971 : 1485 : param->paramcollid = exprCollation((Node *) tent->expr);
5708 1972 : 1485 : param->location = -1;
1973 : :
6682 1974 : 1485 : right_list = lappend(right_list, param);
1975 : : }
1976 : :
1977 : : /*
1978 : : * We could rely on make_row_comparison_op to complain if the list
1979 : : * lengths differ, but we prefer to generate a more specific error
1980 : : * message.
1981 : : */
1982 [ - + ]: 1288 : if (list_length(left_list) < list_length(right_list))
6682 tgl@sss.pgh.pa.us 1983 [ # # ]:UBC 0 : ereport(ERROR,
1984 : : (errcode(ERRCODE_SYNTAX_ERROR),
1985 : : errmsg("subquery has too many columns"),
1986 : : parser_errposition(pstate, sublink->location)));
6682 tgl@sss.pgh.pa.us 1987 [ - + ]:CBC 1288 : if (list_length(left_list) > list_length(right_list))
7025 neilc@samurai.com 1988 [ # # ]:UBC 0 : ereport(ERROR,
1989 : : (errcode(ERRCODE_SYNTAX_ERROR),
1990 : : errmsg("subquery has too few columns"),
1991 : : parser_errposition(pstate, sublink->location)));
1992 : :
1993 : : /*
1994 : : * Identify the combining operator(s) and generate a suitable
1995 : : * row-comparison expression.
1996 : : */
6682 tgl@sss.pgh.pa.us 1997 :CBC 1288 : sublink->testexpr = make_row_comparison_op(pstate,
1998 : : sublink->operName,
1999 : : left_list,
2000 : : right_list,
2001 : : sublink->location);
2002 : : }
2003 : :
7025 neilc@samurai.com 2004 : 18954 : return result;
2005 : : }
2006 : :
2007 : : /*
2008 : : * transformArrayExpr
2009 : : *
2010 : : * If the caller specifies the target type, the resulting array will
2011 : : * be of exactly that type. Otherwise we try to infer a common type
2012 : : * for the elements using select_common_type().
2013 : : */
2014 : : static Node *
5869 tgl@sss.pgh.pa.us 2015 : 3928 : transformArrayExpr(ParseState *pstate, A_ArrayExpr *a,
2016 : : Oid array_type, Oid element_type, int32 typmod)
2017 : : {
7025 neilc@samurai.com 2018 : 3928 : ArrayExpr *newa = makeNode(ArrayExpr);
2019 : 3928 : List *newelems = NIL;
2020 : 3928 : List *newcoercedelems = NIL;
2021 : : ListCell *element;
2022 : : Oid coerce_type;
2023 : : bool coerce_hard;
2024 : :
2025 : : /*
2026 : : * Transform the element expressions
2027 : : *
2028 : : * Assume that the array is one-dimensional unless we find an array-type
2029 : : * element expression.
2030 : : */
5869 tgl@sss.pgh.pa.us 2031 : 3928 : newa->multidims = false;
7025 neilc@samurai.com 2032 [ + + + + : 12866 : foreach(element, a->elements)
+ + ]
2033 : : {
2034 : 8938 : Node *e = (Node *) lfirst(element);
2035 : : Node *newe;
2036 : :
2037 : : /*
2038 : : * If an element is itself an A_ArrayExpr, recurse directly so that we
2039 : : * can pass down any target type we were given.
2040 : : */
5869 tgl@sss.pgh.pa.us 2041 [ + + ]: 8938 : if (IsA(e, A_ArrayExpr))
2042 : : {
2043 : 407 : newe = transformArrayExpr(pstate,
2044 : : (A_ArrayExpr *) e,
2045 : : array_type,
2046 : : element_type,
2047 : : typmod);
2048 : : /* we certainly have an array here */
5708 2049 [ + + - + ]: 407 : Assert(array_type == InvalidOid || array_type == exprType(newe));
5869 2050 : 407 : newa->multidims = true;
2051 : : }
2052 : : else
2053 : : {
4265 2054 : 8531 : newe = transformExprRecurse(pstate, e);
2055 : :
2056 : : /*
2057 : : * Check for sub-array expressions, if we haven't already found
2058 : : * one.
2059 : : */
5708 2060 [ + - + + ]: 8531 : if (!newa->multidims && type_is_array(exprType(newe)))
5869 2061 : 3 : newa->multidims = true;
2062 : : }
2063 : :
7025 neilc@samurai.com 2064 : 8938 : newelems = lappend(newelems, newe);
2065 : : }
2066 : :
2067 : : /*
2068 : : * Select a target type for the elements.
2069 : : *
2070 : : * If we haven't been given a target array type, we must try to deduce a
2071 : : * common type based on the types of the individual elements present.
2072 : : */
5869 tgl@sss.pgh.pa.us 2073 [ + + ]: 3928 : if (OidIsValid(array_type))
2074 : : {
2075 : : /* Caller must ensure array_type matches element_type */
2076 [ - + ]: 405 : Assert(OidIsValid(element_type));
2077 [ + + ]: 405 : coerce_type = (newa->multidims ? array_type : element_type);
2078 : 405 : coerce_hard = true;
2079 : : }
2080 : : else
2081 : : {
2082 : : /* Can't handle an empty array without a target type */
5708 2083 [ + + ]: 3523 : if (newelems == NIL)
5869 2084 [ + - ]: 3 : ereport(ERROR,
2085 : : (errcode(ERRCODE_INDETERMINATE_DATATYPE),
2086 : : errmsg("cannot determine type of empty array"),
2087 : : errhint("Explicitly cast to the desired type, "
2088 : : "for example ARRAY[]::integer[]."),
2089 : : parser_errposition(pstate, a->location)));
2090 : :
2091 : : /* Select a common type for the elements */
5708 2092 : 3520 : coerce_type = select_common_type(pstate, newelems, "ARRAY", NULL);
2093 : :
5869 2094 [ + + ]: 3520 : if (newa->multidims)
2095 : : {
2096 : 196 : array_type = coerce_type;
2097 : 196 : element_type = get_element_type(array_type);
2098 [ - + ]: 196 : if (!OidIsValid(element_type))
5869 tgl@sss.pgh.pa.us 2099 [ # # ]:UBC 0 : ereport(ERROR,
2100 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2101 : : errmsg("could not find element type for data type %s",
2102 : : format_type_be(array_type)),
2103 : : parser_errposition(pstate, a->location)));
2104 : : }
2105 : : else
2106 : : {
5869 tgl@sss.pgh.pa.us 2107 :CBC 3324 : element_type = coerce_type;
2108 : 3324 : array_type = get_array_type(element_type);
2109 [ - + ]: 3324 : if (!OidIsValid(array_type))
5869 tgl@sss.pgh.pa.us 2110 [ # # ]:UBC 0 : ereport(ERROR,
2111 : : (errcode(ERRCODE_UNDEFINED_OBJECT),
2112 : : errmsg("could not find array type for data type %s",
2113 : : format_type_be(element_type)),
2114 : : parser_errposition(pstate, a->location)));
2115 : : }
5869 tgl@sss.pgh.pa.us 2116 :CBC 3520 : coerce_hard = false;
2117 : : }
2118 : :
2119 : : /*
2120 : : * Coerce elements to target type
2121 : : *
2122 : : * If the array has been explicitly cast, then the elements are in turn
2123 : : * explicitly coerced.
2124 : : *
2125 : : * If the array's type was merely derived from the common type of its
2126 : : * elements, then the elements are implicitly coerced to the common type.
2127 : : * This is consistent with other uses of select_common_type().
2128 : : */
7025 neilc@samurai.com 2129 [ + + + + : 12863 : foreach(element, newelems)
+ + ]
2130 : : {
2131 : 8938 : Node *e = (Node *) lfirst(element);
2132 : : Node *newe;
2133 : :
5869 tgl@sss.pgh.pa.us 2134 [ + + ]: 8938 : if (coerce_hard)
2135 : : {
5421 bruce@momjian.us 2136 : 988 : newe = coerce_to_target_type(pstate, e,
2137 : : exprType(e),
2138 : : coerce_type,
2139 : : typmod,
2140 : : COERCION_EXPLICIT,
2141 : : COERCE_EXPLICIT_CAST,
2142 : : -1);
5869 tgl@sss.pgh.pa.us 2143 [ - + ]: 988 : if (newe == NULL)
5869 tgl@sss.pgh.pa.us 2144 [ # # ]:UBC 0 : ereport(ERROR,
2145 : : (errcode(ERRCODE_CANNOT_COERCE),
2146 : : errmsg("cannot cast type %s to %s",
2147 : : format_type_be(exprType(e)),
2148 : : format_type_be(coerce_type)),
2149 : : parser_errposition(pstate, exprLocation(e))));
2150 : : }
2151 : : else
5869 tgl@sss.pgh.pa.us 2152 :CBC 7950 : newe = coerce_to_common_type(pstate, e,
2153 : : coerce_type,
2154 : : "ARRAY");
7025 neilc@samurai.com 2155 : 8938 : newcoercedelems = lappend(newcoercedelems, newe);
2156 : : }
2157 : :
2158 : 3925 : newa->array_typeid = array_type;
2159 : : /* array_collid will be set by parse_collate.c */
2160 : 3925 : newa->element_typeid = element_type;
2161 : 3925 : newa->elements = newcoercedelems;
5708 tgl@sss.pgh.pa.us 2162 : 3925 : newa->location = a->location;
2163 : :
7025 neilc@samurai.com 2164 : 3925 : return (Node *) newa;
2165 : : }
2166 : :
2167 : : static Node *
2700 tgl@sss.pgh.pa.us 2168 : 2651 : transformRowExpr(ParseState *pstate, RowExpr *r, bool allowDefault)
2169 : : {
2170 : : RowExpr *newr;
2171 : : char fname[16];
2172 : : int fnum;
2173 : :
4130 2174 : 2651 : newr = makeNode(RowExpr);
2175 : :
2176 : : /* Transform the field expressions */
2700 2177 : 2651 : newr->args = transformExpressionList(pstate, r->args,
2178 : : pstate->p_expr_kind, allowDefault);
2179 : :
2180 : : /* Disallow more columns than will fit in a tuple */
625 2181 [ - + ]: 2651 : if (list_length(newr->args) > MaxTupleAttributeNumber)
625 tgl@sss.pgh.pa.us 2182 [ # # ]:UBC 0 : ereport(ERROR,
2183 : : (errcode(ERRCODE_TOO_MANY_COLUMNS),
2184 : : errmsg("ROW expressions can have at most %d entries",
2185 : : MaxTupleAttributeNumber),
2186 : : parser_errposition(pstate, r->location)));
2187 : :
2188 : : /* Barring later casting, we consider the type RECORD */
7025 neilc@samurai.com 2189 :CBC 2651 : newr->row_typeid = RECORDOID;
2190 : 2651 : newr->row_format = COERCE_IMPLICIT_CAST;
2191 : :
2192 : : /* ROW() has anonymous columns, so invent some field names */
4443 tgl@sss.pgh.pa.us 2193 : 2651 : newr->colnames = NIL;
1735 2194 [ + + ]: 9163 : for (fnum = 1; fnum <= list_length(newr->args); fnum++)
2195 : : {
2196 : 6512 : snprintf(fname, sizeof(fname), "f%d", fnum);
4443 2197 : 6512 : newr->colnames = lappend(newr->colnames, makeString(pstrdup(fname)));
2198 : : }
2199 : :
5708 2200 : 2651 : newr->location = r->location;
2201 : :
7025 neilc@samurai.com 2202 : 2651 : return (Node *) newr;
2203 : : }
2204 : :
2205 : : static Node *
2206 : 1492 : transformCoalesceExpr(ParseState *pstate, CoalesceExpr *c)
2207 : : {
2208 : 1492 : CoalesceExpr *newc = makeNode(CoalesceExpr);
2497 tgl@sss.pgh.pa.us 2209 : 1492 : Node *last_srf = pstate->p_last_srf;
7025 neilc@samurai.com 2210 : 1492 : List *newargs = NIL;
2211 : 1492 : List *newcoercedargs = NIL;
2212 : : ListCell *args;
2213 : :
2214 [ + - + + : 4461 : foreach(args, c->args)
+ + ]
2215 : : {
2216 : 2969 : Node *e = (Node *) lfirst(args);
2217 : : Node *newe;
2218 : :
4265 tgl@sss.pgh.pa.us 2219 : 2969 : newe = transformExprRecurse(pstate, e);
7025 neilc@samurai.com 2220 : 2969 : newargs = lappend(newargs, newe);
2221 : : }
2222 : :
5708 tgl@sss.pgh.pa.us 2223 : 1492 : newc->coalescetype = select_common_type(pstate, newargs, "COALESCE", NULL);
2224 : : /* coalescecollid will be set by parse_collate.c */
2225 : :
2226 : : /* Convert arguments if necessary */
7025 neilc@samurai.com 2227 [ + - + + : 4461 : foreach(args, newargs)
+ + ]
2228 : : {
2229 : 2969 : Node *e = (Node *) lfirst(args);
2230 : : Node *newe;
2231 : :
2232 : 2969 : newe = coerce_to_common_type(pstate, e,
2233 : : newc->coalescetype,
2234 : : "COALESCE");
2235 : 2969 : newcoercedargs = lappend(newcoercedargs, newe);
2236 : : }
2237 : :
2238 : : /* if any subexpression contained a SRF, complain */
2497 tgl@sss.pgh.pa.us 2239 [ + + ]: 1492 : if (pstate->p_last_srf != last_srf)
2240 [ + - ]: 3 : ereport(ERROR,
2241 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2242 : : /* translator: %s is name of a SQL construct, eg GROUP BY */
2243 : : errmsg("set-returning functions are not allowed in %s",
2244 : : "COALESCE"),
2245 : : errhint("You might be able to move the set-returning function into a LATERAL FROM item."),
2246 : : parser_errposition(pstate,
2247 : : exprLocation(pstate->p_last_srf))));
2248 : :
7025 neilc@samurai.com 2249 : 1489 : newc->args = newcoercedargs;
5708 tgl@sss.pgh.pa.us 2250 : 1489 : newc->location = c->location;
7025 neilc@samurai.com 2251 : 1489 : return (Node *) newc;
2252 : : }
2253 : :
2254 : : static Node *
6867 tgl@sss.pgh.pa.us 2255 : 128 : transformMinMaxExpr(ParseState *pstate, MinMaxExpr *m)
2256 : : {
2257 : 128 : MinMaxExpr *newm = makeNode(MinMaxExpr);
2258 : 128 : List *newargs = NIL;
2259 : 128 : List *newcoercedargs = NIL;
5708 2260 [ + + ]: 128 : const char *funcname = (m->op == IS_GREATEST) ? "GREATEST" : "LEAST";
2261 : : ListCell *args;
2262 : :
6867 2263 : 128 : newm->op = m->op;
2264 [ + - + + : 411 : foreach(args, m->args)
+ + ]
2265 : : {
2266 : 283 : Node *e = (Node *) lfirst(args);
2267 : : Node *newe;
2268 : :
4265 2269 : 283 : newe = transformExprRecurse(pstate, e);
6867 2270 : 283 : newargs = lappend(newargs, newe);
2271 : : }
2272 : :
5708 2273 : 128 : newm->minmaxtype = select_common_type(pstate, newargs, funcname, NULL);
2274 : : /* minmaxcollid and inputcollid will be set by parse_collate.c */
2275 : :
2276 : : /* Convert arguments if necessary */
6867 2277 [ + - + + : 411 : foreach(args, newargs)
+ + ]
2278 : : {
2279 : 283 : Node *e = (Node *) lfirst(args);
2280 : : Node *newe;
2281 : :
2282 : 283 : newe = coerce_to_common_type(pstate, e,
2283 : : newm->minmaxtype,
2284 : : funcname);
2285 : 283 : newcoercedargs = lappend(newcoercedargs, newe);
2286 : : }
2287 : :
2288 : 128 : newm->args = newcoercedargs;
5708 2289 : 128 : newm->location = m->location;
6867 2290 : 128 : return (Node *) newm;
2291 : : }
2292 : :
2293 : : static Node *
333 michael@paquier.xyz 2294 : 1215 : transformSQLValueFunction(ParseState *pstate, SQLValueFunction *svf)
2295 : : {
2296 : : /*
2297 : : * All we need to do is insert the correct result type and (where needed)
2298 : : * validate the typmod, so we just modify the node in-place.
2299 : : */
2300 [ + + + + : 1215 : switch (svf->op)
+ + + + +
+ - ]
2301 : : {
2302 : 132 : case SVFOP_CURRENT_DATE:
2303 : 132 : svf->type = DATEOID;
2304 : 132 : break;
2305 : 12 : case SVFOP_CURRENT_TIME:
2306 : 12 : svf->type = TIMETZOID;
2307 : 12 : break;
2308 : 12 : case SVFOP_CURRENT_TIME_N:
2309 : 12 : svf->type = TIMETZOID;
2310 : 12 : svf->typmod = anytime_typmod_check(true, svf->typmod);
2311 : 12 : break;
2312 : 143 : case SVFOP_CURRENT_TIMESTAMP:
2313 : 143 : svf->type = TIMESTAMPTZOID;
2314 : 143 : break;
2315 : 76 : case SVFOP_CURRENT_TIMESTAMP_N:
2316 : 76 : svf->type = TIMESTAMPTZOID;
2317 : 76 : svf->typmod = anytimestamp_typmod_check(true, svf->typmod);
2318 : 76 : break;
2319 : 12 : case SVFOP_LOCALTIME:
2320 : 12 : svf->type = TIMEOID;
2321 : 12 : break;
2322 : 12 : case SVFOP_LOCALTIME_N:
2323 : 12 : svf->type = TIMEOID;
2324 : 12 : svf->typmod = anytime_typmod_check(false, svf->typmod);
2325 : 12 : break;
2326 : 18 : case SVFOP_LOCALTIMESTAMP:
2327 : 18 : svf->type = TIMESTAMPOID;
2328 : 18 : break;
2329 : 12 : case SVFOP_LOCALTIMESTAMP_N:
2330 : 12 : svf->type = TIMESTAMPOID;
2331 : 12 : svf->typmod = anytimestamp_typmod_check(false, svf->typmod);
2332 : 12 : break;
2333 : 786 : case SVFOP_CURRENT_ROLE:
2334 : : case SVFOP_CURRENT_USER:
2335 : : case SVFOP_USER:
2336 : : case SVFOP_SESSION_USER:
2337 : : case SVFOP_CURRENT_CATALOG:
2338 : : case SVFOP_CURRENT_SCHEMA:
2339 : 786 : svf->type = NAMEOID;
2340 : 786 : break;
2341 : : }
2342 : :
2343 : 1215 : return (Node *) svf;
2344 : : }
2345 : :
2346 : : static Node *
5995 bruce@momjian.us 2347 : 298 : transformXmlExpr(ParseState *pstate, XmlExpr *x)
2348 : : {
2349 : : XmlExpr *newx;
2350 : : ListCell *lc;
2351 : : int i;
2352 : :
4130 tgl@sss.pgh.pa.us 2353 : 298 : newx = makeNode(XmlExpr);
6321 2354 : 298 : newx->op = x->op;
2355 [ + + ]: 298 : if (x->name)
6272 peter_e@gmx.net 2356 : 129 : newx->name = map_sql_identifier_to_xml_name(x->name, false, false);
2357 : : else
6321 tgl@sss.pgh.pa.us 2358 : 169 : newx->name = NULL;
5708 2359 : 298 : newx->xmloption = x->xmloption;
4130 2360 : 298 : newx->type = XMLOID; /* this just marks the node as transformed */
2361 : 298 : newx->typmod = -1;
5708 2362 : 298 : newx->location = x->location;
2363 : :
2364 : : /*
2365 : : * gram.y built the named args as a list of ResTarget. Transform each,
2366 : : * and break the names out as a separate list.
2367 : : */
6321 2368 : 298 : newx->named_args = NIL;
2369 : 298 : newx->arg_names = NIL;
2370 : :
2371 [ + + + + : 408 : foreach(lc, x->named_args)
+ + ]
2372 : : {
2561 2373 : 116 : ResTarget *r = lfirst_node(ResTarget, lc);
2374 : : Node *expr;
2375 : : char *argname;
2376 : :
4265 2377 : 116 : expr = transformExprRecurse(pstate, r->val);
2378 : :
6321 2379 [ + + ]: 116 : if (r->name)
6272 peter_e@gmx.net 2380 : 53 : argname = map_sql_identifier_to_xml_name(r->name, false, false);
6321 tgl@sss.pgh.pa.us 2381 [ + + ]: 63 : else if (IsA(r->val, ColumnRef))
2382 : 60 : argname = map_sql_identifier_to_xml_name(FigureColname(r->val),
2383 : : true, false);
2384 : : else
2385 : : {
2386 [ + - + - ]: 3 : ereport(ERROR,
2387 : : (errcode(ERRCODE_SYNTAX_ERROR),
2388 : : x->op == IS_XMLELEMENT
2389 : : ? errmsg("unnamed XML attribute value must be a column reference")
2390 : : : errmsg("unnamed XML element value must be a column reference"),
2391 : : parser_errposition(pstate, r->location)));
2392 : : argname = NULL; /* keep compiler quiet */
2393 : : }
2394 : :
2395 : : /* reject duplicate argnames in XMLELEMENT only */
5708 2396 [ + + ]: 113 : if (x->op == IS_XMLELEMENT)
2397 : : {
2398 : : ListCell *lc2;
2399 : :
2400 [ + + + + : 60 : foreach(lc2, newx->arg_names)
+ + ]
2401 : : {
2402 [ + + ]: 19 : if (strcmp(argname, strVal(lfirst(lc2))) == 0)
6306 peter_e@gmx.net 2403 [ + - ]: 3 : ereport(ERROR,
2404 : : (errcode(ERRCODE_SYNTAX_ERROR),
2405 : : errmsg("XML attribute name \"%s\" appears more than once",
2406 : : argname),
2407 : : parser_errposition(pstate, r->location)));
2408 : : }
2409 : : }
2410 : :
5708 tgl@sss.pgh.pa.us 2411 : 110 : newx->named_args = lappend(newx->named_args, expr);
2412 : 110 : newx->arg_names = lappend(newx->arg_names, makeString(argname));
2413 : : }
2414 : :
2415 : : /* The other arguments are of varying types depending on the function */
6321 2416 : 292 : newx->args = NIL;
2417 : 292 : i = 0;
2418 [ + + + + : 701 : foreach(lc, x->args)
+ + ]
2419 : : {
2420 : 418 : Node *e = (Node *) lfirst(lc);
2421 : : Node *newe;
2422 : :
4265 2423 : 418 : newe = transformExprRecurse(pstate, e);
6321 2424 [ + + - + : 418 : switch (x->op)
+ + - +
- ]
2425 : : {
2426 : 65 : case IS_XMLCONCAT:
2427 : 65 : newe = coerce_to_specific_type(pstate, newe, XMLOID,
2428 : : "XMLCONCAT");
2429 : 59 : break;
6302 peter_e@gmx.net 2430 : 68 : case IS_XMLELEMENT:
2431 : : /* no coercion necessary */
2432 : 68 : break;
6321 tgl@sss.pgh.pa.us 2433 :UBC 0 : case IS_XMLFOREST:
2434 : 0 : newe = coerce_to_specific_type(pstate, newe, XMLOID,
2435 : : "XMLFOREST");
2436 : 0 : break;
6321 tgl@sss.pgh.pa.us 2437 :CBC 140 : case IS_XMLPARSE:
2438 [ + + ]: 140 : if (i == 0)
2439 : 70 : newe = coerce_to_specific_type(pstate, newe, TEXTOID,
2440 : : "XMLPARSE");
2441 : : else
2442 : 70 : newe = coerce_to_boolean(pstate, newe, "XMLPARSE");
2443 : 140 : break;
2444 : 25 : case IS_XMLPI:
2445 : 25 : newe = coerce_to_specific_type(pstate, newe, TEXTOID,
2446 : : "XMLPI");
2447 : 25 : break;
2448 : 102 : case IS_XMLROOT:
2449 [ + + ]: 102 : if (i == 0)
2450 : 34 : newe = coerce_to_specific_type(pstate, newe, XMLOID,
2451 : : "XMLROOT");
2452 [ + + ]: 68 : else if (i == 1)
2453 : 34 : newe = coerce_to_specific_type(pstate, newe, TEXTOID,
2454 : : "XMLROOT");
2455 : : else
6289 peter_e@gmx.net 2456 : 34 : newe = coerce_to_specific_type(pstate, newe, INT4OID,
2457 : : "XMLROOT");
6321 tgl@sss.pgh.pa.us 2458 : 102 : break;
6280 peter_e@gmx.net 2459 :UBC 0 : case IS_XMLSERIALIZE:
2460 : : /* not handled here */
5708 tgl@sss.pgh.pa.us 2461 : 0 : Assert(false);
2462 : : break;
6300 peter_e@gmx.net 2463 :CBC 18 : case IS_DOCUMENT:
2464 : 18 : newe = coerce_to_specific_type(pstate, newe, XMLOID,
2465 : : "IS DOCUMENT");
2466 : 15 : break;
2467 : : }
6321 tgl@sss.pgh.pa.us 2468 : 409 : newx->args = lappend(newx->args, newe);
2469 : 409 : i++;
2470 : : }
2471 : :
2472 : 283 : return (Node *) newx;
2473 : : }
2474 : :
2475 : : static Node *
5995 bruce@momjian.us 2476 : 95 : transformXmlSerialize(ParseState *pstate, XmlSerialize *xs)
2477 : : {
2478 : : Node *result;
2479 : : XmlExpr *xexpr;
2480 : : Oid targetType;
2481 : : int32 targetTypmod;
2482 : :
6280 peter_e@gmx.net 2483 : 95 : xexpr = makeNode(XmlExpr);
2484 : 95 : xexpr->op = IS_XMLSERIALIZE;
2485 : 95 : xexpr->args = list_make1(coerce_to_specific_type(pstate,
2486 : : transformExprRecurse(pstate, xs->expr),
2487 : : XMLOID,
2488 : : "XMLSERIALIZE"));
2489 : :
4920 2490 : 95 : typenameTypeIdAndMod(pstate, xs->typeName, &targetType, &targetTypmod);
2491 : :
6280 2492 : 95 : xexpr->xmloption = xs->xmloption;
396 tgl@sss.pgh.pa.us 2493 : 95 : xexpr->indent = xs->indent;
5708 2494 : 95 : xexpr->location = xs->location;
2495 : : /* We actually only need these to be able to parse back the expression. */
6280 peter_e@gmx.net 2496 : 95 : xexpr->type = targetType;
2497 : 95 : xexpr->typmod = targetTypmod;
2498 : :
2499 : : /*
2500 : : * The actual target type is determined this way. SQL allows char and
2501 : : * varchar as target types. We allow anything that can be cast implicitly
2502 : : * from text. This way, user-defined text-like data types automatically
2503 : : * fit in.
2504 : : */
5708 tgl@sss.pgh.pa.us 2505 : 95 : result = coerce_to_target_type(pstate, (Node *) xexpr,
2506 : : TEXTOID, targetType, targetTypmod,
2507 : : COERCION_IMPLICIT,
2508 : : COERCE_IMPLICIT_CAST,
2509 : : -1);
2510 [ - + ]: 95 : if (result == NULL)
5708 tgl@sss.pgh.pa.us 2511 [ # # ]:UBC 0 : ereport(ERROR,
2512 : : (errcode(ERRCODE_CANNOT_COERCE),
2513 : : errmsg("cannot cast XMLSERIALIZE result to %s",
2514 : : format_type_be(targetType)),
2515 : : parser_errposition(pstate, xexpr->location)));
5708 tgl@sss.pgh.pa.us 2516 :CBC 95 : return result;
2517 : : }
2518 : :
2519 : : static Node *
7025 neilc@samurai.com 2520 : 408 : transformBooleanTest(ParseState *pstate, BooleanTest *b)
2521 : : {
2522 : : const char *clausename;
2523 : :
2524 [ + + + + : 408 : switch (b->booltesttype)
+ + - ]
2525 : : {
2526 : 187 : case IS_TRUE:
2527 : 187 : clausename = "IS TRUE";
2528 : 187 : break;
2529 : 69 : case IS_NOT_TRUE:
2530 : 69 : clausename = "IS NOT TRUE";
2531 : 69 : break;
2532 : 57 : case IS_FALSE:
2533 : 57 : clausename = "IS FALSE";
2534 : 57 : break;
2535 : 45 : case IS_NOT_FALSE:
2536 : 45 : clausename = "IS NOT FALSE";
2537 : 45 : break;
2538 : 26 : case IS_UNKNOWN:
2539 : 26 : clausename = "IS UNKNOWN";
2540 : 26 : break;
2541 : 24 : case IS_NOT_UNKNOWN:
2542 : 24 : clausename = "IS NOT UNKNOWN";
2543 : 24 : break;
7025 neilc@samurai.com 2544 :UBC 0 : default:
2545 [ # # ]: 0 : elog(ERROR, "unrecognized booltesttype: %d",
2546 : : (int) b->booltesttype);
2547 : : clausename = NULL; /* keep compiler quiet */
2548 : : }
2549 : :
4265 tgl@sss.pgh.pa.us 2550 :CBC 408 : b->arg = (Expr *) transformExprRecurse(pstate, (Node *) b->arg);
2551 : :
7025 neilc@samurai.com 2552 : 816 : b->arg = (Expr *) coerce_to_boolean(pstate,
2553 : 408 : (Node *) b->arg,
2554 : : clausename);
2555 : :
2556 : 408 : return (Node *) b;
2557 : : }
2558 : :
2559 : : static Node *
5995 bruce@momjian.us 2560 : 121 : transformCurrentOfExpr(ParseState *pstate, CurrentOfExpr *cexpr)
2561 : : {
2562 : : /* CURRENT OF can only appear at top level of UPDATE/DELETE */
1564 tgl@sss.pgh.pa.us 2563 [ - + ]: 121 : Assert(pstate->p_target_nsitem != NULL);
2564 : 121 : cexpr->cvarno = pstate->p_target_nsitem->p_rtindex;
2565 : :
2566 : : /*
2567 : : * Check to see if the cursor name matches a parameter of type REFCURSOR.
2568 : : * If so, replace the raw name reference with a parameter reference. (This
2569 : : * is a hack for the convenience of plpgsql.)
2570 : : */
2489 2571 [ + - ]: 121 : if (cexpr->cursor_name != NULL) /* in case already transformed */
2572 : : {
5270 2573 : 121 : ColumnRef *cref = makeNode(ColumnRef);
2574 : 121 : Node *node = NULL;
2575 : :
2576 : : /* Build an unqualified ColumnRef with the given name */
2577 : 121 : cref->fields = list_make1(makeString(cexpr->cursor_name));
2578 : 121 : cref->location = -1;
2579 : :
2580 : : /* See if there is a translation available from a parser hook */
2581 [ + + ]: 121 : if (pstate->p_pre_columnref_hook != NULL)
2411 peter_e@gmx.net 2582 : 6 : node = pstate->p_pre_columnref_hook(pstate, cref);
5270 tgl@sss.pgh.pa.us 2583 [ + - + + ]: 121 : if (node == NULL && pstate->p_post_columnref_hook != NULL)
2411 peter_e@gmx.net 2584 : 6 : node = pstate->p_post_columnref_hook(pstate, cref, NULL);
2585 : :
2586 : : /*
2587 : : * XXX Should we throw an error if we get a translation that isn't a
2588 : : * refcursor Param? For now it seems best to silently ignore false
2589 : : * matches.
2590 : : */
5270 tgl@sss.pgh.pa.us 2591 [ + + + - ]: 121 : if (node != NULL && IsA(node, Param))
2592 : : {
5161 bruce@momjian.us 2593 : 6 : Param *p = (Param *) node;
2594 : :
5270 tgl@sss.pgh.pa.us 2595 [ + - ]: 6 : if (p->paramkind == PARAM_EXTERN &&
2596 [ + - ]: 6 : p->paramtype == REFCURSOROID)
2597 : : {
2598 : : /* Matches, so convert CURRENT OF to a param reference */
2599 : 6 : cexpr->cursor_name = NULL;
2600 : 6 : cexpr->cursor_param = p->paramid;
2601 : : }
2602 : : }
2603 : : }
2604 : :
6152 2605 : 121 : return (Node *) cexpr;
2606 : : }
2607 : :
2608 : : /*
2609 : : * Construct a whole-row reference to represent the notation "relation.*".
2610 : : */
2611 : : static Node *
1571 2612 : 3609 : transformWholeRowRef(ParseState *pstate, ParseNamespaceItem *nsitem,
2613 : : int sublevels_up, int location)
2614 : : {
2615 : : /*
2616 : : * Build the appropriate referencing node. Normally this can be a
2617 : : * whole-row Var, but if the nsitem is a JOIN USING alias then it contains
2618 : : * only a subset of the columns of the underlying join RTE, so that will
2619 : : * not work. Instead we immediately expand the reference into a RowExpr.
2620 : : * Since the JOIN USING's common columns are fully determined at this
2621 : : * point, there seems no harm in expanding it now rather than during
2622 : : * planning.
2623 : : *
2624 : : * Note that if the RTE is a function returning scalar, we create just a
2625 : : * plain reference to the function value, not a composite containing a
2626 : : * single column. This is pretty inconsistent at first sight, but it's
2627 : : * what we've done historically. One argument for it is that "rel" and
2628 : : * "rel.*" mean the same thing for composite relations, so why not for
2629 : : * scalar functions...
2630 : : */
1110 peter@eisentraut.org 2631 [ + + ]: 3609 : if (nsitem->p_names == nsitem->p_rte->eref)
2632 : : {
2633 : : Var *result;
2634 : :
2635 : 3603 : result = makeWholeRowVar(nsitem->p_rte, nsitem->p_rtindex,
2636 : : sublevels_up, true);
2637 : :
2638 : : /* location is not filled in by makeWholeRowVar */
2639 : 3603 : result->location = location;
2640 : :
2641 : : /* mark Var if it's nulled by any outer joins */
440 tgl@sss.pgh.pa.us 2642 : 3603 : markNullableIfNeeded(pstate, result);
2643 : :
2644 : : /* mark relation as requiring whole-row SELECT access */
1110 peter@eisentraut.org 2645 : 3603 : markVarForSelectPriv(pstate, result);
2646 : :
2647 : 3603 : return (Node *) result;
2648 : : }
2649 : : else
2650 : : {
2651 : : RowExpr *rowexpr;
2652 : : List *fields;
2653 : :
2654 : : /*
2655 : : * We want only as many columns as are listed in p_names->colnames,
2656 : : * and we should use those names not whatever possibly-aliased names
2657 : : * are in the RTE. We needn't worry about marking the RTE for SELECT
2658 : : * access, as the common columns are surely so marked already.
2659 : : */
2660 : 6 : expandRTE(nsitem->p_rte, nsitem->p_rtindex,
2661 : : sublevels_up, location, false,
2662 : : NULL, &fields);
2663 : 6 : rowexpr = makeNode(RowExpr);
2664 : 6 : rowexpr->args = list_truncate(fields,
2665 : 6 : list_length(nsitem->p_names->colnames));
2666 : 6 : rowexpr->row_typeid = RECORDOID;
2667 : 6 : rowexpr->row_format = COERCE_IMPLICIT_CAST;
2668 : 6 : rowexpr->colnames = copyObject(nsitem->p_names->colnames);
2669 : 6 : rowexpr->location = location;
2670 : :
2671 : : /* XXX we ought to mark the row as possibly nullable */
2672 : :
2673 : 6 : return (Node *) rowexpr;
2674 : : }
2675 : : }
2676 : :
2677 : : /*
2678 : : * Handle an explicit CAST construct.
2679 : : *
2680 : : * Transform the argument, look up the type name, and apply any necessary
2681 : : * coercion function(s).
2682 : : */
2683 : : static Node *
5708 tgl@sss.pgh.pa.us 2684 : 119810 : transformTypeCast(ParseState *pstate, TypeCast *tc)
2685 : : {
2686 : : Node *result;
2915 2687 : 119810 : Node *arg = tc->arg;
2688 : : Node *expr;
2689 : : Oid inputType;
2690 : : Oid targetType;
2691 : : int32 targetTypmod;
2692 : : int location;
2693 : :
2694 : : /* Look up the type name first */
4920 peter_e@gmx.net 2695 : 119810 : typenameTypeIdAndMod(pstate, tc->typeName, &targetType, &targetTypmod);
2696 : :
2697 : : /*
2698 : : * If the subject of the typecast is an ARRAY[] construct and the target
2699 : : * type is an array type, we invoke transformArrayExpr() directly so that
2700 : : * we can pass down the type information. This avoids some cases where
2701 : : * transformArrayExpr() might not infer the correct type. Otherwise, just
2702 : : * transform the argument normally.
2703 : : */
2915 tgl@sss.pgh.pa.us 2704 [ + + ]: 119810 : if (IsA(arg, A_ArrayExpr))
2705 : : {
2706 : : Oid targetBaseType;
2707 : : int32 targetBaseTypmod;
2708 : : Oid elementType;
2709 : :
2710 : : /*
2711 : : * If target is a domain over array, work with the base array type
2712 : : * here. Below, we'll cast the array type to the domain. In the
2713 : : * usual case that the target is not a domain, the remaining steps
2714 : : * will be a no-op.
2715 : : */
3339 2716 : 346 : targetBaseTypmod = targetTypmod;
2717 : 346 : targetBaseType = getBaseTypeAndTypmod(targetType, &targetBaseTypmod);
2718 : 346 : elementType = get_element_type(targetBaseType);
2719 [ + + ]: 346 : if (OidIsValid(elementType))
2720 : : {
2721 : 341 : expr = transformArrayExpr(pstate,
2722 : : (A_ArrayExpr *) arg,
2723 : : targetBaseType,
2724 : : elementType,
2725 : : targetBaseTypmod);
2726 : : }
2727 : : else
2915 2728 : 5 : expr = transformExprRecurse(pstate, arg);
2729 : : }
2730 : : else
2731 : 119464 : expr = transformExprRecurse(pstate, arg);
2732 : :
3339 2733 : 119801 : inputType = exprType(expr);
8854 2734 [ - + ]: 119801 : if (inputType == InvalidOid)
8854 tgl@sss.pgh.pa.us 2735 :UBC 0 : return expr; /* do nothing if NULL input */
2736 : :
2737 : : /*
2738 : : * Location of the coercion is preferentially the location of the :: or
2739 : : * CAST symbol, but if there is none then use the location of the type
2740 : : * name (this can happen in TypeName 'string' syntax, for instance).
2741 : : */
5708 tgl@sss.pgh.pa.us 2742 :CBC 119801 : location = tc->location;
2743 [ + + ]: 119801 : if (location < 0)
5386 peter_e@gmx.net 2744 : 8195 : location = tc->typeName->location;
2745 : :
5708 tgl@sss.pgh.pa.us 2746 : 119801 : result = coerce_to_target_type(pstate, expr, inputType,
2747 : : targetType, targetTypmod,
2748 : : COERCION_EXPLICIT,
2749 : : COERCE_EXPLICIT_CAST,
2750 : : location);
2751 [ + + ]: 118152 : if (result == NULL)
7575 2752 [ + - ]: 11 : ereport(ERROR,
2753 : : (errcode(ERRCODE_CANNOT_COERCE),
2754 : : errmsg("cannot cast type %s to %s",
2755 : : format_type_be(inputType),
2756 : : format_type_be(targetType)),
2757 : : parser_coercion_errposition(pstate, location, expr)));
2758 : :
5708 2759 : 118141 : return result;
2760 : : }
2761 : :
2762 : : /*
2763 : : * Handle an explicit COLLATE clause.
2764 : : *
2765 : : * Transform the argument, and look up the collation name.
2766 : : */
2767 : : static Node *
4814 peter_e@gmx.net 2768 : 3839 : transformCollateClause(ParseState *pstate, CollateClause *c)
2769 : : {
2770 : : CollateExpr *newc;
2771 : : Oid argtype;
2772 : :
4783 tgl@sss.pgh.pa.us 2773 : 3839 : newc = makeNode(CollateExpr);
4265 2774 : 3839 : newc->arg = (Expr *) transformExprRecurse(pstate, c->arg);
2775 : :
4814 peter_e@gmx.net 2776 : 3839 : argtype = exprType((Node *) newc->arg);
2777 : :
2778 : : /*
2779 : : * The unknown type is not collatable, but coerce_type() takes care of it
2780 : : * separately, so we'll let it go here.
2781 : : */
2782 [ + + + + ]: 3839 : if (!type_is_collatable(argtype) && argtype != UNKNOWNOID)
2783 [ + - ]: 9 : ereport(ERROR,
2784 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2785 : : errmsg("collations are not supported by type %s",
2786 : : format_type_be(argtype)),
2787 : : parser_errposition(pstate, c->location)));
2788 : :
4783 tgl@sss.pgh.pa.us 2789 : 3830 : newc->collOid = LookupCollation(pstate, c->collname, c->location);
4814 peter_e@gmx.net 2790 : 3830 : newc->location = c->location;
2791 : :
2792 : 3830 : return (Node *) newc;
2793 : : }
2794 : :
2795 : : /*
2796 : : * Transform a "row compare-op row" construct
2797 : : *
2798 : : * The inputs are lists of already-transformed expressions.
2799 : : * As with coerce_type, pstate may be NULL if no special unknown-Param
2800 : : * processing is wanted.
2801 : : *
2802 : : * The output may be a single OpExpr, an AND or OR combination of OpExprs,
2803 : : * or a RowCompareExpr. In all cases it is guaranteed to return boolean.
2804 : : * The AND, OR, and RowCompareExpr cases further imply things about the
2805 : : * behavior of the operators (ie, they behave as =, <>, or < <= > >=).
2806 : : */
2807 : : static Node *
6682 tgl@sss.pgh.pa.us 2808 : 1593 : make_row_comparison_op(ParseState *pstate, List *opname,
2809 : : List *largs, List *rargs, int location)
2810 : : {
2811 : : RowCompareExpr *rcexpr;
2812 : : RowCompareType rctype;
2813 : : List *opexprs;
2814 : : List *opnos;
2815 : : List *opfamilies;
2816 : : ListCell *l,
2817 : : *r;
2818 : : List **opinfo_lists;
2819 : : Bitmapset *strats;
2820 : : int nopers;
2821 : : int i;
2822 : :
2823 : 1593 : nopers = list_length(largs);
2824 [ - + ]: 1593 : if (nopers != list_length(rargs))
7279 tgl@sss.pgh.pa.us 2825 [ # # ]:UBC 0 : ereport(ERROR,
2826 : : (errcode(ERRCODE_SYNTAX_ERROR),
2827 : : errmsg("unequal number of entries in row expressions"),
2828 : : parser_errposition(pstate, location)));
2829 : :
2830 : : /*
2831 : : * We can't compare zero-length rows because there is no principled basis
2832 : : * for figuring out what the operator is.
2833 : : */
6682 tgl@sss.pgh.pa.us 2834 [ + + ]:CBC 1593 : if (nopers == 0)
7279 2835 [ + - ]: 3 : ereport(ERROR,
2836 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2837 : : errmsg("cannot compare rows of zero length"),
2838 : : parser_errposition(pstate, location)));
2839 : :
2840 : : /*
2841 : : * Identify all the pairwise operators, using make_op so that behavior is
2842 : : * the same as in the simple scalar case.
2843 : : */
6682 2844 : 1590 : opexprs = NIL;
7263 neilc@samurai.com 2845 [ + - + + : 3706 : forboth(l, largs, r, rargs)
+ - + + +
+ + - +
+ ]
2846 : : {
7168 bruce@momjian.us 2847 : 2122 : Node *larg = (Node *) lfirst(l);
2848 : 2122 : Node *rarg = (Node *) lfirst(r);
2849 : : OpExpr *cmp;
2850 : :
2497 tgl@sss.pgh.pa.us 2851 : 2122 : cmp = castNode(OpExpr, make_op(pstate, opname, larg, rarg,
2852 : : pstate->p_last_srf, location));
2853 : :
2854 : : /*
2855 : : * We don't use coerce_to_boolean here because we insist on the
2856 : : * operator yielding boolean directly, not via coercion. If it
2857 : : * doesn't yield bool it won't be in any index opfamilies...
2858 : : */
6682 2859 [ - + ]: 2116 : if (cmp->opresulttype != BOOLOID)
6682 tgl@sss.pgh.pa.us 2860 [ # # ]:UBC 0 : ereport(ERROR,
2861 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2862 : : errmsg("row comparison operator must yield type boolean, "
2863 : : "not type %s",
2864 : : format_type_be(cmp->opresulttype)),
2865 : : parser_errposition(pstate, location)));
6682 tgl@sss.pgh.pa.us 2866 [ - + ]:CBC 2116 : if (expression_returns_set((Node *) cmp))
6682 tgl@sss.pgh.pa.us 2867 [ # # ]:UBC 0 : ereport(ERROR,
2868 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
2869 : : errmsg("row comparison operator must not return a set"),
2870 : : parser_errposition(pstate, location)));
6682 tgl@sss.pgh.pa.us 2871 :CBC 2116 : opexprs = lappend(opexprs, cmp);
2872 : : }
2873 : :
2874 : : /*
2875 : : * If rows are length 1, just return the single operator. In this case we
2876 : : * don't insist on identifying btree semantics for the operator (but we
2877 : : * still require it to return boolean).
2878 : : */
2879 [ + + ]: 1584 : if (nopers == 1)
2880 : 1162 : return (Node *) linitial(opexprs);
2881 : :
2882 : : /*
2883 : : * Now we must determine which row comparison semantics (= <> < <= > >=)
2884 : : * apply to this set of operators. We look for btree opfamilies
2885 : : * containing the operators, and see which interpretations (strategy
2886 : : * numbers) exist for each operator.
2887 : : */
4666 2888 : 422 : opinfo_lists = (List **) palloc(nopers * sizeof(List *));
6682 2889 : 422 : strats = NULL;
2890 : 422 : i = 0;
2891 [ + - + + : 1376 : foreach(l, opexprs)
+ + ]
2892 : : {
6322 2893 : 954 : Oid opno = ((OpExpr *) lfirst(l))->opno;
2894 : : Bitmapset *this_strats;
2895 : : ListCell *j;
2896 : :
4666 2897 : 954 : opinfo_lists[i] = get_op_btree_interpretation(opno);
2898 : :
2899 : : /*
2900 : : * convert strategy numbers into a Bitmapset to make the intersection
2901 : : * calculation easy.
2902 : : */
6682 2903 : 954 : this_strats = NULL;
4666 2904 [ + + + + : 1975 : foreach(j, opinfo_lists[i])
+ + ]
2905 : : {
2906 : 1021 : OpBtreeInterpretation *opinfo = lfirst(j);
2907 : :
2908 : 1021 : this_strats = bms_add_member(this_strats, opinfo->strategy);
2909 : : }
6682 2910 [ + + ]: 954 : if (i == 0)
2911 : 422 : strats = this_strats;
2912 : : else
2913 : 532 : strats = bms_int_members(strats, this_strats);
2914 : 954 : i++;
2915 : : }
2916 : :
2917 : : /*
2918 : : * If there are multiple common interpretations, we may use any one of
2919 : : * them ... this coding arbitrarily picks the lowest btree strategy
2920 : : * number.
2921 : : */
409 2922 : 422 : i = bms_next_member(strats, -1);
6322 2923 [ + + ]: 422 : if (i < 0)
2924 : : {
2925 : : /* No common interpretation, so fail */
2926 [ + - ]: 3 : ereport(ERROR,
2927 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2928 : : errmsg("could not determine interpretation of row comparison operator %s",
2929 : : strVal(llast(opname))),
2930 : : errhint("Row comparison operators must be associated with btree operator families."),
2931 : : parser_errposition(pstate, location)));
2932 : : }
2933 : 419 : rctype = (RowCompareType) i;
2934 : :
2935 : : /*
2936 : : * For = and <> cases, we just combine the pairwise operators with AND or
2937 : : * OR respectively.
2938 : : */
6682 2939 [ + + ]: 419 : if (rctype == ROWCOMPARE_EQ)
5708 2940 : 178 : return (Node *) makeBoolExpr(AND_EXPR, opexprs, location);
6682 2941 [ + + ]: 241 : if (rctype == ROWCOMPARE_NE)
5708 2942 : 154 : return (Node *) makeBoolExpr(OR_EXPR, opexprs, location);
2943 : :
2944 : : /*
2945 : : * Otherwise we need to choose exactly which opfamily to associate with
2946 : : * each operator.
2947 : : */
6322 2948 : 87 : opfamilies = NIL;
6682 2949 [ + + ]: 288 : for (i = 0; i < nopers; i++)
2950 : : {
6322 2951 : 201 : Oid opfamily = InvalidOid;
2952 : : ListCell *j;
2953 : :
4666 2954 [ + - + - : 201 : foreach(j, opinfo_lists[i])
+ - ]
2955 : : {
2956 : 201 : OpBtreeInterpretation *opinfo = lfirst(j);
2957 : :
2958 [ + - ]: 201 : if (opinfo->strategy == rctype)
2959 : : {
2960 : 201 : opfamily = opinfo->opfamily_id;
6322 2961 : 201 : break;
2962 : : }
2963 : : }
2964 [ + - ]: 201 : if (OidIsValid(opfamily))
2965 : 201 : opfamilies = lappend_oid(opfamilies, opfamily);
2966 : : else /* should not happen */
6682 tgl@sss.pgh.pa.us 2967 [ # # ]:UBC 0 : ereport(ERROR,
2968 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
2969 : : errmsg("could not determine interpretation of row comparison operator %s",
2970 : : strVal(llast(opname))),
2971 : : errdetail("There are multiple equally-plausible candidates."),
2972 : : parser_errposition(pstate, location)));
2973 : : }
2974 : :
2975 : : /*
2976 : : * Now deconstruct the OpExprs and create a RowCompareExpr.
2977 : : *
2978 : : * Note: can't just reuse the passed largs/rargs lists, because of
2979 : : * possibility that make_op inserted coercion operations.
2980 : : */
6682 tgl@sss.pgh.pa.us 2981 :CBC 87 : opnos = NIL;
2982 : 87 : largs = NIL;
2983 : 87 : rargs = NIL;
2984 [ + - + + : 288 : foreach(l, opexprs)
+ + ]
2985 : : {
2986 : 201 : OpExpr *cmp = (OpExpr *) lfirst(l);
2987 : :
2988 : 201 : opnos = lappend_oid(opnos, cmp->opno);
2989 : 201 : largs = lappend(largs, linitial(cmp->args));
2990 : 201 : rargs = lappend(rargs, lsecond(cmp->args));
2991 : : }
2992 : :
2993 : 87 : rcexpr = makeNode(RowCompareExpr);
2994 : 87 : rcexpr->rctype = rctype;
2995 : 87 : rcexpr->opnos = opnos;
6322 2996 : 87 : rcexpr->opfamilies = opfamilies;
4753 bruce@momjian.us 2997 : 87 : rcexpr->inputcollids = NIL; /* assign_expr_collations will fix this */
6682 tgl@sss.pgh.pa.us 2998 : 87 : rcexpr->largs = largs;
2999 : 87 : rcexpr->rargs = rargs;
3000 : :
3001 : 87 : return (Node *) rcexpr;
3002 : : }
3003 : :
3004 : : /*
3005 : : * Transform a "row IS DISTINCT FROM row" construct
3006 : : *
3007 : : * The input RowExprs are already transformed
3008 : : */
3009 : : static Node *
7279 3010 : 3 : make_row_distinct_op(ParseState *pstate, List *opname,
3011 : : RowExpr *lrow, RowExpr *rrow,
3012 : : int location)
3013 : : {
3014 : 3 : Node *result = NULL;
6712 3015 : 3 : List *largs = lrow->args;
3016 : 3 : List *rargs = rrow->args;
3017 : : ListCell *l,
3018 : : *r;
3019 : :
7259 neilc@samurai.com 3020 [ - + ]: 3 : if (list_length(largs) != list_length(rargs))
7279 tgl@sss.pgh.pa.us 3021 [ # # ]:UBC 0 : ereport(ERROR,
3022 : : (errcode(ERRCODE_SYNTAX_ERROR),
3023 : : errmsg("unequal number of entries in row expressions"),
3024 : : parser_errposition(pstate, location)));
3025 : :
7263 neilc@samurai.com 3026 [ + - + + :CBC 12 : forboth(l, largs, r, rargs)
+ - + + +
+ + - +
+ ]
3027 : : {
7168 bruce@momjian.us 3028 : 9 : Node *larg = (Node *) lfirst(l);
3029 : 9 : Node *rarg = (Node *) lfirst(r);
3030 : : Node *cmp;
3031 : :
6606 tgl@sss.pgh.pa.us 3032 : 9 : cmp = (Node *) make_distinct_op(pstate, opname, larg, rarg, location);
7279 3033 [ + + ]: 9 : if (result == NULL)
3034 : 3 : result = cmp;
3035 : : else
3036 : 6 : result = (Node *) makeBoolExpr(OR_EXPR,
5708 3037 : 6 : list_make2(result, cmp),
3038 : : location);
3039 : : }
3040 : :
7279 3041 [ - + ]: 3 : if (result == NULL)
3042 : : {
3043 : : /* zero-length rows? Generate constant FALSE */
7279 tgl@sss.pgh.pa.us 3044 :UBC 0 : result = makeBoolConst(false, false);
3045 : : }
3046 : :
7279 tgl@sss.pgh.pa.us 3047 :CBC 3 : return result;
3048 : : }
3049 : :
3050 : : /*
3051 : : * make the node for an IS DISTINCT FROM operator
3052 : : */
3053 : : static Expr *
6606 3054 : 455 : make_distinct_op(ParseState *pstate, List *opname, Node *ltree, Node *rtree,
3055 : : int location)
3056 : : {
3057 : : Expr *result;
3058 : :
2497 3059 : 455 : result = make_op(pstate, opname, ltree, rtree,
3060 : : pstate->p_last_srf, location);
7279 3061 [ - + ]: 455 : if (((OpExpr *) result)->opresulttype != BOOLOID)
7279 tgl@sss.pgh.pa.us 3062 [ # # ]:UBC 0 : ereport(ERROR,
3063 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
3064 : : errmsg("IS DISTINCT FROM requires = operator to yield boolean"),
3065 : : parser_errposition(pstate, location)));
2497 tgl@sss.pgh.pa.us 3066 [ - + ]:CBC 455 : if (((OpExpr *) result)->opretset)
2497 tgl@sss.pgh.pa.us 3067 [ # # ]:UBC 0 : ereport(ERROR,
3068 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
3069 : : /* translator: %s is name of a SQL construct, eg NULLIF */
3070 : : errmsg("%s must not return a set", "IS DISTINCT FROM"),
3071 : : parser_errposition(pstate, location)));
3072 : :
3073 : : /*
3074 : : * We rely on DistinctExpr and OpExpr being same struct
3075 : : */
7279 tgl@sss.pgh.pa.us 3076 :CBC 455 : NodeSetTag(result, T_DistinctExpr);
3077 : :
3078 : 455 : return result;
3079 : : }
3080 : :
3081 : : /*
3082 : : * Produce a NullTest node from an IS [NOT] DISTINCT FROM NULL construct
3083 : : *
3084 : : * "arg" is the untransformed other argument
3085 : : */
3086 : : static Node *
2817 3087 : 15 : make_nulltest_from_distinct(ParseState *pstate, A_Expr *distincta, Node *arg)
3088 : : {
3089 : 15 : NullTest *nt = makeNode(NullTest);
3090 : :
3091 : 15 : nt->arg = (Expr *) transformExprRecurse(pstate, arg);
3092 : : /* the argument can be any type, so don't coerce it */
3093 [ + + ]: 15 : if (distincta->kind == AEXPR_NOT_DISTINCT)
3094 : 6 : nt->nulltesttype = IS_NULL;
3095 : : else
3096 : 9 : nt->nulltesttype = IS_NOT_NULL;
3097 : : /* argisrow = false is correct whether or not arg is composite */
3098 : 15 : nt->argisrow = false;
3099 : 15 : nt->location = distincta->location;
3100 : 15 : return (Node *) nt;
3101 : : }
3102 : :
3103 : : /*
3104 : : * Produce a string identifying an expression by kind.
3105 : : *
3106 : : * Note: when practical, use a simple SQL keyword for the result. If that
3107 : : * doesn't work well, check call sites to see whether custom error message
3108 : : * strings are required.
3109 : : */
3110 : : const char *
4265 3111 : 39 : ParseExprKindName(ParseExprKind exprKind)
3112 : : {
3113 [ - - - - : 39 : switch (exprKind)
- - + - -
+ - - - -
- - - + -
+ - - + -
+ + - - -
- - - - -
- - - - -
- - ]
3114 : : {
4265 tgl@sss.pgh.pa.us 3115 :UBC 0 : case EXPR_KIND_NONE:
3116 : 0 : return "invalid expression context";
3117 : 0 : case EXPR_KIND_OTHER:
3118 : 0 : return "extension expression";
3119 : 0 : case EXPR_KIND_JOIN_ON:
3120 : 0 : return "JOIN/ON";
3121 : 0 : case EXPR_KIND_JOIN_USING:
3122 : 0 : return "JOIN/USING";
3123 : 0 : case EXPR_KIND_FROM_SUBSELECT:
3124 : 0 : return "sub-SELECT in FROM";
3125 : 0 : case EXPR_KIND_FROM_FUNCTION:
3126 : 0 : return "function in FROM";
4265 tgl@sss.pgh.pa.us 3127 :CBC 12 : case EXPR_KIND_WHERE:
3128 : 12 : return "WHERE";
3182 mail@joeconway.com 3129 :UBC 0 : case EXPR_KIND_POLICY:
3130 : 0 : return "POLICY";
4265 tgl@sss.pgh.pa.us 3131 : 0 : case EXPR_KIND_HAVING:
3132 : 0 : return "HAVING";
3925 noah@leadboat.com 3133 :CBC 6 : case EXPR_KIND_FILTER:
3134 : 6 : return "FILTER";
4265 tgl@sss.pgh.pa.us 3135 :UBC 0 : case EXPR_KIND_WINDOW_PARTITION:
3136 : 0 : return "window PARTITION BY";
3137 : 0 : case EXPR_KIND_WINDOW_ORDER:
3138 : 0 : return "window ORDER BY";
3139 : 0 : case EXPR_KIND_WINDOW_FRAME_RANGE:
3140 : 0 : return "window RANGE";
3141 : 0 : case EXPR_KIND_WINDOW_FRAME_ROWS:
3142 : 0 : return "window ROWS";
2258 3143 : 0 : case EXPR_KIND_WINDOW_FRAME_GROUPS:
3144 : 0 : return "window GROUPS";
4265 3145 : 0 : case EXPR_KIND_SELECT_TARGET:
3146 : 0 : return "SELECT";
3147 : 0 : case EXPR_KIND_INSERT_TARGET:
3148 : 0 : return "INSERT";
4265 tgl@sss.pgh.pa.us 3149 :CBC 3 : case EXPR_KIND_UPDATE_SOURCE:
3150 : : case EXPR_KIND_UPDATE_TARGET:
3151 : 3 : return "UPDATE";
748 alvherre@alvh.no-ip. 3152 :UBC 0 : case EXPR_KIND_MERGE_WHEN:
3153 : 0 : return "MERGE WHEN";
4265 tgl@sss.pgh.pa.us 3154 :CBC 6 : case EXPR_KIND_GROUP_BY:
3155 : 6 : return "GROUP BY";
4265 tgl@sss.pgh.pa.us 3156 :UBC 0 : case EXPR_KIND_ORDER_BY:
3157 : 0 : return "ORDER BY";
3158 : 0 : case EXPR_KIND_DISTINCT_ON:
3159 : 0 : return "DISTINCT ON";
4265 tgl@sss.pgh.pa.us 3160 :CBC 3 : case EXPR_KIND_LIMIT:
3161 : 3 : return "LIMIT";
4265 tgl@sss.pgh.pa.us 3162 :UBC 0 : case EXPR_KIND_OFFSET:
3163 : 0 : return "OFFSET";
4265 tgl@sss.pgh.pa.us 3164 :CBC 6 : case EXPR_KIND_RETURNING:
3165 : : case EXPR_KIND_MERGE_RETURNING:
3166 : 6 : return "RETURNING";
3167 : 3 : case EXPR_KIND_VALUES:
3168 : : case EXPR_KIND_VALUES_SINGLE:
3169 : 3 : return "VALUES";
4265 tgl@sss.pgh.pa.us 3170 :UBC 0 : case EXPR_KIND_CHECK_CONSTRAINT:
3171 : : case EXPR_KIND_DOMAIN_CHECK:
3172 : 0 : return "CHECK";
3173 : 0 : case EXPR_KIND_COLUMN_DEFAULT:
3174 : : case EXPR_KIND_FUNCTION_DEFAULT:
3175 : 0 : return "DEFAULT";
3176 : 0 : case EXPR_KIND_INDEX_EXPRESSION:
3177 : 0 : return "index expression";
3178 : 0 : case EXPR_KIND_INDEX_PREDICATE:
3179 : 0 : return "index predicate";
1115 tomas.vondra@postgre 3180 : 0 : case EXPR_KIND_STATS_EXPRESSION:
3181 : 0 : return "statistics expression";
4265 tgl@sss.pgh.pa.us 3182 : 0 : case EXPR_KIND_ALTER_COL_TRANSFORM:
3183 : 0 : return "USING";
3184 : 0 : case EXPR_KIND_EXECUTE_PARAMETER:
3185 : 0 : return "EXECUTE";
3186 : 0 : case EXPR_KIND_TRIGGER_WHEN:
3187 : 0 : return "WHEN";
1906 peter@eisentraut.org 3188 : 0 : case EXPR_KIND_PARTITION_BOUND:
3189 : 0 : return "partition bound";
2685 rhaas@postgresql.org 3190 : 0 : case EXPR_KIND_PARTITION_EXPRESSION:
3191 : 0 : return "PARTITION BY";
2255 tgl@sss.pgh.pa.us 3192 : 0 : case EXPR_KIND_CALL_ARGUMENT:
2327 peter_e@gmx.net 3193 : 0 : return "CALL";
1912 tomas.vondra@postgre 3194 : 0 : case EXPR_KIND_COPY_WHERE:
3195 : 0 : return "WHERE";
1842 peter@eisentraut.org 3196 : 0 : case EXPR_KIND_GENERATED_COLUMN:
3197 : 0 : return "GENERATED AS";
1168 3198 : 0 : case EXPR_KIND_CYCLE_MARK:
3199 : 0 : return "CYCLE";
3200 : :
3201 : : /*
3202 : : * There is intentionally no default: case here, so that the
3203 : : * compiler will warn if we add a new ParseExprKind without
3204 : : * extending this switch. If we do see an unrecognized value at
3205 : : * runtime, we'll fall through to the "unrecognized" return.
3206 : : */
3207 : : }
4265 tgl@sss.pgh.pa.us 3208 : 0 : return "unrecognized expression kind";
3209 : : }
3210 : :
3211 : : /*
3212 : : * Make string Const node from JSON encoding name.
3213 : : *
3214 : : * UTF8 is default encoding.
3215 : : */
3216 : : static Const *
382 alvherre@alvh.no-ip. 3217 :CBC 105 : getJsonEncodingConst(JsonFormat *format)
3218 : : {
3219 : : JsonEncoding encoding;
3220 : : const char *enc;
3221 : 105 : Name encname = palloc(sizeof(NameData));
3222 : :
3223 [ + - ]: 105 : if (!format ||
3224 [ + + ]: 105 : format->format_type == JS_FORMAT_DEFAULT ||
3225 [ + + ]: 75 : format->encoding == JS_ENC_DEFAULT)
3226 : 93 : encoding = JS_ENC_UTF8;
3227 : : else
3228 : 12 : encoding = format->encoding;
3229 : :
3230 [ - - + - ]: 105 : switch (encoding)
3231 : : {
382 alvherre@alvh.no-ip. 3232 :UBC 0 : case JS_ENC_UTF16:
3233 : 0 : enc = "UTF16";
3234 : 0 : break;
3235 : 0 : case JS_ENC_UTF32:
3236 : 0 : enc = "UTF32";
3237 : 0 : break;
382 alvherre@alvh.no-ip. 3238 :CBC 105 : case JS_ENC_UTF8:
3239 : 105 : enc = "UTF8";
3240 : 105 : break;
382 alvherre@alvh.no-ip. 3241 :UBC 0 : default:
3242 [ # # ]: 0 : elog(ERROR, "invalid JSON encoding: %d", encoding);
3243 : : break;
3244 : : }
3245 : :
382 alvherre@alvh.no-ip. 3246 :CBC 105 : namestrcpy(encname, enc);
3247 : :
3248 : 105 : return makeConst(NAMEOID, -1, InvalidOid, NAMEDATALEN,
3249 : : NameGetDatum(encname), false, false);
3250 : : }
3251 : :
3252 : : /*
3253 : : * Make bytea => text conversion using specified JSON format encoding.
3254 : : */
3255 : : static Node *
3256 : 60 : makeJsonByteaToTextConversion(Node *expr, JsonFormat *format, int location)
3257 : : {
3258 : 60 : Const *encoding = getJsonEncodingConst(format);
3259 : 60 : FuncExpr *fexpr = makeFuncExpr(F_CONVERT_FROM, TEXTOID,
3260 : 60 : list_make2(expr, encoding),
3261 : : InvalidOid, InvalidOid,
3262 : : COERCE_EXPLICIT_CALL);
3263 : :
3264 : 60 : fexpr->location = location;
3265 : :
3266 : 60 : return (Node *) fexpr;
3267 : : }
3268 : :
3269 : : /*
3270 : : * Transform JSON value expression using specified input JSON format or
3271 : : * default format otherwise, coercing to the targettype if needed.
3272 : : *
3273 : : * Returned expression is either ve->raw_expr coerced to text (if needed) or
3274 : : * a JsonValueExpr with formatted_expr set to the coerced copy of raw_expr
3275 : : * if the specified format and the targettype requires it.
3276 : : */
3277 : : static Node *
276 amitlan@postgresql.o 3278 : 2510 : transformJsonValueExpr(ParseState *pstate, const char *constructName,
3279 : : JsonValueExpr *ve, JsonFormatType default_format,
3280 : : Oid targettype, bool isarg)
3281 : : {
382 alvherre@alvh.no-ip. 3282 : 2510 : Node *expr = transformExprRecurse(pstate, (Node *) ve->raw_expr);
3283 : : Node *rawexpr;
3284 : : JsonFormatType format;
3285 : : Oid exprtype;
3286 : : int location;
3287 : : char typcategory;
3288 : : bool typispreferred;
3289 : :
3290 [ + + ]: 2510 : if (exprType(expr) == UNKNOWNOID)
282 amitlan@postgresql.o 3291 : 278 : expr = coerce_to_specific_type(pstate, expr, TEXTOID, constructName);
3292 : :
382 alvherre@alvh.no-ip. 3293 : 2510 : rawexpr = expr;
3294 : 2510 : exprtype = exprType(expr);
3295 : 2510 : location = exprLocation(expr);
3296 : :
3297 : 2510 : get_type_category_preferred(exprtype, &typcategory, &typispreferred);
3298 : :
3299 [ + + ]: 2510 : if (ve->format->format_type != JS_FORMAT_DEFAULT)
3300 : : {
3301 [ + + + + ]: 93 : if (ve->format->encoding != JS_ENC_DEFAULT && exprtype != BYTEAOID)
3302 [ + - ]: 13 : ereport(ERROR,
3303 : : errcode(ERRCODE_DATATYPE_MISMATCH),
3304 : : errmsg("JSON ENCODING clause is only allowed for bytea input type"),
3305 : : parser_errposition(pstate, ve->format->location));
3306 : :
3307 [ + + + + ]: 80 : if (exprtype == JSONOID || exprtype == JSONBOID)
3308 : 6 : format = JS_FORMAT_DEFAULT; /* do not format json[b] types */
3309 : : else
3310 : 74 : format = ve->format->format_type;
3311 : : }
24 amitlan@postgresql.o 3312 [ + + ]:GNC 2417 : else if (isarg)
3313 : : {
3314 : : /*
3315 : : * Special treatment for PASSING arguments.
3316 : : *
3317 : : * Pass types supported by GetJsonPathVar() / JsonItemFromDatum()
3318 : : * directly without converting to json[b].
3319 : : */
3320 [ + + ]: 582 : switch (exprtype)
3321 : : {
3322 : 447 : case BOOLOID:
3323 : : case NUMERICOID:
3324 : : case INT2OID:
3325 : : case INT4OID:
3326 : : case INT8OID:
3327 : : case FLOAT4OID:
3328 : : case FLOAT8OID:
3329 : : case TEXTOID:
3330 : : case VARCHAROID:
3331 : : case DATEOID:
3332 : : case TIMEOID:
3333 : : case TIMETZOID:
3334 : : case TIMESTAMPOID:
3335 : : case TIMESTAMPTZOID:
3336 : 447 : return expr;
3337 : :
3338 : 135 : default:
3339 [ - + ]: 135 : if (typcategory == TYPCATEGORY_STRING)
24 amitlan@postgresql.o 3340 :UNC 0 : return expr;
3341 : : /* else convert argument to json[b] type */
24 amitlan@postgresql.o 3342 :GNC 135 : break;
3343 : : }
3344 : :
3345 : 135 : format = default_format;
3346 : : }
382 alvherre@alvh.no-ip. 3347 [ + + + + ]:CBC 1835 : else if (exprtype == JSONOID || exprtype == JSONBOID)
3348 : 1311 : format = JS_FORMAT_DEFAULT; /* do not format json[b] types */
3349 : : else
3350 : 524 : format = default_format;
3351 : :
269 amitlan@postgresql.o 3352 [ + + + + ]:GNC 2050 : if (format != JS_FORMAT_DEFAULT ||
3353 [ + + ]: 1289 : (OidIsValid(targettype) && exprtype != targettype))
3354 : : {
3355 : : Node *coerced;
3356 : 298 : bool only_allow_cast = OidIsValid(targettype);
3357 : :
3358 : : /*
3359 : : * PASSING args are handled appropriately by GetJsonPathVar() /
3360 : : * JsonItemFromDatum().
3361 : : */
24 3362 [ + + ]: 298 : if (!isarg &&
3363 [ + + + + ]: 163 : !only_allow_cast &&
269 3364 [ + + ]: 65 : exprtype != BYTEAOID && typcategory != TYPCATEGORY_STRING)
382 alvherre@alvh.no-ip. 3365 [ + - - + :CBC 3 : ereport(ERROR,
+ - ]
3366 : : errcode(ERRCODE_DATATYPE_MISMATCH),
3367 : : ve->format->format_type == JS_FORMAT_DEFAULT ?
3368 : : errmsg("cannot use non-string types with implicit FORMAT JSON clause") :
3369 : : errmsg("cannot use non-string types with explicit FORMAT JSON clause"),
3370 : : parser_errposition(pstate, ve->format->location >= 0 ?
3371 : : ve->format->location : location));
3372 : :
3373 : : /* Convert encoded JSON text from bytea. */
3374 [ + + + + ]: 295 : if (format == JS_FORMAT_JSON && exprtype == BYTEAOID)
3375 : : {
3376 : 30 : expr = makeJsonByteaToTextConversion(expr, ve->format, location);
3377 : 30 : exprtype = TEXTOID;
3378 : : }
3379 : :
269 amitlan@postgresql.o 3380 [ + + ]:GNC 295 : if (!OidIsValid(targettype))
3381 [ + + ]: 218 : targettype = format == JS_FORMAT_JSONB ? JSONBOID : JSONOID;
3382 : :
3383 : : /* Try to coerce to the target type. */
382 alvherre@alvh.no-ip. 3384 :CBC 295 : coerced = coerce_to_target_type(pstate, expr, exprtype,
3385 : : targettype, -1,
3386 : : COERCION_EXPLICIT,
3387 : : COERCE_EXPLICIT_CAST,
3388 : : location);
3389 : :
3390 [ + + ]: 295 : if (!coerced)
3391 : : {
3392 : : /* If coercion failed, use to_json()/to_jsonb() functions. */
3393 : : FuncExpr *fexpr;
3394 : : Oid fnoid;
3395 : :
3396 : : /*
3397 : : * Though only allow a cast when the target type is specified by
3398 : : * the caller.
3399 : : */
269 amitlan@postgresql.o 3400 [ + + ]:GNC 15 : if (only_allow_cast)
3401 [ + - ]: 3 : ereport(ERROR,
3402 : : (errcode(ERRCODE_CANNOT_COERCE),
3403 : : errmsg("cannot cast type %s to %s",
3404 : : format_type_be(exprtype),
3405 : : format_type_be(targettype)),
3406 : : parser_errposition(pstate, location)));
3407 : :
3408 [ - + ]: 12 : fnoid = targettype == JSONOID ? F_TO_JSON : F_TO_JSONB;
3409 : 12 : fexpr = makeFuncExpr(fnoid, targettype, list_make1(expr),
3410 : : InvalidOid, InvalidOid, COERCE_EXPLICIT_CALL);
3411 : :
382 alvherre@alvh.no-ip. 3412 :GBC 12 : fexpr->location = location;
3413 : :
3414 : 12 : coerced = (Node *) fexpr;
3415 : : }
3416 : :
282 amitlan@postgresql.o 3417 [ - + ]:CBC 292 : if (coerced == expr)
382 alvherre@alvh.no-ip. 3418 :UBC 0 : expr = rawexpr;
3419 : : else
3420 : : {
382 alvherre@alvh.no-ip. 3421 :CBC 292 : ve = copyObject(ve);
3422 : 292 : ve->raw_expr = (Expr *) rawexpr;
3423 : 292 : ve->formatted_expr = (Expr *) coerced;
3424 : :
3425 : 292 : expr = (Node *) ve;
3426 : : }
3427 : : }
3428 : :
3429 : : /* If returning a JsonValueExpr, formatted_expr must have been set. */
268 amitlan@postgresql.o 3430 [ + + - + ]: 2044 : Assert(!IsA(expr, JsonValueExpr) ||
3431 : : ((JsonValueExpr *) expr)->formatted_expr != NULL);
3432 : :
382 alvherre@alvh.no-ip. 3433 : 2044 : return expr;
3434 : : }
3435 : :
3436 : : /*
3437 : : * Checks specified output format for its applicability to the target type.
3438 : : */
3439 : : static void
3440 : 122 : checkJsonOutputFormat(ParseState *pstate, const JsonFormat *format,
3441 : : Oid targettype, bool allow_format_for_non_strings)
3442 : : {
3443 [ + + ]: 122 : if (!allow_format_for_non_strings &&
382 alvherre@alvh.no-ip. 3444 [ + - + + ]:GBC 72 : format->format_type != JS_FORMAT_DEFAULT &&
3445 [ + + ]: 63 : (targettype != BYTEAOID &&
3446 [ + + ]: 60 : targettype != JSONOID &&
3447 : : targettype != JSONBOID))
3448 : : {
3449 : : char typcategory;
3450 : : bool typispreferred;
3451 : :
3452 : 45 : get_type_category_preferred(targettype, &typcategory, &typispreferred);
3453 : :
3454 [ - + ]: 45 : if (typcategory != TYPCATEGORY_STRING)
382 alvherre@alvh.no-ip. 3455 [ # # ]:UBC 0 : ereport(ERROR,
3456 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3457 : : parser_errposition(pstate, format->location),
3458 : : errmsg("cannot use JSON format with non-string output types"));
3459 : : }
3460 : :
382 alvherre@alvh.no-ip. 3461 [ + - ]:CBC 122 : if (format->format_type == JS_FORMAT_JSON)
3462 : : {
3463 : 244 : JsonEncoding enc = format->encoding != JS_ENC_DEFAULT ?
331 tgl@sss.pgh.pa.us 3464 [ + + ]: 122 : format->encoding : JS_ENC_UTF8;
3465 : :
382 alvherre@alvh.no-ip. 3466 [ + + ]: 122 : if (targettype != BYTEAOID &&
3467 [ + + ]: 89 : format->encoding != JS_ENC_DEFAULT)
3468 [ + - ]: 6 : ereport(ERROR,
3469 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3470 : : parser_errposition(pstate, format->location),
3471 : : errmsg("cannot set JSON encoding for non-bytea output types"));
3472 : :
3473 [ + + ]: 116 : if (enc != JS_ENC_UTF8)
3474 [ + - ]: 12 : ereport(ERROR,
3475 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3476 : : errmsg("unsupported JSON encoding"),
3477 : : errhint("Only UTF8 JSON encoding is supported."),
3478 : : parser_errposition(pstate, format->location));
3479 : : }
3480 : 104 : }
3481 : :
3482 : : /*
3483 : : * Transform JSON output clause.
3484 : : *
3485 : : * Assigns target type oid and modifier.
3486 : : * Assigns default format or checks specified format for its applicability to
3487 : : * the target type.
3488 : : */
3489 : : static JsonReturning *
3490 : 1738 : transformJsonOutput(ParseState *pstate, const JsonOutput *output,
3491 : : bool allow_format)
3492 : : {
3493 : : JsonReturning *ret;
3494 : :
3495 : : /* if output clause is not specified, make default clause value */
3496 [ + + ]: 1738 : if (!output)
3497 : : {
3498 : 775 : ret = makeNode(JsonReturning);
3499 : :
3500 : 775 : ret->format = makeJsonFormat(JS_FORMAT_DEFAULT, JS_ENC_DEFAULT, -1);
3501 : 775 : ret->typid = InvalidOid;
3502 : 775 : ret->typmod = -1;
3503 : :
3504 : 775 : return ret;
3505 : : }
3506 : :
3507 : 963 : ret = copyObject(output->returning);
3508 : :
3509 : 963 : typenameTypeIdAndMod(pstate, output->typeName, &ret->typid, &ret->typmod);
3510 : :
3511 [ - + ]: 963 : if (output->typeName->setof)
382 alvherre@alvh.no-ip. 3512 [ # # ]:UBC 0 : ereport(ERROR,
3513 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3514 : : errmsg("returning SETOF types is not supported in SQL/JSON functions"));
3515 : :
24 amitlan@postgresql.o 3516 [ + + ]:GNC 963 : if (get_typtype(ret->typid) == TYPTYPE_PSEUDO)
3517 [ + - ]: 6 : ereport(ERROR,
3518 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3519 : : errmsg("returning pseudo-types is not supported in SQL/JSON functions"));
3520 : :
382 alvherre@alvh.no-ip. 3521 [ + + ]:CBC 957 : if (ret->format->format_type == JS_FORMAT_DEFAULT)
3522 : : /* assign JSONB format when returning jsonb, or JSON format otherwise */
3523 : 835 : ret->format->format_type =
3524 [ + + ]: 835 : ret->typid == JSONBOID ? JS_FORMAT_JSONB : JS_FORMAT_JSON;
3525 : : else
3526 : 122 : checkJsonOutputFormat(pstate, ret->format, ret->typid, allow_format);
3527 : :
3528 : 939 : return ret;
3529 : : }
3530 : :
3531 : : /*
3532 : : * Transform JSON output clause of JSON constructor functions.
3533 : : *
3534 : : * Derive RETURNING type, if not specified, from argument types.
3535 : : */
3536 : : static JsonReturning *
3537 : 433 : transformJsonConstructorOutput(ParseState *pstate, JsonOutput *output,
3538 : : List *args)
3539 : : {
3540 : 433 : JsonReturning *returning = transformJsonOutput(pstate, output, true);
3541 : :
3542 [ + + ]: 415 : if (!OidIsValid(returning->typid))
3543 : : {
3544 : : ListCell *lc;
3545 : 206 : bool have_jsonb = false;
3546 : :
3547 [ + + + + : 687 : foreach(lc, args)
+ + ]
3548 : : {
3549 : 493 : Node *expr = lfirst(lc);
3550 : 493 : Oid typid = exprType(expr);
3551 : :
3552 : 493 : have_jsonb |= typid == JSONBOID;
3553 : :
3554 [ + + ]: 493 : if (have_jsonb)
3555 : 12 : break;
3556 : : }
3557 : :
3558 [ + + ]: 206 : if (have_jsonb)
3559 : : {
3560 : 12 : returning->typid = JSONBOID;
3561 : 12 : returning->format->format_type = JS_FORMAT_JSONB;
3562 : : }
3563 : : else
3564 : : {
3565 : : /* XXX TEXT is default by the standard, but we return JSON */
3566 : 194 : returning->typid = JSONOID;
3567 : 194 : returning->format->format_type = JS_FORMAT_JSON;
3568 : : }
3569 : :
3570 : 206 : returning->typmod = -1;
3571 : : }
3572 : :
3573 : 415 : return returning;
3574 : : }
3575 : :
3576 : : /*
3577 : : * Coerce json[b]-valued function expression to the output type.
3578 : : */
3579 : : static Node *
3580 : 844 : coerceJsonFuncExpr(ParseState *pstate, Node *expr,
3581 : : const JsonReturning *returning, bool report_error)
3582 : : {
3583 : : Node *res;
3584 : : int location;
3585 : 844 : Oid exprtype = exprType(expr);
3586 : :
3587 : : /* if output type is not specified or equals to function type, return */
3588 [ + - + + ]: 844 : if (!OidIsValid(returning->typid) || returning->typid == exprtype)
3589 : 446 : return expr;
3590 : :
3591 : 398 : location = exprLocation(expr);
3592 : :
3593 [ + - ]: 398 : if (location < 0)
3594 : 398 : location = returning->format->location;
3595 : :
3596 : : /* special case for RETURNING bytea FORMAT json */
3597 [ + + ]: 398 : if (returning->format->format_type == JS_FORMAT_JSON &&
3598 [ + + ]: 347 : returning->typid == BYTEAOID)
3599 : : {
3600 : : /* encode json text into bytea using pg_convert_to() */
3601 : 45 : Node *texpr = coerce_to_specific_type(pstate, expr, TEXTOID,
3602 : : "JSON_FUNCTION");
3603 : 45 : Const *enc = getJsonEncodingConst(returning->format);
3604 : 45 : FuncExpr *fexpr = makeFuncExpr(F_CONVERT_TO, BYTEAOID,
3605 : 45 : list_make2(texpr, enc),
3606 : : InvalidOid, InvalidOid,
3607 : : COERCE_EXPLICIT_CALL);
3608 : :
3609 : 45 : fexpr->location = location;
3610 : :
3611 : 45 : return (Node *) fexpr;
3612 : : }
3613 : :
3614 : : /* try to coerce expression to the output type */
3615 : 353 : res = coerce_to_target_type(pstate, expr, exprtype,
3616 : 353 : returning->typid, returning->typmod,
3617 : : COERCION_EXPLICIT,
3618 : : COERCE_EXPLICIT_CAST,
3619 : : location);
3620 : :
3621 [ + + - + ]: 353 : if (!res && report_error)
382 alvherre@alvh.no-ip. 3622 [ # # ]:UBC 0 : ereport(ERROR,
3623 : : errcode(ERRCODE_CANNOT_COERCE),
3624 : : errmsg("cannot cast type %s to %s",
3625 : : format_type_be(exprtype),
3626 : : format_type_be(returning->typid)),
3627 : : parser_coercion_errposition(pstate, location, expr));
3628 : :
382 alvherre@alvh.no-ip. 3629 :CBC 353 : return res;
3630 : : }
3631 : :
3632 : : /*
3633 : : * Make a JsonConstructorExpr node.
3634 : : */
3635 : : static Node *
3636 : 562 : makeJsonConstructorExpr(ParseState *pstate, JsonConstructorType type,
3637 : : List *args, Expr *fexpr, JsonReturning *returning,
3638 : : bool unique, bool absent_on_null, int location)
3639 : : {
3640 : 562 : JsonConstructorExpr *jsctor = makeNode(JsonConstructorExpr);
3641 : : Node *placeholder;
3642 : : Node *coercion;
3643 : :
3644 : 562 : jsctor->args = args;
3645 : 562 : jsctor->func = fexpr;
3646 : 562 : jsctor->type = type;
3647 : 562 : jsctor->returning = returning;
3648 : 562 : jsctor->unique = unique;
3649 : 562 : jsctor->absent_on_null = absent_on_null;
3650 : 562 : jsctor->location = location;
3651 : :
3652 : : /*
3653 : : * Coerce to the RETURNING type and format, if needed. We abuse
3654 : : * CaseTestExpr here as placeholder to pass the result of either
3655 : : * evaluating 'fexpr' or whatever is produced by ExecEvalJsonConstructor()
3656 : : * that is of type JSON or JSONB to the coercion function.
3657 : : */
3658 [ + + ]: 562 : if (fexpr)
3659 : : {
282 amitlan@postgresql.o 3660 : 171 : CaseTestExpr *cte = makeNode(CaseTestExpr);
3661 : :
3662 : 171 : cte->typeId = exprType((Node *) fexpr);
3663 : 171 : cte->typeMod = exprTypmod((Node *) fexpr);
3664 : 171 : cte->collation = exprCollation((Node *) fexpr);
3665 : :
3666 : 171 : placeholder = (Node *) cte;
3667 : : }
3668 : : else
3669 : : {
382 alvherre@alvh.no-ip. 3670 : 391 : CaseTestExpr *cte = makeNode(CaseTestExpr);
3671 : :
381 3672 : 782 : cte->typeId = returning->format->format_type == JS_FORMAT_JSONB ?
3673 [ + + ]: 391 : JSONBOID : JSONOID;
382 3674 : 391 : cte->typeMod = -1;
3675 : 391 : cte->collation = InvalidOid;
3676 : :
3677 : 391 : placeholder = (Node *) cte;
3678 : : }
3679 : :
3680 : 562 : coercion = coerceJsonFuncExpr(pstate, placeholder, returning, true);
3681 : :
3682 [ + + ]: 562 : if (coercion != placeholder)
3683 : 116 : jsctor->coercion = (Expr *) coercion;
3684 : :
3685 : 562 : return (Node *) jsctor;
3686 : : }
3687 : :
3688 : : /*
3689 : : * Transform JSON_OBJECT() constructor.
3690 : : *
3691 : : * JSON_OBJECT() is transformed into json[b]_build_object[_ext]() call
3692 : : * depending on the output JSON format. The first two arguments of
3693 : : * json[b]_build_object_ext() are absent_on_null and check_unique.
3694 : : *
3695 : : * Then function call result is coerced to the target type.
3696 : : */
3697 : : static Node *
3698 : 185 : transformJsonObjectConstructor(ParseState *pstate, JsonObjectConstructor *ctor)
3699 : : {
3700 : : JsonReturning *returning;
3701 : 185 : List *args = NIL;
3702 : :
3703 : : /* transform key-value pairs, if any */
3704 [ + + ]: 185 : if (ctor->exprs)
3705 : : {
3706 : : ListCell *lc;
3707 : :
3708 : : /* transform and append key-value arguments */
3709 [ + - + + : 382 : foreach(lc, ctor->exprs)
+ + ]
3710 : : {
3711 : 253 : JsonKeyValue *kv = castNode(JsonKeyValue, lfirst(lc));
3712 : 253 : Node *key = transformExprRecurse(pstate, (Node *) kv->key);
282 amitlan@postgresql.o 3713 : 253 : Node *val = transformJsonValueExpr(pstate, "JSON_OBJECT()",
3714 : : kv->value,
3715 : : JS_FORMAT_DEFAULT,
3716 : : InvalidOid, false);
3717 : :
382 alvherre@alvh.no-ip. 3718 : 241 : args = lappend(args, key);
3719 : 241 : args = lappend(args, val);
3720 : : }
3721 : : }
3722 : :
3723 : 173 : returning = transformJsonConstructorOutput(pstate, ctor->output, args);
3724 : :
3725 : 328 : return makeJsonConstructorExpr(pstate, JSCTOR_JSON_OBJECT, args, NULL,
3726 : 164 : returning, ctor->unique,
3727 : 164 : ctor->absent_on_null, ctor->location);
3728 : : }
3729 : :
3730 : : /*
3731 : : * Transform JSON_ARRAY(query [FORMAT] [RETURNING] [ON NULL]) into
3732 : : * (SELECT JSON_ARRAYAGG(a [FORMAT] [RETURNING] [ON NULL]) FROM (query) q(a))
3733 : : */
3734 : : static Node *
3735 : 27 : transformJsonArrayQueryConstructor(ParseState *pstate,
3736 : : JsonArrayQueryConstructor *ctor)
3737 : : {
3738 : 27 : SubLink *sublink = makeNode(SubLink);
3739 : 27 : SelectStmt *select = makeNode(SelectStmt);
3740 : 27 : RangeSubselect *range = makeNode(RangeSubselect);
3741 : 27 : Alias *alias = makeNode(Alias);
3742 : 27 : ResTarget *target = makeNode(ResTarget);
3743 : 27 : JsonArrayAgg *agg = makeNode(JsonArrayAgg);
3744 : 27 : ColumnRef *colref = makeNode(ColumnRef);
3745 : : Query *query;
3746 : : ParseState *qpstate;
3747 : :
3748 : : /* Transform query only for counting target list entries. */
3749 : 27 : qpstate = make_parsestate(pstate);
3750 : :
3751 : 27 : query = transformStmt(qpstate, ctor->query);
3752 : :
3753 [ + + ]: 27 : if (count_nonjunk_tlist_entries(query->targetList) != 1)
3754 [ + - ]: 9 : ereport(ERROR,
3755 : : errcode(ERRCODE_SYNTAX_ERROR),
3756 : : errmsg("subquery must return only one column"),
3757 : : parser_errposition(pstate, ctor->location));
3758 : :
3759 : 18 : free_parsestate(qpstate);
3760 : :
3761 : 18 : colref->fields = list_make2(makeString(pstrdup("q")),
3762 : : makeString(pstrdup("a")));
3763 : 18 : colref->location = ctor->location;
3764 : :
3765 : : /*
3766 : : * No formatting necessary, so set formatted_expr to be the same as
3767 : : * raw_expr.
3768 : : */
268 amitlan@postgresql.o 3769 : 18 : agg->arg = makeJsonValueExpr((Expr *) colref, (Expr *) colref,
3770 : : ctor->format);
382 alvherre@alvh.no-ip. 3771 : 18 : agg->absent_on_null = ctor->absent_on_null;
3772 : 18 : agg->constructor = makeNode(JsonAggConstructor);
3773 : 18 : agg->constructor->agg_order = NIL;
3774 : 18 : agg->constructor->output = ctor->output;
3775 : 18 : agg->constructor->location = ctor->location;
3776 : :
3777 : 18 : target->name = NULL;
3778 : 18 : target->indirection = NIL;
3779 : 18 : target->val = (Node *) agg;
3780 : 18 : target->location = ctor->location;
3781 : :
3782 : 18 : alias->aliasname = pstrdup("q");
3783 : 18 : alias->colnames = list_make1(makeString(pstrdup("a")));
3784 : :
3785 : 18 : range->lateral = false;
3786 : 18 : range->subquery = ctor->query;
3787 : 18 : range->alias = alias;
3788 : :
3789 : 18 : select->targetList = list_make1(target);
3790 : 18 : select->fromClause = list_make1(range);
3791 : :
3792 : 18 : sublink->subLinkType = EXPR_SUBLINK;
3793 : 18 : sublink->subLinkId = 0;
3794 : 18 : sublink->testexpr = NULL;
3795 : 18 : sublink->operName = NIL;
3796 : 18 : sublink->subselect = (Node *) select;
3797 : 18 : sublink->location = ctor->location;
3798 : :
3799 : 18 : return transformExprRecurse(pstate, (Node *) sublink);
3800 : : }
3801 : :
3802 : : /*
3803 : : * Common code for JSON_OBJECTAGG and JSON_ARRAYAGG transformation.
3804 : : */
3805 : : static Node *
3806 : 171 : transformJsonAggConstructor(ParseState *pstate, JsonAggConstructor *agg_ctor,
3807 : : JsonReturning *returning, List *args,
3808 : : Oid aggfnoid, Oid aggtype,
3809 : : JsonConstructorType ctor_type,
3810 : : bool unique, bool absent_on_null)
3811 : : {
3812 : : Node *node;
3813 : : Expr *aggfilter;
3814 : :
381 3815 : 342 : aggfilter = agg_ctor->agg_filter ? (Expr *)
3816 : 21 : transformWhereClause(pstate, agg_ctor->agg_filter,
3817 [ + + ]: 171 : EXPR_KIND_FILTER, "FILTER") : NULL;
3818 : :
382 3819 [ + + ]: 171 : if (agg_ctor->over)
3820 : : {
3821 : : /* window function */
3822 : 24 : WindowFunc *wfunc = makeNode(WindowFunc);
3823 : :
3824 : 24 : wfunc->winfnoid = aggfnoid;
3825 : 24 : wfunc->wintype = aggtype;
3826 : : /* wincollid and inputcollid will be set by parse_collate.c */
3827 : 24 : wfunc->args = args;
381 3828 : 24 : wfunc->aggfilter = aggfilter;
3829 : : /* winref will be set by transformWindowFuncCall */
382 3830 : 24 : wfunc->winstar = false;
3831 : 24 : wfunc->winagg = true;
3832 : 24 : wfunc->location = agg_ctor->location;
3833 : :
3834 : : /*
3835 : : * ordered aggs not allowed in windows yet
3836 : : */
3837 [ - + ]: 24 : if (agg_ctor->agg_order != NIL)
382 alvherre@alvh.no-ip. 3838 [ # # ]:UBC 0 : ereport(ERROR,
3839 : : errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
3840 : : errmsg("aggregate ORDER BY is not implemented for window functions"),
3841 : : parser_errposition(pstate, agg_ctor->location));
3842 : :
3843 : : /* parse_agg.c does additional window-func-specific processing */
382 alvherre@alvh.no-ip. 3844 :CBC 24 : transformWindowFuncCall(pstate, wfunc, agg_ctor->over);
3845 : :
3846 : 24 : node = (Node *) wfunc;
3847 : : }
3848 : : else
3849 : : {
3850 : 147 : Aggref *aggref = makeNode(Aggref);
3851 : :
3852 : 147 : aggref->aggfnoid = aggfnoid;
3853 : 147 : aggref->aggtype = aggtype;
3854 : :
3855 : : /* aggcollid and inputcollid will be set by parse_collate.c */
3856 : : /* aggtranstype will be set by planner */
3857 : : /* aggargtypes will be set by transformAggregateCall */
3858 : : /* aggdirectargs and args will be set by transformAggregateCall */
3859 : : /* aggorder and aggdistinct will be set by transformAggregateCall */
3860 : 147 : aggref->aggfilter = aggfilter;
3861 : 147 : aggref->aggstar = false;
3862 : 147 : aggref->aggvariadic = false;
3863 : 147 : aggref->aggkind = AGGKIND_NORMAL;
381 3864 : 147 : aggref->aggpresorted = false;
3865 : : /* agglevelsup will be set by transformAggregateCall */
382 3866 : 147 : aggref->aggsplit = AGGSPLIT_SIMPLE; /* planner might change this */
381 3867 : 147 : aggref->aggno = -1; /* planner will set aggno and aggtransno */
3868 : 147 : aggref->aggtransno = -1;
382 3869 : 147 : aggref->location = agg_ctor->location;
3870 : :
3871 : 147 : transformAggregateCall(pstate, aggref, args, agg_ctor->agg_order, false);
3872 : :
3873 : 147 : node = (Node *) aggref;
3874 : : }
3875 : :
3876 : 171 : return makeJsonConstructorExpr(pstate, ctor_type, NIL, (Expr *) node,
3877 : : returning, unique, absent_on_null,
3878 : : agg_ctor->location);
3879 : : }
3880 : :
3881 : : /*
3882 : : * Transform JSON_OBJECTAGG() aggregate function.
3883 : : *
3884 : : * JSON_OBJECTAGG() is transformed into
3885 : : * json[b]_objectagg[_unique][_strict](key, value) call depending on
3886 : : * the output JSON format. Then the function call result is coerced to the
3887 : : * target output type.
3888 : : */
3889 : : static Node *
3890 : 78 : transformJsonObjectAgg(ParseState *pstate, JsonObjectAgg *agg)
3891 : : {
3892 : : JsonReturning *returning;
3893 : : Node *key;
3894 : : Node *val;
3895 : : List *args;
3896 : : Oid aggfnoid;
3897 : : Oid aggtype;
3898 : :
3899 : 78 : key = transformExprRecurse(pstate, (Node *) agg->arg->key);
282 amitlan@postgresql.o 3900 : 78 : val = transformJsonValueExpr(pstate, "JSON_OBJECTAGG()",
3901 : 78 : agg->arg->value,
3902 : : JS_FORMAT_DEFAULT,
3903 : : InvalidOid, false);
382 alvherre@alvh.no-ip. 3904 : 78 : args = list_make2(key, val);
3905 : :
3906 : 78 : returning = transformJsonConstructorOutput(pstate, agg->constructor->output,
3907 : : args);
3908 : :
3909 [ + + ]: 78 : if (returning->format->format_type == JS_FORMAT_JSONB)
3910 : : {
3911 [ + + ]: 27 : if (agg->absent_on_null)
3912 [ + + ]: 9 : if (agg->unique)
381 3913 : 6 : aggfnoid = F_JSONB_OBJECT_AGG_UNIQUE_STRICT;
3914 : : else
3915 : 3 : aggfnoid = F_JSONB_OBJECT_AGG_STRICT;
382 3916 [ + + ]: 18 : else if (agg->unique)
381 3917 : 3 : aggfnoid = F_JSONB_OBJECT_AGG_UNIQUE;
3918 : : else
3919 : 15 : aggfnoid = F_JSONB_OBJECT_AGG;
3920 : :
382 3921 : 27 : aggtype = JSONBOID;
3922 : : }
3923 : : else
3924 : : {
3925 [ + + ]: 51 : if (agg->absent_on_null)
3926 [ + + ]: 18 : if (agg->unique)
381 3927 : 9 : aggfnoid = F_JSON_OBJECT_AGG_UNIQUE_STRICT;
3928 : : else
3929 : 9 : aggfnoid = F_JSON_OBJECT_AGG_STRICT;
382 3930 [ + + ]: 33 : else if (agg->unique)
381 3931 : 18 : aggfnoid = F_JSON_OBJECT_AGG_UNIQUE;
3932 : : else
3933 : 15 : aggfnoid = F_JSON_OBJECT_AGG;
3934 : :
382 3935 : 51 : aggtype = JSONOID;
3936 : : }
3937 : :
3938 : 156 : return transformJsonAggConstructor(pstate, agg->constructor, returning,
3939 : : args, aggfnoid, aggtype,
3940 : : JSCTOR_JSON_OBJECTAGG,
3941 : 78 : agg->unique, agg->absent_on_null);
3942 : : }
3943 : :
3944 : : /*
3945 : : * Transform JSON_ARRAYAGG() aggregate function.
3946 : : *
3947 : : * JSON_ARRAYAGG() is transformed into json[b]_agg[_strict]() call depending
3948 : : * on the output JSON format and absent_on_null. Then the function call result
3949 : : * is coerced to the target output type.
3950 : : */
3951 : : static Node *
3952 : 93 : transformJsonArrayAgg(ParseState *pstate, JsonArrayAgg *agg)
3953 : : {
3954 : : JsonReturning *returning;
3955 : : Node *arg;
3956 : : Oid aggfnoid;
3957 : : Oid aggtype;
3958 : :
24 amitlan@postgresql.o 3959 :GNC 93 : arg = transformJsonValueExpr(pstate, "JSON_ARRAYAGG()", agg->arg,
3960 : : JS_FORMAT_DEFAULT, InvalidOid, false);
3961 : :
382 alvherre@alvh.no-ip. 3962 :CBC 93 : returning = transformJsonConstructorOutput(pstate, agg->constructor->output,
3963 : 93 : list_make1(arg));
3964 : :
3965 [ + + ]: 93 : if (returning->format->format_type == JS_FORMAT_JSONB)
3966 : : {
381 3967 [ + + ]: 36 : aggfnoid = agg->absent_on_null ? F_JSONB_AGG_STRICT : F_JSONB_AGG;
382 3968 : 36 : aggtype = JSONBOID;
3969 : : }
3970 : : else
3971 : : {
381 3972 [ + + ]: 57 : aggfnoid = agg->absent_on_null ? F_JSON_AGG_STRICT : F_JSON_AGG;
382 3973 : 57 : aggtype = JSONOID;
3974 : : }
3975 : :
3976 : 93 : return transformJsonAggConstructor(pstate, agg->constructor, returning,
381 3977 : 93 : list_make1(arg), aggfnoid, aggtype,
3978 : : JSCTOR_JSON_ARRAYAGG,
382 3979 : 93 : false, agg->absent_on_null);
3980 : : }
3981 : :
3982 : : /*
3983 : : * Transform JSON_ARRAY() constructor.
3984 : : *
3985 : : * JSON_ARRAY() is transformed into json[b]_build_array[_ext]() call
3986 : : * depending on the output JSON format. The first argument of
3987 : : * json[b]_build_array_ext() is absent_on_null.
3988 : : *
3989 : : * Then function call result is coerced to the target type.
3990 : : */
3991 : : static Node *
3992 : 89 : transformJsonArrayConstructor(ParseState *pstate, JsonArrayConstructor *ctor)
3993 : : {
3994 : : JsonReturning *returning;
3995 : 89 : List *args = NIL;
3996 : :
3997 : : /* transform element expressions, if any */
3998 [ + + ]: 89 : if (ctor->exprs)
3999 : : {
4000 : : ListCell *lc;
4001 : :
4002 : : /* transform and append element arguments */
4003 [ + - + + : 159 : foreach(lc, ctor->exprs)
+ + ]
4004 : : {
4005 : 111 : JsonValueExpr *jsval = castNode(JsonValueExpr, lfirst(lc));
282 amitlan@postgresql.o 4006 : 111 : Node *val = transformJsonValueExpr(pstate, "JSON_ARRAY()",
4007 : : jsval, JS_FORMAT_DEFAULT,
4008 : : InvalidOid, false);
4009 : :
382 alvherre@alvh.no-ip. 4010 : 111 : args = lappend(args, val);
4011 : : }
4012 : : }
4013 : :
4014 : 89 : returning = transformJsonConstructorOutput(pstate, ctor->output, args);
4015 : :
4016 : 160 : return makeJsonConstructorExpr(pstate, JSCTOR_JSON_ARRAY, args, NULL,
4017 : 80 : returning, false, ctor->absent_on_null,
4018 : : ctor->location);
4019 : : }
4020 : :
4021 : : static Node *
380 4022 : 178 : transformJsonParseArg(ParseState *pstate, Node *jsexpr, JsonFormat *format,
4023 : : Oid *exprtype)
4024 : : {
4025 : 178 : Node *raw_expr = transformExprRecurse(pstate, jsexpr);
4026 : 178 : Node *expr = raw_expr;
4027 : :
4028 : 178 : *exprtype = exprType(expr);
4029 : :
4030 : : /* prepare input document */
4031 [ + + ]: 178 : if (*exprtype == BYTEAOID)
4032 : : {
4033 : : JsonValueExpr *jve;
4034 : :
282 amitlan@postgresql.o 4035 : 30 : expr = raw_expr;
380 alvherre@alvh.no-ip. 4036 : 30 : expr = makeJsonByteaToTextConversion(expr, format, exprLocation(expr));
4037 : 30 : *exprtype = TEXTOID;
4038 : :
268 amitlan@postgresql.o 4039 : 30 : jve = makeJsonValueExpr((Expr *) raw_expr, (Expr *) expr, format);
380 alvherre@alvh.no-ip. 4040 : 30 : expr = (Node *) jve;
4041 : : }
4042 : : else
4043 : : {
4044 : : char typcategory;
4045 : : bool typispreferred;
4046 : :
4047 : 148 : get_type_category_preferred(*exprtype, &typcategory, &typispreferred);
4048 : :
4049 [ + + + + ]: 148 : if (*exprtype == UNKNOWNOID || typcategory == TYPCATEGORY_STRING)
4050 : : {
4051 : 87 : expr = coerce_to_target_type(pstate, (Node *) expr, *exprtype,
4052 : : TEXTOID, -1,
4053 : : COERCION_IMPLICIT,
4054 : : COERCE_IMPLICIT_CAST, -1);
4055 : 87 : *exprtype = TEXTOID;
4056 : : }
4057 : :
4058 [ - + ]: 148 : if (format->encoding != JS_ENC_DEFAULT)
380 alvherre@alvh.no-ip. 4059 [ # # ]:UBC 0 : ereport(ERROR,
4060 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
4061 : : parser_errposition(pstate, format->location),
4062 : : errmsg("cannot use JSON FORMAT ENCODING clause for non-bytea input types")));
4063 : : }
4064 : :
380 alvherre@alvh.no-ip. 4065 :CBC 178 : return expr;
4066 : : }
4067 : :
4068 : : /*
4069 : : * Transform IS JSON predicate.
4070 : : */
4071 : : static Node *
4072 : 167 : transformJsonIsPredicate(ParseState *pstate, JsonIsPredicate *pred)
4073 : : {
4074 : : Oid exprtype;
4075 : 167 : Node *expr = transformJsonParseArg(pstate, pred->expr, pred->format,
4076 : : &exprtype);
4077 : :
4078 : : /* make resulting expression */
4079 [ + + + + : 167 : if (exprtype != TEXTOID && exprtype != JSONOID && exprtype != JSONBOID)
+ + ]
4080 [ + - ]: 3 : ereport(ERROR,
4081 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4082 : : errmsg("cannot use type %s in IS JSON predicate",
4083 : : format_type_be(exprtype))));
4084 : :
4085 : : /* This intentionally(?) drops the format clause. */
4086 : 328 : return makeJsonIsPredicate(expr, NULL, pred->item_type,
4087 : 164 : pred->unique_keys, pred->location);
4088 : : }
4089 : :
4090 : : /*
4091 : : * Transform the RETURNING clause of a JSON_*() expression if there is one and
4092 : : * create one if not.
4093 : : */
4094 : : static JsonReturning *
269 amitlan@postgresql.o 4095 :GNC 123 : transformJsonReturning(ParseState *pstate, JsonOutput *output, const char *fname)
4096 : : {
4097 : : JsonReturning *returning;
4098 : :
4099 [ - + ]: 123 : if (output)
4100 : : {
269 amitlan@postgresql.o 4101 :UNC 0 : returning = transformJsonOutput(pstate, output, false);
4102 : :
4103 [ # # ]: 0 : Assert(OidIsValid(returning->typid));
4104 : :
4105 [ # # # # ]: 0 : if (returning->typid != JSONOID && returning->typid != JSONBOID)
4106 [ # # ]: 0 : ereport(ERROR,
4107 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4108 : : errmsg("cannot use RETURNING type %s in %s",
4109 : : format_type_be(returning->typid), fname),
4110 : : parser_errposition(pstate, output->typeName->location)));
4111 : : }
4112 : : else
4113 : : {
4114 : : /* Output type is JSON by default. */
269 amitlan@postgresql.o 4115 :GNC 123 : Oid targettype = JSONOID;
4116 : 123 : JsonFormatType format = JS_FORMAT_JSON;
4117 : :
4118 : 123 : returning = makeNode(JsonReturning);
4119 : 123 : returning->format = makeJsonFormat(format, JS_ENC_DEFAULT, -1);
4120 : 123 : returning->typid = targettype;
4121 : 123 : returning->typmod = -1;
4122 : : }
4123 : :
4124 : 123 : return returning;
4125 : : }
4126 : :
4127 : : /*
4128 : : * Transform a JSON() expression.
4129 : : *
4130 : : * JSON() is transformed into a JsonConstructorExpr of type JSCTOR_JSON_PARSE,
4131 : : * which validates the input expression value as JSON.
4132 : : */
4133 : : static Node *
4134 : 74 : transformJsonParseExpr(ParseState *pstate, JsonParseExpr *jsexpr)
4135 : : {
4136 : 74 : JsonOutput *output = jsexpr->output;
4137 : : JsonReturning *returning;
4138 : : Node *arg;
4139 : :
4140 : 74 : returning = transformJsonReturning(pstate, output, "JSON()");
4141 : :
4142 [ + + ]: 74 : if (jsexpr->unique_keys)
4143 : : {
4144 : : /*
4145 : : * Coerce string argument to text and then to json[b] in the executor
4146 : : * node with key uniqueness check.
4147 : : */
4148 : 11 : JsonValueExpr *jve = jsexpr->expr;
4149 : : Oid arg_type;
4150 : :
4151 : 11 : arg = transformJsonParseArg(pstate, (Node *) jve->raw_expr, jve->format,
4152 : : &arg_type);
4153 : :
4154 [ + + ]: 11 : if (arg_type != TEXTOID)
4155 [ + - ]: 4 : ereport(ERROR,
4156 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4157 : : errmsg("cannot use non-string types with WITH UNIQUE KEYS clause"),
4158 : : parser_errposition(pstate, jsexpr->location)));
4159 : : }
4160 : : else
4161 : : {
4162 : : /*
4163 : : * Coerce argument to target type using CAST for compatibility with PG
4164 : : * function-like CASTs.
4165 : : */
4166 : 63 : arg = transformJsonValueExpr(pstate, "JSON()", jsexpr->expr,
4167 : : JS_FORMAT_JSON, returning->typid, false);
4168 : : }
4169 : :
4170 : 63 : return makeJsonConstructorExpr(pstate, JSCTOR_JSON_PARSE, list_make1(arg), NULL,
4171 : 63 : returning, jsexpr->unique_keys, false,
4172 : : jsexpr->location);
4173 : : }
4174 : :
4175 : : /*
4176 : : * Transform a JSON_SCALAR() expression.
4177 : : *
4178 : : * JSON_SCALAR() is transformed into a JsonConstructorExpr of type
4179 : : * JSCTOR_JSON_SCALAR, which converts the input SQL scalar value into
4180 : : * a json[b] value.
4181 : : */
4182 : : static Node *
4183 : 49 : transformJsonScalarExpr(ParseState *pstate, JsonScalarExpr *jsexpr)
4184 : : {
4185 : 49 : Node *arg = transformExprRecurse(pstate, (Node *) jsexpr->expr);
4186 : 49 : JsonOutput *output = jsexpr->output;
4187 : : JsonReturning *returning;
4188 : :
4189 : 49 : returning = transformJsonReturning(pstate, output, "JSON_SCALAR()");
4190 : :
4191 [ + + ]: 49 : if (exprType(arg) == UNKNOWNOID)
4192 : 11 : arg = coerce_to_specific_type(pstate, arg, TEXTOID, "JSON_SCALAR");
4193 : :
4194 : 49 : return makeJsonConstructorExpr(pstate, JSCTOR_JSON_SCALAR, list_make1(arg), NULL,
4195 : : returning, false, false, jsexpr->location);
4196 : : }
4197 : :
4198 : : /*
4199 : : * Transform a JSON_SERIALIZE() expression.
4200 : : *
4201 : : * JSON_SERIALIZE() is transformed into a JsonConstructorExpr of type
4202 : : * JSCTOR_JSON_SERIALIZE which converts the input JSON value into a character
4203 : : * or bytea string.
4204 : : */
4205 : : static Node *
4206 : 39 : transformJsonSerializeExpr(ParseState *pstate, JsonSerializeExpr *expr)
4207 : : {
4208 : : JsonReturning *returning;
4209 : 39 : Node *arg = transformJsonValueExpr(pstate, "JSON_SERIALIZE()",
4210 : : expr->expr,
4211 : : JS_FORMAT_JSON,
4212 : : InvalidOid, false);
4213 : :
4214 [ + + ]: 39 : if (expr->output)
4215 : : {
4216 : 14 : returning = transformJsonOutput(pstate, expr->output, true);
4217 : :
4218 [ + + ]: 14 : if (returning->typid != BYTEAOID)
4219 : : {
4220 : : char typcategory;
4221 : : bool typispreferred;
4222 : :
4223 : 8 : get_type_category_preferred(returning->typid, &typcategory,
4224 : : &typispreferred);
4225 [ + + ]: 8 : if (typcategory != TYPCATEGORY_STRING)
4226 [ + - ]: 4 : ereport(ERROR,
4227 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4228 : : errmsg("cannot use RETURNING type %s in %s",
4229 : : format_type_be(returning->typid),
4230 : : "JSON_SERIALIZE()"),
4231 : : errhint("Try returning a string type or bytea.")));
4232 : : }
4233 : : }
4234 : : else
4235 : : {
4236 : : /* RETURNING TEXT FORMAT JSON is by default */
4237 : 25 : returning = makeNode(JsonReturning);
4238 : 25 : returning->format = makeJsonFormat(JS_FORMAT_JSON, JS_ENC_DEFAULT, -1);
4239 : 25 : returning->typid = TEXTOID;
4240 : 25 : returning->typmod = -1;
4241 : : }
4242 : :
4243 : 35 : return makeJsonConstructorExpr(pstate, JSCTOR_JSON_SERIALIZE, list_make1(arg),
4244 : : NULL, returning, false, false, expr->location);
4245 : : }
4246 : :
4247 : : /*
4248 : : * Transform JSON_VALUE, JSON_QUERY, JSON_EXISTS, JSON_TABLE functions into
4249 : : * a JsonExpr node.
4250 : : */
4251 : : static Node *
24 4252 : 1303 : transformJsonFuncExpr(ParseState *pstate, JsonFuncExpr *func)
4253 : : {
4254 : : JsonExpr *jsexpr;
4255 : : Node *path_spec;
4256 : 1303 : const char *func_name = NULL;
4257 : : JsonFormatType default_format;
4258 : :
4259 [ + + + + : 1303 : switch (func->op)
- ]
4260 : : {
4261 : 123 : case JSON_EXISTS_OP:
4262 : 123 : func_name = "JSON_EXISTS";
4263 : 123 : default_format = JS_FORMAT_DEFAULT;
4264 : 123 : break;
4265 : 567 : case JSON_QUERY_OP:
4266 : 567 : func_name = "JSON_QUERY";
4267 : 567 : default_format = JS_FORMAT_JSONB;
4268 : 567 : break;
4269 : 440 : case JSON_VALUE_OP:
4270 : 440 : func_name = "JSON_VALUE";
4271 : 440 : default_format = JS_FORMAT_DEFAULT;
4272 : 440 : break;
10 4273 : 173 : case JSON_TABLE_OP:
4274 : 173 : func_name = "JSON_TABLE";
4275 : 173 : default_format = JS_FORMAT_JSONB;
4276 : 173 : break;
24 amitlan@postgresql.o 4277 :UNC 0 : default:
4278 [ # # ]: 0 : elog(ERROR, "invalid JsonFuncExpr op %d", (int) func->op);
4279 : : default_format = JS_FORMAT_DEFAULT; /* keep compiler quiet */
4280 : : break;
4281 : : }
4282 : :
4283 : : /*
4284 : : * Even though the syntax allows it, FORMAT JSON specification in
4285 : : * RETURNING is meaningless except for JSON_QUERY(). Flag if not
4286 : : * JSON_QUERY().
4287 : : */
24 amitlan@postgresql.o 4288 [ + + + + ]:GNC 1303 : if (func->output && func->op != JSON_QUERY_OP)
4289 : : {
4290 : 395 : JsonFormat *format = func->output->returning->format;
4291 : :
4292 [ + + ]: 395 : if (format->format_type != JS_FORMAT_DEFAULT ||
4293 [ - + ]: 392 : format->encoding != JS_ENC_DEFAULT)
4294 [ + - ]: 3 : ereport(ERROR,
4295 : : errcode(ERRCODE_SYNTAX_ERROR),
4296 : : errmsg("cannot specify FORMAT JSON in RETURNING clause of %s()",
4297 : : func_name),
4298 : : parser_errposition(pstate, format->location));
4299 : : }
4300 : :
4301 : : /* OMIT QUOTES is meaningless when strings are wrapped. */
4302 [ + + ]: 1300 : if (func->op == JSON_QUERY_OP &&
6 4303 [ + + ]: 567 : func->quotes == JS_QUOTES_OMIT &&
24 4304 [ + + ]: 81 : (func->wrapper == JSW_CONDITIONAL ||
4305 [ + + ]: 78 : func->wrapper == JSW_UNCONDITIONAL))
4306 [ + - ]: 9 : ereport(ERROR,
4307 : : errcode(ERRCODE_SYNTAX_ERROR),
4308 : : errmsg("SQL/JSON QUOTES behavior must not be specified when WITH WRAPPER is used"),
4309 : : parser_errposition(pstate, func->location));
4310 : :
4311 : 1291 : jsexpr = makeNode(JsonExpr);
4312 : 1291 : jsexpr->location = func->location;
4313 : 1291 : jsexpr->op = func->op;
4314 : :
4315 : : /*
4316 : : * jsonpath machinery can only handle jsonb documents, so coerce the input
4317 : : * if not already of jsonb type.
4318 : : */
4319 : 1291 : jsexpr->formatted_expr = transformJsonValueExpr(pstate, func_name,
4320 : : func->context_item,
4321 : : default_format,
4322 : : JSONBOID,
4323 : : false);
4324 : 1291 : jsexpr->format = func->context_item->format;
4325 : :
4326 : 1291 : path_spec = transformExprRecurse(pstate, func->pathspec);
4327 : 1291 : path_spec = coerce_to_target_type(pstate, path_spec, exprType(path_spec),
4328 : : JSONPATHOID, -1,
4329 : : COERCION_EXPLICIT, COERCE_IMPLICIT_CAST,
4330 : : exprLocation(path_spec));
4331 [ - + ]: 1291 : if (path_spec == NULL)
24 amitlan@postgresql.o 4332 [ # # ]:UNC 0 : ereport(ERROR,
4333 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4334 : : errmsg("JSON path expression must be of type %s, not of type %s",
4335 : : "jsonpath", format_type_be(exprType(path_spec))),
4336 : : parser_errposition(pstate, exprLocation(path_spec))));
24 amitlan@postgresql.o 4337 :GNC 1291 : jsexpr->path_spec = path_spec;
4338 : :
4339 : : /* Transform and coerce the PASSING arguments to jsonb. */
4340 : 1291 : transformJsonPassingArgs(pstate, func_name,
4341 : : JS_FORMAT_JSONB,
4342 : : func->passing,
4343 : : &jsexpr->passing_values,
4344 : : &jsexpr->passing_names);
4345 : :
4346 : : /* Transform the JsonOutput into JsonReturning. */
4347 : 1291 : jsexpr->returning = transformJsonOutput(pstate, func->output, false);
4348 : :
4349 [ + + + + : 1285 : switch (func->op)
- ]
4350 : : {
4351 : 123 : case JSON_EXISTS_OP:
4352 : : /* JSON_EXISTS returns boolean by default. */
4353 [ + + ]: 123 : if (!OidIsValid(jsexpr->returning->typid))
4354 : : {
4355 : 81 : jsexpr->returning->typid = BOOLOID;
4356 : 81 : jsexpr->returning->typmod = -1;
4357 : : }
4358 : :
4359 : : /* JSON_TABLE() COLUMNS can specify a non-boolean type. */
10 4360 [ + + ]: 123 : if (jsexpr->returning->typid != BOOLOID)
4361 : : {
4362 : : Node *coercion_expr;
4363 : 36 : CaseTestExpr *placeholder = makeNode(CaseTestExpr);
4364 : 36 : int location = exprLocation((Node *) jsexpr);
4365 : :
4366 : : /*
4367 : : * We abuse CaseTestExpr here as placeholder to pass the
4368 : : * result of evaluating JSON_EXISTS to the coercion
4369 : : * expression.
4370 : : */
4371 : 36 : placeholder->typeId = BOOLOID;
4372 : 36 : placeholder->typeMod = -1;
4373 : 36 : placeholder->collation = InvalidOid;
4374 : :
4375 : : coercion_expr =
4376 : 36 : coerce_to_target_type(pstate, (Node *) placeholder, BOOLOID,
4377 : 36 : jsexpr->returning->typid,
4378 : 36 : jsexpr->returning->typmod,
4379 : : COERCION_EXPLICIT,
4380 : : COERCE_IMPLICIT_CAST,
4381 : : location);
4382 : :
4383 [ + + ]: 36 : if (coercion_expr == NULL)
4384 [ + - ]: 15 : ereport(ERROR,
4385 : : (errcode(ERRCODE_CANNOT_COERCE),
4386 : : errmsg("cannot cast type %s to %s",
4387 : : format_type_be(BOOLOID),
4388 : : format_type_be(jsexpr->returning->typid)),
4389 : : parser_coercion_errposition(pstate, location, (Node *) jsexpr)));
4390 : :
4391 [ + - ]: 21 : if (coercion_expr != (Node *) placeholder)
4392 : 21 : jsexpr->coercion_expr = coercion_expr;
4393 : : }
4394 : :
24 4395 : 108 : jsexpr->on_error = transformJsonBehavior(pstate, func->on_error,
4396 : : JSON_BEHAVIOR_FALSE,
4397 : : jsexpr->returning);
4398 : 108 : break;
4399 : :
4400 : 555 : case JSON_QUERY_OP:
4401 : : /* JSON_QUERY returns jsonb by default. */
4402 [ + + ]: 555 : if (!OidIsValid(jsexpr->returning->typid))
4403 : : {
4404 : 228 : JsonReturning *ret = jsexpr->returning;
4405 : :
4406 : 228 : ret->typid = JSONBOID;
4407 : 228 : ret->typmod = -1;
4408 : : }
4409 : :
4410 : : /*
4411 : : * Keep quotes on scalar strings by default, omitting them only if
4412 : : * OMIT QUOTES is specified.
4413 : : */
4414 : 555 : jsexpr->omit_quotes = (func->quotes == JS_QUOTES_OMIT);
4415 : 555 : jsexpr->wrapper = func->wrapper;
4416 : :
4417 : 555 : coerceJsonExprOutput(pstate, jsexpr);
4418 : :
4419 [ + + ]: 555 : if (func->on_empty)
4420 : 42 : jsexpr->on_empty = transformJsonBehavior(pstate,
4421 : : func->on_empty,
4422 : : JSON_BEHAVIOR_NULL,
4423 : : jsexpr->returning);
4424 : 555 : jsexpr->on_error = transformJsonBehavior(pstate, func->on_error,
4425 : : JSON_BEHAVIOR_NULL,
4426 : : jsexpr->returning);
4427 : 531 : break;
4428 : :
4429 : 434 : case JSON_VALUE_OP:
4430 : : /* JSON_VALUE returns text by default. */
4431 [ + + ]: 434 : if (!OidIsValid(jsexpr->returning->typid))
4432 : : {
4433 : 87 : jsexpr->returning->typid = TEXTOID;
4434 : 87 : jsexpr->returning->typmod = -1;
4435 : : }
4436 : :
4437 : : /*
4438 : : * Override whatever transformJsonOutput() set these to, which
4439 : : * assumes that output type to be jsonb.
4440 : : */
4441 : 434 : jsexpr->returning->format->format_type = JS_FORMAT_DEFAULT;
4442 : 434 : jsexpr->returning->format->encoding = JS_ENC_DEFAULT;
4443 : :
4444 : : /* Always omit quotes from scalar strings. */
4445 : 434 : jsexpr->omit_quotes = true;
4446 : :
4447 : 434 : coerceJsonExprOutput(pstate, jsexpr);
4448 : :
4449 [ + + ]: 434 : if (func->on_empty)
4450 : 57 : jsexpr->on_empty = transformJsonBehavior(pstate,
4451 : : func->on_empty,
4452 : : JSON_BEHAVIOR_NULL,
4453 : : jsexpr->returning);
4454 : 434 : jsexpr->on_error = transformJsonBehavior(pstate, func->on_error,
4455 : : JSON_BEHAVIOR_NULL,
4456 : : jsexpr->returning);
4457 : 434 : break;
4458 : :
10 4459 : 173 : case JSON_TABLE_OP:
4460 [ + - ]: 173 : if (!OidIsValid(jsexpr->returning->typid))
4461 : : {
4462 : 173 : jsexpr->returning->typid = exprType(jsexpr->formatted_expr);
4463 : 173 : jsexpr->returning->typmod = -1;
4464 : : }
4465 : 173 : jsexpr->on_error = transformJsonBehavior(pstate, func->on_error,
4466 : : JSON_BEHAVIOR_EMPTY,
4467 : : jsexpr->returning);
4468 : 173 : break;
4469 : :
24 amitlan@postgresql.o 4470 :UNC 0 : default:
4471 [ # # ]: 0 : elog(ERROR, "invalid JsonFuncExpr op %d", (int) func->op);
4472 : : break;
4473 : : }
4474 : :
24 amitlan@postgresql.o 4475 :GNC 1246 : return (Node *) jsexpr;
4476 : : }
4477 : :
4478 : : /*
4479 : : * Transform a SQL/JSON PASSING clause.
4480 : : */
4481 : : static void
4482 : 1291 : transformJsonPassingArgs(ParseState *pstate, const char *constructName,
4483 : : JsonFormatType format, List *args,
4484 : : List **passing_values, List **passing_names)
4485 : : {
4486 : : ListCell *lc;
4487 : :
4488 : 1291 : *passing_values = NIL;
4489 : 1291 : *passing_names = NIL;
4490 : :
4491 [ + + + + : 1873 : foreach(lc, args)
+ + ]
4492 : : {
4493 : 582 : JsonArgument *arg = castNode(JsonArgument, lfirst(lc));
4494 : 582 : Node *expr = transformJsonValueExpr(pstate, constructName,
4495 : : arg->val, format,
4496 : : InvalidOid, true);
4497 : :
4498 : 582 : *passing_values = lappend(*passing_values, expr);
4499 : 582 : *passing_names = lappend(*passing_names, makeString(arg->name));
4500 : : }
4501 : 1291 : }
4502 : :
4503 : : /*
4504 : : * Set up to coerce the result value of JSON_VALUE() / JSON_QUERY() to the
4505 : : * RETURNING type (default or user-specified), if needed.
4506 : : */
4507 : : static void
4508 : 989 : coerceJsonExprOutput(ParseState *pstate, JsonExpr *jsexpr)
4509 : : {
4510 : 989 : JsonReturning *returning = jsexpr->returning;
4511 : 989 : Node *context_item = jsexpr->formatted_expr;
4512 : : int default_typmod;
4513 : : Oid default_typid;
4514 : 989 : bool omit_quotes =
4515 [ + + + + ]: 989 : jsexpr->op == JSON_QUERY_OP && jsexpr->omit_quotes;
4516 : 989 : Node *coercion_expr = NULL;
4517 : :
4518 [ - + ]: 989 : Assert(returning);
4519 : :
4520 : : /*
4521 : : * Check for cases where the coercion should be handled at runtime, that
4522 : : * is, without using a cast expression.
4523 : : */
4524 [ + + ]: 989 : if (jsexpr->op == JSON_VALUE_OP)
4525 : : {
4526 : : /*
4527 : : * Use cast expressions for types with typmod and domain types.
4528 : : */
4529 [ + + + + ]: 856 : if (returning->typmod == -1 &&
4530 : 422 : get_typtype(returning->typid) != TYPTYPE_DOMAIN)
4531 : : {
4532 : 395 : jsexpr->use_io_coercion = true;
4533 : 395 : return;
4534 : : }
4535 : : }
4536 [ + - ]: 555 : else if (jsexpr->op == JSON_QUERY_OP)
4537 : : {
4538 : : /*
4539 : : * Cast functions from jsonb to the following types (jsonb_bool() et
4540 : : * al) don't handle errors softly, so coerce either by calling
4541 : : * json_populate_type() or the type's input function so that any
4542 : : * errors are handled appropriately. The latter only if OMIT QUOTES is
4543 : : * true.
4544 : : */
4545 [ + + ]: 555 : switch (returning->typid)
4546 : : {
4547 : 45 : case BOOLOID:
4548 : : case NUMERICOID:
4549 : : case INT2OID:
4550 : : case INT4OID:
4551 : : case INT8OID:
4552 : : case FLOAT4OID:
4553 : : case FLOAT8OID:
4554 [ + + ]: 45 : if (jsexpr->omit_quotes)
4555 : 9 : jsexpr->use_io_coercion = true;
4556 : : else
4557 : 36 : jsexpr->use_json_coercion = true;
4558 : 45 : return;
4559 : 510 : default:
4560 : 510 : break;
4561 : : }
4562 : : }
4563 : :
4564 : : /* Look up a cast expression. */
4565 : :
4566 : : /*
4567 : : * For JSON_VALUE() and for JSON_QUERY() when OMIT QUOTES is true,
4568 : : * ExecEvalJsonExprPath() will convert a quote-stripped source value to
4569 : : * its text representation, so use TEXTOID as the source type.
4570 : : */
4571 [ + + + + ]: 549 : if (omit_quotes || jsexpr->op == JSON_VALUE_OP)
4572 : : {
4573 : 102 : default_typid = TEXTOID;
4574 : 102 : default_typmod = -1;
4575 : : }
4576 : : else
4577 : : {
4578 : 447 : default_typid = exprType(context_item);
4579 : 447 : default_typmod = exprTypmod(context_item);
4580 : : }
4581 : :
4582 [ + + ]: 549 : if (returning->typid != default_typid ||
4583 [ - + ]: 267 : returning->typmod != default_typmod)
4584 : : {
4585 : : /*
4586 : : * We abuse CaseTestExpr here as placeholder to pass the result of
4587 : : * jsonpath evaluation as input to the coercion expression.
4588 : : */
4589 : 282 : CaseTestExpr *placeholder = makeNode(CaseTestExpr);
4590 : :
4591 : 282 : placeholder->typeId = default_typid;
4592 : 282 : placeholder->typeMod = default_typmod;
4593 : :
4594 : 282 : coercion_expr = coerceJsonFuncExpr(pstate, (Node *) placeholder,
4595 : : returning, false);
4596 [ - + ]: 282 : if (coercion_expr == (Node *) placeholder)
24 amitlan@postgresql.o 4597 :UNC 0 : coercion_expr = NULL;
4598 : : }
4599 : :
24 amitlan@postgresql.o 4600 :GNC 549 : jsexpr->coercion_expr = coercion_expr;
4601 : :
4602 [ + + ]: 549 : if (coercion_expr == NULL)
4603 : : {
4604 : : /*
4605 : : * Either no cast was found or coercion is unnecessary but still must
4606 : : * convert the string value to the output type.
4607 : : */
4608 [ + + - + ]: 345 : if (omit_quotes || jsexpr->op == JSON_VALUE_OP)
4609 : 15 : jsexpr->use_io_coercion = true;
4610 : : else
4611 : 330 : jsexpr->use_json_coercion = true;
4612 : : }
4613 : :
4614 [ + + - + ]: 549 : Assert(jsexpr->coercion_expr != NULL ||
4615 : : (jsexpr->use_io_coercion != jsexpr->use_json_coercion));
4616 : : }
4617 : :
4618 : : /*
4619 : : * Transform a JSON BEHAVIOR clause.
4620 : : */
4621 : : static JsonBehavior *
4622 : 1369 : transformJsonBehavior(ParseState *pstate, JsonBehavior *behavior,
4623 : : JsonBehaviorType default_behavior,
4624 : : JsonReturning *returning)
4625 : : {
4626 : 1369 : JsonBehaviorType btype = default_behavior;
4627 : 1369 : Node *expr = NULL;
4628 : 1369 : bool coerce_at_runtime = false;
4629 : 1369 : int location = -1;
4630 : :
4631 [ + + ]: 1369 : if (behavior)
4632 : : {
4633 : 363 : btype = behavior->btype;
4634 : 363 : location = behavior->location;
4635 [ + + ]: 363 : if (btype == JSON_BEHAVIOR_DEFAULT)
4636 : : {
4637 : 111 : expr = transformExprRecurse(pstate, behavior->expr);
4638 [ + + + + ]: 111 : if (!IsA(expr, Const) && !IsA(expr, FuncExpr) &&
4639 [ + + ]: 15 : !IsA(expr, OpExpr))
4640 [ + - ]: 6 : ereport(ERROR,
4641 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4642 : : errmsg("can only specify a constant, non-aggregate function, or operator expression for DEFAULT"),
4643 : : parser_errposition(pstate, exprLocation(expr))));
4644 [ + + ]: 105 : if (contain_var_clause(expr))
4645 [ + - ]: 3 : ereport(ERROR,
4646 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4647 : : errmsg("DEFAULT expression must not contain column references"),
4648 : : parser_errposition(pstate, exprLocation(expr))));
4649 [ + + ]: 102 : if (expression_returns_set(expr))
4650 [ + - ]: 3 : ereport(ERROR,
4651 : : (errcode(ERRCODE_DATATYPE_MISMATCH),
4652 : : errmsg("DEFAULT expression must not return a set"),
4653 : : parser_errposition(pstate, exprLocation(expr))));
4654 : : }
4655 : : }
4656 : :
4657 [ + + + + ]: 1357 : if (expr == NULL && btype != JSON_BEHAVIOR_ERROR)
4658 : 1072 : expr = GetJsonBehaviorConst(btype, location);
4659 : :
4660 [ + + ]: 1357 : if (expr)
4661 : : {
4662 : 1171 : Node *coerced_expr = expr;
4663 [ + + + + ]: 1171 : bool isnull = (IsA(expr, Const) && ((Const *) expr)->constisnull);
4664 : :
4665 : : /*
4666 : : * Coerce NULLs and "internal" (that is, not specified by the user)
4667 : : * jsonb-valued expressions at runtime using json_populate_type().
4668 : : *
4669 : : * For other (user-specified) non-NULL values, try to find a cast and
4670 : : * error out if one is not found.
4671 : : */
4672 [ + + + + ]: 1408 : if (isnull ||
4673 [ - + ]: 285 : (exprType(expr) == JSONBOID &&
4674 : : btype == default_behavior))
4675 : 934 : coerce_at_runtime = true;
4676 : : else
4677 : : coerced_expr =
4678 : 237 : coerce_to_target_type(pstate, expr, exprType(expr),
4679 : : returning->typid, returning->typmod,
4680 : : COERCION_EXPLICIT, COERCE_EXPLICIT_CAST,
4681 : : exprLocation((Node *) behavior));
4682 : :
4683 [ + + ]: 1171 : if (coerced_expr == NULL)
4684 [ + - ]: 12 : ereport(ERROR,
4685 : : errcode(ERRCODE_CANNOT_COERCE),
4686 : : errmsg("cannot cast behavior expression of type %s to %s",
4687 : : format_type_be(exprType(expr)),
4688 : : format_type_be(returning->typid)),
4689 : : parser_errposition(pstate, exprLocation(expr)));
4690 : : else
4691 : 1159 : expr = coerced_expr;
4692 : : }
4693 : :
4694 [ + + ]: 1345 : if (behavior)
4695 : 339 : behavior->expr = expr;
4696 : : else
4697 : 1006 : behavior = makeJsonBehavior(btype, expr, location);
4698 : :
4699 : 1345 : behavior->coerce = coerce_at_runtime;
4700 : :
4701 : 1345 : return behavior;
4702 : : }
4703 : :
4704 : : /*
4705 : : * Returns a Const node holding the value for the given non-ERROR
4706 : : * JsonBehaviorType.
4707 : : */
4708 : : static Node *
4709 : 1072 : GetJsonBehaviorConst(JsonBehaviorType btype, int location)
4710 : : {
4711 : 1072 : Datum val = (Datum) 0;
4712 : 1072 : Oid typid = JSONBOID;
4713 : 1072 : int len = -1;
4714 : 1072 : bool isbyval = false;
4715 : 1072 : bool isnull = false;
4716 : : Const *con;
4717 : :
4718 [ + + + + : 1072 : switch (btype)
+ - - ]
4719 : : {
4720 : 18 : case JSON_BEHAVIOR_EMPTY_ARRAY:
4721 : 18 : val = DirectFunctionCall1(jsonb_in, CStringGetDatum("[]"));
4722 : 18 : break;
4723 : :
4724 : 27 : case JSON_BEHAVIOR_EMPTY_OBJECT:
4725 : 27 : val = DirectFunctionCall1(jsonb_in, CStringGetDatum("{}"));
4726 : 27 : break;
4727 : :
4728 : 3 : case JSON_BEHAVIOR_TRUE:
4729 : 3 : val = BoolGetDatum(true);
4730 : 3 : typid = BOOLOID;
4731 : 3 : len = sizeof(bool);
4732 : 3 : isbyval = true;
4733 : 3 : break;
4734 : :
4735 : 93 : case JSON_BEHAVIOR_FALSE:
4736 : 93 : val = BoolGetDatum(false);
4737 : 93 : typid = BOOLOID;
4738 : 93 : len = sizeof(bool);
4739 : 93 : isbyval = true;
4740 : 93 : break;
4741 : :
4742 : 931 : case JSON_BEHAVIOR_NULL:
4743 : : case JSON_BEHAVIOR_UNKNOWN:
4744 : : case JSON_BEHAVIOR_EMPTY:
4745 : 931 : val = (Datum) 0;
4746 : 931 : isnull = true;
4747 : 931 : typid = INT4OID;
4748 : 931 : len = sizeof(int32);
4749 : 931 : isbyval = true;
4750 : 931 : break;
4751 : :
4752 : : /* These two behavior types are handled by the caller. */
24 amitlan@postgresql.o 4753 :UNC 0 : case JSON_BEHAVIOR_DEFAULT:
4754 : : case JSON_BEHAVIOR_ERROR:
4755 : 0 : Assert(false);
4756 : : break;
4757 : :
4758 : 0 : default:
4759 [ # # ]: 0 : elog(ERROR, "unrecognized SQL/JSON behavior %d", btype);
4760 : : break;
4761 : : }
4762 : :
24 amitlan@postgresql.o 4763 :GNC 1072 : con = makeConst(typid, -1, InvalidOid, len, val, isnull, isbyval);
4764 : 1072 : con->location = location;
4765 : :
4766 : 1072 : return (Node *) con;
4767 : : }
|
