Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * deparse.c
4 : : * Query deparser for postgres_fdw
5 : : *
6 : : * This file includes functions that examine query WHERE clauses to see
7 : : * whether they're safe to send to the remote server for execution, as
8 : : * well as functions to construct the query text to be sent. The latter
9 : : * functionality is annoyingly duplicative of ruleutils.c, but there are
10 : : * enough special considerations that it seems best to keep this separate.
11 : : * One saving grace is that we only need deparse logic for node types that
12 : : * we consider safe to send.
13 : : *
14 : : * We assume that the remote session's search_path is exactly "pg_catalog",
15 : : * and thus we need schema-qualify all and only names outside pg_catalog.
16 : : *
17 : : * We do not consider that it is ever safe to send COLLATE expressions to
18 : : * the remote server: it might not have the same collation names we do.
19 : : * (Later we might consider it safe to send COLLATE "C", but even that would
20 : : * fail on old remote servers.) An expression is considered safe to send
21 : : * only if all operator/function input collations used in it are traceable to
22 : : * Var(s) of the foreign table. That implies that if the remote server gets
23 : : * a different answer than we do, the foreign table's columns are not marked
24 : : * with collations that match the remote table's columns, which we can
25 : : * consider to be user error.
26 : : *
27 : : * Portions Copyright (c) 2012-2024, PostgreSQL Global Development Group
28 : : *
29 : : * IDENTIFICATION
30 : : * contrib/postgres_fdw/deparse.c
31 : : *
32 : : *-------------------------------------------------------------------------
33 : : */
34 : : #include "postgres.h"
35 : :
36 : : #include "access/htup_details.h"
37 : : #include "access/sysattr.h"
38 : : #include "access/table.h"
39 : : #include "catalog/pg_aggregate.h"
40 : : #include "catalog/pg_authid.h"
41 : : #include "catalog/pg_collation.h"
42 : : #include "catalog/pg_namespace.h"
43 : : #include "catalog/pg_operator.h"
44 : : #include "catalog/pg_opfamily.h"
45 : : #include "catalog/pg_proc.h"
46 : : #include "catalog/pg_ts_config.h"
47 : : #include "catalog/pg_ts_dict.h"
48 : : #include "catalog/pg_type.h"
49 : : #include "commands/defrem.h"
50 : : #include "commands/tablecmds.h"
51 : : #include "nodes/makefuncs.h"
52 : : #include "nodes/nodeFuncs.h"
53 : : #include "nodes/plannodes.h"
54 : : #include "optimizer/optimizer.h"
55 : : #include "optimizer/prep.h"
56 : : #include "optimizer/tlist.h"
57 : : #include "parser/parsetree.h"
58 : : #include "postgres_fdw.h"
59 : : #include "utils/builtins.h"
60 : : #include "utils/lsyscache.h"
61 : : #include "utils/rel.h"
62 : : #include "utils/syscache.h"
63 : : #include "utils/typcache.h"
64 : :
65 : : /*
66 : : * Global context for foreign_expr_walker's search of an expression tree.
67 : : */
68 : : typedef struct foreign_glob_cxt
69 : : {
70 : : PlannerInfo *root; /* global planner state */
71 : : RelOptInfo *foreignrel; /* the foreign relation we are planning for */
72 : : Relids relids; /* relids of base relations in the underlying
73 : : * scan */
74 : : } foreign_glob_cxt;
75 : :
76 : : /*
77 : : * Local (per-tree-level) context for foreign_expr_walker's search.
78 : : * This is concerned with identifying collations used in the expression.
79 : : */
80 : : typedef enum
81 : : {
82 : : FDW_COLLATE_NONE, /* expression is of a noncollatable type, or
83 : : * it has default collation that is not
84 : : * traceable to a foreign Var */
85 : : FDW_COLLATE_SAFE, /* collation derives from a foreign Var */
86 : : FDW_COLLATE_UNSAFE, /* collation is non-default and derives from
87 : : * something other than a foreign Var */
88 : : } FDWCollateState;
89 : :
90 : : typedef struct foreign_loc_cxt
91 : : {
92 : : Oid collation; /* OID of current collation, if any */
93 : : FDWCollateState state; /* state of current collation choice */
94 : : } foreign_loc_cxt;
95 : :
96 : : /*
97 : : * Context for deparseExpr
98 : : */
99 : : typedef struct deparse_expr_cxt
100 : : {
101 : : PlannerInfo *root; /* global planner state */
102 : : RelOptInfo *foreignrel; /* the foreign relation we are planning for */
103 : : RelOptInfo *scanrel; /* the underlying scan relation. Same as
104 : : * foreignrel, when that represents a join or
105 : : * a base relation. */
106 : : StringInfo buf; /* output buffer to append to */
107 : : List **params_list; /* exprs that will become remote Params */
108 : : } deparse_expr_cxt;
109 : :
110 : : #define REL_ALIAS_PREFIX "r"
111 : : /* Handy macro to add relation name qualification */
112 : : #define ADD_REL_QUALIFIER(buf, varno) \
113 : : appendStringInfo((buf), "%s%d.", REL_ALIAS_PREFIX, (varno))
114 : : #define SUBQUERY_REL_ALIAS_PREFIX "s"
115 : : #define SUBQUERY_COL_ALIAS_PREFIX "c"
116 : :
117 : : /*
118 : : * Functions to determine whether an expression can be evaluated safely on
119 : : * remote server.
120 : : */
121 : : static bool foreign_expr_walker(Node *node,
122 : : foreign_glob_cxt *glob_cxt,
123 : : foreign_loc_cxt *outer_cxt,
124 : : foreign_loc_cxt *case_arg_cxt);
125 : : static char *deparse_type_name(Oid type_oid, int32 typemod);
126 : :
127 : : /*
128 : : * Functions to construct string representation of a node tree.
129 : : */
130 : : static void deparseTargetList(StringInfo buf,
131 : : RangeTblEntry *rte,
132 : : Index rtindex,
133 : : Relation rel,
134 : : bool is_returning,
135 : : Bitmapset *attrs_used,
136 : : bool qualify_col,
137 : : List **retrieved_attrs);
138 : : static void deparseExplicitTargetList(List *tlist,
139 : : bool is_returning,
140 : : List **retrieved_attrs,
141 : : deparse_expr_cxt *context);
142 : : static void deparseSubqueryTargetList(deparse_expr_cxt *context);
143 : : static void deparseReturningList(StringInfo buf, RangeTblEntry *rte,
144 : : Index rtindex, Relation rel,
145 : : bool trig_after_row,
146 : : List *withCheckOptionList,
147 : : List *returningList,
148 : : List **retrieved_attrs);
149 : : static void deparseColumnRef(StringInfo buf, int varno, int varattno,
150 : : RangeTblEntry *rte, bool qualify_col);
151 : : static void deparseRelation(StringInfo buf, Relation rel);
152 : : static void deparseExpr(Expr *node, deparse_expr_cxt *context);
153 : : static void deparseVar(Var *node, deparse_expr_cxt *context);
154 : : static void deparseConst(Const *node, deparse_expr_cxt *context, int showtype);
155 : : static void deparseParam(Param *node, deparse_expr_cxt *context);
156 : : static void deparseSubscriptingRef(SubscriptingRef *node, deparse_expr_cxt *context);
157 : : static void deparseFuncExpr(FuncExpr *node, deparse_expr_cxt *context);
158 : : static void deparseOpExpr(OpExpr *node, deparse_expr_cxt *context);
159 : : static bool isPlainForeignVar(Expr *node, deparse_expr_cxt *context);
160 : : static void deparseOperatorName(StringInfo buf, Form_pg_operator opform);
161 : : static void deparseDistinctExpr(DistinctExpr *node, deparse_expr_cxt *context);
162 : : static void deparseScalarArrayOpExpr(ScalarArrayOpExpr *node,
163 : : deparse_expr_cxt *context);
164 : : static void deparseRelabelType(RelabelType *node, deparse_expr_cxt *context);
165 : : static void deparseBoolExpr(BoolExpr *node, deparse_expr_cxt *context);
166 : : static void deparseNullTest(NullTest *node, deparse_expr_cxt *context);
167 : : static void deparseCaseExpr(CaseExpr *node, deparse_expr_cxt *context);
168 : : static void deparseArrayExpr(ArrayExpr *node, deparse_expr_cxt *context);
169 : : static void printRemoteParam(int paramindex, Oid paramtype, int32 paramtypmod,
170 : : deparse_expr_cxt *context);
171 : : static void printRemotePlaceholder(Oid paramtype, int32 paramtypmod,
172 : : deparse_expr_cxt *context);
173 : : static void deparseSelectSql(List *tlist, bool is_subquery, List **retrieved_attrs,
174 : : deparse_expr_cxt *context);
175 : : static void deparseLockingClause(deparse_expr_cxt *context);
176 : : static void appendOrderByClause(List *pathkeys, bool has_final_sort,
177 : : deparse_expr_cxt *context);
178 : : static void appendLimitClause(deparse_expr_cxt *context);
179 : : static void appendConditions(List *exprs, deparse_expr_cxt *context);
180 : : static void deparseFromExprForRel(StringInfo buf, PlannerInfo *root,
181 : : RelOptInfo *foreignrel, bool use_alias,
182 : : Index ignore_rel, List **ignore_conds,
183 : : List **additional_conds,
184 : : List **params_list);
185 : : static void appendWhereClause(List *exprs, List *additional_conds,
186 : : deparse_expr_cxt *context);
187 : : static void deparseFromExpr(List *quals, deparse_expr_cxt *context);
188 : : static void deparseRangeTblRef(StringInfo buf, PlannerInfo *root,
189 : : RelOptInfo *foreignrel, bool make_subquery,
190 : : Index ignore_rel, List **ignore_conds,
191 : : List **additional_conds, List **params_list);
192 : : static void deparseAggref(Aggref *node, deparse_expr_cxt *context);
193 : : static void appendGroupByClause(List *tlist, deparse_expr_cxt *context);
194 : : static void appendOrderBySuffix(Oid sortop, Oid sortcoltype, bool nulls_first,
195 : : deparse_expr_cxt *context);
196 : : static void appendAggOrderBy(List *orderList, List *targetList,
197 : : deparse_expr_cxt *context);
198 : : static void appendFunctionName(Oid funcid, deparse_expr_cxt *context);
199 : : static Node *deparseSortGroupClause(Index ref, List *tlist, bool force_colno,
200 : : deparse_expr_cxt *context);
201 : :
202 : : /*
203 : : * Helper functions
204 : : */
205 : : static bool is_subquery_var(Var *node, RelOptInfo *foreignrel,
206 : : int *relno, int *colno);
207 : : static void get_relation_column_alias_ids(Var *node, RelOptInfo *foreignrel,
208 : : int *relno, int *colno);
209 : :
210 : :
211 : : /*
212 : : * Examine each qual clause in input_conds, and classify them into two groups,
213 : : * which are returned as two lists:
214 : : * - remote_conds contains expressions that can be evaluated remotely
215 : : * - local_conds contains expressions that can't be evaluated remotely
216 : : */
217 : : void
4070 tgl@sss.pgh.pa.us 218 :CBC 2384 : classifyConditions(PlannerInfo *root,
219 : : RelOptInfo *baserel,
220 : : List *input_conds,
221 : : List **remote_conds,
222 : : List **local_conds)
223 : : {
224 : : ListCell *lc;
225 : :
226 : 2384 : *remote_conds = NIL;
227 : 2384 : *local_conds = NIL;
228 : :
3691 229 [ + + + + : 3101 : foreach(lc, input_conds)
+ + ]
230 : : {
2560 231 : 717 : RestrictInfo *ri = lfirst_node(RestrictInfo, lc);
232 : :
4042 233 [ + + ]: 717 : if (is_foreign_expr(root, baserel, ri->clause))
234 : 634 : *remote_conds = lappend(*remote_conds, ri);
235 : : else
4070 236 : 83 : *local_conds = lappend(*local_conds, ri);
237 : : }
238 : 2384 : }
239 : :
240 : : /*
241 : : * Returns true if given expr is safe to evaluate on the foreign server.
242 : : */
243 : : bool
244 : 3639 : is_foreign_expr(PlannerInfo *root,
245 : : RelOptInfo *baserel,
246 : : Expr *expr)
247 : : {
248 : : foreign_glob_cxt glob_cxt;
249 : : foreign_loc_cxt loc_cxt;
2732 rhaas@postgresql.org 250 : 3639 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) (baserel->fdw_private);
251 : :
252 : : /*
253 : : * Check that the expression consists of nodes that are safe to execute
254 : : * remotely.
255 : : */
4050 tgl@sss.pgh.pa.us 256 : 3639 : glob_cxt.root = root;
257 : 3639 : glob_cxt.foreignrel = baserel;
258 : :
259 : : /*
260 : : * For an upper relation, use relids from its underneath scan relation,
261 : : * because the upperrel's own relids currently aren't set to anything
262 : : * meaningful by the core code. For other relation, use their own relids.
263 : : */
2568 rhaas@postgresql.org 264 [ + + + + ]: 3639 : if (IS_UPPER_REL(baserel))
2732 265 : 448 : glob_cxt.relids = fpinfo->outerrel->relids;
266 : : else
267 : 3191 : glob_cxt.relids = baserel->relids;
4050 tgl@sss.pgh.pa.us 268 : 3639 : loc_cxt.collation = InvalidOid;
269 : 3639 : loc_cxt.state = FDW_COLLATE_NONE;
989 270 [ + + ]: 3639 : if (!foreign_expr_walker((Node *) expr, &glob_cxt, &loc_cxt, NULL))
4070 271 : 150 : return false;
272 : :
273 : : /*
274 : : * If the expression has a valid collation that does not arise from a
275 : : * foreign var, the expression can not be sent over.
276 : : */
3085 rhaas@postgresql.org 277 [ + + ]: 3489 : if (loc_cxt.state == FDW_COLLATE_UNSAFE)
278 : 2 : return false;
279 : :
280 : : /*
281 : : * An expression which includes any mutable functions can't be sent over
282 : : * because its result is not stable. For example, sending now() remote
283 : : * side could cause confusion from clock offsets. Future versions might
284 : : * be able to make this choice with more granularity. (We check this last
285 : : * because it requires a lot of expensive catalog lookups.)
286 : : */
4070 tgl@sss.pgh.pa.us 287 [ + + ]: 3487 : if (contain_mutable_functions((Node *) expr))
288 : 24 : return false;
289 : :
290 : : /* OK to evaluate on the remote server */
291 : 3463 : return true;
292 : : }
293 : :
294 : : /*
295 : : * Check if expression is safe to execute remotely, and return true if so.
296 : : *
297 : : * In addition, *outer_cxt is updated with collation information.
298 : : *
299 : : * case_arg_cxt is NULL if this subexpression is not inside a CASE-with-arg.
300 : : * Otherwise, it points to the collation info derived from the arg expression,
301 : : * which must be consulted by any CaseTestExpr.
302 : : *
303 : : * We must check that the expression contains only node types we can deparse,
304 : : * that all types/functions/operators are safe to send (they are "shippable"),
305 : : * and that all collations used in the expression derive from Vars of the
306 : : * foreign table. Because of the latter, the logic is pretty close to
307 : : * assign_collations_walker() in parse_collate.c, though we can assume here
308 : : * that the given expression is valid. Note function mutability is not
309 : : * currently considered here.
310 : : */
311 : : static bool
4050 312 : 9048 : foreign_expr_walker(Node *node,
313 : : foreign_glob_cxt *glob_cxt,
314 : : foreign_loc_cxt *outer_cxt,
315 : : foreign_loc_cxt *case_arg_cxt)
316 : : {
4070 317 : 9048 : bool check_type = true;
318 : : PgFdwRelationInfo *fpinfo;
319 : : foreign_loc_cxt inner_cxt;
320 : : Oid collation;
321 : : FDWCollateState state;
322 : :
323 : : /* Need do nothing for empty subexpressions */
324 [ + + ]: 9048 : if (node == NULL)
4050 325 : 322 : return true;
326 : :
327 : : /* May need server info from baserel's fdw_private struct */
3085 328 : 8726 : fpinfo = (PgFdwRelationInfo *) (glob_cxt->foreignrel->fdw_private);
329 : :
330 : : /* Set up inner_cxt for possible recursion to child nodes */
4050 331 : 8726 : inner_cxt.collation = InvalidOid;
332 : 8726 : inner_cxt.state = FDW_COLLATE_NONE;
333 : :
4070 334 [ + + + + : 8726 : switch (nodeTag(node))
+ + + + +
+ + + + +
+ + ]
335 : : {
336 : 4026 : case T_Var:
337 : : {
4050 338 : 4026 : Var *var = (Var *) node;
339 : :
340 : : /*
341 : : * If the Var is from the foreign table, we consider its
342 : : * collation (if any) safe to use. If it is from another
343 : : * table, we treat its collation the same way as we would a
344 : : * Param's collation, ie it's not safe for it to have a
345 : : * non-default collation.
346 : : */
2732 rhaas@postgresql.org 347 [ + + ]: 4026 : if (bms_is_member(var->varno, glob_cxt->relids) &&
4042 tgl@sss.pgh.pa.us 348 [ + - ]: 3603 : var->varlevelsup == 0)
349 : : {
350 : : /* Var belongs to foreign table */
351 : :
352 : : /*
353 : : * System columns other than ctid should not be sent to
354 : : * the remote, since we don't make any effort to ensure
355 : : * that local and remote values match (tableoid, in
356 : : * particular, almost certainly doesn't match).
357 : : */
3431 358 [ + + ]: 3603 : if (var->varattno < 0 &&
1972 andres@anarazel.de 359 [ + + ]: 10 : var->varattno != SelfItemPointerAttributeNumber)
3431 tgl@sss.pgh.pa.us 360 : 8 : return false;
361 : :
362 : : /* Else check the collation */
4042 363 : 3595 : collation = var->varcollid;
364 : 3595 : state = OidIsValid(collation) ? FDW_COLLATE_SAFE : FDW_COLLATE_NONE;
365 : : }
366 : : else
367 : : {
368 : : /* Var belongs to some other table */
3125 369 : 423 : collation = var->varcollid;
370 [ + + + - ]: 423 : if (collation == InvalidOid ||
371 : : collation == DEFAULT_COLLATION_OID)
372 : : {
373 : : /*
374 : : * It's noncollatable, or it's safe to combine with a
375 : : * collatable foreign Var, so set state to NONE.
376 : : */
377 : 423 : state = FDW_COLLATE_NONE;
378 : : }
379 : : else
380 : : {
381 : : /*
382 : : * Do not fail right away, since the Var might appear
383 : : * in a collation-insensitive context.
384 : : */
3125 tgl@sss.pgh.pa.us 385 :UBC 0 : state = FDW_COLLATE_UNSAFE;
386 : : }
387 : : }
388 : : }
4070 tgl@sss.pgh.pa.us 389 :CBC 4018 : break;
390 : 899 : case T_Const:
391 : : {
4050 392 : 899 : Const *c = (Const *) node;
393 : :
394 : : /*
395 : : * Constants of regproc and related types can't be shipped
396 : : * unless the referenced object is shippable. But NULL's ok.
397 : : * (See also the related code in dependency.c.)
398 : : */
637 399 [ + + ]: 899 : if (!c->constisnull)
400 : : {
401 [ - - - - : 890 : switch (c->consttype)
- + - - -
+ ]
402 : : {
637 tgl@sss.pgh.pa.us 403 :UBC 0 : case REGPROCOID:
404 : : case REGPROCEDUREOID:
405 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
406 : : ProcedureRelationId, fpinfo))
407 : 0 : return false;
408 : 0 : break;
409 : 0 : case REGOPEROID:
410 : : case REGOPERATOROID:
411 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
412 : : OperatorRelationId, fpinfo))
413 : 0 : return false;
414 : 0 : break;
415 : 0 : case REGCLASSOID:
416 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
417 : : RelationRelationId, fpinfo))
418 : 0 : return false;
419 : 0 : break;
420 : 0 : case REGTYPEOID:
421 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
422 : : TypeRelationId, fpinfo))
423 : 0 : return false;
424 : 0 : break;
425 : 0 : case REGCOLLATIONOID:
426 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
427 : : CollationRelationId, fpinfo))
428 : 0 : return false;
429 : 0 : break;
637 tgl@sss.pgh.pa.us 430 :CBC 4 : case REGCONFIGOID:
431 : :
432 : : /*
433 : : * For text search objects only, we weaken the
434 : : * normal shippability criterion to allow all OIDs
435 : : * below FirstNormalObjectId. Without this, none
436 : : * of the initdb-installed TS configurations would
437 : : * be shippable, which would be quite annoying.
438 : : */
439 [ + - ]: 4 : if (DatumGetObjectId(c->constvalue) >= FirstNormalObjectId &&
440 [ + + ]: 4 : !is_shippable(DatumGetObjectId(c->constvalue),
441 : : TSConfigRelationId, fpinfo))
442 : 2 : return false;
443 : 2 : break;
637 tgl@sss.pgh.pa.us 444 :UBC 0 : case REGDICTIONARYOID:
445 [ # # ]: 0 : if (DatumGetObjectId(c->constvalue) >= FirstNormalObjectId &&
446 [ # # ]: 0 : !is_shippable(DatumGetObjectId(c->constvalue),
447 : : TSDictionaryRelationId, fpinfo))
448 : 0 : return false;
449 : 0 : break;
450 : 0 : case REGNAMESPACEOID:
451 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
452 : : NamespaceRelationId, fpinfo))
453 : 0 : return false;
454 : 0 : break;
455 : 0 : case REGROLEOID:
456 [ # # ]: 0 : if (!is_shippable(DatumGetObjectId(c->constvalue),
457 : : AuthIdRelationId, fpinfo))
458 : 0 : return false;
459 : 0 : break;
460 : : }
461 : : }
462 : :
463 : : /*
464 : : * If the constant has nondefault collation, either it's of a
465 : : * non-builtin type, or it reflects folding of a CollateExpr.
466 : : * It's unsafe to send to the remote unless it's used in a
467 : : * non-collation-sensitive context.
468 : : */
3125 tgl@sss.pgh.pa.us 469 :CBC 897 : collation = c->constcollid;
470 [ + + + + ]: 897 : if (collation == InvalidOid ||
471 : : collation == DEFAULT_COLLATION_OID)
472 : 895 : state = FDW_COLLATE_NONE;
473 : : else
474 : 2 : state = FDW_COLLATE_UNSAFE;
475 : : }
4070 476 : 897 : break;
477 : 25 : case T_Param:
478 : : {
479 : 25 : Param *p = (Param *) node;
480 : :
481 : : /*
482 : : * If it's a MULTIEXPR Param, punt. We can't tell from here
483 : : * whether the referenced sublink/subplan contains any remote
484 : : * Vars; if it does, handling that is too complicated to
485 : : * consider supporting at present. Fortunately, MULTIEXPR
486 : : * Params are not reduced to plain PARAM_EXEC until the end of
487 : : * planning, so we can easily detect this case. (Normal
488 : : * PARAM_EXEC Params are safe to ship because their values
489 : : * come from somewhere else in the plan tree; but a MULTIEXPR
490 : : * references a sub-select elsewhere in the same targetlist,
491 : : * so we'd be on the hook to evaluate it somehow if we wanted
492 : : * to handle such cases as direct foreign updates.)
493 : : */
1540 494 [ + + ]: 25 : if (p->paramkind == PARAM_MULTIEXPR)
495 : 3 : return false;
496 : :
497 : : /*
498 : : * Collation rule is same as for Consts and non-foreign Vars.
499 : : */
3125 500 : 22 : collation = p->paramcollid;
501 [ + + + - ]: 22 : if (collation == InvalidOid ||
502 : : collation == DEFAULT_COLLATION_OID)
503 : 22 : state = FDW_COLLATE_NONE;
504 : : else
3125 tgl@sss.pgh.pa.us 505 :UBC 0 : state = FDW_COLLATE_UNSAFE;
506 : : }
4070 tgl@sss.pgh.pa.us 507 :CBC 22 : break;
1899 alvherre@alvh.no-ip. 508 : 1 : case T_SubscriptingRef:
509 : : {
510 : 1 : SubscriptingRef *sr = (SubscriptingRef *) node;
511 : :
512 : : /* Assignment should not be in restrictions. */
513 [ - + ]: 1 : if (sr->refassgnexpr != NULL)
4050 tgl@sss.pgh.pa.us 514 :UBC 0 : return false;
515 : :
516 : : /*
517 : : * Recurse into the remaining subexpressions. The container
518 : : * subscripts will not affect collation of the SubscriptingRef
519 : : * result, so do those first and reset inner_cxt afterwards.
520 : : */
1899 alvherre@alvh.no-ip. 521 [ - + ]:CBC 1 : if (!foreign_expr_walker((Node *) sr->refupperindexpr,
522 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 523 :UBC 0 : return false;
1222 tgl@sss.pgh.pa.us 524 :CBC 1 : inner_cxt.collation = InvalidOid;
525 : 1 : inner_cxt.state = FDW_COLLATE_NONE;
1899 alvherre@alvh.no-ip. 526 [ - + ]: 1 : if (!foreign_expr_walker((Node *) sr->reflowerindexpr,
527 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 528 :UBC 0 : return false;
1222 tgl@sss.pgh.pa.us 529 :CBC 1 : inner_cxt.collation = InvalidOid;
530 : 1 : inner_cxt.state = FDW_COLLATE_NONE;
1899 alvherre@alvh.no-ip. 531 [ - + ]: 1 : if (!foreign_expr_walker((Node *) sr->refexpr,
532 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 533 :UBC 0 : return false;
534 : :
535 : : /*
536 : : * Container subscripting typically yields same collation as
537 : : * refexpr's, but in case it doesn't, use same logic as for
538 : : * function nodes.
539 : : */
1899 alvherre@alvh.no-ip. 540 :CBC 1 : collation = sr->refcollid;
4050 tgl@sss.pgh.pa.us 541 [ + - ]: 1 : if (collation == InvalidOid)
542 : 1 : state = FDW_COLLATE_NONE;
4050 tgl@sss.pgh.pa.us 543 [ # # ]:UBC 0 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
544 [ # # ]: 0 : collation == inner_cxt.collation)
545 : 0 : state = FDW_COLLATE_SAFE;
3125 546 [ # # ]: 0 : else if (collation == DEFAULT_COLLATION_OID)
547 : 0 : state = FDW_COLLATE_NONE;
548 : : else
4050 549 : 0 : state = FDW_COLLATE_UNSAFE;
550 : : }
4070 tgl@sss.pgh.pa.us 551 :CBC 1 : break;
552 : 134 : case T_FuncExpr:
553 : : {
4050 554 : 134 : FuncExpr *fe = (FuncExpr *) node;
555 : :
556 : : /*
557 : : * If function used by the expression is not shippable, it
558 : : * can't be sent to remote because it might have incompatible
559 : : * semantics on remote side.
560 : : */
3085 561 [ + + ]: 134 : if (!is_shippable(fe->funcid, ProcedureRelationId, fpinfo))
4050 562 : 8 : return false;
563 : :
564 : : /*
565 : : * Recurse to input subexpressions.
566 : : */
567 [ + + ]: 126 : if (!foreign_expr_walker((Node *) fe->args,
568 : : glob_cxt, &inner_cxt, case_arg_cxt))
569 : 13 : return false;
570 : :
571 : : /*
572 : : * If function's input collation is not derived from a foreign
573 : : * Var, it can't be sent to remote.
574 : : */
575 [ + + ]: 113 : if (fe->inputcollid == InvalidOid)
576 : : /* OK, inputs are all noncollatable */ ;
577 [ + + ]: 33 : else if (inner_cxt.state != FDW_COLLATE_SAFE ||
578 [ - + ]: 20 : fe->inputcollid != inner_cxt.collation)
579 : 13 : return false;
580 : :
581 : : /*
582 : : * Detect whether node is introducing a collation not derived
583 : : * from a foreign Var. (If so, we just mark it unsafe for now
584 : : * rather than immediately returning false, since the parent
585 : : * node might not care.)
586 : : */
587 : 100 : collation = fe->funccollid;
588 [ + + ]: 100 : if (collation == InvalidOid)
589 : 81 : state = FDW_COLLATE_NONE;
590 [ + + ]: 19 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
591 [ + - ]: 18 : collation == inner_cxt.collation)
592 : 18 : state = FDW_COLLATE_SAFE;
3125 593 [ - + ]: 1 : else if (collation == DEFAULT_COLLATION_OID)
3125 tgl@sss.pgh.pa.us 594 :UBC 0 : state = FDW_COLLATE_NONE;
595 : : else
4050 tgl@sss.pgh.pa.us 596 :CBC 1 : state = FDW_COLLATE_UNSAFE;
597 : : }
4070 598 : 100 : break;
599 : 1526 : case T_OpExpr:
600 : : case T_DistinctExpr: /* struct-equivalent to OpExpr */
601 : : {
4050 602 : 1526 : OpExpr *oe = (OpExpr *) node;
603 : :
604 : : /*
605 : : * Similarly, only shippable operators can be sent to remote.
606 : : * (If the operator is shippable, we assume its underlying
607 : : * function is too.)
608 : : */
3085 609 [ + + ]: 1526 : if (!is_shippable(oe->opno, OperatorRelationId, fpinfo))
4050 610 : 47 : return false;
611 : :
612 : : /*
613 : : * Recurse to input subexpressions.
614 : : */
615 [ + + ]: 1479 : if (!foreign_expr_walker((Node *) oe->args,
616 : : glob_cxt, &inner_cxt, case_arg_cxt))
617 : 63 : return false;
618 : :
619 : : /*
620 : : * If operator's input collation is not derived from a foreign
621 : : * Var, it can't be sent to remote.
622 : : */
623 [ + + ]: 1416 : if (oe->inputcollid == InvalidOid)
624 : : /* OK, inputs are all noncollatable */ ;
625 [ + + ]: 69 : else if (inner_cxt.state != FDW_COLLATE_SAFE ||
626 [ - + ]: 63 : oe->inputcollid != inner_cxt.collation)
627 : 6 : return false;
628 : :
629 : : /* Result-collation handling is same as for functions */
630 : 1410 : collation = oe->opcollid;
631 [ + + ]: 1410 : if (collation == InvalidOid)
632 : 1396 : state = FDW_COLLATE_NONE;
633 [ + - ]: 14 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
634 [ + - ]: 14 : collation == inner_cxt.collation)
635 : 14 : state = FDW_COLLATE_SAFE;
3125 tgl@sss.pgh.pa.us 636 [ # # ]:UBC 0 : else if (collation == DEFAULT_COLLATION_OID)
637 : 0 : state = FDW_COLLATE_NONE;
638 : : else
4050 639 : 0 : state = FDW_COLLATE_UNSAFE;
640 : : }
4070 tgl@sss.pgh.pa.us 641 :CBC 1410 : break;
642 : 3 : case T_ScalarArrayOpExpr:
643 : : {
4050 644 : 3 : ScalarArrayOpExpr *oe = (ScalarArrayOpExpr *) node;
645 : :
646 : : /*
647 : : * Again, only shippable operators can be sent to remote.
648 : : */
3085 649 [ - + ]: 3 : if (!is_shippable(oe->opno, OperatorRelationId, fpinfo))
4050 tgl@sss.pgh.pa.us 650 :UBC 0 : return false;
651 : :
652 : : /*
653 : : * Recurse to input subexpressions.
654 : : */
4050 tgl@sss.pgh.pa.us 655 [ - + ]:CBC 3 : if (!foreign_expr_walker((Node *) oe->args,
656 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 657 :UBC 0 : return false;
658 : :
659 : : /*
660 : : * If operator's input collation is not derived from a foreign
661 : : * Var, it can't be sent to remote.
662 : : */
4050 tgl@sss.pgh.pa.us 663 [ - + ]:CBC 3 : if (oe->inputcollid == InvalidOid)
664 : : /* OK, inputs are all noncollatable */ ;
4050 tgl@sss.pgh.pa.us 665 [ # # ]:UBC 0 : else if (inner_cxt.state != FDW_COLLATE_SAFE ||
666 [ # # ]: 0 : oe->inputcollid != inner_cxt.collation)
667 : 0 : return false;
668 : :
669 : : /* Output is always boolean and so noncollatable. */
4050 tgl@sss.pgh.pa.us 670 :CBC 3 : collation = InvalidOid;
671 : 3 : state = FDW_COLLATE_NONE;
672 : : }
4070 673 : 3 : break;
674 : 68 : case T_RelabelType:
675 : : {
4050 676 : 68 : RelabelType *r = (RelabelType *) node;
677 : :
678 : : /*
679 : : * Recurse to input subexpression.
680 : : */
681 [ - + ]: 68 : if (!foreign_expr_walker((Node *) r->arg,
682 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 683 :UBC 0 : return false;
684 : :
685 : : /*
686 : : * RelabelType must not introduce a collation not derived from
687 : : * an input foreign Var (same logic as for a real function).
688 : : */
4050 tgl@sss.pgh.pa.us 689 :CBC 68 : collation = r->resultcollid;
690 [ - + ]: 68 : if (collation == InvalidOid)
4050 tgl@sss.pgh.pa.us 691 :UBC 0 : state = FDW_COLLATE_NONE;
4050 tgl@sss.pgh.pa.us 692 [ + + ]:CBC 68 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
693 [ + + ]: 64 : collation == inner_cxt.collation)
694 : 58 : state = FDW_COLLATE_SAFE;
3125 695 [ + + ]: 10 : else if (collation == DEFAULT_COLLATION_OID)
696 : 3 : state = FDW_COLLATE_NONE;
697 : : else
4050 698 : 7 : state = FDW_COLLATE_UNSAFE;
699 : : }
700 : 68 : break;
4070 701 : 36 : case T_BoolExpr:
702 : : {
4050 703 : 36 : BoolExpr *b = (BoolExpr *) node;
704 : :
705 : : /*
706 : : * Recurse to input subexpressions.
707 : : */
708 [ - + ]: 36 : if (!foreign_expr_walker((Node *) b->args,
709 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 710 :UBC 0 : return false;
711 : :
712 : : /* Output is always boolean and so noncollatable. */
4050 tgl@sss.pgh.pa.us 713 :CBC 36 : collation = InvalidOid;
714 : 36 : state = FDW_COLLATE_NONE;
715 : : }
716 : 36 : break;
4070 717 : 27 : case T_NullTest:
718 : : {
4050 719 : 27 : NullTest *nt = (NullTest *) node;
720 : :
721 : : /*
722 : : * Recurse to input subexpressions.
723 : : */
724 [ - + ]: 27 : if (!foreign_expr_walker((Node *) nt->arg,
725 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 726 :UBC 0 : return false;
727 : :
728 : : /* Output is always boolean and so noncollatable. */
4050 tgl@sss.pgh.pa.us 729 :CBC 27 : collation = InvalidOid;
730 : 27 : state = FDW_COLLATE_NONE;
731 : : }
732 : 27 : break;
989 733 : 13 : case T_CaseExpr:
734 : : {
735 : 13 : CaseExpr *ce = (CaseExpr *) node;
736 : : foreign_loc_cxt arg_cxt;
737 : : foreign_loc_cxt tmp_cxt;
738 : : ListCell *lc;
739 : :
740 : : /*
741 : : * Recurse to CASE's arg expression, if any. Its collation
742 : : * has to be saved aside for use while examining CaseTestExprs
743 : : * within the WHEN expressions.
744 : : */
745 : 13 : arg_cxt.collation = InvalidOid;
746 : 13 : arg_cxt.state = FDW_COLLATE_NONE;
747 [ + + ]: 13 : if (ce->arg)
748 : : {
749 [ - + ]: 7 : if (!foreign_expr_walker((Node *) ce->arg,
750 : : glob_cxt, &arg_cxt, case_arg_cxt))
989 tgl@sss.pgh.pa.us 751 :UBC 0 : return false;
752 : : }
753 : :
754 : : /* Examine the CaseWhen subexpressions. */
989 tgl@sss.pgh.pa.us 755 [ + - + + :CBC 29 : foreach(lc, ce->args)
+ + ]
756 : : {
757 : 17 : CaseWhen *cw = lfirst_node(CaseWhen, lc);
758 : :
759 [ + + ]: 17 : if (ce->arg)
760 : : {
761 : : /*
762 : : * In a CASE-with-arg, the parser should have produced
763 : : * WHEN clauses of the form "CaseTestExpr = RHS",
764 : : * possibly with an implicit coercion inserted above
765 : : * the CaseTestExpr. However in an expression that's
766 : : * been through the optimizer, the WHEN clause could
767 : : * be almost anything (since the equality operator
768 : : * could have been expanded into an inline function).
769 : : * In such cases forbid pushdown, because
770 : : * deparseCaseExpr can't handle it.
771 : : */
772 : 11 : Node *whenExpr = (Node *) cw->expr;
773 : : List *opArgs;
774 : :
775 [ - + ]: 11 : if (!IsA(whenExpr, OpExpr))
989 tgl@sss.pgh.pa.us 776 :UBC 0 : return false;
777 : :
989 tgl@sss.pgh.pa.us 778 :CBC 11 : opArgs = ((OpExpr *) whenExpr)->args;
779 [ + - ]: 11 : if (list_length(opArgs) != 2 ||
780 [ - + ]: 11 : !IsA(strip_implicit_coercions(linitial(opArgs)),
781 : : CaseTestExpr))
989 tgl@sss.pgh.pa.us 782 :UBC 0 : return false;
783 : : }
784 : :
785 : : /*
786 : : * Recurse to WHEN expression, passing down the arg info.
787 : : * Its collation doesn't affect the result (really, it
788 : : * should be boolean and thus not have a collation).
789 : : */
989 tgl@sss.pgh.pa.us 790 :CBC 17 : tmp_cxt.collation = InvalidOid;
791 : 17 : tmp_cxt.state = FDW_COLLATE_NONE;
792 [ + + ]: 17 : if (!foreign_expr_walker((Node *) cw->expr,
793 : : glob_cxt, &tmp_cxt, &arg_cxt))
794 : 1 : return false;
795 : :
796 : : /* Recurse to THEN expression. */
797 [ - + ]: 16 : if (!foreign_expr_walker((Node *) cw->result,
798 : : glob_cxt, &inner_cxt, case_arg_cxt))
989 tgl@sss.pgh.pa.us 799 :UBC 0 : return false;
800 : : }
801 : :
802 : : /* Recurse to ELSE expression. */
989 tgl@sss.pgh.pa.us 803 [ - + ]:CBC 12 : if (!foreign_expr_walker((Node *) ce->defresult,
804 : : glob_cxt, &inner_cxt, case_arg_cxt))
989 tgl@sss.pgh.pa.us 805 :UBC 0 : return false;
806 : :
807 : : /*
808 : : * Detect whether node is introducing a collation not derived
809 : : * from a foreign Var. (If so, we just mark it unsafe for now
810 : : * rather than immediately returning false, since the parent
811 : : * node might not care.) This is the same as for function
812 : : * nodes, except that the input collation is derived from only
813 : : * the THEN and ELSE subexpressions.
814 : : */
989 tgl@sss.pgh.pa.us 815 :CBC 12 : collation = ce->casecollid;
816 [ + - ]: 12 : if (collation == InvalidOid)
817 : 12 : state = FDW_COLLATE_NONE;
989 tgl@sss.pgh.pa.us 818 [ # # ]:UBC 0 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
819 [ # # ]: 0 : collation == inner_cxt.collation)
820 : 0 : state = FDW_COLLATE_SAFE;
821 [ # # ]: 0 : else if (collation == DEFAULT_COLLATION_OID)
822 : 0 : state = FDW_COLLATE_NONE;
823 : : else
824 : 0 : state = FDW_COLLATE_UNSAFE;
825 : : }
989 tgl@sss.pgh.pa.us 826 :CBC 12 : break;
827 : 11 : case T_CaseTestExpr:
828 : : {
829 : 11 : CaseTestExpr *c = (CaseTestExpr *) node;
830 : :
831 : : /* Punt if we seem not to be inside a CASE arg WHEN. */
832 [ - + ]: 11 : if (!case_arg_cxt)
989 tgl@sss.pgh.pa.us 833 :UBC 0 : return false;
834 : :
835 : : /*
836 : : * Otherwise, any nondefault collation attached to the
837 : : * CaseTestExpr node must be derived from foreign Var(s) in
838 : : * the CASE arg.
839 : : */
989 tgl@sss.pgh.pa.us 840 :CBC 11 : collation = c->collation;
841 [ + + ]: 11 : if (collation == InvalidOid)
842 : 8 : state = FDW_COLLATE_NONE;
843 [ + + ]: 3 : else if (case_arg_cxt->state == FDW_COLLATE_SAFE &&
844 [ + - ]: 2 : collation == case_arg_cxt->collation)
845 : 2 : state = FDW_COLLATE_SAFE;
846 [ - + ]: 1 : else if (collation == DEFAULT_COLLATION_OID)
989 tgl@sss.pgh.pa.us 847 :UBC 0 : state = FDW_COLLATE_NONE;
848 : : else
989 tgl@sss.pgh.pa.us 849 :CBC 1 : state = FDW_COLLATE_UNSAFE;
850 : : }
851 : 11 : break;
4070 852 : 4 : case T_ArrayExpr:
853 : : {
4050 854 : 4 : ArrayExpr *a = (ArrayExpr *) node;
855 : :
856 : : /*
857 : : * Recurse to input subexpressions.
858 : : */
859 [ - + ]: 4 : if (!foreign_expr_walker((Node *) a->elements,
860 : : glob_cxt, &inner_cxt, case_arg_cxt))
4050 tgl@sss.pgh.pa.us 861 :UBC 0 : return false;
862 : :
863 : : /*
864 : : * ArrayExpr must not introduce a collation not derived from
865 : : * an input foreign Var (same logic as for a function).
866 : : */
4050 tgl@sss.pgh.pa.us 867 :CBC 4 : collation = a->array_collid;
868 [ + - ]: 4 : if (collation == InvalidOid)
869 : 4 : state = FDW_COLLATE_NONE;
4050 tgl@sss.pgh.pa.us 870 [ # # ]:UBC 0 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
871 [ # # ]: 0 : collation == inner_cxt.collation)
872 : 0 : state = FDW_COLLATE_SAFE;
3125 873 [ # # ]: 0 : else if (collation == DEFAULT_COLLATION_OID)
874 : 0 : state = FDW_COLLATE_NONE;
875 : : else
4050 876 : 0 : state = FDW_COLLATE_UNSAFE;
877 : : }
4070 tgl@sss.pgh.pa.us 878 :CBC 4 : break;
879 : 1648 : case T_List:
880 : : {
4050 881 : 1648 : List *l = (List *) node;
882 : : ListCell *lc;
883 : :
884 : : /*
885 : : * Recurse to component subexpressions.
886 : : */
887 [ + - + + : 4684 : foreach(lc, l)
+ + ]
888 : : {
889 [ + + ]: 3125 : if (!foreign_expr_walker((Node *) lfirst(lc),
890 : : glob_cxt, &inner_cxt, case_arg_cxt))
891 : 89 : return false;
892 : : }
893 : :
894 : : /*
895 : : * When processing a list, collation state just bubbles up
896 : : * from the list elements.
897 : : */
898 : 1559 : collation = inner_cxt.collation;
899 : 1559 : state = inner_cxt.state;
900 : :
901 : : /* Don't apply exprType() to the list. */
902 : 1559 : check_type = false;
903 : : }
4070 904 : 1559 : break;
2732 rhaas@postgresql.org 905 : 275 : case T_Aggref:
906 : : {
907 : 275 : Aggref *agg = (Aggref *) node;
908 : : ListCell *lc;
909 : :
910 : : /* Not safe to pushdown when not in grouping context */
2568 911 [ + + - + ]: 275 : if (!IS_UPPER_REL(glob_cxt->foreignrel))
2732 rhaas@postgresql.org 912 :UBC 0 : return false;
913 : :
914 : : /* Only non-split aggregates are pushable. */
2732 rhaas@postgresql.org 915 [ - + ]:CBC 275 : if (agg->aggsplit != AGGSPLIT_SIMPLE)
2732 rhaas@postgresql.org 916 :UBC 0 : return false;
917 : :
918 : : /* As usual, it must be shippable. */
2732 rhaas@postgresql.org 919 [ + + ]:CBC 275 : if (!is_shippable(agg->aggfnoid, ProcedureRelationId, fpinfo))
920 : 4 : return false;
921 : :
922 : : /*
923 : : * Recurse to input args. aggdirectargs, aggorder and
924 : : * aggdistinct are all present in args, so no need to check
925 : : * their shippability explicitly.
926 : : */
927 [ + + + + : 490 : foreach(lc, agg->args)
+ + ]
928 : : {
929 : 231 : Node *n = (Node *) lfirst(lc);
930 : :
931 : : /* If TargetEntry, extract the expression from it */
932 [ + - ]: 231 : if (IsA(n, TargetEntry))
933 : : {
934 : 231 : TargetEntry *tle = (TargetEntry *) n;
935 : :
936 : 231 : n = (Node *) tle->expr;
937 : : }
938 : :
989 tgl@sss.pgh.pa.us 939 [ + + ]: 231 : if (!foreign_expr_walker(n,
940 : : glob_cxt, &inner_cxt, case_arg_cxt))
2732 rhaas@postgresql.org 941 : 12 : return false;
942 : : }
943 : :
944 : : /*
945 : : * For aggorder elements, check whether the sort operator, if
946 : : * specified, is shippable or not.
947 : : */
948 [ + + ]: 259 : if (agg->aggorder)
949 : : {
950 [ + - + + : 70 : foreach(lc, agg->aggorder)
+ + ]
951 : : {
952 : 38 : SortGroupClause *srt = (SortGroupClause *) lfirst(lc);
953 : : Oid sortcoltype;
954 : : TypeCacheEntry *typentry;
955 : : TargetEntry *tle;
956 : :
957 : 38 : tle = get_sortgroupref_tle(srt->tleSortGroupRef,
958 : : agg->args);
959 : 38 : sortcoltype = exprType((Node *) tle->expr);
960 : 38 : typentry = lookup_type_cache(sortcoltype,
961 : : TYPECACHE_LT_OPR | TYPECACHE_GT_OPR);
962 : : /* Check shippability of non-default sort operator. */
963 [ + + ]: 38 : if (srt->sortop != typentry->lt_opr &&
964 [ + + ]: 14 : srt->sortop != typentry->gt_opr &&
965 [ + + ]: 6 : !is_shippable(srt->sortop, OperatorRelationId,
966 : : fpinfo))
967 : 4 : return false;
968 : : }
969 : : }
970 : :
971 : : /* Check aggregate filter */
972 [ + + ]: 255 : if (!foreign_expr_walker((Node *) agg->aggfilter,
973 : : glob_cxt, &inner_cxt, case_arg_cxt))
974 : 2 : return false;
975 : :
976 : : /*
977 : : * If aggregate's input collation is not derived from a
978 : : * foreign Var, it can't be sent to remote.
979 : : */
980 [ + + ]: 253 : if (agg->inputcollid == InvalidOid)
981 : : /* OK, inputs are all noncollatable */ ;
982 [ + - ]: 22 : else if (inner_cxt.state != FDW_COLLATE_SAFE ||
983 [ - + ]: 22 : agg->inputcollid != inner_cxt.collation)
2732 rhaas@postgresql.org 984 :UBC 0 : return false;
985 : :
986 : : /*
987 : : * Detect whether node is introducing a collation not derived
988 : : * from a foreign Var. (If so, we just mark it unsafe for now
989 : : * rather than immediately returning false, since the parent
990 : : * node might not care.)
991 : : */
2732 rhaas@postgresql.org 992 :CBC 253 : collation = agg->aggcollid;
993 [ + + ]: 253 : if (collation == InvalidOid)
994 : 252 : state = FDW_COLLATE_NONE;
995 [ + - ]: 1 : else if (inner_cxt.state == FDW_COLLATE_SAFE &&
996 [ + - ]: 1 : collation == inner_cxt.collation)
997 : 1 : state = FDW_COLLATE_SAFE;
2732 rhaas@postgresql.org 998 [ # # ]:UBC 0 : else if (collation == DEFAULT_COLLATION_OID)
999 : 0 : state = FDW_COLLATE_NONE;
1000 : : else
1001 : 0 : state = FDW_COLLATE_UNSAFE;
1002 : : }
2732 rhaas@postgresql.org 1003 :CBC 253 : break;
4070 tgl@sss.pgh.pa.us 1004 : 30 : default:
1005 : :
1006 : : /*
1007 : : * If it's anything else, assume it's unsafe. This list can be
1008 : : * expanded later, but don't forget to add deparse support below.
1009 : : */
4050 1010 : 30 : return false;
1011 : : }
1012 : :
1013 : : /*
1014 : : * If result type of given expression is not shippable, it can't be sent
1015 : : * to remote because it might have incompatible semantics on remote side.
1016 : : */
3085 1017 [ + + + + ]: 8421 : if (check_type && !is_shippable(exprType(node), TypeRelationId, fpinfo))
4050 1018 : 25 : return false;
1019 : :
1020 : : /*
1021 : : * Now, merge my collation information into my parent's state.
1022 : : */
1023 [ + + ]: 8396 : if (state > outer_cxt->state)
1024 : : {
1025 : : /* Override previous parent state */
1026 : 480 : outer_cxt->collation = collation;
1027 : 480 : outer_cxt->state = state;
1028 : : }
1029 [ + + ]: 7916 : else if (state == outer_cxt->state)
1030 : : {
1031 : : /* Merge, or detect error if there's a collation conflict */
1032 [ + + + - ]: 7865 : switch (state)
1033 : : {
1034 : 7853 : case FDW_COLLATE_NONE:
1035 : : /* Nothing + nothing is still nothing */
1036 : 7853 : break;
1037 : 11 : case FDW_COLLATE_SAFE:
1038 [ - + ]: 11 : if (collation != outer_cxt->collation)
1039 : : {
1040 : : /*
1041 : : * Non-default collation always beats default.
1042 : : */
4050 tgl@sss.pgh.pa.us 1043 [ # # ]:UBC 0 : if (outer_cxt->collation == DEFAULT_COLLATION_OID)
1044 : : {
1045 : : /* Override previous parent state */
1046 : 0 : outer_cxt->collation = collation;
1047 : : }
1048 [ # # ]: 0 : else if (collation != DEFAULT_COLLATION_OID)
1049 : : {
1050 : : /*
1051 : : * Conflict; show state as indeterminate. We don't
1052 : : * want to "return false" right away, since parent
1053 : : * node might not care about collation.
1054 : : */
1055 : 0 : outer_cxt->state = FDW_COLLATE_UNSAFE;
1056 : : }
1057 : : }
4050 tgl@sss.pgh.pa.us 1058 :CBC 11 : break;
1059 : 1 : case FDW_COLLATE_UNSAFE:
1060 : : /* We're still conflicted ... */
1061 : 1 : break;
1062 : : }
1063 : : }
1064 : :
1065 : : /* It looks OK */
1066 : 8396 : return true;
1067 : : }
1068 : :
1069 : : /*
1070 : : * Returns true if given expr is something we'd have to send the value of
1071 : : * to the foreign server.
1072 : : *
1073 : : * This should return true when the expression is a shippable node that
1074 : : * deparseExpr would add to context->params_list. Note that we don't care
1075 : : * if the expression *contains* such a node, only whether one appears at top
1076 : : * level. We need this to detect cases where setrefs.c would recognize a
1077 : : * false match between an fdw_exprs item (which came from the params_list)
1078 : : * and an entry in fdw_scan_tlist (which we're considering putting the given
1079 : : * expression into).
1080 : : */
1081 : : bool
1814 1082 : 271 : is_foreign_param(PlannerInfo *root,
1083 : : RelOptInfo *baserel,
1084 : : Expr *expr)
1085 : : {
1086 [ - + ]: 271 : if (expr == NULL)
1814 tgl@sss.pgh.pa.us 1087 :UBC 0 : return false;
1088 : :
1814 tgl@sss.pgh.pa.us 1089 [ + + + ]:CBC 271 : switch (nodeTag(expr))
1090 : : {
1091 : 76 : case T_Var:
1092 : : {
1093 : : /* It would have to be sent unless it's a foreign Var */
1094 : 76 : Var *var = (Var *) expr;
1095 : 76 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) (baserel->fdw_private);
1096 : : Relids relids;
1097 : :
1098 [ + + + - ]: 76 : if (IS_UPPER_REL(baserel))
1099 : 76 : relids = fpinfo->outerrel->relids;
1100 : : else
1814 tgl@sss.pgh.pa.us 1101 :UBC 0 : relids = baserel->relids;
1102 : :
1814 tgl@sss.pgh.pa.us 1103 [ + - + - ]:CBC 76 : if (bms_is_member(var->varno, relids) && var->varlevelsup == 0)
1104 : 76 : return false; /* foreign Var, so not a param */
1105 : : else
1814 tgl@sss.pgh.pa.us 1106 :UBC 0 : return true; /* it'd have to be a param */
1107 : : break;
1108 : : }
1814 tgl@sss.pgh.pa.us 1109 :CBC 4 : case T_Param:
1110 : : /* Params always have to be sent to the foreign server */
1111 : 4 : return true;
1112 : 191 : default:
1113 : 191 : break;
1114 : : }
1115 : 191 : return false;
1116 : : }
1117 : :
1118 : : /*
1119 : : * Returns true if it's safe to push down the sort expression described by
1120 : : * 'pathkey' to the foreign server.
1121 : : */
1122 : : bool
745 1123 : 759 : is_foreign_pathkey(PlannerInfo *root,
1124 : : RelOptInfo *baserel,
1125 : : PathKey *pathkey)
1126 : : {
1127 : 759 : EquivalenceClass *pathkey_ec = pathkey->pk_eclass;
1128 : 759 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) baserel->fdw_private;
1129 : :
1130 : : /*
1131 : : * is_foreign_expr would detect volatile expressions as well, but checking
1132 : : * ec_has_volatile here saves some cycles.
1133 : : */
1134 [ + + ]: 759 : if (pathkey_ec->ec_has_volatile)
1135 : 4 : return false;
1136 : :
1137 : : /* can't push down the sort if the pathkey's opfamily is not shippable */
1138 [ + + ]: 755 : if (!is_shippable(pathkey->pk_opfamily, OperatorFamilyRelationId, fpinfo))
1139 : 3 : return false;
1140 : :
1141 : : /* can push if a suitable EC member exists */
1142 : 752 : return (find_em_for_rel(root, pathkey_ec, baserel) != NULL);
1143 : : }
1144 : :
1145 : : /*
1146 : : * Convert type OID + typmod info into a type name we can ship to the remote
1147 : : * server. Someplace else had better have verified that this type name is
1148 : : * expected to be known on the remote end.
1149 : : *
1150 : : * This is almost just format_type_with_typemod(), except that if left to its
1151 : : * own devices, that function will make schema-qualification decisions based
1152 : : * on the local search_path, which is wrong. We must schema-qualify all
1153 : : * type names that are not in pg_catalog. We assume here that built-in types
1154 : : * are all in pg_catalog and need not be qualified; otherwise, qualify.
1155 : : */
1156 : : static char *
3085 1157 : 527 : deparse_type_name(Oid type_oid, int32 typemod)
1158 : : {
2236 1159 : 527 : bits16 flags = FORMAT_TYPE_TYPEMOD_GIVEN;
1160 : :
2248 alvherre@alvh.no-ip. 1161 [ - + ]: 527 : if (!is_builtin(type_oid))
2248 alvherre@alvh.no-ip. 1162 :UBC 0 : flags |= FORMAT_TYPE_FORCE_QUALIFY;
1163 : :
2248 alvherre@alvh.no-ip. 1164 :CBC 527 : return format_type_extended(type_oid, typemod, flags);
1165 : : }
1166 : :
1167 : : /*
1168 : : * Build the targetlist for given relation to be deparsed as SELECT clause.
1169 : : *
1170 : : * The output targetlist contains the columns that need to be fetched from the
1171 : : * foreign server for the given relation. If foreignrel is an upper relation,
1172 : : * then the output targetlist can also contain expressions to be evaluated on
1173 : : * foreign server.
1174 : : */
1175 : : List *
2987 rhaas@postgresql.org 1176 : 774 : build_tlist_to_deparse(RelOptInfo *foreignrel)
1177 : : {
1178 : 774 : List *tlist = NIL;
1179 : 774 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
1180 : : ListCell *lc;
1181 : :
1182 : : /*
1183 : : * For an upper relation, we have already built the target list while
1184 : : * checking shippability, so just return that.
1185 : : */
2568 1186 [ + + + + ]: 774 : if (IS_UPPER_REL(foreignrel))
2732 1187 : 161 : return fpinfo->grouped_tlist;
1188 : :
1189 : : /*
1190 : : * We require columns specified in foreignrel->reltarget->exprs and those
1191 : : * required for evaluating the local conditions.
1192 : : */
2861 1193 : 613 : tlist = add_to_flat_tlist(tlist,
2489 tgl@sss.pgh.pa.us 1194 : 613 : pull_var_clause((Node *) foreignrel->reltarget->exprs,
1195 : : PVC_RECURSE_PLACEHOLDERS));
2560 1196 [ + + + + : 637 : foreach(lc, fpinfo->local_conds)
+ + ]
1197 : : {
1198 : 24 : RestrictInfo *rinfo = lfirst_node(RestrictInfo, lc);
1199 : :
1200 : 24 : tlist = add_to_flat_tlist(tlist,
1201 : 24 : pull_var_clause((Node *) rinfo->clause,
1202 : : PVC_RECURSE_PLACEHOLDERS));
1203 : : }
1204 : :
2987 rhaas@postgresql.org 1205 : 613 : return tlist;
1206 : : }
1207 : :
1208 : : /*
1209 : : * Deparse SELECT statement for given relation into buf.
1210 : : *
1211 : : * tlist contains the list of desired columns to be fetched from foreign server.
1212 : : * For a base relation fpinfo->attrs_used is used to construct SELECT clause,
1213 : : * hence the tlist is ignored for a base relation.
1214 : : *
1215 : : * remote_conds is the list of conditions to be deparsed into the WHERE clause
1216 : : * (or, in the case of upper relations, into the HAVING clause).
1217 : : *
1218 : : * If params_list is not NULL, it receives a list of Params and other-relation
1219 : : * Vars used in the clauses; these values must be transmitted to the remote
1220 : : * server as parameter values.
1221 : : *
1222 : : * If params_list is NULL, we're generating the query for EXPLAIN purposes,
1223 : : * so Params and other-relation Vars should be replaced by dummy values.
1224 : : *
1225 : : * pathkeys is the list of pathkeys to order the result by.
1226 : : *
1227 : : * is_subquery is the flag to indicate whether to deparse the specified
1228 : : * relation as a subquery.
1229 : : *
1230 : : * List of columns selected is returned in retrieved_attrs.
1231 : : */
1232 : : void
2997 1233 : 2235 : deparseSelectStmtForRel(StringInfo buf, PlannerInfo *root, RelOptInfo *rel,
1234 : : List *tlist, List *remote_conds, List *pathkeys,
1235 : : bool has_final_sort, bool has_limit, bool is_subquery,
1236 : : List **retrieved_attrs, List **params_list)
1237 : : {
1238 : : deparse_expr_cxt context;
2732 1239 : 2235 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) rel->fdw_private;
1240 : : List *quals;
1241 : :
1242 : : /*
1243 : : * We handle relations for foreign tables, joins between those and upper
1244 : : * relations.
1245 : : */
2568 1246 [ + + + + : 2235 : Assert(IS_JOIN_REL(rel) || IS_SIMPLE_REL(rel) || IS_UPPER_REL(rel));
+ + + + +
+ - + ]
1247 : :
1248 : : /* Fill portions of context common to upper, join and base relation */
2997 1249 : 2235 : context.buf = buf;
1250 : 2235 : context.root = root;
1251 : 2235 : context.foreignrel = rel;
2568 1252 [ + + + + ]: 2235 : context.scanrel = IS_UPPER_REL(rel) ? fpinfo->outerrel : rel;
2997 1253 : 2235 : context.params_list = params_list;
1254 : :
1255 : : /* Construct SELECT clause */
2586 1256 : 2235 : deparseSelectSql(tlist, is_subquery, retrieved_attrs, &context);
1257 : :
1258 : : /*
1259 : : * For upper relations, the WHERE clause is built from the remote
1260 : : * conditions of the underlying scan relation; otherwise, we can use the
1261 : : * supplied list of remote conditions directly.
1262 : : */
2568 1263 [ + + + + ]: 2235 : if (IS_UPPER_REL(rel))
2987 1264 : 161 : {
1265 : : PgFdwRelationInfo *ofpinfo;
1266 : :
2732 1267 : 161 : ofpinfo = (PgFdwRelationInfo *) fpinfo->outerrel->fdw_private;
1268 : 161 : quals = ofpinfo->remote_conds;
1269 : : }
1270 : : else
1271 : 2074 : quals = remote_conds;
1272 : :
1273 : : /* Construct FROM and WHERE clauses */
1274 : 2235 : deparseFromExpr(quals, &context);
1275 : :
2568 1276 [ + + + + ]: 2235 : if (IS_UPPER_REL(rel))
1277 : : {
1278 : : /* Append GROUP BY clause */
2732 1279 : 161 : appendGroupByClause(tlist, &context);
1280 : :
1281 : : /* Append HAVING clause */
1282 [ + + ]: 161 : if (remote_conds)
1283 : : {
2434 peter_e@gmx.net 1284 : 18 : appendStringInfoString(buf, " HAVING ");
2732 rhaas@postgresql.org 1285 : 18 : appendConditions(remote_conds, &context);
1286 : : }
1287 : : }
1288 : :
1289 : : /* Add ORDER BY clause if we found any useful pathkeys */
2997 1290 [ + + ]: 2235 : if (pathkeys)
1839 efujita@postgresql.o 1291 : 713 : appendOrderByClause(pathkeys, has_final_sort, &context);
1292 : :
1293 : : /* Add LIMIT clause if necessary */
1294 [ + + ]: 2235 : if (has_limit)
1295 : 141 : appendLimitClause(&context);
1296 : :
1297 : : /* Add any necessary FOR UPDATE/SHARE. */
2997 rhaas@postgresql.org 1298 : 2235 : deparseLockingClause(&context);
1299 : 2235 : }
1300 : :
1301 : : /*
1302 : : * Construct a simple SELECT statement that retrieves desired columns
1303 : : * of the specified foreign table, and append it to "buf". The output
1304 : : * contains just "SELECT ... ".
1305 : : *
1306 : : * We also create an integer List of the columns being retrieved, which is
1307 : : * returned to *retrieved_attrs, unless we deparse the specified relation
1308 : : * as a subquery.
1309 : : *
1310 : : * tlist is the list of desired columns. is_subquery is the flag to
1311 : : * indicate whether to deparse the specified relation as a subquery.
1312 : : * Read prologue of deparseSelectStmtForRel() for details.
1313 : : */
1314 : : static void
2586 1315 : 2235 : deparseSelectSql(List *tlist, bool is_subquery, List **retrieved_attrs,
1316 : : deparse_expr_cxt *context)
1317 : : {
2997 1318 : 2235 : StringInfo buf = context->buf;
1319 : 2235 : RelOptInfo *foreignrel = context->foreignrel;
1320 : 2235 : PlannerInfo *root = context->root;
2987 1321 : 2235 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
1322 : :
1323 : : /*
1324 : : * Construct SELECT list
1325 : : */
4053 tgl@sss.pgh.pa.us 1326 : 2235 : appendStringInfoString(buf, "SELECT ");
1327 : :
2586 rhaas@postgresql.org 1328 [ + + ]: 2235 : if (is_subquery)
1329 : : {
1330 : : /*
1331 : : * For a relation that is deparsed as a subquery, emit expressions
1332 : : * specified in the relation's reltarget. Note that since this is for
1333 : : * the subquery, no need to care about *retrieved_attrs.
1334 : : */
1335 : 36 : deparseSubqueryTargetList(context);
1336 : : }
2568 1337 [ + + + + : 2199 : else if (IS_JOIN_REL(foreignrel) || IS_UPPER_REL(foreignrel))
+ + + + ]
1338 : : {
1339 : : /*
1340 : : * For a join or upper relation the input tlist gives the list of
1341 : : * columns required to be fetched from the foreign server.
1342 : : */
2258 1343 : 774 : deparseExplicitTargetList(tlist, false, retrieved_attrs, context);
1344 : : }
1345 : : else
1346 : : {
1347 : : /*
1348 : : * For a base relation fpinfo->attrs_used gives the list of columns
1349 : : * required to be fetched from the foreign server.
1350 : : */
2987 1351 [ + - ]: 1425 : RangeTblEntry *rte = planner_rt_fetch(foreignrel->relid, root);
1352 : :
1353 : : /*
1354 : : * Core code already has some lock on each rel being planned, so we
1355 : : * can use NoLock here.
1356 : : */
1910 andres@anarazel.de 1357 : 1425 : Relation rel = table_open(rte->relid, NoLock);
1358 : :
2175 rhaas@postgresql.org 1359 : 1425 : deparseTargetList(buf, rte, foreignrel->relid, rel, false,
1360 : : fpinfo->attrs_used, false, retrieved_attrs);
1910 andres@anarazel.de 1361 : 1425 : table_close(rel, NoLock);
1362 : : }
2732 rhaas@postgresql.org 1363 : 2235 : }
1364 : :
1365 : : /*
1366 : : * Construct a FROM clause and, if needed, a WHERE clause, and append those to
1367 : : * "buf".
1368 : : *
1369 : : * quals is the list of clauses to be included in the WHERE clause.
1370 : : * (These may or may not include RestrictInfo decoration.)
1371 : : */
1372 : : static void
1373 : 2235 : deparseFromExpr(List *quals, deparse_expr_cxt *context)
1374 : : {
1375 : 2235 : StringInfo buf = context->buf;
1376 : 2235 : RelOptInfo *scanrel = context->scanrel;
131 akorotkov@postgresql 1377 :GNC 2235 : List *additional_conds = NIL;
1378 : :
1379 : : /* For upper relations, scanrel must be either a joinrel or a baserel */
2568 rhaas@postgresql.org 1380 [ + + + + :CBC 2235 : Assert(!IS_UPPER_REL(context->foreignrel) ||
+ + + - +
+ - + ]
1381 : : IS_JOIN_REL(scanrel) || IS_SIMPLE_REL(scanrel));
1382 : :
1383 : : /* Construct FROM clause */
4053 tgl@sss.pgh.pa.us 1384 : 2235 : appendStringInfoString(buf, " FROM ");
2732 rhaas@postgresql.org 1385 : 2235 : deparseFromExprForRel(buf, context->root, scanrel,
2114 michael@paquier.xyz 1386 : 2235 : (bms_membership(scanrel->relids) == BMS_MULTIPLE),
1387 : : (Index) 0, NULL, &additional_conds,
1388 : : context->params_list);
131 akorotkov@postgresql 1389 :GNC 2235 : appendWhereClause(quals, additional_conds, context);
1390 [ + + ]: 2235 : if (additional_conds != NIL)
1391 : 128 : list_free_deep(additional_conds);
4053 tgl@sss.pgh.pa.us 1392 :CBC 2235 : }
1393 : :
1394 : : /*
1395 : : * Emit a target list that retrieves the columns specified in attrs_used.
1396 : : * This is used for both SELECT and RETURNING targetlists; the is_returning
1397 : : * parameter is true only for a RETURNING targetlist.
1398 : : *
1399 : : * The tlist text is appended to buf, and we also create an integer List
1400 : : * of the columns being retrieved, which is returned to *retrieved_attrs.
1401 : : *
1402 : : * If qualify_col is true, add relation alias before the column name.
1403 : : */
1404 : : static void
1405 : 1767 : deparseTargetList(StringInfo buf,
1406 : : RangeTblEntry *rte,
1407 : : Index rtindex,
1408 : : Relation rel,
1409 : : bool is_returning,
1410 : : Bitmapset *attrs_used,
1411 : : bool qualify_col,
1412 : : List **retrieved_attrs)
1413 : : {
1414 : 1767 : TupleDesc tupdesc = RelationGetDescr(rel);
1415 : : bool have_wholerow;
1416 : : bool first;
1417 : : int i;
1418 : :
4041 1419 : 1767 : *retrieved_attrs = NIL;
1420 : :
1421 : : /* If there's a whole-row reference, we'll need all the columns. */
4069 1422 : 1767 : have_wholerow = bms_is_member(0 - FirstLowInvalidHeapAttributeNumber,
1423 : : attrs_used);
1424 : :
4070 1425 : 1767 : first = true;
4053 1426 [ + + ]: 12669 : for (i = 1; i <= tupdesc->natts; i++)
1427 : : {
2429 andres@anarazel.de 1428 : 10902 : Form_pg_attribute attr = TupleDescAttr(tupdesc, i - 1);
1429 : :
1430 : : /* Ignore dropped attributes. */
4053 tgl@sss.pgh.pa.us 1431 [ + + ]: 10902 : if (attr->attisdropped)
4070 1432 : 981 : continue;
1433 : :
4069 1434 [ + + + + ]: 16792 : if (have_wholerow ||
4053 1435 : 6871 : bms_is_member(i - FirstLowInvalidHeapAttributeNumber,
1436 : : attrs_used))
1437 : : {
4041 1438 [ + + ]: 6134 : if (!first)
1439 : 4485 : appendStringInfoString(buf, ", ");
2992 rhaas@postgresql.org 1440 [ + + ]: 1649 : else if (is_returning)
1441 : 104 : appendStringInfoString(buf, " RETURNING ");
4041 tgl@sss.pgh.pa.us 1442 : 6134 : first = false;
1443 : :
2175 rhaas@postgresql.org 1444 : 6134 : deparseColumnRef(buf, rtindex, i, rte, qualify_col);
1445 : :
4041 tgl@sss.pgh.pa.us 1446 : 6134 : *retrieved_attrs = lappend_int(*retrieved_attrs, i);
1447 : : }
1448 : : }
1449 : :
1450 : : /*
1451 : : * Add ctid if needed. We currently don't support retrieving any other
1452 : : * system columns.
1453 : : */
4053 1454 [ + + ]: 1767 : if (bms_is_member(SelfItemPointerAttributeNumber - FirstLowInvalidHeapAttributeNumber,
1455 : : attrs_used))
1456 : : {
1457 [ + + ]: 280 : if (!first)
1458 : 212 : appendStringInfoString(buf, ", ");
2992 rhaas@postgresql.org 1459 [ - + ]: 68 : else if (is_returning)
2992 rhaas@postgresql.org 1460 :UBC 0 : appendStringInfoString(buf, " RETURNING ");
4053 tgl@sss.pgh.pa.us 1461 :CBC 280 : first = false;
1462 : :
2987 rhaas@postgresql.org 1463 [ - + ]: 280 : if (qualify_col)
2987 rhaas@postgresql.org 1464 :UBC 0 : ADD_REL_QUALIFIER(buf, rtindex);
4053 tgl@sss.pgh.pa.us 1465 :CBC 280 : appendStringInfoString(buf, "ctid");
1466 : :
4041 1467 : 280 : *retrieved_attrs = lappend_int(*retrieved_attrs,
1468 : : SelfItemPointerAttributeNumber);
1469 : : }
1470 : :
1471 : : /* Don't generate bad syntax if no undropped columns */
2992 rhaas@postgresql.org 1472 [ + + + + ]: 1767 : if (first && !is_returning)
4053 tgl@sss.pgh.pa.us 1473 : 44 : appendStringInfoString(buf, "NULL");
4070 1474 : 1767 : }
1475 : :
1476 : : /*
1477 : : * Deparse the appropriate locking clause (FOR UPDATE or FOR SHARE) for a
1478 : : * given relation (context->scanrel).
1479 : : */
1480 : : static void
2997 rhaas@postgresql.org 1481 : 2235 : deparseLockingClause(deparse_expr_cxt *context)
1482 : : {
1483 : 2235 : StringInfo buf = context->buf;
1484 : 2235 : PlannerInfo *root = context->root;
2732 1485 : 2235 : RelOptInfo *rel = context->scanrel;
2586 1486 : 2235 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) rel->fdw_private;
2987 1487 : 2235 : int relid = -1;
1488 : :
1489 [ + + ]: 5591 : while ((relid = bms_next_member(rel->relids, relid)) >= 0)
1490 : : {
1491 : : /*
1492 : : * Ignore relation if it appears in a lower subquery. Locking clause
1493 : : * for such a relation is included in the subquery if necessary.
1494 : : */
2586 1495 [ + + ]: 3356 : if (bms_is_member(relid, fpinfo->lower_subquery_rels))
1496 : 52 : continue;
1497 : :
1498 : : /*
1499 : : * Add FOR UPDATE/SHARE if appropriate. We apply locking during the
1500 : : * initial row fetch, rather than later on as is done for local
1501 : : * tables. The extra roundtrips involved in trying to duplicate the
1502 : : * local semantics exactly don't seem worthwhile (see also comments
1503 : : * for RowMarkType).
1504 : : *
1505 : : * Note: because we actually run the query as a cursor, this assumes
1506 : : * that DECLARE CURSOR ... FOR UPDATE is supported, which it isn't
1507 : : * before 8.3.
1508 : : */
1110 tgl@sss.pgh.pa.us 1509 [ + + ]: 3304 : if (bms_is_member(relid, root->all_result_relids) &&
2987 rhaas@postgresql.org 1510 [ + + ]: 277 : (root->parse->commandType == CMD_UPDATE ||
1511 [ + + ]: 120 : root->parse->commandType == CMD_DELETE))
1512 : : {
1513 : : /* Relation is UPDATE/DELETE target, so use FOR UPDATE */
1514 : 275 : appendStringInfoString(buf, " FOR UPDATE");
1515 : :
1516 : : /* Add the relation alias if we are here for a join relation */
2568 1517 [ + + - + ]: 275 : if (IS_JOIN_REL(rel))
2987 1518 : 42 : appendStringInfo(buf, " OF %s%d", REL_ALIAS_PREFIX, relid);
1519 : : }
1520 : : else
1521 : : {
1522 : 3029 : PlanRowMark *rc = get_plan_rowmark(root->rowMarks, relid);
1523 : :
1524 [ + + ]: 3029 : if (rc)
1525 : : {
1526 : : /*
1527 : : * Relation is specified as a FOR UPDATE/SHARE target, so
1528 : : * handle that. (But we could also see LCS_NONE, meaning this
1529 : : * isn't a target relation after all.)
1530 : : *
1531 : : * For now, just ignore any [NO] KEY specification, since (a)
1532 : : * it's not clear what that means for a remote table that we
1533 : : * don't have complete information about, and (b) it wouldn't
1534 : : * work anyway on older remote servers. Likewise, we don't
1535 : : * worry about NOWAIT.
1536 : : */
1537 [ + + + - ]: 328 : switch (rc->strength)
1538 : : {
1539 : 176 : case LCS_NONE:
1540 : : /* No locking needed */
1541 : 176 : break;
1542 : 38 : case LCS_FORKEYSHARE:
1543 : : case LCS_FORSHARE:
1544 : 38 : appendStringInfoString(buf, " FOR SHARE");
1545 : 38 : break;
1546 : 114 : case LCS_FORNOKEYUPDATE:
1547 : : case LCS_FORUPDATE:
1548 : 114 : appendStringInfoString(buf, " FOR UPDATE");
1549 : 114 : break;
1550 : : }
1551 : :
1552 : : /* Add the relation alias if we are here for a join relation */
2114 michael@paquier.xyz 1553 [ + + ]: 328 : if (bms_membership(rel->relids) == BMS_MULTIPLE &&
2987 rhaas@postgresql.org 1554 [ + + ]: 190 : rc->strength != LCS_NONE)
1555 : 104 : appendStringInfo(buf, " OF %s%d", REL_ALIAS_PREFIX, relid);
1556 : : }
1557 : : }
1558 : : }
2999 1559 : 2235 : }
1560 : :
1561 : : /*
1562 : : * Deparse conditions from the provided list and append them to buf.
1563 : : *
1564 : : * The conditions in the list are assumed to be ANDed. This function is used to
1565 : : * deparse WHERE clauses, JOIN .. ON clauses and HAVING clauses.
1566 : : *
1567 : : * Depending on the caller, the list elements might be either RestrictInfos
1568 : : * or bare clauses.
1569 : : */
1570 : : static void
2987 1571 : 1760 : appendConditions(List *exprs, deparse_expr_cxt *context)
1572 : : {
1573 : : int nestlevel;
1574 : : ListCell *lc;
2997 1575 : 1760 : bool is_first = true;
1576 : 1760 : StringInfo buf = context->buf;
1577 : :
1578 : : /* Make sure any constants in the exprs are printed portably */
4052 tgl@sss.pgh.pa.us 1579 : 1760 : nestlevel = set_transmission_modes();
1580 : :
4070 1581 [ + - + + : 4045 : foreach(lc, exprs)
+ + ]
1582 : : {
2987 rhaas@postgresql.org 1583 : 2285 : Expr *expr = (Expr *) lfirst(lc);
1584 : :
1585 : : /* Extract clause from RestrictInfo, if required */
1586 [ + + ]: 2285 : if (IsA(expr, RestrictInfo))
2560 tgl@sss.pgh.pa.us 1587 : 1944 : expr = ((RestrictInfo *) expr)->clause;
1588 : :
1589 : : /* Connect expressions with "AND" and parenthesize each condition. */
2987 rhaas@postgresql.org 1590 [ + + ]: 2285 : if (!is_first)
4053 tgl@sss.pgh.pa.us 1591 : 525 : appendStringInfoString(buf, " AND ");
1592 : :
4070 1593 : 2285 : appendStringInfoChar(buf, '(');
2987 rhaas@postgresql.org 1594 : 2285 : deparseExpr(expr, context);
4070 tgl@sss.pgh.pa.us 1595 : 2285 : appendStringInfoChar(buf, ')');
1596 : :
1597 : 2285 : is_first = false;
1598 : : }
1599 : :
4052 1600 : 1760 : reset_transmission_modes(nestlevel);
4070 1601 : 1760 : }
1602 : :
1603 : : /*
1604 : : * Append WHERE clause, containing conditions from exprs and additional_conds,
1605 : : * to context->buf.
1606 : : */
1607 : : static void
131 akorotkov@postgresql 1608 :GNC 2499 : appendWhereClause(List *exprs, List *additional_conds, deparse_expr_cxt *context)
1609 : : {
1610 : 2499 : StringInfo buf = context->buf;
1611 : 2499 : bool need_and = false;
1612 : : ListCell *lc;
1613 : :
1614 [ + + + + ]: 2499 : if (exprs != NIL || additional_conds != NIL)
1615 : 1139 : appendStringInfoString(buf, " WHERE ");
1616 : :
1617 : : /*
1618 : : * If there are some filters, append them.
1619 : : */
1620 [ + + ]: 2499 : if (exprs != NIL)
1621 : : {
1622 : 1036 : appendConditions(exprs, context);
1623 : 1036 : need_and = true;
1624 : : }
1625 : :
1626 : : /*
1627 : : * If there are some EXISTS conditions, coming from SEMI-JOINS, append
1628 : : * them.
1629 : : */
1630 [ + + + + : 2659 : foreach(lc, additional_conds)
+ + ]
1631 : : {
1632 [ + + ]: 160 : if (need_and)
1633 : 57 : appendStringInfoString(buf, " AND ");
1634 : 160 : appendStringInfoString(buf, (char *) lfirst(lc));
1635 : 160 : need_and = true;
1636 : : }
1637 : 2499 : }
1638 : :
1639 : : /* Output join name for given join type */
1640 : : const char *
2987 rhaas@postgresql.org 1641 :CBC 1057 : get_jointype_name(JoinType jointype)
1642 : : {
1643 [ + + - + : 1057 : switch (jointype)
+ - ]
1644 : : {
1645 : 711 : case JOIN_INNER:
1646 : 711 : return "INNER";
1647 : :
1648 : 200 : case JOIN_LEFT:
1649 : 200 : return "LEFT";
1650 : :
2987 rhaas@postgresql.org 1651 :UBC 0 : case JOIN_RIGHT:
1652 : 0 : return "RIGHT";
1653 : :
2987 rhaas@postgresql.org 1654 :CBC 108 : case JOIN_FULL:
1655 : 108 : return "FULL";
1656 : :
131 akorotkov@postgresql 1657 :GNC 38 : case JOIN_SEMI:
1658 : 38 : return "SEMI";
1659 : :
2987 rhaas@postgresql.org 1660 :UBC 0 : default:
1661 : : /* Shouldn't come here, but protect from buggy code. */
1662 [ # # ]: 0 : elog(ERROR, "unsupported join type %d", jointype);
1663 : : }
1664 : :
1665 : : /* Keep compiler happy */
1666 : : return NULL;
1667 : : }
1668 : :
1669 : : /*
1670 : : * Deparse given targetlist and append it to context->buf.
1671 : : *
1672 : : * tlist is list of TargetEntry's which in turn contain Var nodes.
1673 : : *
1674 : : * retrieved_attrs is the list of continuously increasing integers starting
1675 : : * from 1. It has same number of entries as tlist.
1676 : : *
1677 : : * This is used for both SELECT and RETURNING targetlists; the is_returning
1678 : : * parameter is true only for a RETURNING targetlist.
1679 : : */
1680 : : static void
2258 rhaas@postgresql.org 1681 :CBC 782 : deparseExplicitTargetList(List *tlist,
1682 : : bool is_returning,
1683 : : List **retrieved_attrs,
1684 : : deparse_expr_cxt *context)
1685 : : {
1686 : : ListCell *lc;
2987 1687 : 782 : StringInfo buf = context->buf;
1688 : 782 : int i = 0;
1689 : :
1690 : 782 : *retrieved_attrs = NIL;
1691 : :
1692 [ + + + + : 4613 : foreach(lc, tlist)
+ + ]
1693 : : {
2561 tgl@sss.pgh.pa.us 1694 : 3831 : TargetEntry *tle = lfirst_node(TargetEntry, lc);
1695 : :
2987 rhaas@postgresql.org 1696 [ + + ]: 3831 : if (i > 0)
1697 : 3070 : appendStringInfoString(buf, ", ");
2258 1698 [ + + ]: 761 : else if (is_returning)
1699 : 2 : appendStringInfoString(buf, " RETURNING ");
1700 : :
2732 1701 : 3831 : deparseExpr((Expr *) tle->expr, context);
1702 : :
2987 1703 : 3831 : *retrieved_attrs = lappend_int(*retrieved_attrs, i + 1);
1704 : 3831 : i++;
1705 : : }
1706 : :
2258 1707 [ + + + + ]: 782 : if (i == 0 && !is_returning)
2987 1708 : 15 : appendStringInfoString(buf, "NULL");
1709 : 782 : }
1710 : :
1711 : : /*
1712 : : * Emit expressions specified in the given relation's reltarget.
1713 : : *
1714 : : * This is used for deparsing the given relation as a subquery.
1715 : : */
1716 : : static void
2586 1717 : 36 : deparseSubqueryTargetList(deparse_expr_cxt *context)
1718 : : {
1719 : 36 : StringInfo buf = context->buf;
1720 : 36 : RelOptInfo *foreignrel = context->foreignrel;
1721 : : bool first;
1722 : : ListCell *lc;
1723 : :
1724 : : /* Should only be called in these cases. */
2568 1725 [ + + + - : 36 : Assert(IS_SIMPLE_REL(foreignrel) || IS_JOIN_REL(foreignrel));
- + - - ]
1726 : :
2586 1727 : 36 : first = true;
1728 [ + + + + : 92 : foreach(lc, foreignrel->reltarget->exprs)
+ + ]
1729 : : {
1730 : 56 : Node *node = (Node *) lfirst(lc);
1731 : :
1732 [ + + ]: 56 : if (!first)
1733 : 24 : appendStringInfoString(buf, ", ");
1734 : 56 : first = false;
1735 : :
1736 : 56 : deparseExpr((Expr *) node, context);
1737 : : }
1738 : :
1739 : : /* Don't generate bad syntax if no expressions */
1740 [ + + ]: 36 : if (first)
1741 : 4 : appendStringInfoString(buf, "NULL");
1742 : 36 : }
1743 : :
1744 : : /*
1745 : : * Construct FROM clause for given relation
1746 : : *
1747 : : * The function constructs ... JOIN ... ON ... for join relation. For a base
1748 : : * relation it just returns schema-qualified tablename, with the appropriate
1749 : : * alias if so requested.
1750 : : *
1751 : : * 'ignore_rel' is either zero or the RT index of a target relation. In the
1752 : : * latter case the function constructs FROM clause of UPDATE or USING clause
1753 : : * of DELETE; it deparses the join relation as if the relation never contained
1754 : : * the target relation, and creates a List of conditions to be deparsed into
1755 : : * the top-level WHERE clause, which is returned to *ignore_conds.
1756 : : *
1757 : : * 'additional_conds' is a pointer to a list of strings to be appended to
1758 : : * the WHERE clause, coming from lower-level SEMI-JOINs.
1759 : : */
1760 : : static void
2987 1761 : 4033 : deparseFromExprForRel(StringInfo buf, PlannerInfo *root, RelOptInfo *foreignrel,
1762 : : bool use_alias, Index ignore_rel, List **ignore_conds,
1763 : : List **additional_conds, List **params_list)
1764 : : {
1765 : 4033 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
1766 : :
2568 1767 [ + + + + ]: 4033 : if (IS_JOIN_REL(foreignrel))
2987 1768 : 909 : {
1769 : : StringInfoData join_sql_o;
1770 : : StringInfoData join_sql_i;
2258 1771 : 917 : RelOptInfo *outerrel = fpinfo->outerrel;
1772 : 917 : RelOptInfo *innerrel = fpinfo->innerrel;
1773 : 917 : bool outerrel_is_target = false;
1774 : 917 : bool innerrel_is_target = false;
131 akorotkov@postgresql 1775 :GNC 917 : List *additional_conds_i = NIL;
1776 : 917 : List *additional_conds_o = NIL;
1777 : :
2258 rhaas@postgresql.org 1778 [ + + + - ]:CBC 917 : if (ignore_rel > 0 && bms_is_member(ignore_rel, foreignrel->relids))
1779 : : {
1780 : : /*
1781 : : * If this is an inner join, add joinclauses to *ignore_conds and
1782 : : * set it to empty so that those can be deparsed into the WHERE
1783 : : * clause. Note that since the target relation can never be
1784 : : * within the nullable side of an outer join, those could safely
1785 : : * be pulled up into the WHERE clause (see foreign_join_ok()).
1786 : : * Note also that since the target relation is only inner-joined
1787 : : * to any other relation in the query, all conditions in the join
1788 : : * tree mentioning the target relation could be deparsed into the
1789 : : * WHERE clause by doing this recursively.
1790 : : */
1791 [ + + ]: 12 : if (fpinfo->jointype == JOIN_INNER)
1792 : : {
1793 : 20 : *ignore_conds = list_concat(*ignore_conds,
1707 tgl@sss.pgh.pa.us 1794 : 10 : fpinfo->joinclauses);
2258 rhaas@postgresql.org 1795 : 10 : fpinfo->joinclauses = NIL;
1796 : : }
1797 : :
1798 : : /*
1799 : : * Check if either of the input relations is the target relation.
1800 : : */
1801 [ + + ]: 12 : if (outerrel->relid == ignore_rel)
1802 : 8 : outerrel_is_target = true;
1803 [ - + ]: 4 : else if (innerrel->relid == ignore_rel)
2258 rhaas@postgresql.org 1804 :UBC 0 : innerrel_is_target = true;
1805 : : }
1806 : :
1807 : : /* Deparse outer relation if not the target relation. */
2258 rhaas@postgresql.org 1808 [ + + ]:CBC 917 : if (!outerrel_is_target)
1809 : : {
1810 : 909 : initStringInfo(&join_sql_o);
1811 : 909 : deparseRangeTblRef(&join_sql_o, root, outerrel,
1812 : 909 : fpinfo->make_outerrel_subquery,
1813 : : ignore_rel, ignore_conds, &additional_conds_o,
1814 : : params_list);
1815 : :
1816 : : /*
1817 : : * If inner relation is the target relation, skip deparsing it.
1818 : : * Note that since the join of the target relation with any other
1819 : : * relation in the query is an inner join and can never be within
1820 : : * the nullable side of an outer join, the join could be
1821 : : * interchanged with higher-level joins (cf. identity 1 on outer
1822 : : * join reordering shown in src/backend/optimizer/README), which
1823 : : * means it's safe to skip the target-relation deparsing here.
1824 : : */
1825 [ - + ]: 909 : if (innerrel_is_target)
1826 : : {
2258 rhaas@postgresql.org 1827 [ # # ]:UBC 0 : Assert(fpinfo->jointype == JOIN_INNER);
1828 [ # # ]: 0 : Assert(fpinfo->joinclauses == NIL);
1727 drowley@postgresql.o 1829 : 0 : appendBinaryStringInfo(buf, join_sql_o.data, join_sql_o.len);
1830 : : /* Pass EXISTS conditions to upper level */
131 akorotkov@postgresql 1831 [ # # ]:UNC 0 : if (additional_conds_o != NIL)
1832 : : {
1833 [ # # ]: 0 : Assert(*additional_conds == NIL);
1834 : 0 : *additional_conds = additional_conds_o;
1835 : : }
2258 rhaas@postgresql.org 1836 :UBC 0 : return;
1837 : : }
1838 : : }
1839 : :
1840 : : /* Deparse inner relation if not the target relation. */
2258 rhaas@postgresql.org 1841 [ + - ]:CBC 917 : if (!innerrel_is_target)
1842 : : {
1843 : 917 : initStringInfo(&join_sql_i);
1844 : 917 : deparseRangeTblRef(&join_sql_i, root, innerrel,
1845 : 917 : fpinfo->make_innerrel_subquery,
1846 : : ignore_rel, ignore_conds, &additional_conds_i,
1847 : : params_list);
1848 : :
1849 : : /*
1850 : : * SEMI-JOIN is deparsed as the EXISTS subquery. It references
1851 : : * outer and inner relations, so it should be evaluated as the
1852 : : * condition in the upper-level WHERE clause. We deparse the
1853 : : * condition and pass it to upper level callers as an
1854 : : * additional_conds list. Upper level callers are responsible for
1855 : : * inserting conditions from the list where appropriate.
1856 : : */
131 akorotkov@postgresql 1857 [ + + ]:GNC 917 : if (fpinfo->jointype == JOIN_SEMI)
1858 : : {
1859 : : deparse_expr_cxt context;
1860 : : StringInfoData str;
1861 : :
1862 : : /* Construct deparsed condition from this SEMI-JOIN */
1863 : 160 : initStringInfo(&str);
1864 : 160 : appendStringInfo(&str, "EXISTS (SELECT NULL FROM %s",
1865 : : join_sql_i.data);
1866 : :
1867 : 160 : context.buf = &str;
1868 : 160 : context.foreignrel = foreignrel;
1869 : 160 : context.scanrel = foreignrel;
1870 : 160 : context.root = root;
1871 : 160 : context.params_list = params_list;
1872 : :
1873 : : /*
1874 : : * Append SEMI-JOIN clauses and EXISTS conditions from lower
1875 : : * levels to the current EXISTS subquery
1876 : : */
1877 : 160 : appendWhereClause(fpinfo->joinclauses, additional_conds_i, &context);
1878 : :
1879 : : /*
1880 : : * EXISTS conditions, coming from lower join levels, have just
1881 : : * been processed.
1882 : : */
1883 [ + + ]: 160 : if (additional_conds_i != NIL)
1884 : : {
1885 : 16 : list_free_deep(additional_conds_i);
1886 : 16 : additional_conds_i = NIL;
1887 : : }
1888 : :
1889 : : /* Close parentheses for EXISTS subquery */
4 drowley@postgresql.o 1890 : 160 : appendStringInfoChar(&str, ')');
1891 : :
131 akorotkov@postgresql 1892 : 160 : *additional_conds = lappend(*additional_conds, str.data);
1893 : : }
1894 : :
1895 : : /*
1896 : : * If outer relation is the target relation, skip deparsing it.
1897 : : * See the above note about safety.
1898 : : */
2258 rhaas@postgresql.org 1899 [ + + ]:CBC 917 : if (outerrel_is_target)
1900 : : {
1901 [ - + ]: 8 : Assert(fpinfo->jointype == JOIN_INNER);
1902 [ - + ]: 8 : Assert(fpinfo->joinclauses == NIL);
1727 drowley@postgresql.o 1903 : 8 : appendBinaryStringInfo(buf, join_sql_i.data, join_sql_i.len);
1904 : : /* Pass EXISTS conditions to the upper call */
131 akorotkov@postgresql 1905 [ - + ]:GNC 8 : if (additional_conds_i != NIL)
1906 : : {
131 akorotkov@postgresql 1907 [ # # ]:UNC 0 : Assert(*additional_conds == NIL);
1908 : 0 : *additional_conds = additional_conds_i;
1909 : : }
2258 rhaas@postgresql.org 1910 :CBC 8 : return;
1911 : : }
1912 : : }
1913 : :
1914 : : /* Neither of the relations is the target relation. */
1915 [ + - - + ]: 909 : Assert(!outerrel_is_target && !innerrel_is_target);
1916 : :
1917 : : /*
1918 : : * For semijoin FROM clause is deparsed as an outer relation. An inner
1919 : : * relation and join clauses are converted to EXISTS condition and
1920 : : * passed to the upper level.
1921 : : */
131 akorotkov@postgresql 1922 [ + + ]:GNC 909 : if (fpinfo->jointype == JOIN_SEMI)
1923 : : {
4 drowley@postgresql.o 1924 : 160 : appendBinaryStringInfo(buf, join_sql_o.data, join_sql_o.len);
1925 : : }
1926 : : else
1927 : : {
1928 : : /*
1929 : : * For a join relation FROM clause, entry is deparsed as
1930 : : *
1931 : : * ((outer relation) <join type> (inner relation) ON
1932 : : * (joinclauses))
1933 : : */
131 akorotkov@postgresql 1934 : 749 : appendStringInfo(buf, "(%s %s JOIN %s ON ", join_sql_o.data,
1935 : : get_jointype_name(fpinfo->jointype), join_sql_i.data);
1936 : :
1937 : : /* Append join clause; (TRUE) if no join clause */
1938 [ + + ]: 749 : if (fpinfo->joinclauses)
1939 : : {
1940 : : deparse_expr_cxt context;
1941 : :
1942 : 706 : context.buf = buf;
1943 : 706 : context.foreignrel = foreignrel;
1944 : 706 : context.scanrel = foreignrel;
1945 : 706 : context.root = root;
1946 : 706 : context.params_list = params_list;
1947 : :
1948 : 706 : appendStringInfoChar(buf, '(');
1949 : 706 : appendConditions(fpinfo->joinclauses, &context);
1950 : 706 : appendStringInfoChar(buf, ')');
1951 : : }
1952 : : else
1953 : 43 : appendStringInfoString(buf, "(TRUE)");
1954 : :
1955 : : /* End the FROM clause entry. */
2434 peter_e@gmx.net 1956 : 749 : appendStringInfoChar(buf, ')');
1957 : : }
1958 : :
1959 : : /*
1960 : : * Construct additional_conds to be passed to the upper caller from
1961 : : * current level additional_conds and additional_conds, coming from
1962 : : * inner and outer rels.
1963 : : */
131 akorotkov@postgresql 1964 [ + + ]: 909 : if (additional_conds_o != NIL)
1965 : : {
1966 : 48 : *additional_conds = list_concat(*additional_conds,
1967 : : additional_conds_o);
1968 : 48 : list_free(additional_conds_o);
1969 : : }
1970 : :
1971 [ - + ]: 909 : if (additional_conds_i != NIL)
1972 : : {
131 akorotkov@postgresql 1973 :UNC 0 : *additional_conds = list_concat(*additional_conds,
1974 : : additional_conds_i);
1975 : 0 : list_free(additional_conds_i);
1976 : : }
1977 : : }
1978 : : else
1979 : : {
2987 rhaas@postgresql.org 1980 [ + - ]:CBC 3116 : RangeTblEntry *rte = planner_rt_fetch(foreignrel->relid, root);
1981 : :
1982 : : /*
1983 : : * Core code already has some lock on each rel being planned, so we
1984 : : * can use NoLock here.
1985 : : */
1910 andres@anarazel.de 1986 : 3116 : Relation rel = table_open(rte->relid, NoLock);
1987 : :
2987 rhaas@postgresql.org 1988 : 3116 : deparseRelation(buf, rel);
1989 : :
1990 : : /*
1991 : : * Add a unique alias to avoid any conflict in relation names due to
1992 : : * pulled up subqueries in the query being built for a pushed down
1993 : : * join.
1994 : : */
1995 [ + + ]: 3116 : if (use_alias)
1996 : 1524 : appendStringInfo(buf, " %s%d", REL_ALIAS_PREFIX, foreignrel->relid);
1997 : :
1910 andres@anarazel.de 1998 : 3116 : table_close(rel, NoLock);
1999 : : }
2000 : : }
2001 : :
2002 : : /*
2003 : : * Append FROM clause entry for the given relation into buf.
2004 : : * Conditions from lower-level SEMI-JOINs are appended to additional_conds
2005 : : * and should be added to upper level WHERE clause.
2006 : : */
2007 : : static void
2586 rhaas@postgresql.org 2008 : 1826 : deparseRangeTblRef(StringInfo buf, PlannerInfo *root, RelOptInfo *foreignrel,
2009 : : bool make_subquery, Index ignore_rel, List **ignore_conds,
2010 : : List **additional_conds, List **params_list)
2011 : : {
2012 : 1826 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
2013 : :
2014 : : /* Should only be called in these cases. */
2568 2015 [ + + + + : 1826 : Assert(IS_SIMPLE_REL(foreignrel) || IS_JOIN_REL(foreignrel));
+ + - + ]
2016 : :
2586 2017 [ - + ]: 1826 : Assert(fpinfo->local_conds == NIL);
2018 : :
2019 : : /* If make_subquery is true, deparse the relation as a subquery. */
2020 [ + + ]: 1826 : if (make_subquery)
2021 : : {
2022 : : List *retrieved_attrs;
2023 : : int ncols;
2024 : :
2025 : : /*
2026 : : * The given relation shouldn't contain the target relation, because
2027 : : * this should only happen for input relations for a full join, and
2028 : : * such relations can never contain an UPDATE/DELETE target.
2029 : : */
2258 2030 [ - + - - ]: 36 : Assert(ignore_rel == 0 ||
2031 : : !bms_is_member(ignore_rel, foreignrel->relids));
2032 : :
2033 : : /* Deparse the subquery representing the relation. */
2586 2034 : 36 : appendStringInfoChar(buf, '(');
2035 : 36 : deparseSelectStmtForRel(buf, root, foreignrel, NIL,
2036 : : fpinfo->remote_conds, NIL,
2037 : : false, false, true,
2038 : : &retrieved_attrs, params_list);
2039 : 36 : appendStringInfoChar(buf, ')');
2040 : :
2041 : : /* Append the relation alias. */
2042 : 36 : appendStringInfo(buf, " %s%d", SUBQUERY_REL_ALIAS_PREFIX,
2043 : : fpinfo->relation_index);
2044 : :
2045 : : /*
2046 : : * Append the column aliases if needed. Note that the subquery emits
2047 : : * expressions specified in the relation's reltarget (see
2048 : : * deparseSubqueryTargetList).
2049 : : */
2050 : 36 : ncols = list_length(foreignrel->reltarget->exprs);
2051 [ + + ]: 36 : if (ncols > 0)
2052 : : {
2053 : : int i;
2054 : :
2055 : 32 : appendStringInfoChar(buf, '(');
2056 [ + + ]: 88 : for (i = 1; i <= ncols; i++)
2057 : : {
2058 [ + + ]: 56 : if (i > 1)
2059 : 24 : appendStringInfoString(buf, ", ");
2060 : :
2061 : 56 : appendStringInfo(buf, "%s%d", SUBQUERY_COL_ALIAS_PREFIX, i);
2062 : : }
2063 : 32 : appendStringInfoChar(buf, ')');
2064 : : }
2065 : : }
2066 : : else
2258 2067 : 1790 : deparseFromExprForRel(buf, root, foreignrel, true, ignore_rel,
2068 : : ignore_conds, additional_conds,
2069 : : params_list);
2586 2070 : 1826 : }
2071 : :
2072 : : /*
2073 : : * deparse remote INSERT statement
2074 : : *
2075 : : * The statement text is appended to buf, and we also create an integer List
2076 : : * of the columns being retrieved by WITH CHECK OPTION or RETURNING (if any),
2077 : : * which is returned to *retrieved_attrs.
2078 : : *
2079 : : * This also stores end position of the VALUES clause, so that we can rebuild
2080 : : * an INSERT for a batch of rows later.
2081 : : */
2082 : : void
2175 2083 : 141 : deparseInsertSql(StringInfo buf, RangeTblEntry *rte,
2084 : : Index rtindex, Relation rel,
2085 : : List *targetAttrs, bool doNothing,
2086 : : List *withCheckOptionList, List *returningList,
2087 : : List **retrieved_attrs, int *values_end_len)
2088 : : {
983 efujita@postgresql.o 2089 : 141 : TupleDesc tupdesc = RelationGetDescr(rel);
2090 : : AttrNumber pindex;
2091 : : bool first;
2092 : : ListCell *lc;
2093 : :
4053 tgl@sss.pgh.pa.us 2094 : 141 : appendStringInfoString(buf, "INSERT INTO ");
4051 2095 : 141 : deparseRelation(buf, rel);
2096 : :
4052 2097 [ + + ]: 141 : if (targetAttrs)
2098 : : {
3818 rhaas@postgresql.org 2099 : 140 : appendStringInfoChar(buf, '(');
2100 : :
4052 tgl@sss.pgh.pa.us 2101 : 140 : first = true;
2102 [ + - + + : 520 : foreach(lc, targetAttrs)
+ + ]
2103 : : {
2104 : 380 : int attnum = lfirst_int(lc);
2105 : :
2106 [ + + ]: 380 : if (!first)
2107 : 240 : appendStringInfoString(buf, ", ");
2108 : 380 : first = false;
2109 : :
2175 rhaas@postgresql.org 2110 : 380 : deparseColumnRef(buf, rtindex, attnum, rte, false);
2111 : : }
2112 : :
4052 tgl@sss.pgh.pa.us 2113 : 140 : appendStringInfoString(buf, ") VALUES (");
2114 : :
2115 : 140 : pindex = 1;
2116 : 140 : first = true;
2117 [ + - + + : 520 : foreach(lc, targetAttrs)
+ + ]
2118 : : {
983 efujita@postgresql.o 2119 : 380 : int attnum = lfirst_int(lc);
2120 : 380 : Form_pg_attribute attr = TupleDescAttr(tupdesc, attnum - 1);
2121 : :
4052 tgl@sss.pgh.pa.us 2122 [ + + ]: 380 : if (!first)
2123 : 240 : appendStringInfoString(buf, ", ");
2124 : 380 : first = false;
2125 : :
983 efujita@postgresql.o 2126 [ + + ]: 380 : if (attr->attgenerated)
2127 : 5 : appendStringInfoString(buf, "DEFAULT");
2128 : : else
2129 : : {
2130 : 375 : appendStringInfo(buf, "$%d", pindex);
2131 : 375 : pindex++;
2132 : : }
2133 : : }
2134 : :
3818 rhaas@postgresql.org 2135 : 140 : appendStringInfoChar(buf, ')');
2136 : : }
2137 : : else
4052 tgl@sss.pgh.pa.us 2138 : 1 : appendStringInfoString(buf, " DEFAULT VALUES");
1180 tomas.vondra@postgre 2139 : 141 : *values_end_len = buf->len;
2140 : :
3264 andres@anarazel.de 2141 [ + + ]: 141 : if (doNothing)
2142 : 3 : appendStringInfoString(buf, " ON CONFLICT DO NOTHING");
2143 : :
2175 rhaas@postgresql.org 2144 : 141 : deparseReturningList(buf, rte, rtindex, rel,
2489 tgl@sss.pgh.pa.us 2145 [ + + + + ]: 141 : rel->trigdesc && rel->trigdesc->trig_insert_after_row,
2107 jdavis@postgresql.or 2146 :ECB (141) : withCheckOptionList, returningList, retrieved_attrs);
4053 tgl@sss.pgh.pa.us 2147 :CBC 141 : }
2148 : :
2149 : : /*
2150 : : * rebuild remote INSERT statement
2151 : : *
2152 : : * Provided a number of rows in a batch, builds INSERT statement with the
2153 : : * right number of parameters.
2154 : : */
2155 : : void
983 efujita@postgresql.o 2156 : 26 : rebuildInsertSql(StringInfo buf, Relation rel,
2157 : : char *orig_query, List *target_attrs,
2158 : : int values_end_len, int num_params,
2159 : : int num_rows)
2160 : : {
2161 : 26 : TupleDesc tupdesc = RelationGetDescr(rel);
2162 : : int i;
2163 : : int pindex;
2164 : : bool first;
2165 : : ListCell *lc;
2166 : :
2167 : : /* Make sure the values_end_len is sensible */
1180 tomas.vondra@postgre 2168 [ + - - + ]: 26 : Assert((values_end_len > 0) && (values_end_len <= strlen(orig_query)));
2169 : :
2170 : : /* Copy up to the end of the first record from the original query */
2171 : 26 : appendBinaryStringInfo(buf, orig_query, values_end_len);
2172 : :
2173 : : /*
2174 : : * Add records to VALUES clause (we already have parameters for the first
2175 : : * row, so start at the right offset).
2176 : : */
983 efujita@postgresql.o 2177 : 26 : pindex = num_params + 1;
1180 tomas.vondra@postgre 2178 [ + + ]: 103 : for (i = 0; i < num_rows; i++)
2179 : : {
2180 : 77 : appendStringInfoString(buf, ", (");
2181 : :
2182 : 77 : first = true;
983 efujita@postgresql.o 2183 [ + - + + : 225 : foreach(lc, target_attrs)
+ + ]
2184 : : {
2185 : 148 : int attnum = lfirst_int(lc);
2186 : 148 : Form_pg_attribute attr = TupleDescAttr(tupdesc, attnum - 1);
2187 : :
1180 tomas.vondra@postgre 2188 [ + + ]: 148 : if (!first)
2189 : 71 : appendStringInfoString(buf, ", ");
2190 : 148 : first = false;
2191 : :
983 efujita@postgresql.o 2192 [ + + ]: 148 : if (attr->attgenerated)
2193 : 1 : appendStringInfoString(buf, "DEFAULT");
2194 : : else
2195 : : {
2196 : 147 : appendStringInfo(buf, "$%d", pindex);
2197 : 147 : pindex++;
2198 : : }
2199 : : }
2200 : :
1180 tomas.vondra@postgre 2201 : 77 : appendStringInfoChar(buf, ')');
2202 : : }
2203 : :
2204 : : /* Copy stuff after VALUES clause from the original query */
2205 : 26 : appendStringInfoString(buf, orig_query + values_end_len);
2206 : 26 : }
2207 : :
2208 : : /*
2209 : : * deparse remote UPDATE statement
2210 : : *
2211 : : * The statement text is appended to buf, and we also create an integer List
2212 : : * of the columns being retrieved by WITH CHECK OPTION or RETURNING (if any),
2213 : : * which is returned to *retrieved_attrs.
2214 : : */
2215 : : void
2175 rhaas@postgresql.org 2216 : 50 : deparseUpdateSql(StringInfo buf, RangeTblEntry *rte,
2217 : : Index rtindex, Relation rel,
2218 : : List *targetAttrs,
2219 : : List *withCheckOptionList, List *returningList,
2220 : : List **retrieved_attrs)
2221 : : {
983 efujita@postgresql.o 2222 : 50 : TupleDesc tupdesc = RelationGetDescr(rel);
2223 : : AttrNumber pindex;
2224 : : bool first;
2225 : : ListCell *lc;
2226 : :
4053 tgl@sss.pgh.pa.us 2227 : 50 : appendStringInfoString(buf, "UPDATE ");
4051 2228 : 50 : deparseRelation(buf, rel);
4053 2229 : 50 : appendStringInfoString(buf, " SET ");
2230 : :
2231 : 50 : pindex = 2; /* ctid is always the first param */
2232 : 50 : first = true;
2233 [ + - + + : 123 : foreach(lc, targetAttrs)
+ + ]
2234 : : {
2235 : 73 : int attnum = lfirst_int(lc);
983 efujita@postgresql.o 2236 : 73 : Form_pg_attribute attr = TupleDescAttr(tupdesc, attnum - 1);
2237 : :
4053 tgl@sss.pgh.pa.us 2238 [ + + ]: 73 : if (!first)
2239 : 23 : appendStringInfoString(buf, ", ");
2240 : 73 : first = false;
2241 : :
2175 rhaas@postgresql.org 2242 : 73 : deparseColumnRef(buf, rtindex, attnum, rte, false);
983 efujita@postgresql.o 2243 [ + + ]: 73 : if (attr->attgenerated)
2244 : 2 : appendStringInfoString(buf, " = DEFAULT");
2245 : : else
2246 : : {
2247 : 71 : appendStringInfo(buf, " = $%d", pindex);
2248 : 71 : pindex++;
2249 : : }
2250 : : }
4053 tgl@sss.pgh.pa.us 2251 : 50 : appendStringInfoString(buf, " WHERE ctid = $1");
2252 : :
2175 rhaas@postgresql.org 2253 : 50 : deparseReturningList(buf, rte, rtindex, rel,
2489 tgl@sss.pgh.pa.us 2254 [ + + + + ]: 50 : rel->trigdesc && rel->trigdesc->trig_update_after_row,
2107 jdavis@postgresql.or 2255 :ECB (50) : withCheckOptionList, returningList, retrieved_attrs);
4053 tgl@sss.pgh.pa.us 2256 :CBC 50 : }
2257 : :
2258 : : /*
2259 : : * deparse remote UPDATE statement
2260 : : *
2261 : : * 'buf' is the output buffer to append the statement to
2262 : : * 'rtindex' is the RT index of the associated target relation
2263 : : * 'rel' is the relation descriptor for the target relation
2264 : : * 'foreignrel' is the RelOptInfo for the target relation or the join relation
2265 : : * containing all base relations in the query
2266 : : * 'targetlist' is the tlist of the underlying foreign-scan plan node
2267 : : * (note that this only contains new-value expressions and junk attrs)
2268 : : * 'targetAttrs' is the target columns of the UPDATE
2269 : : * 'remote_conds' is the qual clauses that must be evaluated remotely
2270 : : * '*params_list' is an output list of exprs that will become remote Params
2271 : : * 'returningList' is the RETURNING targetlist
2272 : : * '*retrieved_attrs' is an output list of integers of columns being retrieved
2273 : : * by RETURNING (if any)
2274 : : */
2275 : : void
2949 rhaas@postgresql.org 2276 : 45 : deparseDirectUpdateSql(StringInfo buf, PlannerInfo *root,
2277 : : Index rtindex, Relation rel,
2278 : : RelOptInfo *foreignrel,
2279 : : List *targetlist,
2280 : : List *targetAttrs,
2281 : : List *remote_conds,
2282 : : List **params_list,
2283 : : List *returningList,
2284 : : List **retrieved_attrs)
2285 : : {
2286 : : deparse_expr_cxt context;
2287 : : int nestlevel;
2288 : : bool first;
2175 2289 [ + - ]: 45 : RangeTblEntry *rte = planner_rt_fetch(rtindex, root);
2290 : : ListCell *lc,
2291 : : *lc2;
131 akorotkov@postgresql 2292 :GNC 45 : List *additional_conds = NIL;
2293 : :
2294 : : /* Set up context struct for recursion */
2949 rhaas@postgresql.org 2295 :CBC 45 : context.root = root;
2258 2296 : 45 : context.foreignrel = foreignrel;
2297 : 45 : context.scanrel = foreignrel;
2949 2298 : 45 : context.buf = buf;
2299 : 45 : context.params_list = params_list;
2300 : :
2301 : 45 : appendStringInfoString(buf, "UPDATE ");
2302 : 45 : deparseRelation(buf, rel);
2258 2303 [ + + ]: 45 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2304 : 4 : appendStringInfo(buf, " %s%d", REL_ALIAS_PREFIX, rtindex);
2949 2305 : 45 : appendStringInfoString(buf, " SET ");
2306 : :
2307 : : /* Make sure any constants in the exprs are printed portably */
2308 : 45 : nestlevel = set_transmission_modes();
2309 : :
2310 : 45 : first = true;
1110 tgl@sss.pgh.pa.us 2311 [ + - + - : 98 : forboth(lc, targetlist, lc2, targetAttrs)
+ - + + +
- + + +
+ ]
2312 : : {
2313 : 53 : TargetEntry *tle = lfirst_node(TargetEntry, lc);
2314 : 53 : int attnum = lfirst_int(lc2);
2315 : :
2316 : : /* update's new-value expressions shouldn't be resjunk */
2317 [ - + ]: 53 : Assert(!tle->resjunk);
2318 : :
2949 rhaas@postgresql.org 2319 [ + + ]: 53 : if (!first)
2320 : 8 : appendStringInfoString(buf, ", ");
2321 : 53 : first = false;
2322 : :
2175 2323 : 53 : deparseColumnRef(buf, rtindex, attnum, rte, false);
2949 2324 : 53 : appendStringInfoString(buf, " = ");
2325 : 53 : deparseExpr((Expr *) tle->expr, &context);
2326 : : }
2327 : :
2328 : 45 : reset_transmission_modes(nestlevel);
2329 : :
2258 2330 [ + + ]: 45 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2331 : : {
2332 : 4 : List *ignore_conds = NIL;
2333 : :
2334 : :
1746 drowley@postgresql.o 2335 : 4 : appendStringInfoString(buf, " FROM ");
2258 rhaas@postgresql.org 2336 : 4 : deparseFromExprForRel(buf, root, foreignrel, true, rtindex,
2337 : : &ignore_conds, &additional_conds, params_list);
2338 : 4 : remote_conds = list_concat(remote_conds, ignore_conds);
2339 : : }
2340 : :
131 akorotkov@postgresql 2341 :GNC 45 : appendWhereClause(remote_conds, additional_conds, &context);
2342 : :
2343 [ - + ]: 45 : if (additional_conds != NIL)
131 akorotkov@postgresql 2344 :UNC 0 : list_free_deep(additional_conds);
2345 : :
2258 rhaas@postgresql.org 2346 [ + + ]:CBC 45 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2347 : 4 : deparseExplicitTargetList(returningList, true, retrieved_attrs,
2348 : : &context);
2349 : : else
2175 2350 : 41 : deparseReturningList(buf, rte, rtindex, rel, false,
2351 : : NIL, returningList, retrieved_attrs);
2949 2352 : 45 : }
2353 : :
2354 : : /*
2355 : : * deparse remote DELETE statement
2356 : : *
2357 : : * The statement text is appended to buf, and we also create an integer List
2358 : : * of the columns being retrieved by RETURNING (if any), which is returned
2359 : : * to *retrieved_attrs.
2360 : : */
2361 : : void
2175 2362 : 16 : deparseDeleteSql(StringInfo buf, RangeTblEntry *rte,
2363 : : Index rtindex, Relation rel,
2364 : : List *returningList,
2365 : : List **retrieved_attrs)
2366 : : {
4053 tgl@sss.pgh.pa.us 2367 : 16 : appendStringInfoString(buf, "DELETE FROM ");
4051 2368 : 16 : deparseRelation(buf, rel);
4053 2369 : 16 : appendStringInfoString(buf, " WHERE ctid = $1");
2370 : :
2175 rhaas@postgresql.org 2371 : 16 : deparseReturningList(buf, rte, rtindex, rel,
2489 tgl@sss.pgh.pa.us 2372 [ + + + + ]: 16 : rel->trigdesc && rel->trigdesc->trig_delete_after_row,
2107 jdavis@postgresql.or 2373 :ECB (16) : NIL, returningList, retrieved_attrs);
4053 tgl@sss.pgh.pa.us 2374 :CBC 16 : }
2375 : :
2376 : : /*
2377 : : * deparse remote DELETE statement
2378 : : *
2379 : : * 'buf' is the output buffer to append the statement to
2380 : : * 'rtindex' is the RT index of the associated target relation
2381 : : * 'rel' is the relation descriptor for the target relation
2382 : : * 'foreignrel' is the RelOptInfo for the target relation or the join relation
2383 : : * containing all base relations in the query
2384 : : * 'remote_conds' is the qual clauses that must be evaluated remotely
2385 : : * '*params_list' is an output list of exprs that will become remote Params
2386 : : * 'returningList' is the RETURNING targetlist
2387 : : * '*retrieved_attrs' is an output list of integers of columns being retrieved
2388 : : * by RETURNING (if any)
2389 : : */
2390 : : void
2949 rhaas@postgresql.org 2391 : 59 : deparseDirectDeleteSql(StringInfo buf, PlannerInfo *root,
2392 : : Index rtindex, Relation rel,
2393 : : RelOptInfo *foreignrel,
2394 : : List *remote_conds,
2395 : : List **params_list,
2396 : : List *returningList,
2397 : : List **retrieved_attrs)
2398 : : {
2399 : : deparse_expr_cxt context;
131 akorotkov@postgresql 2400 :GNC 59 : List *additional_conds = NIL;
2401 : :
2402 : : /* Set up context struct for recursion */
2949 rhaas@postgresql.org 2403 :CBC 59 : context.root = root;
2258 2404 : 59 : context.foreignrel = foreignrel;
2405 : 59 : context.scanrel = foreignrel;
2949 2406 : 59 : context.buf = buf;
2407 : 59 : context.params_list = params_list;
2408 : :
2409 : 59 : appendStringInfoString(buf, "DELETE FROM ");
2410 : 59 : deparseRelation(buf, rel);
2258 2411 [ + + ]: 59 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2412 : 4 : appendStringInfo(buf, " %s%d", REL_ALIAS_PREFIX, rtindex);
2413 : :
2414 [ + + ]: 59 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2415 : : {
2416 : 4 : List *ignore_conds = NIL;
2417 : :
1746 drowley@postgresql.o 2418 : 4 : appendStringInfoString(buf, " USING ");
2258 rhaas@postgresql.org 2419 : 4 : deparseFromExprForRel(buf, root, foreignrel, true, rtindex,
2420 : : &ignore_conds, &additional_conds, params_list);
2421 : 4 : remote_conds = list_concat(remote_conds, ignore_conds);
2422 : : }
2423 : :
131 akorotkov@postgresql 2424 :GNC 59 : appendWhereClause(remote_conds, additional_conds, &context);
2425 : :
2426 [ - + ]: 59 : if (additional_conds != NIL)
131 akorotkov@postgresql 2427 :UNC 0 : list_free_deep(additional_conds);
2428 : :
2258 rhaas@postgresql.org 2429 [ + + ]:CBC 59 : if (foreignrel->reloptkind == RELOPT_JOINREL)
2430 : 4 : deparseExplicitTargetList(returningList, true, retrieved_attrs,
2431 : : &context);
2432 : : else
2175 2433 [ + - ]: 55 : deparseReturningList(buf, planner_rt_fetch(rtindex, root),
2434 : : rtindex, rel, false,
2435 : : NIL, returningList, retrieved_attrs);
2949 2436 : 59 : }
2437 : :
2438 : : /*
2439 : : * Add a RETURNING clause, if needed, to an INSERT/UPDATE/DELETE.
2440 : : */
2441 : : static void
2175 2442 : 303 : deparseReturningList(StringInfo buf, RangeTblEntry *rte,
2443 : : Index rtindex, Relation rel,
2444 : : bool trig_after_row,
2445 : : List *withCheckOptionList,
2446 : : List *returningList,
2447 : : List **retrieved_attrs)
2448 : : {
3675 noah@leadboat.com 2449 : 303 : Bitmapset *attrs_used = NULL;
2450 : :
2451 [ + + ]: 303 : if (trig_after_row)
2452 : : {
2453 : : /* whole-row reference acquires all non-system columns */
2454 : 24 : attrs_used =
2455 : 24 : bms_make_singleton(0 - FirstLowInvalidHeapAttributeNumber);
2456 : : }
2457 : :
2107 jdavis@postgresql.or 2458 [ + + ]: 303 : if (withCheckOptionList != NIL)
2459 : : {
2460 : : /*
2461 : : * We need the attrs, non-system and system, mentioned in the local
2462 : : * query's WITH CHECK OPTION list.
2463 : : *
2464 : : * Note: we do this to ensure that WCO constraints will be evaluated
2465 : : * on the data actually inserted/updated on the remote side, which
2466 : : * might differ from the data supplied by the core code, for example
2467 : : * as a result of remote triggers.
2468 : : */
2469 : 19 : pull_varattnos((Node *) withCheckOptionList, rtindex,
2470 : : &attrs_used);
2471 : : }
2472 : :
3675 noah@leadboat.com 2473 [ + + ]: 303 : if (returningList != NIL)
2474 : : {
2475 : : /*
2476 : : * We need the attrs, non-system and system, mentioned in the local
2477 : : * query's RETURNING list.
2478 : : */
2479 : 68 : pull_varattnos((Node *) returningList, rtindex,
2480 : : &attrs_used);
2481 : : }
2482 : :
2483 [ + + ]: 303 : if (attrs_used != NULL)
2175 rhaas@postgresql.org 2484 : 110 : deparseTargetList(buf, rte, rtindex, rel, true, attrs_used, false,
2485 : : retrieved_attrs);
2486 : : else
3675 noah@leadboat.com 2487 : 193 : *retrieved_attrs = NIL;
4053 tgl@sss.pgh.pa.us 2488 : 303 : }
2489 : :
2490 : : /*
2491 : : * Construct SELECT statement to acquire size in blocks of given relation.
2492 : : *
2493 : : * Note: we use local definition of block size, not remote definition.
2494 : : * This is perhaps debatable.
2495 : : *
2496 : : * Note: pg_relation_size() exists in 8.1 and later.
2497 : : */
2498 : : void
4069 2499 : 40 : deparseAnalyzeSizeSql(StringInfo buf, Relation rel)
2500 : : {
2501 : : StringInfoData relname;
2502 : :
2503 : : /* We'll need the remote relation name as a literal. */
2504 : 40 : initStringInfo(&relname);
4051 2505 : 40 : deparseRelation(&relname, rel);
2506 : :
3818 rhaas@postgresql.org 2507 : 40 : appendStringInfoString(buf, "SELECT pg_catalog.pg_relation_size(");
4069 tgl@sss.pgh.pa.us 2508 : 40 : deparseStringLiteral(buf, relname.data);
2509 : 40 : appendStringInfo(buf, "::pg_catalog.regclass) / %d", BLCKSZ);
2510 : 40 : }
2511 : :
2512 : : /*
2513 : : * Construct SELECT statement to acquire the number of rows and the relkind of
2514 : : * a relation.
2515 : : *
2516 : : * Note: we just return the remote server's reltuples value, which might
2517 : : * be off a good deal, but it doesn't seem worth working harder. See
2518 : : * comments in postgresAcquireSampleRowsFunc.
2519 : : */
2520 : : void
463 tomas.vondra@postgre 2521 : 40 : deparseAnalyzeInfoSql(StringInfo buf, Relation rel)
2522 : : {
2523 : : StringInfoData relname;
2524 : :
2525 : : /* We'll need the remote relation name as a literal. */
471 2526 : 40 : initStringInfo(&relname);
2527 : 40 : deparseRelation(&relname, rel);
2528 : :
463 2529 : 40 : appendStringInfoString(buf, "SELECT reltuples, relkind FROM pg_catalog.pg_class WHERE oid = ");
471 2530 : 40 : deparseStringLiteral(buf, relname.data);
2531 : 40 : appendStringInfoString(buf, "::pg_catalog.regclass");
2532 : 40 : }
2533 : :
2534 : : /*
2535 : : * Construct SELECT statement to acquire sample rows of given relation.
2536 : : *
2537 : : * SELECT command is appended to buf, and list of columns retrieved
2538 : : * is returned to *retrieved_attrs.
2539 : : *
2540 : : * We only support sampling methods we can decide based on server version.
2541 : : * Allowing custom TSM modules (like tsm_system_rows) might be useful, but it
2542 : : * would require detecting which extensions are installed, to allow automatic
2543 : : * fall-back. Moreover, the methods may use different parameters like number
2544 : : * of rows (and not sampling rate). So we leave this for future improvements.
2545 : : *
2546 : : * Using random() to sample rows on the remote server has the advantage that
2547 : : * this works on all PostgreSQL versions (unlike TABLESAMPLE), and that it
2548 : : * does the sampling on the remote side (without transferring everything and
2549 : : * then discarding most rows).
2550 : : *
2551 : : * The disadvantage is that we still have to read all rows and evaluate the
2552 : : * random(), while TABLESAMPLE (at least with the "system" method) may skip.
2553 : : * It's not that different from the "bernoulli" method, though.
2554 : : *
2555 : : * We could also do "ORDER BY random() LIMIT x", which would always pick
2556 : : * the expected number of rows, but it requires sorting so it may be much
2557 : : * more expensive (particularly on large tables, which is what the
2558 : : * remote sampling is meant to improve).
2559 : : */
2560 : : void
2561 : 40 : deparseAnalyzeSql(StringInfo buf, Relation rel,
2562 : : PgFdwSamplingMethod sample_method, double sample_frac,
2563 : : List **retrieved_attrs)
2564 : : {
4070 tgl@sss.pgh.pa.us 2565 : 40 : Oid relid = RelationGetRelid(rel);
2566 : 40 : TupleDesc tupdesc = RelationGetDescr(rel);
2567 : : int i;
2568 : : char *colname;
2569 : : List *options;
2570 : : ListCell *lc;
2571 : 40 : bool first = true;
2572 : :
4041 2573 : 40 : *retrieved_attrs = NIL;
2574 : :
4053 2575 : 40 : appendStringInfoString(buf, "SELECT ");
4070 2576 [ + + ]: 175 : for (i = 0; i < tupdesc->natts; i++)
2577 : : {
2578 : : /* Ignore dropped columns. */
2429 andres@anarazel.de 2579 [ + + ]: 135 : if (TupleDescAttr(tupdesc, i)->attisdropped)
4070 tgl@sss.pgh.pa.us 2580 : 3 : continue;
2581 : :
4053 2582 [ + + ]: 132 : if (!first)
2583 : 92 : appendStringInfoString(buf, ", ");
2584 : 132 : first = false;
2585 : :
2586 : : /* Use attribute name or column_name option. */
2429 andres@anarazel.de 2587 : 132 : colname = NameStr(TupleDescAttr(tupdesc, i)->attname);
4070 tgl@sss.pgh.pa.us 2588 : 132 : options = GetForeignColumnOptions(relid, i + 1);
2589 : :
2590 [ + + + - : 132 : foreach(lc, options)
+ + ]
2591 : : {
2592 : 3 : DefElem *def = (DefElem *) lfirst(lc);
2593 : :
2594 [ + - ]: 3 : if (strcmp(def->defname, "column_name") == 0)
2595 : : {
2596 : 3 : colname = defGetString(def);
2597 : 3 : break;
2598 : : }
2599 : : }
2600 : :
2601 : 132 : appendStringInfoString(buf, quote_identifier(colname));
2602 : :
4041 2603 : 132 : *retrieved_attrs = lappend_int(*retrieved_attrs, i + 1);
2604 : : }
2605 : :
2606 : : /* Don't generate bad syntax for zero-column relation. */
4070 2607 [ - + ]: 40 : if (first)
4053 tgl@sss.pgh.pa.us 2608 :UBC 0 : appendStringInfoString(buf, "NULL");
2609 : :
2610 : : /*
2611 : : * Construct FROM clause, and perhaps WHERE clause too, depending on the
2612 : : * selected sampling method.
2613 : : */
4053 tgl@sss.pgh.pa.us 2614 :CBC 40 : appendStringInfoString(buf, " FROM ");
4051 2615 : 40 : deparseRelation(buf, rel);
2616 : :
471 tomas.vondra@postgre 2617 [ + - - - : 40 : switch (sample_method)
- - ]
2618 : : {
2619 : 40 : case ANALYZE_SAMPLE_OFF:
2620 : : /* nothing to do here */
2621 : 40 : break;
2622 : :
471 tomas.vondra@postgre 2623 :UBC 0 : case ANALYZE_SAMPLE_RANDOM:
2624 : 0 : appendStringInfo(buf, " WHERE pg_catalog.random() < %f", sample_frac);
2625 : 0 : break;
2626 : :
2627 : 0 : case ANALYZE_SAMPLE_SYSTEM:
2628 : 0 : appendStringInfo(buf, " TABLESAMPLE SYSTEM(%f)", (100.0 * sample_frac));
2629 : 0 : break;
2630 : :
2631 : 0 : case ANALYZE_SAMPLE_BERNOULLI:
2632 : 0 : appendStringInfo(buf, " TABLESAMPLE BERNOULLI(%f)", (100.0 * sample_frac));
2633 : 0 : break;
2634 : :
2635 : 0 : case ANALYZE_SAMPLE_AUTO:
2636 : : /* should have been resolved into actual method */
2637 [ # # ]: 0 : elog(ERROR, "unexpected sampling method");
2638 : : break;
2639 : : }
4070 tgl@sss.pgh.pa.us 2640 :CBC 40 : }
2641 : :
2642 : : /*
2643 : : * Construct a simple "TRUNCATE rel" statement
2644 : : */
2645 : : void
1102 fujii@postgresql.org 2646 : 12 : deparseTruncateSql(StringInfo buf,
2647 : : List *rels,
2648 : : DropBehavior behavior,
2649 : : bool restart_seqs)
2650 : : {
2651 : : ListCell *cell;
2652 : :
2653 : 12 : appendStringInfoString(buf, "TRUNCATE ");
2654 : :
1083 2655 [ + - + + : 26 : foreach(cell, rels)
+ + ]
2656 : : {
2657 : 14 : Relation rel = lfirst(cell);
2658 : :
2659 [ + + ]: 14 : if (cell != list_head(rels))
1102 2660 : 2 : appendStringInfoString(buf, ", ");
2661 : :
2662 : 14 : deparseRelation(buf, rel);
2663 : : }
2664 : :
2665 [ - + ]: 12 : appendStringInfo(buf, " %s IDENTITY",
2666 : : restart_seqs ? "RESTART" : "CONTINUE");
2667 : :
2668 [ + + ]: 12 : if (behavior == DROP_RESTRICT)
2669 : 10 : appendStringInfoString(buf, " RESTRICT");
2670 [ + - ]: 2 : else if (behavior == DROP_CASCADE)
2671 : 2 : appendStringInfoString(buf, " CASCADE");
2672 : 12 : }
2673 : :
2674 : : /*
2675 : : * Construct name to use for given column, and emit it into buf.
2676 : : * If it has a column_name FDW option, use that instead of attribute name.
2677 : : *
2678 : : * If qualify_col is true, qualify column name with the alias of relation.
2679 : : */
2680 : : static void
2175 rhaas@postgresql.org 2681 : 14502 : deparseColumnRef(StringInfo buf, int varno, int varattno, RangeTblEntry *rte,
2682 : : bool qualify_col)
2683 : : {
2684 : : /* We support fetching the remote side's CTID and OID. */
2987 2685 [ + + ]: 14502 : if (varattno == SelfItemPointerAttributeNumber)
2686 : : {
2687 [ + + ]: 48 : if (qualify_col)
2688 : 46 : ADD_REL_QUALIFIER(buf, varno);
2689 : 48 : appendStringInfoString(buf, "ctid");
2690 : : }
2921 2691 [ - + ]: 14454 : else if (varattno < 0)
2692 : : {
2693 : : /*
2694 : : * All other system attributes are fetched as 0, except for table OID,
2695 : : * which is fetched as the local table OID. However, we must be
2696 : : * careful; the table could be beneath an outer join, in which case it
2697 : : * must go to NULL whenever the rest of the row does.
2698 : : */
2866 rhaas@postgresql.org 2699 :UBC 0 : Oid fetchval = 0;
2700 : :
2921 2701 [ # # ]: 0 : if (varattno == TableOidAttributeNumber)
2702 : 0 : fetchval = rte->relid;
2703 : :
2704 [ # # ]: 0 : if (qualify_col)
2705 : : {
2851 2706 : 0 : appendStringInfoString(buf, "CASE WHEN (");
2921 2707 : 0 : ADD_REL_QUALIFIER(buf, varno);
2844 2708 : 0 : appendStringInfo(buf, "*)::text IS NOT NULL THEN %u END", fetchval);
2709 : : }
2710 : : else
2921 2711 : 0 : appendStringInfo(buf, "%u", fetchval);
2712 : : }
2987 rhaas@postgresql.org 2713 [ + + ]:CBC 14454 : else if (varattno == 0)
2714 : : {
2715 : : /* Whole row reference */
2716 : : Relation rel;
2717 : : Bitmapset *attrs_used;
2718 : :
2719 : : /* Required only to be passed down to deparseTargetList(). */
2720 : : List *retrieved_attrs;
2721 : :
2722 : : /*
2723 : : * The lock on the relation will be held by upper callers, so it's
2724 : : * fine to open it with no lock here.
2725 : : */
1910 andres@anarazel.de 2726 : 232 : rel = table_open(rte->relid, NoLock);
2727 : :
2728 : : /*
2729 : : * The local name of the foreign table can not be recognized by the
2730 : : * foreign server and the table it references on foreign server might
2731 : : * have different column ordering or different columns than those
2732 : : * declared locally. Hence we have to deparse whole-row reference as
2733 : : * ROW(columns referenced locally). Construct this by deparsing a
2734 : : * "whole row" attribute.
2735 : : */
2987 rhaas@postgresql.org 2736 : 232 : attrs_used = bms_add_member(NULL,
2737 : : 0 - FirstLowInvalidHeapAttributeNumber);
2738 : :
2739 : : /*
2740 : : * In case the whole-row reference is under an outer join then it has
2741 : : * to go NULL whenever the rest of the row goes NULL. Deparsing a join
2742 : : * query would always involve multiple relations, thus qualify_col
2743 : : * would be true.
2744 : : */
2921 2745 [ + + ]: 232 : if (qualify_col)
2746 : : {
2851 2747 : 228 : appendStringInfoString(buf, "CASE WHEN (");
2921 2748 : 228 : ADD_REL_QUALIFIER(buf, varno);
2434 peter_e@gmx.net 2749 : 228 : appendStringInfoString(buf, "*)::text IS NOT NULL THEN ");
2750 : : }
2751 : :
2987 rhaas@postgresql.org 2752 : 232 : appendStringInfoString(buf, "ROW(");
2175 2753 : 232 : deparseTargetList(buf, rte, varno, rel, false, attrs_used, qualify_col,
2754 : : &retrieved_attrs);
2434 peter_e@gmx.net 2755 : 232 : appendStringInfoChar(buf, ')');
2756 : :
2757 : : /* Complete the CASE WHEN statement started above. */
2921 rhaas@postgresql.org 2758 [ + + ]: 232 : if (qualify_col)
2434 peter_e@gmx.net 2759 : 228 : appendStringInfoString(buf, " END");
2760 : :
1910 andres@anarazel.de 2761 : 232 : table_close(rel, NoLock);
2987 rhaas@postgresql.org 2762 : 232 : bms_free(attrs_used);
2763 : : }
2764 : : else
2765 : : {
2766 : 14222 : char *colname = NULL;
2767 : : List *options;
2768 : : ListCell *lc;
2769 : :
2770 : : /* varno must not be any of OUTER_VAR, INNER_VAR and INDEX_VAR. */
2771 [ - + ]: 14222 : Assert(!IS_SPECIAL_VARNO(varno));
2772 : :
2773 : : /*
2774 : : * If it's a column of a foreign table, and it has the column_name FDW
2775 : : * option, use that value.
2776 : : */
2777 : 14222 : options = GetForeignColumnOptions(rte->relid, varattno);
2778 [ + + + - : 14222 : foreach(lc, options)
+ + ]
2779 : : {
2780 : 3281 : DefElem *def = (DefElem *) lfirst(lc);
2781 : :
2782 [ + - ]: 3281 : if (strcmp(def->defname, "column_name") == 0)
2783 : : {
2784 : 3281 : colname = defGetString(def);
2785 : 3281 : break;
2786 : : }
2787 : : }
2788 : :
2789 : : /*
2790 : : * If it's a column of a regular table or it doesn't have column_name
2791 : : * FDW option, use attribute name.
2792 : : */
2793 [ + + ]: 14222 : if (colname == NULL)
2253 alvherre@alvh.no-ip. 2794 : 10941 : colname = get_attname(rte->relid, varattno, false);
2795 : :
2987 rhaas@postgresql.org 2796 [ + + ]: 14222 : if (qualify_col)
2797 : 7286 : ADD_REL_QUALIFIER(buf, varno);
2798 : :
2799 : 14222 : appendStringInfoString(buf, quote_identifier(colname));
2800 : : }
4070 tgl@sss.pgh.pa.us 2801 : 14502 : }
2802 : :
2803 : : /*
2804 : : * Append remote name of specified foreign table to buf.
2805 : : * Use value of table_name FDW option (if any) instead of relation's name.
2806 : : * Similarly, schema_name FDW option overrides schema name.
2807 : : */
2808 : : static void
4051 2809 : 3561 : deparseRelation(StringInfo buf, Relation rel)
2810 : : {
2811 : : ForeignTable *table;
4070 2812 : 3561 : const char *nspname = NULL;
2813 : 3561 : const char *relname = NULL;
2814 : : ListCell *lc;
2815 : :
2816 : : /* obtain additional catalog information. */
4051 2817 : 3561 : table = GetForeignTable(RelationGetRelid(rel));
2818 : :
2819 : : /*
2820 : : * Use value of FDW options if any, instead of the name of object itself.
2821 : : */
4070 2822 [ + - + + : 11647 : foreach(lc, table->options)
+ + ]
2823 : : {
2824 : 8086 : DefElem *def = (DefElem *) lfirst(lc);
2825 : :
2826 [ + + ]: 8086 : if (strcmp(def->defname, "schema_name") == 0)
2827 : 2461 : nspname = defGetString(def);
2828 [ + + ]: 5625 : else if (strcmp(def->defname, "table_name") == 0)
2829 : 3561 : relname = defGetString(def);
2830 : : }
2831 : :
2832 : : /*
2833 : : * Note: we could skip printing the schema name if it's pg_catalog, but
2834 : : * that doesn't seem worth the trouble.
2835 : : */
2836 [ + + ]: 3561 : if (nspname == NULL)
4051 2837 : 1100 : nspname = get_namespace_name(RelationGetNamespace(rel));
4070 2838 [ - + ]: 3561 : if (relname == NULL)
4051 tgl@sss.pgh.pa.us 2839 :UBC 0 : relname = RelationGetRelationName(rel);
2840 : :
4070 tgl@sss.pgh.pa.us 2841 :CBC 3561 : appendStringInfo(buf, "%s.%s",
2842 : : quote_identifier(nspname), quote_identifier(relname));
2843 : 3561 : }
2844 : :
2845 : : /*
2846 : : * Append a SQL string literal representing "val" to buf.
2847 : : */
2848 : : void
2849 : 322 : deparseStringLiteral(StringInfo buf, const char *val)
2850 : : {
2851 : : const char *valptr;
2852 : :
2853 : : /*
2854 : : * Rather than making assumptions about the remote server's value of
2855 : : * standard_conforming_strings, always use E'foo' syntax if there are any
2856 : : * backslashes. This will fail on remote servers before 8.1, but those
2857 : : * are long out of support.
2858 : : */
2859 [ + + ]: 322 : if (strchr(val, '\\') != NULL)
2860 : 1 : appendStringInfoChar(buf, ESCAPE_STRING_SYNTAX);
2861 : 322 : appendStringInfoChar(buf, '\'');
2862 [ + + ]: 2926 : for (valptr = val; *valptr; valptr++)
2863 : : {
2864 : 2604 : char ch = *valptr;
2865 : :
2866 [ + + + + ]: 2604 : if (SQL_STR_DOUBLE(ch, true))
2867 : 2 : appendStringInfoChar(buf, ch);
2868 : 2604 : appendStringInfoChar(buf, ch);
2869 : : }
2870 : 322 : appendStringInfoChar(buf, '\'');
2871 : 322 : }
2872 : :
2873 : : /*
2874 : : * Deparse given expression into context->buf.
2875 : : *
2876 : : * This function must support all the same node types that foreign_expr_walker
2877 : : * accepts.
2878 : : *
2879 : : * Note: unlike ruleutils.c, we just use a simple hard-wired parenthesization
2880 : : * scheme: anything more complex than a Var, Const, function call or cast
2881 : : * should be self-parenthesized.
2882 : : */
2883 : : static void
4042 2884 : 11824 : deparseExpr(Expr *node, deparse_expr_cxt *context)
2885 : : {
4070 2886 [ - + ]: 11824 : if (node == NULL)
4070 tgl@sss.pgh.pa.us 2887 :UBC 0 : return;
2888 : :
4070 tgl@sss.pgh.pa.us 2889 [ + + + + :CBC 11824 : switch (nodeTag(node))
+ + + + +
+ + + + +
- ]
2890 : : {
2891 : 8183 : case T_Var:
4042 2892 : 8183 : deparseVar((Var *) node, context);
4070 2893 : 8183 : break;
2894 : 570 : case T_Const:
2732 rhaas@postgresql.org 2895 : 570 : deparseConst((Const *) node, context, 0);
4070 tgl@sss.pgh.pa.us 2896 : 570 : break;
2897 : 28 : case T_Param:
4042 2898 : 28 : deparseParam((Param *) node, context);
4070 2899 : 28 : break;
1899 alvherre@alvh.no-ip. 2900 : 1 : case T_SubscriptingRef:
2901 : 1 : deparseSubscriptingRef((SubscriptingRef *) node, context);
4070 tgl@sss.pgh.pa.us 2902 : 1 : break;
2903 : 58 : case T_FuncExpr:
4042 2904 : 58 : deparseFuncExpr((FuncExpr *) node, context);
4070 2905 : 58 : break;
2906 : 2597 : case T_OpExpr:
4042 2907 : 2597 : deparseOpExpr((OpExpr *) node, context);
4070 2908 : 2597 : break;
2909 : 1 : case T_DistinctExpr:
4042 2910 : 1 : deparseDistinctExpr((DistinctExpr *) node, context);
4070 2911 : 1 : break;
2912 : 4 : case T_ScalarArrayOpExpr:
4042 2913 : 4 : deparseScalarArrayOpExpr((ScalarArrayOpExpr *) node, context);
4070 2914 : 4 : break;
2915 : 33 : case T_RelabelType:
4042 2916 : 33 : deparseRelabelType((RelabelType *) node, context);
4070 2917 : 33 : break;
2918 : 38 : case T_BoolExpr:
4042 2919 : 38 : deparseBoolExpr((BoolExpr *) node, context);
4070 2920 : 38 : break;
2921 : 28 : case T_NullTest:
4042 2922 : 28 : deparseNullTest((NullTest *) node, context);
4070 2923 : 28 : break;
989 2924 : 21 : case T_CaseExpr:
2925 : 21 : deparseCaseExpr((CaseExpr *) node, context);
2926 : 21 : break;
4070 2927 : 4 : case T_ArrayExpr:
4042 2928 : 4 : deparseArrayExpr((ArrayExpr *) node, context);
4070 2929 : 4 : break;
2732 rhaas@postgresql.org 2930 : 258 : case T_Aggref:
2931 : 258 : deparseAggref((Aggref *) node, context);
2932 : 258 : break;
4070 tgl@sss.pgh.pa.us 2933 :UBC 0 : default:
2934 [ # # ]: 0 : elog(ERROR, "unsupported expression type for deparse: %d",
2935 : : (int) nodeTag(node));
2936 : : break;
2937 : : }
2938 : : }
2939 : :
2940 : : /*
2941 : : * Deparse given Var node into context->buf.
2942 : : *
2943 : : * If the Var belongs to the foreign relation, just print its remote name.
2944 : : * Otherwise, it's effectively a Param (and will in fact be a Param at
2945 : : * run time). Handle it the same way we handle plain Params --- see
2946 : : * deparseParam for comments.
2947 : : */
2948 : : static void
4042 tgl@sss.pgh.pa.us 2949 :CBC 8183 : deparseVar(Var *node, deparse_expr_cxt *context)
2950 : : {
2732 rhaas@postgresql.org 2951 : 8183 : Relids relids = context->scanrel->relids;
2952 : : int relno;
2953 : : int colno;
2954 : :
2955 : : /* Qualify columns when multiple relations are involved. */
2114 michael@paquier.xyz 2956 : 8183 : bool qualify_col = (bms_membership(relids) == BMS_MULTIPLE);
2957 : :
2958 : : /*
2959 : : * If the Var belongs to the foreign relation that is deparsed as a
2960 : : * subquery, use the relation and column alias to the Var provided by the
2961 : : * subquery, instead of the remote name.
2962 : : */
2586 rhaas@postgresql.org 2963 [ + + ]: 8183 : if (is_subquery_var(node, context->scanrel, &relno, &colno))
2964 : : {
2965 : 116 : appendStringInfo(context->buf, "%s%d.%s%d",
2966 : : SUBQUERY_REL_ALIAS_PREFIX, relno,
2967 : : SUBQUERY_COL_ALIAS_PREFIX, colno);
2968 : 116 : return;
2969 : : }
2970 : :
2732 2971 [ + + + - ]: 8067 : if (bms_is_member(node->varno, relids) && node->varlevelsup == 0)
2987 2972 : 7862 : deparseColumnRef(context->buf, node->varno, node->varattno,
2175 2973 [ + - ]: 7862 : planner_rt_fetch(node->varno, context->root),
2974 : : qualify_col);
2975 : : else
2976 : : {
2977 : : /* Treat like a Param */
4042 tgl@sss.pgh.pa.us 2978 [ + + ]: 205 : if (context->params_list)
2979 : : {
2980 : 11 : int pindex = 0;
2981 : : ListCell *lc;
2982 : :
2983 : : /* find its index in params_list */
2984 [ - + - - : 11 : foreach(lc, *context->params_list)
- + ]
2985 : : {
4042 tgl@sss.pgh.pa.us 2986 :UBC 0 : pindex++;
2987 [ # # ]: 0 : if (equal(node, (Node *) lfirst(lc)))
2988 : 0 : break;
2989 : : }
4042 tgl@sss.pgh.pa.us 2990 [ + - ]:CBC 11 : if (lc == NULL)
2991 : : {
2992 : : /* not in list, so add it */
2993 : 11 : pindex++;
2994 : 11 : *context->params_list = lappend(*context->params_list, node);
2995 : : }
2996 : :
3651 2997 : 11 : printRemoteParam(pindex, node->vartype, node->vartypmod, context);
2998 : : }
2999 : : else
3000 : : {
3001 : 194 : printRemotePlaceholder(node->vartype, node->vartypmod, context);
3002 : : }
3003 : : }
3004 : : }
3005 : :
3006 : : /*
3007 : : * Deparse given constant value into context->buf.
3008 : : *
3009 : : * This function has to be kept in sync with ruleutils.c's get_const_expr.
3010 : : *
3011 : : * As in that function, showtype can be -1 to never show "::typename"
3012 : : * decoration, +1 to always show it, or 0 to show it only if the constant
3013 : : * wouldn't be assumed to be the right type by default.
3014 : : *
3015 : : * In addition, this code allows showtype to be -2 to indicate that we should
3016 : : * not show "::typename" decoration if the constant is printed as an untyped
3017 : : * literal or NULL (while in other cases, behaving as for showtype == 0).
3018 : : */
3019 : : static void
2732 rhaas@postgresql.org 3020 : 1760 : deparseConst(Const *node, deparse_expr_cxt *context, int showtype)
3021 : : {
4042 tgl@sss.pgh.pa.us 3022 : 1760 : StringInfo buf = context->buf;
3023 : : Oid typoutput;
3024 : : bool typIsVarlena;
3025 : : char *extval;
4070 3026 : 1760 : bool isfloat = false;
884 3027 : 1760 : bool isstring = false;
3028 : : bool needlabel;
3029 : :
4070 3030 [ + + ]: 1760 : if (node->constisnull)
3031 : : {
3818 rhaas@postgresql.org 3032 : 19 : appendStringInfoString(buf, "NULL");
2732 3033 [ + - ]: 19 : if (showtype >= 0)
3034 : 19 : appendStringInfo(buf, "::%s",
3035 : : deparse_type_name(node->consttype,
3036 : : node->consttypmod));
4070 tgl@sss.pgh.pa.us 3037 : 19 : return;
3038 : : }
3039 : :
3040 : 1741 : getTypeOutputInfo(node->consttype,
3041 : : &typoutput, &typIsVarlena);
3042 : 1741 : extval = OidOutputFunctionCall(typoutput, node->constvalue);
3043 : :
3044 [ + - + + ]: 1741 : switch (node->consttype)
3045 : : {
3046 : 1658 : case INT2OID:
3047 : : case INT4OID:
3048 : : case INT8OID:
3049 : : case OIDOID:
3050 : : case FLOAT4OID:
3051 : : case FLOAT8OID:
3052 : : case NUMERICOID:
3053 : : {
3054 : : /*
3055 : : * No need to quote unless it's a special value such as 'NaN'.
3056 : : * See comments in get_const_expr().
3057 : : */
3058 [ + - ]: 1658 : if (strspn(extval, "0123456789+-eE.") == strlen(extval))
3059 : : {
3060 [ + - + + ]: 1658 : if (extval[0] == '+' || extval[0] == '-')
3061 : 1 : appendStringInfo(buf, "(%s)", extval);
3062 : : else
3063 : 1657 : appendStringInfoString(buf, extval);
3064 [ + + ]: 1658 : if (strcspn(extval, "eE.") != strlen(extval))
3065 : 2 : isfloat = true; /* it looks like a float */
3066 : : }
3067 : : else
4070 tgl@sss.pgh.pa.us 3068 :UBC 0 : appendStringInfo(buf, "'%s'", extval);
3069 : : }
4070 tgl@sss.pgh.pa.us 3070 :CBC 1658 : break;
4070 tgl@sss.pgh.pa.us 3071 :UBC 0 : case BITOID:
3072 : : case VARBITOID:
3073 : 0 : appendStringInfo(buf, "B'%s'", extval);
3074 : 0 : break;
4070 tgl@sss.pgh.pa.us 3075 :CBC 2 : case BOOLOID:
3076 [ + - ]: 2 : if (strcmp(extval, "t") == 0)
3077 : 2 : appendStringInfoString(buf, "true");
3078 : : else
4070 tgl@sss.pgh.pa.us 3079 :UBC 0 : appendStringInfoString(buf, "false");
4070 tgl@sss.pgh.pa.us 3080 :CBC 2 : break;
3081 : 81 : default:
3082 : 81 : deparseStringLiteral(buf, extval);
884 3083 : 81 : isstring = true;
4070 3084 : 81 : break;
3085 : : }
3086 : :
2732 rhaas@postgresql.org 3087 : 1741 : pfree(extval);
3088 : :
884 tgl@sss.pgh.pa.us 3089 [ - + ]: 1741 : if (showtype == -1)
884 tgl@sss.pgh.pa.us 3090 :UBC 0 : return; /* never print type label */
3091 : :
3092 : : /*
3093 : : * For showtype == 0, append ::typename unless the constant will be
3094 : : * implicitly typed as the right type when it is read in.
3095 : : *
3096 : : * XXX this code has to be kept in sync with the behavior of the parser,
3097 : : * especially make_const.
3098 : : */
4070 tgl@sss.pgh.pa.us 3099 [ + + + ]:CBC 1741 : switch (node->consttype)
3100 : : {
3101 : 1421 : case BOOLOID:
3102 : : case INT4OID:
3103 : : case UNKNOWNOID:
3104 : 1421 : needlabel = false;
3105 : 1421 : break;
3106 : 21 : case NUMERICOID:
3107 [ + + - + ]: 21 : needlabel = !isfloat || (node->consttypmod >= 0);
3108 : 21 : break;
3109 : 299 : default:
884 3110 [ + + ]: 299 : if (showtype == -2)
3111 : : {
3112 : : /* label unless we printed it as an untyped string */
3113 : 43 : needlabel = !isstring;
3114 : : }
3115 : : else
3116 : 256 : needlabel = true;
4070 3117 : 299 : break;
3118 : : }
2732 rhaas@postgresql.org 3119 [ + + - + ]: 1741 : if (needlabel || showtype > 0)
4070 tgl@sss.pgh.pa.us 3120 : 275 : appendStringInfo(buf, "::%s",
3121 : : deparse_type_name(node->consttype,
3122 : : node->consttypmod));
3123 : : }
3124 : :
3125 : : /*
3126 : : * Deparse given Param node.
3127 : : *
3128 : : * If we're generating the query "for real", add the Param to
3129 : : * context->params_list if it's not already present, and then use its index
3130 : : * in that list as the remote parameter number. During EXPLAIN, there's
3131 : : * no need to identify a parameter number.
3132 : : */
3133 : : static void
4042 3134 : 28 : deparseParam(Param *node, deparse_expr_cxt *context)
3135 : : {
3136 [ + + ]: 28 : if (context->params_list)
3137 : : {
3138 : 18 : int pindex = 0;
3139 : : ListCell *lc;
3140 : :
3141 : : /* find its index in params_list */
3142 [ + + + + : 20 : foreach(lc, *context->params_list)
+ + ]
3143 : : {
3144 : 2 : pindex++;
3145 [ - + ]: 2 : if (equal(node, (Node *) lfirst(lc)))
4042 tgl@sss.pgh.pa.us 3146 :UBC 0 : break;
3147 : : }
4042 tgl@sss.pgh.pa.us 3148 [ + - ]:CBC 18 : if (lc == NULL)
3149 : : {
3150 : : /* not in list, so add it */
3151 : 18 : pindex++;
3152 : 18 : *context->params_list = lappend(*context->params_list, node);
3153 : : }
3154 : :
3651 3155 : 18 : printRemoteParam(pindex, node->paramtype, node->paramtypmod, context);
3156 : : }
3157 : : else
3158 : : {
3159 : 10 : printRemotePlaceholder(node->paramtype, node->paramtypmod, context);
3160 : : }
4070 3161 : 28 : }
3162 : :
3163 : : /*
3164 : : * Deparse a container subscript expression.
3165 : : */
3166 : : static void
1899 alvherre@alvh.no-ip. 3167 : 1 : deparseSubscriptingRef(SubscriptingRef *node, deparse_expr_cxt *context)
3168 : : {
4042 tgl@sss.pgh.pa.us 3169 : 1 : StringInfo buf = context->buf;
3170 : : ListCell *lowlist_item;
3171 : : ListCell *uplist_item;
3172 : :
3173 : : /* Always parenthesize the expression. */
4070 3174 : 1 : appendStringInfoChar(buf, '(');
3175 : :
3176 : : /*
3177 : : * Deparse referenced array expression first. If that expression includes
3178 : : * a cast, we have to parenthesize to prevent the array subscript from
3179 : : * being taken as typename decoration. We can avoid that in the typical
3180 : : * case of subscripting a Var, but otherwise do it.
3181 : : */
3182 [ - + ]: 1 : if (IsA(node->refexpr, Var))
4042 tgl@sss.pgh.pa.us 3183 :UBC 0 : deparseExpr(node->refexpr, context);
3184 : : else
3185 : : {
4070 tgl@sss.pgh.pa.us 3186 :CBC 1 : appendStringInfoChar(buf, '(');
4042 3187 : 1 : deparseExpr(node->refexpr, context);
4070 3188 : 1 : appendStringInfoChar(buf, ')');
3189 : : }
3190 : :
3191 : : /* Deparse subscript expressions. */
3192 : 1 : lowlist_item = list_head(node->reflowerindexpr); /* could be NULL */
3193 [ + - + + : 2 : foreach(uplist_item, node->refupperindexpr)
+ + ]
3194 : : {
3195 : 1 : appendStringInfoChar(buf, '[');
3196 [ - + ]: 1 : if (lowlist_item)
3197 : : {
4042 tgl@sss.pgh.pa.us 3198 :UBC 0 : deparseExpr(lfirst(lowlist_item), context);
4070 3199 : 0 : appendStringInfoChar(buf, ':');
1735 3200 : 0 : lowlist_item = lnext(node->reflowerindexpr, lowlist_item);
3201 : : }
4042 tgl@sss.pgh.pa.us 3202 :CBC 1 : deparseExpr(lfirst(uplist_item), context);
4070 3203 : 1 : appendStringInfoChar(buf, ']');
3204 : : }
3205 : :
3206 : 1 : appendStringInfoChar(buf, ')');
3207 : 1 : }
3208 : :
3209 : : /*
3210 : : * Deparse a function call.
3211 : : */
3212 : : static void
4042 3213 : 58 : deparseFuncExpr(FuncExpr *node, deparse_expr_cxt *context)
3214 : : {
3215 : 58 : StringInfo buf = context->buf;
3216 : : bool use_variadic;
3217 : : bool first;
3218 : : ListCell *arg;
3219 : :
3220 : : /*
3221 : : * If the function call came from an implicit coercion, then just show the
3222 : : * first argument.
3223 : : */
4069 3224 [ + + ]: 58 : if (node->funcformat == COERCE_IMPLICIT_CAST)
3225 : : {
4042 3226 : 21 : deparseExpr((Expr *) linitial(node->args), context);
4069 3227 : 21 : return;
3228 : : }
3229 : :
3230 : : /*
3231 : : * If the function call came from a cast, then show the first argument
3232 : : * plus an explicit cast operation.
3233 : : */
3234 [ - + ]: 37 : if (node->funcformat == COERCE_EXPLICIT_CAST)
3235 : : {
4069 tgl@sss.pgh.pa.us 3236 :UBC 0 : Oid rettype = node->funcresulttype;
3237 : : int32 coercedTypmod;
3238 : :
3239 : : /* Get the typmod if this is a length-coercion function */
3240 : 0 : (void) exprIsLengthCoercion((Node *) node, &coercedTypmod);
3241 : :
4042 3242 : 0 : deparseExpr((Expr *) linitial(node->args), context);
4069 3243 : 0 : appendStringInfo(buf, "::%s",
3244 : : deparse_type_name(rettype, coercedTypmod));
3245 : 0 : return;
3246 : : }
3247 : :
3248 : : /* Check if need to print VARIADIC (cf. ruleutils.c) */
3664 tgl@sss.pgh.pa.us 3249 :CBC 37 : use_variadic = node->funcvariadic;
3250 : :
3251 : : /*
3252 : : * Normal function: display as proname(args).
3253 : : */
2732 rhaas@postgresql.org 3254 : 37 : appendFunctionName(node->funcid, context);
3255 : 37 : appendStringInfoChar(buf, '(');
3256 : :
3257 : : /* ... and all the arguments */
4070 tgl@sss.pgh.pa.us 3258 : 37 : first = true;
3259 [ + - + + : 78 : foreach(arg, node->args)
+ + ]
3260 : : {
3261 [ + + ]: 41 : if (!first)
3262 : 4 : appendStringInfoString(buf, ", ");
1735 3263 [ - + - - ]: 41 : if (use_variadic && lnext(node->args, arg) == NULL)
4070 tgl@sss.pgh.pa.us 3264 :UBC 0 : appendStringInfoString(buf, "VARIADIC ");
4042 tgl@sss.pgh.pa.us 3265 :CBC 41 : deparseExpr((Expr *) lfirst(arg), context);
4070 3266 : 41 : first = false;
3267 : : }
3268 : 37 : appendStringInfoChar(buf, ')');
3269 : : }
3270 : :
3271 : : /*
3272 : : * Deparse given operator expression. To avoid problems around
3273 : : * priority of operations, we always parenthesize the arguments.
3274 : : */
3275 : : static void
4042 3276 : 2597 : deparseOpExpr(OpExpr *node, deparse_expr_cxt *context)
3277 : : {
3278 : 2597 : StringInfo buf = context->buf;
3279 : : HeapTuple tuple;
3280 : : Form_pg_operator form;
3281 : : Expr *right;
884 3282 : 2597 : bool canSuppressRightConstCast = false;
3283 : : char oprkind;
3284 : :
3285 : : /* Retrieve information about the operator from system catalog. */
4070 3286 : 2597 : tuple = SearchSysCache1(OPEROID, ObjectIdGetDatum(node->opno));
3287 [ - + ]: 2597 : if (!HeapTupleIsValid(tuple))
4070 tgl@sss.pgh.pa.us 3288 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for operator %u", node->opno);
4070 tgl@sss.pgh.pa.us 3289 :CBC 2597 : form = (Form_pg_operator) GETSTRUCT(tuple);
3290 : 2597 : oprkind = form->oprkind;
3291 : :
3292 : : /* Sanity check. */
1305 3293 [ + + - + : 2597 : Assert((oprkind == 'l' && list_length(node->args) == 1) ||
+ - - + ]
3294 : : (oprkind == 'b' && list_length(node->args) == 2));
3295 : :
884 3296 : 2597 : right = llast(node->args);
3297 : :
3298 : : /* Always parenthesize the expression. */
4070 3299 : 2597 : appendStringInfoChar(buf, '(');
3300 : :
3301 : : /* Deparse left operand, if any. */
1305 3302 [ + + ]: 2597 : if (oprkind == 'b')
3303 : : {
884 3304 : 2594 : Expr *left = linitial(node->args);
3305 : 2594 : Oid leftType = exprType((Node *) left);
3306 : 2594 : Oid rightType = exprType((Node *) right);
3307 : 2594 : bool canSuppressLeftConstCast = false;
3308 : :
3309 : : /*
3310 : : * When considering a binary operator, if one operand is a Const that
3311 : : * can be printed as a bare string literal or NULL (i.e., it will look
3312 : : * like type UNKNOWN to the remote parser), the Const normally
3313 : : * receives an explicit cast to the operator's input type. However,
3314 : : * in Const-to-Var comparisons where both operands are of the same
3315 : : * type, we prefer to suppress the explicit cast, leaving the Const's
3316 : : * type resolution up to the remote parser. The remote's resolution
3317 : : * heuristic will assume that an unknown input type being compared to
3318 : : * a known input type is of that known type as well.
3319 : : *
3320 : : * This hack allows some cases to succeed where a remote column is
3321 : : * declared with a different type in the local (foreign) table. By
3322 : : * emitting "foreigncol = 'foo'" not "foreigncol = 'foo'::text" or the
3323 : : * like, we allow the remote parser to pick an "=" operator that's
3324 : : * compatible with whatever type the remote column really is, such as
3325 : : * an enum.
3326 : : *
3327 : : * We allow cast suppression to happen only when the other operand is
3328 : : * a plain foreign Var. Although the remote's unknown-type heuristic
3329 : : * would apply to other cases just as well, we would be taking a
3330 : : * bigger risk that the inferred type is something unexpected. With
3331 : : * this restriction, if anything goes wrong it's the user's fault for
3332 : : * not declaring the local column with the same type as the remote
3333 : : * column.
3334 : : */
3335 [ + + ]: 2594 : if (leftType == rightType)
3336 : : {
3337 [ + + ]: 2578 : if (IsA(left, Const))
3338 : 3 : canSuppressLeftConstCast = isPlainForeignVar(right, context);
3339 [ + + ]: 2575 : else if (IsA(right, Const))
3340 : 1437 : canSuppressRightConstCast = isPlainForeignVar(left, context);
3341 : : }
3342 : :
3343 [ + + ]: 2594 : if (canSuppressLeftConstCast)
3344 : 2 : deparseConst((Const *) left, context, -2);
3345 : : else
3346 : 2592 : deparseExpr(left, context);
3347 : :
4070 3348 : 2594 : appendStringInfoChar(buf, ' ');
3349 : : }
3350 : :
3351 : : /* Deparse operator name. */
4069 3352 : 2597 : deparseOperatorName(buf, form);
3353 : :
3354 : : /* Deparse right operand. */
1305 3355 : 2597 : appendStringInfoChar(buf, ' ');
3356 : :
884 3357 [ + + ]: 2597 : if (canSuppressRightConstCast)
3358 : 1188 : deparseConst((Const *) right, context, -2);
3359 : : else
3360 : 1409 : deparseExpr(right, context);
3361 : :
4070 3362 : 2597 : appendStringInfoChar(buf, ')');
3363 : :
3364 : 2597 : ReleaseSysCache(tuple);
3365 : 2597 : }
3366 : :
3367 : : /*
3368 : : * Will "node" deparse as a plain foreign Var?
3369 : : */
3370 : : static bool
884 3371 : 1440 : isPlainForeignVar(Expr *node, deparse_expr_cxt *context)
3372 : : {
3373 : : /*
3374 : : * We allow the foreign Var to have an implicit RelabelType, mainly so
3375 : : * that this'll work with varchar columns. Note that deparseRelabelType
3376 : : * will not print such a cast, so we're not breaking the restriction that
3377 : : * the expression print as a plain Var. We won't risk it for an implicit
3378 : : * cast that requires a function, nor for non-implicit RelabelType; such
3379 : : * cases seem too likely to involve semantics changes compared to what
3380 : : * would happen on the remote side.
3381 : : */
3382 [ + + ]: 1440 : if (IsA(node, RelabelType) &&
3383 [ + - ]: 5 : ((RelabelType *) node)->relabelformat == COERCE_IMPLICIT_CAST)
3384 : 5 : node = ((RelabelType *) node)->arg;
3385 : :
3386 [ + + ]: 1440 : if (IsA(node, Var))
3387 : : {
3388 : : /*
3389 : : * The Var must be one that'll deparse as a foreign column reference
3390 : : * (cf. deparseVar).
3391 : : */
3392 : 1190 : Var *var = (Var *) node;
3393 : 1190 : Relids relids = context->scanrel->relids;
3394 : :
3395 [ + - + - ]: 1190 : if (bms_is_member(var->varno, relids) && var->varlevelsup == 0)
3396 : 1190 : return true;
3397 : : }
3398 : :
3399 : 250 : return false;
3400 : : }
3401 : :
3402 : : /*
3403 : : * Print the name of an operator.
3404 : : */
3405 : : static void
4069 3406 : 2609 : deparseOperatorName(StringInfo buf, Form_pg_operator opform)
3407 : : {
3408 : : char *opname;
3409 : :
3410 : : /* opname is not a SQL identifier, so we should not quote it. */
3411 : 2609 : opname = NameStr(opform->oprname);
3412 : :
3413 : : /* Print schema name only if it's not pg_catalog */
3414 [ + + ]: 2609 : if (opform->oprnamespace != PG_CATALOG_NAMESPACE)
3415 : : {
3416 : : const char *opnspname;
3417 : :
3418 : 13 : opnspname = get_namespace_name(opform->oprnamespace);
3419 : : /* Print fully qualified operator name. */
3420 : 13 : appendStringInfo(buf, "OPERATOR(%s.%s)",
3421 : : quote_identifier(opnspname), opname);
3422 : : }
3423 : : else
3424 : : {
3425 : : /* Just print operator name. */
3818 rhaas@postgresql.org 3426 : 2596 : appendStringInfoString(buf, opname);
3427 : : }
4069 tgl@sss.pgh.pa.us 3428 : 2609 : }
3429 : :
3430 : : /*
3431 : : * Deparse IS DISTINCT FROM.
3432 : : */
3433 : : static void
4042 3434 : 1 : deparseDistinctExpr(DistinctExpr *node, deparse_expr_cxt *context)
3435 : : {
3436 : 1 : StringInfo buf = context->buf;
3437 : :
4070 3438 [ - + ]: 1 : Assert(list_length(node->args) == 2);
3439 : :
3440 : 1 : appendStringInfoChar(buf, '(');
4042 3441 : 1 : deparseExpr(linitial(node->args), context);
4053 3442 : 1 : appendStringInfoString(buf, " IS DISTINCT FROM ");
4042 3443 : 1 : deparseExpr(lsecond(node->args), context);
4070 3444 : 1 : appendStringInfoChar(buf, ')');
3445 : 1 : }
3446 : :
3447 : : /*
3448 : : * Deparse given ScalarArrayOpExpr expression. To avoid problems
3449 : : * around priority of operations, we always parenthesize the arguments.
3450 : : */
3451 : : static void
4042 3452 : 4 : deparseScalarArrayOpExpr(ScalarArrayOpExpr *node, deparse_expr_cxt *context)
3453 : : {
3454 : 4 : StringInfo buf = context->buf;
3455 : : HeapTuple tuple;
3456 : : Form_pg_operator form;
3457 : : Expr *arg1;
3458 : : Expr *arg2;
3459 : :
3460 : : /* Retrieve information about the operator from system catalog. */
4070 3461 : 4 : tuple = SearchSysCache1(OPEROID, ObjectIdGetDatum(node->opno));
3462 [ - + ]: 4 : if (!HeapTupleIsValid(tuple))
4070 tgl@sss.pgh.pa.us 3463 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for operator %u", node->opno);
4070 tgl@sss.pgh.pa.us 3464 :CBC 4 : form = (Form_pg_operator) GETSTRUCT(tuple);
3465 : :
3466 : : /* Sanity check. */
3467 [ - + ]: 4 : Assert(list_length(node->args) == 2);
3468 : :
3469 : : /* Always parenthesize the expression. */
3470 : 4 : appendStringInfoChar(buf, '(');
3471 : :
3472 : : /* Deparse left operand. */
3473 : 4 : arg1 = linitial(node->args);
4042 3474 : 4 : deparseExpr(arg1, context);
4069 3475 : 4 : appendStringInfoChar(buf, ' ');
3476 : :
3477 : : /* Deparse operator name plus decoration. */
3478 : 4 : deparseOperatorName(buf, form);
3479 [ + - ]: 4 : appendStringInfo(buf, " %s (", node->useOr ? "ANY" : "ALL");
3480 : :
3481 : : /* Deparse right operand. */
4070 3482 : 4 : arg2 = lsecond(node->args);
4042 3483 : 4 : deparseExpr(arg2, context);
3484 : :
4070 3485 : 4 : appendStringInfoChar(buf, ')');
3486 : :
3487 : : /* Always parenthesize the expression. */
3488 : 4 : appendStringInfoChar(buf, ')');
3489 : :
3490 : 4 : ReleaseSysCache(tuple);
3491 : 4 : }
3492 : :
3493 : : /*
3494 : : * Deparse a RelabelType (binary-compatible cast) node.
3495 : : */
3496 : : static void
4042 3497 : 33 : deparseRelabelType(RelabelType *node, deparse_expr_cxt *context)
3498 : : {
3499 : 33 : deparseExpr(node->arg, context);
4069 3500 [ - + ]: 33 : if (node->relabelformat != COERCE_IMPLICIT_CAST)
4042 tgl@sss.pgh.pa.us 3501 :UBC 0 : appendStringInfo(context->buf, "::%s",
3502 : : deparse_type_name(node->resulttype,
3503 : : node->resulttypmod));
4070 tgl@sss.pgh.pa.us 3504 :CBC 33 : }
3505 : :
3506 : : /*
3507 : : * Deparse a BoolExpr node.
3508 : : */
3509 : : static void
4042 3510 : 38 : deparseBoolExpr(BoolExpr *node, deparse_expr_cxt *context)
3511 : : {
3512 : 38 : StringInfo buf = context->buf;
4070 3513 : 38 : const char *op = NULL; /* keep compiler quiet */
3514 : : bool first;
3515 : : ListCell *lc;
3516 : :
3517 [ + + - - ]: 38 : switch (node->boolop)
3518 : : {
3519 : 18 : case AND_EXPR:
3520 : 18 : op = "AND";
3521 : 18 : break;
3522 : 20 : case OR_EXPR:
3523 : 20 : op = "OR";
3524 : 20 : break;
4070 tgl@sss.pgh.pa.us 3525 :UBC 0 : case NOT_EXPR:
4053 3526 : 0 : appendStringInfoString(buf, "(NOT ");
4042 3527 : 0 : deparseExpr(linitial(node->args), context);
4070 3528 : 0 : appendStringInfoChar(buf, ')');
3529 : 0 : return;
3530 : : }
3531 : :
4070 tgl@sss.pgh.pa.us 3532 :CBC 38 : appendStringInfoChar(buf, '(');
3533 : 38 : first = true;
3534 [ + - + + : 114 : foreach(lc, node->args)
+ + ]
3535 : : {
3536 [ + + ]: 76 : if (!first)
3537 : 38 : appendStringInfo(buf, " %s ", op);
4042 3538 : 76 : deparseExpr((Expr *) lfirst(lc), context);
4070 3539 : 76 : first = false;
3540 : : }
3541 : 38 : appendStringInfoChar(buf, ')');
3542 : : }
3543 : :
3544 : : /*
3545 : : * Deparse IS [NOT] NULL expression.
3546 : : */
3547 : : static void
4042 3548 : 28 : deparseNullTest(NullTest *node, deparse_expr_cxt *context)
3549 : : {
3550 : 28 : StringInfo buf = context->buf;
3551 : :
4070 3552 : 28 : appendStringInfoChar(buf, '(');
4042 3553 : 28 : deparseExpr(node->arg, context);
3554 : :
3555 : : /*
3556 : : * For scalar inputs, we prefer to print as IS [NOT] NULL, which is
3557 : : * shorter and traditional. If it's a rowtype input but we're applying a
3558 : : * scalar test, must print IS [NOT] DISTINCT FROM NULL to be semantically
3559 : : * correct.
3560 : : */
2817 3561 [ + - + - ]: 28 : if (node->argisrow || !type_is_rowtype(exprType((Node *) node->arg)))
3562 : : {
3563 [ + + ]: 28 : if (node->nulltesttype == IS_NULL)
3564 : 19 : appendStringInfoString(buf, " IS NULL)");
3565 : : else
3566 : 9 : appendStringInfoString(buf, " IS NOT NULL)");
3567 : : }
3568 : : else
3569 : : {
2817 tgl@sss.pgh.pa.us 3570 [ # # ]:UBC 0 : if (node->nulltesttype == IS_NULL)
3571 : 0 : appendStringInfoString(buf, " IS NOT DISTINCT FROM NULL)");
3572 : : else
3573 : 0 : appendStringInfoString(buf, " IS DISTINCT FROM NULL)");
3574 : : }
4070 tgl@sss.pgh.pa.us 3575 :CBC 28 : }
3576 : :
3577 : : /*
3578 : : * Deparse CASE expression
3579 : : */
3580 : : static void
989 3581 : 21 : deparseCaseExpr(CaseExpr *node, deparse_expr_cxt *context)
3582 : : {
3583 : 21 : StringInfo buf = context->buf;
3584 : : ListCell *lc;
3585 : :
3586 : 21 : appendStringInfoString(buf, "(CASE");
3587 : :
3588 : : /* If this is a CASE arg WHEN then emit the arg expression */
3589 [ + + ]: 21 : if (node->arg != NULL)
3590 : : {
3591 : 9 : appendStringInfoChar(buf, ' ');
3592 : 9 : deparseExpr(node->arg, context);
3593 : : }
3594 : :
3595 : : /* Add each condition/result of the CASE clause */
3596 [ + - + + : 49 : foreach(lc, node->args)
+ + ]
3597 : : {
3598 : 28 : CaseWhen *whenclause = (CaseWhen *) lfirst(lc);
3599 : :
3600 : : /* WHEN */
3601 : 28 : appendStringInfoString(buf, " WHEN ");
3602 [ + + ]: 28 : if (node->arg == NULL) /* CASE WHEN */
3603 : 12 : deparseExpr(whenclause->expr, context);
3604 : : else /* CASE arg WHEN */
3605 : : {
3606 : : /* Ignore the CaseTestExpr and equality operator. */
3607 : 16 : deparseExpr(lsecond(castNode(OpExpr, whenclause->expr)->args),
3608 : : context);
3609 : : }
3610 : :
3611 : : /* THEN */
3612 : 28 : appendStringInfoString(buf, " THEN ");
3613 : 28 : deparseExpr(whenclause->result, context);
3614 : : }
3615 : :
3616 : : /* add ELSE if present */
3617 [ + - ]: 21 : if (node->defresult != NULL)
3618 : : {
3619 : 21 : appendStringInfoString(buf, " ELSE ");
3620 : 21 : deparseExpr(node->defresult, context);
3621 : : }
3622 : :
3623 : : /* append END */
3624 : 21 : appendStringInfoString(buf, " END)");
3625 : 21 : }
3626 : :
3627 : : /*
3628 : : * Deparse ARRAY[...] construct.
3629 : : */
3630 : : static void
4042 3631 : 4 : deparseArrayExpr(ArrayExpr *node, deparse_expr_cxt *context)
3632 : : {
3633 : 4 : StringInfo buf = context->buf;
4070 3634 : 4 : bool first = true;
3635 : : ListCell *lc;
3636 : :
4053 3637 : 4 : appendStringInfoString(buf, "ARRAY[");
4070 3638 [ + - + + : 12 : foreach(lc, node->elements)
+ + ]
3639 : : {
3640 [ + + ]: 8 : if (!first)
4053 3641 : 4 : appendStringInfoString(buf, ", ");
4042 3642 : 8 : deparseExpr(lfirst(lc), context);
4070 3643 : 8 : first = false;
3644 : : }
3645 : 4 : appendStringInfoChar(buf, ']');
3646 : :
3647 : : /* If the array is empty, we need an explicit cast to the array type. */
3648 [ - + ]: 4 : if (node->elements == NIL)
4070 tgl@sss.pgh.pa.us 3649 :UBC 0 : appendStringInfo(buf, "::%s",
3650 : : deparse_type_name(node->array_typeid, -1));
4070 tgl@sss.pgh.pa.us 3651 :CBC 4 : }
3652 : :
3653 : : /*
3654 : : * Deparse an Aggref node.
3655 : : */
3656 : : static void
2732 rhaas@postgresql.org 3657 : 258 : deparseAggref(Aggref *node, deparse_expr_cxt *context)
3658 : : {
3659 : 258 : StringInfo buf = context->buf;
3660 : : bool use_variadic;
3661 : :
3662 : : /* Only basic, non-split aggregation accepted. */
3663 [ - + ]: 258 : Assert(node->aggsplit == AGGSPLIT_SIMPLE);
3664 : :
3665 : : /* Check if need to print VARIADIC (cf. ruleutils.c) */
3666 : 258 : use_variadic = node->aggvariadic;
3667 : :
3668 : : /* Find aggregate name from aggfnoid which is a pg_proc entry */
3669 : 258 : appendFunctionName(node->aggfnoid, context);
3670 : 258 : appendStringInfoChar(buf, '(');
3671 : :
3672 : : /* Add DISTINCT */
2434 peter_e@gmx.net 3673 [ + + ]: 258 : appendStringInfoString(buf, (node->aggdistinct != NIL) ? "DISTINCT " : "");
3674 : :
2732 rhaas@postgresql.org 3675 [ + + ]: 258 : if (AGGKIND_IS_ORDERED_SET(node->aggkind))
3676 : : {
3677 : : /* Add WITHIN GROUP (ORDER BY ..) */
3678 : : ListCell *arg;
3679 : 8 : bool first = true;
3680 : :
3681 [ - + ]: 8 : Assert(!node->aggvariadic);
3682 [ - + ]: 8 : Assert(node->aggorder != NIL);
3683 : :
3684 [ + - + + : 18 : foreach(arg, node->aggdirectargs)
+ + ]
3685 : : {
3686 [ + + ]: 10 : if (!first)
3687 : 2 : appendStringInfoString(buf, ", ");
3688 : 10 : first = false;
3689 : :
3690 : 10 : deparseExpr((Expr *) lfirst(arg), context);
3691 : : }
3692 : :
3693 : 8 : appendStringInfoString(buf, ") WITHIN GROUP (ORDER BY ");
3694 : 8 : appendAggOrderBy(node->aggorder, node->args, context);
3695 : : }
3696 : : else
3697 : : {
3698 : : /* aggstar can be set only in zero-argument aggregates */
3699 [ + + ]: 250 : if (node->aggstar)
3700 : 67 : appendStringInfoChar(buf, '*');
3701 : : else
3702 : : {
3703 : : ListCell *arg;
3704 : 183 : bool first = true;
3705 : :
3706 : : /* Add all the arguments */
3707 [ + - + + : 370 : foreach(arg, node->args)
+ + ]
3708 : : {
3709 : 187 : TargetEntry *tle = (TargetEntry *) lfirst(arg);
3710 : 187 : Node *n = (Node *) tle->expr;
3711 : :
3712 [ + + ]: 187 : if (tle->resjunk)
3713 : 4 : continue;
3714 : :
3715 [ - + ]: 183 : if (!first)
2732 rhaas@postgresql.org 3716 :UBC 0 : appendStringInfoString(buf, ", ");
2732 rhaas@postgresql.org 3717 :CBC 183 : first = false;
3718 : :
3719 : : /* Add VARIADIC */
1735 tgl@sss.pgh.pa.us 3720 [ + + + - ]: 183 : if (use_variadic && lnext(node->args, arg) == NULL)
2732 rhaas@postgresql.org 3721 : 2 : appendStringInfoString(buf, "VARIADIC ");
3722 : :
3723 : 183 : deparseExpr((Expr *) n, context);
3724 : : }
3725 : : }
3726 : :
3727 : : /* Add ORDER BY */
3728 [ + + ]: 250 : if (node->aggorder != NIL)
3729 : : {
3730 : 22 : appendStringInfoString(buf, " ORDER BY ");
3731 : 22 : appendAggOrderBy(node->aggorder, node->args, context);
3732 : : }
3733 : : }
3734 : :
3735 : : /* Add FILTER (WHERE ..) */
3736 [ + + ]: 258 : if (node->aggfilter != NULL)
3737 : : {
3738 : 12 : appendStringInfoString(buf, ") FILTER (WHERE ");
3739 : 12 : deparseExpr((Expr *) node->aggfilter, context);
3740 : : }
3741 : :
3742 : 258 : appendStringInfoChar(buf, ')');
3743 : 258 : }
3744 : :
3745 : : /*
3746 : : * Append ORDER BY within aggregate function.
3747 : : */
3748 : : static void
3749 : 30 : appendAggOrderBy(List *orderList, List *targetList, deparse_expr_cxt *context)
3750 : : {
3751 : 30 : StringInfo buf = context->buf;
3752 : : ListCell *lc;
3753 : 30 : bool first = true;
3754 : :
3755 [ + - + + : 62 : foreach(lc, orderList)
+ + ]
3756 : : {
3757 : 32 : SortGroupClause *srt = (SortGroupClause *) lfirst(lc);
3758 : : Node *sortexpr;
3759 : :
3760 [ + + ]: 32 : if (!first)
3761 : 2 : appendStringInfoString(buf, ", ");
3762 : 32 : first = false;
3763 : :
3764 : : /* Deparse the sort expression proper. */
3765 : 32 : sortexpr = deparseSortGroupClause(srt->tleSortGroupRef, targetList,
3766 : : false, context);
3767 : : /* Add decoration as needed. */
745 tgl@sss.pgh.pa.us 3768 : 32 : appendOrderBySuffix(srt->sortop, exprType(sortexpr), srt->nulls_first,
3769 : : context);
3770 : : }
3771 : 30 : }
3772 : :
3773 : : /*
3774 : : * Append the ASC, DESC, USING <OPERATOR> and NULLS FIRST / NULLS LAST parts
3775 : : * of an ORDER BY clause.
3776 : : */
3777 : : static void
3778 : 872 : appendOrderBySuffix(Oid sortop, Oid sortcoltype, bool nulls_first,
3779 : : deparse_expr_cxt *context)
3780 : : {
3781 : 872 : StringInfo buf = context->buf;
3782 : : TypeCacheEntry *typentry;
3783 : :
3784 : : /* See whether operator is default < or > for sort expr's datatype. */
3785 : 872 : typentry = lookup_type_cache(sortcoltype,
3786 : : TYPECACHE_LT_OPR | TYPECACHE_GT_OPR);
3787 : :
3788 [ + + ]: 872 : if (sortop == typentry->lt_opr)
3789 : 849 : appendStringInfoString(buf, " ASC");
3790 [ + + ]: 23 : else if (sortop == typentry->gt_opr)
3791 : 15 : appendStringInfoString(buf, " DESC");
3792 : : else
3793 : : {
3794 : : HeapTuple opertup;
3795 : : Form_pg_operator operform;
3796 : :
3797 : 8 : appendStringInfoString(buf, " USING ");
3798 : :
3799 : : /* Append operator name. */
3800 : 8 : opertup = SearchSysCache1(OPEROID, ObjectIdGetDatum(sortop));
3801 [ - + ]: 8 : if (!HeapTupleIsValid(opertup))
745 tgl@sss.pgh.pa.us 3802 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for operator %u", sortop);
745 tgl@sss.pgh.pa.us 3803 :CBC 8 : operform = (Form_pg_operator) GETSTRUCT(opertup);
3804 : 8 : deparseOperatorName(buf, operform);
3805 : 8 : ReleaseSysCache(opertup);
3806 : : }
3807 : :
3808 [ + + ]: 872 : if (nulls_first)
3809 : 11 : appendStringInfoString(buf, " NULLS FIRST");
3810 : : else
3811 : 861 : appendStringInfoString(buf, " NULLS LAST");
2732 rhaas@postgresql.org 3812 : 872 : }
3813 : :
3814 : : /*
3815 : : * Print the representation of a parameter to be sent to the remote side.
3816 : : *
3817 : : * Note: we always label the Param's type explicitly rather than relying on
3818 : : * transmitting a numeric type OID in PQsendQueryParams(). This allows us to
3819 : : * avoid assuming that types have the same OIDs on the remote side as they
3820 : : * do locally --- they need only have the same names.
3821 : : */
3822 : : static void
3651 tgl@sss.pgh.pa.us 3823 : 29 : printRemoteParam(int paramindex, Oid paramtype, int32 paramtypmod,
3824 : : deparse_expr_cxt *context)
3825 : : {
3826 : 29 : StringInfo buf = context->buf;
3085 3827 : 29 : char *ptypename = deparse_type_name(paramtype, paramtypmod);
3828 : :
3651 3829 : 29 : appendStringInfo(buf, "$%d::%s", paramindex, ptypename);
3830 : 29 : }
3831 : :
3832 : : /*
3833 : : * Print the representation of a placeholder for a parameter that will be
3834 : : * sent to the remote side at execution time.
3835 : : *
3836 : : * This is used when we're just trying to EXPLAIN the remote query.
3837 : : * We don't have the actual value of the runtime parameter yet, and we don't
3838 : : * want the remote planner to generate a plan that depends on such a value
3839 : : * anyway. Thus, we can't do something simple like "$1::paramtype".
3840 : : * Instead, we emit "((SELECT null::paramtype)::paramtype)".
3841 : : * In all extant versions of Postgres, the planner will see that as an unknown
3842 : : * constant value, which is what we want. This might need adjustment if we
3843 : : * ever make the planner flatten scalar subqueries. Note: the reason for the
3844 : : * apparently useless outer cast is to ensure that the representation as a
3845 : : * whole will be parsed as an a_expr and not a select_with_parens; the latter
3846 : : * would do the wrong thing in the context "x = ANY(...)".
3847 : : */
3848 : : static void
3849 : 204 : printRemotePlaceholder(Oid paramtype, int32 paramtypmod,
3850 : : deparse_expr_cxt *context)
3851 : : {
3852 : 204 : StringInfo buf = context->buf;
3085 3853 : 204 : char *ptypename = deparse_type_name(paramtype, paramtypmod);
3854 : :
3651 3855 : 204 : appendStringInfo(buf, "((SELECT null::%s)::%s)", ptypename, ptypename);
3856 : 204 : }
3857 : :
3858 : : /*
3859 : : * Deparse GROUP BY clause.
3860 : : */
3861 : : static void
2732 rhaas@postgresql.org 3862 : 161 : appendGroupByClause(List *tlist, deparse_expr_cxt *context)
3863 : : {
3864 : 161 : StringInfo buf = context->buf;
3865 : 161 : Query *query = context->root->parse;
3866 : : ListCell *lc;
3867 : 161 : bool first = true;
3868 : :
3869 : : /* Nothing to be done, if there's no GROUP BY clause in the query. */
3870 [ + + ]: 161 : if (!query->groupClause)
3871 : 60 : return;
3872 : :
2434 peter_e@gmx.net 3873 : 101 : appendStringInfoString(buf, " GROUP BY ");
3874 : :
3875 : : /*
3876 : : * Queries with grouping sets are not pushed down, so we don't expect
3877 : : * grouping sets here.
3878 : : */
2732 rhaas@postgresql.org 3879 [ - + ]: 101 : Assert(!query->groupingSets);
3880 : :
3881 : : /*
3882 : : * We intentionally print query->groupClause not processed_groupClause,
3883 : : * leaving it to the remote planner to get rid of any redundant GROUP BY
3884 : : * items again. This is necessary in case processed_groupClause reduced
3885 : : * to empty, and in any case the redundancy situation on the remote might
3886 : : * be different than what we think here.
3887 : : */
3888 [ + - + + : 214 : foreach(lc, query->groupClause)
+ + ]
3889 : : {
3890 : 113 : SortGroupClause *grp = (SortGroupClause *) lfirst(lc);
3891 : :
3892 [ + + ]: 113 : if (!first)
3893 : 12 : appendStringInfoString(buf, ", ");
3894 : 113 : first = false;
3895 : :
2284 tgl@sss.pgh.pa.us 3896 : 113 : deparseSortGroupClause(grp->tleSortGroupRef, tlist, true, context);
3897 : : }
3898 : : }
3899 : :
3900 : : /*
3901 : : * Deparse ORDER BY clause defined by the given pathkeys.
3902 : : *
3903 : : * The clause should use Vars from context->scanrel if !has_final_sort,
3904 : : * or from context->foreignrel's targetlist if has_final_sort.
3905 : : *
3906 : : * We find a suitable pathkey expression (some earlier step
3907 : : * should have verified that there is one) and deparse it.
3908 : : */
3909 : : static void
1839 efujita@postgresql.o 3910 : 713 : appendOrderByClause(List *pathkeys, bool has_final_sort,
3911 : : deparse_expr_cxt *context)
3912 : : {
3913 : : ListCell *lcell;
3914 : : int nestlevel;
2997 rhaas@postgresql.org 3915 : 713 : StringInfo buf = context->buf;
34 drowley@postgresql.o 3916 : 713 : bool gotone = false;
3917 : :
3918 : : /* Make sure any constants in the exprs are printed portably */
3085 rhaas@postgresql.org 3919 : 713 : nestlevel = set_transmission_modes();
3920 : :
3921 [ + - + + : 1559 : foreach(lcell, pathkeys)
+ + ]
3922 : : {
tgl@sss.pgh.pa.us 3923 : 846 : PathKey *pathkey = lfirst(lcell);
3924 : : EquivalenceMember *em;
3925 : : Expr *em_expr;
3926 : : Oid oprid;
3927 : :
1839 efujita@postgresql.o 3928 [ + + ]: 846 : if (has_final_sort)
3929 : : {
3930 : : /*
3931 : : * By construction, context->foreignrel is the input relation to
3932 : : * the final sort.
3933 : : */
745 tgl@sss.pgh.pa.us 3934 : 203 : em = find_em_for_rel_target(context->root,
3935 : : pathkey->pk_eclass,
3936 : : context->foreignrel);
3937 : : }
3938 : : else
3939 : 643 : em = find_em_for_rel(context->root,
3940 : : pathkey->pk_eclass,
3941 : : context->scanrel);
3942 : :
3943 : : /*
3944 : : * We don't expect any error here; it would mean that shippability
3945 : : * wasn't verified earlier. For the same reason, we don't recheck
3946 : : * shippability of the sort operator.
3947 : : */
3948 [ - + ]: 846 : if (em == NULL)
745 tgl@sss.pgh.pa.us 3949 [ # # ]:UBC 0 : elog(ERROR, "could not find pathkey item to sort");
3950 : :
745 tgl@sss.pgh.pa.us 3951 :CBC 846 : em_expr = em->em_expr;
3952 : :
3953 : : /*
3954 : : * If the member is a Const expression then we needn't add it to the
3955 : : * ORDER BY clause. This can happen in UNION ALL queries where the
3956 : : * union child targetlist has a Const. Adding these would be
3957 : : * wasteful, but also, for INT columns, an integer literal would be
3958 : : * seen as an ordinal column position rather than a value to sort by.
3959 : : * deparseConst() does have code to handle this, but it seems less
3960 : : * effort on all accounts just to skip these for ORDER BY clauses.
3961 : : */
34 drowley@postgresql.o 3962 [ + + ]: 846 : if (IsA(em_expr, Const))
3963 : 6 : continue;
3964 : :
3965 [ + + ]: 840 : if (!gotone)
3966 : : {
3967 : 710 : appendStringInfoString(buf, " ORDER BY ");
3968 : 710 : gotone = true;
3969 : : }
3970 : : else
3971 : 130 : appendStringInfoString(buf, ", ");
3972 : :
3973 : : /*
3974 : : * Lookup the operator corresponding to the strategy in the opclass.
3975 : : * The datatype used by the opfamily is not necessarily the same as
3976 : : * the expression type (for array types for example).
3977 : : */
745 tgl@sss.pgh.pa.us 3978 : 840 : oprid = get_opfamily_member(pathkey->pk_opfamily,
3979 : : em->em_datatype,
3980 : : em->em_datatype,
3981 : 840 : pathkey->pk_strategy);
3982 [ - + ]: 840 : if (!OidIsValid(oprid))
745 tgl@sss.pgh.pa.us 3983 [ # # ]:UBC 0 : elog(ERROR, "missing operator %d(%u,%u) in opfamily %u",
3984 : : pathkey->pk_strategy, em->em_datatype, em->em_datatype,
3985 : : pathkey->pk_opfamily);
3986 : :
2997 rhaas@postgresql.org 3987 :CBC 840 : deparseExpr(em_expr, context);
3988 : :
3989 : : /*
3990 : : * Here we need to use the expression's actual type to discover
3991 : : * whether the desired operator will be the default or not.
3992 : : */
745 tgl@sss.pgh.pa.us 3993 : 840 : appendOrderBySuffix(oprid, exprType((Node *) em_expr),
3994 : 840 : pathkey->pk_nulls_first, context);
3995 : :
3996 : : }
3085 rhaas@postgresql.org 3997 : 713 : reset_transmission_modes(nestlevel);
3998 : 713 : }
3999 : :
4000 : : /*
4001 : : * Deparse LIMIT/OFFSET clause.
4002 : : */
4003 : : static void
1839 efujita@postgresql.o 4004 : 141 : appendLimitClause(deparse_expr_cxt *context)
4005 : : {
4006 : 141 : PlannerInfo *root = context->root;
4007 : 141 : StringInfo buf = context->buf;
4008 : : int nestlevel;
4009 : :
4010 : : /* Make sure any constants in the exprs are printed portably */
4011 : 141 : nestlevel = set_transmission_modes();
4012 : :
4013 [ + - ]: 141 : if (root->parse->limitCount)
4014 : : {
4015 : 141 : appendStringInfoString(buf, " LIMIT ");
4016 : 141 : deparseExpr((Expr *) root->parse->limitCount, context);
4017 : : }
4018 [ + + ]: 141 : if (root->parse->limitOffset)
4019 : : {
4020 : 75 : appendStringInfoString(buf, " OFFSET ");
4021 : 75 : deparseExpr((Expr *) root->parse->limitOffset, context);
4022 : : }
4023 : :
4024 : 141 : reset_transmission_modes(nestlevel);
4025 : 141 : }
4026 : :
4027 : : /*
4028 : : * appendFunctionName
4029 : : * Deparses function name from given function oid.
4030 : : */
4031 : : static void
2732 rhaas@postgresql.org 4032 : 295 : appendFunctionName(Oid funcid, deparse_expr_cxt *context)
4033 : : {
4034 : 295 : StringInfo buf = context->buf;
4035 : : HeapTuple proctup;
4036 : : Form_pg_proc procform;
4037 : : const char *proname;
4038 : :
4039 : 295 : proctup = SearchSysCache1(PROCOID, ObjectIdGetDatum(funcid));
4040 [ - + ]: 295 : if (!HeapTupleIsValid(proctup))
2732 rhaas@postgresql.org 4041 [ # # ]:UBC 0 : elog(ERROR, "cache lookup failed for function %u", funcid);
2732 rhaas@postgresql.org 4042 :CBC 295 : procform = (Form_pg_proc) GETSTRUCT(proctup);
4043 : :
4044 : : /* Print schema name only if it's not pg_catalog */
4045 [ + + ]: 295 : if (procform->pronamespace != PG_CATALOG_NAMESPACE)
4046 : : {
4047 : : const char *schemaname;
4048 : :
4049 : 6 : schemaname = get_namespace_name(procform->pronamespace);
4050 : 6 : appendStringInfo(buf, "%s.", quote_identifier(schemaname));
4051 : : }
4052 : :
4053 : : /* Always print the function name */
4054 : 295 : proname = NameStr(procform->proname);
2434 peter_e@gmx.net 4055 : 295 : appendStringInfoString(buf, quote_identifier(proname));
4056 : :
2732 rhaas@postgresql.org 4057 : 295 : ReleaseSysCache(proctup);
4058 : 295 : }
4059 : :
4060 : : /*
4061 : : * Appends a sort or group clause.
4062 : : *
4063 : : * Like get_rule_sortgroupclause(), returns the expression tree, so caller
4064 : : * need not find it again.
4065 : : */
4066 : : static Node *
2284 tgl@sss.pgh.pa.us 4067 : 145 : deparseSortGroupClause(Index ref, List *tlist, bool force_colno,
4068 : : deparse_expr_cxt *context)
4069 : : {
2732 rhaas@postgresql.org 4070 : 145 : StringInfo buf = context->buf;
4071 : : TargetEntry *tle;
4072 : : Expr *expr;
4073 : :
4074 : 145 : tle = get_sortgroupref_tle(ref, tlist);
4075 : 145 : expr = tle->expr;
4076 : :
2284 tgl@sss.pgh.pa.us 4077 [ + + ]: 145 : if (force_colno)
4078 : : {
4079 : : /* Use column-number form when requested by caller. */
4080 [ - + ]: 113 : Assert(!tle->resjunk);
4081 : 113 : appendStringInfo(buf, "%d", tle->resno);
4082 : : }
4083 [ + - - + ]: 32 : else if (expr && IsA(expr, Const))
4084 : : {
4085 : : /*
4086 : : * Force a typecast here so that we don't emit something like "GROUP
4087 : : * BY 2", which will be misconstrued as a column position rather than
4088 : : * a constant.
4089 : : */
2732 rhaas@postgresql.org 4090 :UBC 0 : deparseConst((Const *) expr, context, 1);
4091 : : }
2732 rhaas@postgresql.org 4092 [ + - + + ]:CBC 32 : else if (!expr || IsA(expr, Var))
4093 : 18 : deparseExpr(expr, context);
4094 : : else
4095 : : {
4096 : : /* Always parenthesize the expression. */
2434 peter_e@gmx.net 4097 : 14 : appendStringInfoChar(buf, '(');
2732 rhaas@postgresql.org 4098 : 14 : deparseExpr(expr, context);
2434 peter_e@gmx.net 4099 : 14 : appendStringInfoChar(buf, ')');
4100 : : }
4101 : :
2732 rhaas@postgresql.org 4102 : 145 : return (Node *) expr;
4103 : : }
4104 : :
4105 : :
4106 : : /*
4107 : : * Returns true if given Var is deparsed as a subquery output column, in
4108 : : * which case, *relno and *colno are set to the IDs for the relation and
4109 : : * column alias to the Var provided by the subquery.
4110 : : */
4111 : : static bool
2586 4112 : 8183 : is_subquery_var(Var *node, RelOptInfo *foreignrel, int *relno, int *colno)
4113 : : {
4114 : 8183 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
4115 : 8183 : RelOptInfo *outerrel = fpinfo->outerrel;
4116 : 8183 : RelOptInfo *innerrel = fpinfo->innerrel;
4117 : :
4118 : : /* Should only be called in these cases. */
2568 4119 [ + + + + : 8183 : Assert(IS_SIMPLE_REL(foreignrel) || IS_JOIN_REL(foreignrel));
+ + - + ]
4120 : :
4121 : : /*
4122 : : * If the given relation isn't a join relation, it doesn't have any lower
4123 : : * subqueries, so the Var isn't a subquery output column.
4124 : : */
4125 [ + + + + ]: 8183 : if (!IS_JOIN_REL(foreignrel))
2586 4126 : 1981 : return false;
4127 : :
4128 : : /*
4129 : : * If the Var doesn't belong to any lower subqueries, it isn't a subquery
4130 : : * output column.
4131 : : */
4132 [ + + ]: 6202 : if (!bms_is_member(node->varno, fpinfo->lower_subquery_rels))
4133 : 6086 : return false;
4134 : :
4135 [ + + ]: 116 : if (bms_is_member(node->varno, outerrel->relids))
4136 : : {
4137 : : /*
4138 : : * If outer relation is deparsed as a subquery, the Var is an output
4139 : : * column of the subquery; get the IDs for the relation/column alias.
4140 : : */
4141 [ + - ]: 42 : if (fpinfo->make_outerrel_subquery)
4142 : : {
4143 : 42 : get_relation_column_alias_ids(node, outerrel, relno, colno);
4144 : 42 : return true;
4145 : : }
4146 : :
4147 : : /* Otherwise, recurse into the outer relation. */
2586 rhaas@postgresql.org 4148 :UBC 0 : return is_subquery_var(node, outerrel, relno, colno);
4149 : : }
4150 : : else
4151 : : {
2586 rhaas@postgresql.org 4152 [ - + ]:CBC 74 : Assert(bms_is_member(node->varno, innerrel->relids));
4153 : :
4154 : : /*
4155 : : * If inner relation is deparsed as a subquery, the Var is an output
4156 : : * column of the subquery; get the IDs for the relation/column alias.
4157 : : */
4158 [ + - ]: 74 : if (fpinfo->make_innerrel_subquery)
4159 : : {
4160 : 74 : get_relation_column_alias_ids(node, innerrel, relno, colno);
4161 : 74 : return true;
4162 : : }
4163 : :
4164 : : /* Otherwise, recurse into the inner relation. */
2586 rhaas@postgresql.org 4165 :UBC 0 : return is_subquery_var(node, innerrel, relno, colno);
4166 : : }
4167 : : }
4168 : :
4169 : : /*
4170 : : * Get the IDs for the relation and column alias to given Var belonging to
4171 : : * given relation, which are returned into *relno and *colno.
4172 : : */
4173 : : static void
2586 rhaas@postgresql.org 4174 :CBC 116 : get_relation_column_alias_ids(Var *node, RelOptInfo *foreignrel,
4175 : : int *relno, int *colno)
4176 : : {
4177 : 116 : PgFdwRelationInfo *fpinfo = (PgFdwRelationInfo *) foreignrel->fdw_private;
4178 : : int i;
4179 : : ListCell *lc;
4180 : :
4181 : : /* Get the relation alias ID */
4182 : 116 : *relno = fpinfo->relation_index;
4183 : :
4184 : : /* Get the column alias ID */
4185 : 116 : i = 1;
4186 [ + - + - : 152 : foreach(lc, foreignrel->reltarget->exprs)
+ - ]
4187 : : {
440 tgl@sss.pgh.pa.us 4188 : 152 : Var *tlvar = (Var *) lfirst(lc);
4189 : :
4190 : : /*
4191 : : * Match reltarget entries only on varno/varattno. Ideally there
4192 : : * would be some cross-check on varnullingrels, but it's unclear what
4193 : : * to do exactly; we don't have enough context to know what that value
4194 : : * should be.
4195 : : */
4196 [ + - ]: 152 : if (IsA(tlvar, Var) &&
4197 [ + + ]: 152 : tlvar->varno == node->varno &&
4198 [ + + ]: 144 : tlvar->varattno == node->varattno)
4199 : : {
2586 rhaas@postgresql.org 4200 : 116 : *colno = i;
4201 : 116 : return;
4202 : : }
4203 : 36 : i++;
4204 : : }
4205 : :
4206 : : /* Shouldn't get here */
2586 rhaas@postgresql.org 4207 [ # # ]:UBC 0 : elog(ERROR, "unexpected expression in subquery output");
4208 : : }
|
