mirror of
https://github.com/postgres/postgres.git
synced 2026-03-12 21:54:00 -04:00
363a6e8c6f
The callback functions ReplaceVarsFromTargetList_callback and pullup_replace_vars_callback are both used to replace Vars in an expression tree that reference a particular RTE with items from a targetlist, and they both need to expand whole-tuple references and deal with OLD/NEW RETURNING list Vars. As a result, currently there is significant code duplication between these two functions. This patch introduces a new function, ReplaceVarFromTargetList, to perform the replacement and calls it from both callback functions, thereby eliminating code duplication. Author: Dean Rasheed <dean.a.rasheed@gmail.com> Author: Richard Guo <guofenglinux@gmail.com> Reviewed-by: Jian He <jian.universality@gmail.com> Discussion: https://postgr.es/m/CAEZATCWhr=FM4X5kCPvVs-g2XEk+ceLsNtBK_zZMkqFn9vUjsw@mail.gmail.com
107 lines
3.9 KiB
C
107 lines
3.9 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* rewriteManip.h
|
|
* Querytree manipulation subroutines for query rewriter.
|
|
*
|
|
*
|
|
* Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
|
|
* Portions Copyright (c) 1994, Regents of the University of California
|
|
*
|
|
* src/include/rewrite/rewriteManip.h
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#ifndef REWRITEMANIP_H
|
|
#define REWRITEMANIP_H
|
|
|
|
#include "nodes/parsenodes.h"
|
|
#include "nodes/pathnodes.h"
|
|
|
|
struct AttrMap; /* avoid including attmap.h here */
|
|
|
|
|
|
typedef struct replace_rte_variables_context replace_rte_variables_context;
|
|
|
|
typedef Node *(*replace_rte_variables_callback) (Var *var,
|
|
replace_rte_variables_context *context);
|
|
|
|
struct replace_rte_variables_context
|
|
{
|
|
replace_rte_variables_callback callback; /* callback function */
|
|
void *callback_arg; /* context data for callback function */
|
|
int target_varno; /* RTE index to search for */
|
|
int sublevels_up; /* (current) nesting depth */
|
|
bool inserted_sublink; /* have we inserted a SubLink? */
|
|
};
|
|
|
|
typedef enum ReplaceVarsNoMatchOption
|
|
{
|
|
REPLACEVARS_REPORT_ERROR, /* throw error if no match */
|
|
REPLACEVARS_CHANGE_VARNO, /* change the Var's varno, nothing else */
|
|
REPLACEVARS_SUBSTITUTE_NULL, /* replace with a NULL Const */
|
|
} ReplaceVarsNoMatchOption;
|
|
|
|
|
|
extern Relids adjust_relid_set(Relids relids, int oldrelid, int newrelid);
|
|
extern void CombineRangeTables(List **dst_rtable, List **dst_perminfos,
|
|
List *src_rtable, List *src_perminfos);
|
|
extern void OffsetVarNodes(Node *node, int offset, int sublevels_up);
|
|
extern void ChangeVarNodes(Node *node, int rt_index, int new_index,
|
|
int sublevels_up);
|
|
extern void ChangeVarNodesExtended(Node *node, int rt_index, int new_index,
|
|
int sublevels_up, bool change_RangeTblRef);
|
|
extern void IncrementVarSublevelsUp(Node *node, int delta_sublevels_up,
|
|
int min_sublevels_up);
|
|
extern void IncrementVarSublevelsUp_rtable(List *rtable,
|
|
int delta_sublevels_up, int min_sublevels_up);
|
|
|
|
extern bool rangeTableEntry_used(Node *node, int rt_index,
|
|
int sublevels_up);
|
|
|
|
extern Query *getInsertSelectQuery(Query *parsetree, Query ***subquery_ptr);
|
|
|
|
extern void AddQual(Query *parsetree, Node *qual);
|
|
extern void AddInvertedQual(Query *parsetree, Node *qual);
|
|
|
|
extern bool contain_aggs_of_level(Node *node, int levelsup);
|
|
extern int locate_agg_of_level(Node *node, int levelsup);
|
|
extern bool contain_windowfuncs(Node *node);
|
|
extern int locate_windowfunc(Node *node);
|
|
extern bool checkExprHasSubLink(Node *node);
|
|
|
|
extern Node *add_nulling_relids(Node *node,
|
|
const Bitmapset *target_relids,
|
|
const Bitmapset *added_relids);
|
|
extern Node *remove_nulling_relids(Node *node,
|
|
const Bitmapset *removable_relids,
|
|
const Bitmapset *except_relids);
|
|
|
|
extern Node *replace_rte_variables(Node *node,
|
|
int target_varno, int sublevels_up,
|
|
replace_rte_variables_callback callback,
|
|
void *callback_arg,
|
|
bool *outer_hasSubLinks);
|
|
extern Node *replace_rte_variables_mutator(Node *node,
|
|
replace_rte_variables_context *context);
|
|
|
|
extern Node *map_variable_attnos(Node *node,
|
|
int target_varno, int sublevels_up,
|
|
const struct AttrMap *attno_map,
|
|
Oid to_rowtype, bool *found_whole_row);
|
|
|
|
extern Node *ReplaceVarFromTargetList(Var *var,
|
|
RangeTblEntry *target_rte,
|
|
List *targetlist,
|
|
int result_relation,
|
|
ReplaceVarsNoMatchOption nomatch_option,
|
|
int nomatch_varno);
|
|
extern Node *ReplaceVarsFromTargetList(Node *node,
|
|
int target_varno, int sublevels_up,
|
|
RangeTblEntry *target_rte,
|
|
List *targetlist,
|
|
int result_relation,
|
|
ReplaceVarsNoMatchOption nomatch_option,
|
|
int nomatch_varno,
|
|
bool *outer_hasSubLinks);
|
|
|
|
#endif /* REWRITEMANIP_H */
|