166 bool is_borrowed_rel);
196 int n_total_subplans);
295 while (dispatch != NULL)
320 if (partdesc->
nparts == 0 ||
329 (
errcode(ERRCODE_CHECK_VIOLATION),
330 errmsg(
"no partition of relation \"%s\" found for row",
333 errdetail(
"Partition key of the failing row contains %s.",
338 is_leaf = partdesc->
is_leaf[partidx];
359 partdesc->
oids[partidx],
378 rootResultRelInfo, partidx);
402 dispatch = pd[dispatch->
indexes[partidx]];
415 partdesc->
oids[partidx],
422 dispatch = subdispatch;
437 if (tempslot != NULL)
512 bool found_whole_row;
538 if (partrel->rd_rel->relhasindex &&
539 leaf_part_rri->ri_IndexRelationDescs == NULL)
604 wcoExprs =
lappend(wcoExprs, wcoExpr);
607 leaf_part_rri->ri_WithCheckOptions = wcoList;
608 leaf_part_rri->ri_WithCheckOptionExprs = wcoExprs;
646 if (part_attmap == NULL)
651 returningList = (
List *)
659 leaf_part_rri->ri_returningList = returningList;
671 leaf_part_rri->ri_projectReturning =
678 leaf_part_rri, partidx,
false);
702 foreach(lc, childIdxs)
712 arbiterIndexes =
lappend_oid(arbiterIndexes, childIdx);
725 elog(
ERROR,
"invalid arbiter index list");
726 leaf_part_rri->ri_onConflictArbiterIndexes = arbiterIndexes;
741 leaf_part_rri->ri_onConflict = onconfl;
788 if (part_attmap == NULL)
793 onconflset = (
List *)
800 onconflset = (
List *)
888 if (part_attmap == NULL)
894 if (
unlikely(!leaf_part_rri->ri_projectNewInfoValid))
905 leaf_part_rri->ri_MergeJoinCondition =
908 foreach(lc, firstMergeActionList)
919 leaf_part_rri->ri_MergeActions[
action->matchKind] =
923 switch (
action->commandType)
934 leaf_part_rri->ri_newTupleSlot,
954 leaf_part_rri->ri_newTupleSlot,
961 elog(
ERROR,
"unknown action in MERGE WHEN clause");
977 return leaf_part_rri;
993 bool is_borrowed_rel)
1080 dispatch->
indexes[partidx] = rri_index;
1133 partdesc->
nparts *
sizeof(int));
1138 if (parent_pd != NULL)
1219 parent_pd->
indexes[partidx] = dispatchidx;
1329 if (partexpr_item == NULL)
1330 elog(
ERROR,
"wrong number of partition key expressions");
1340 if (partexpr_item != NULL)
1341 elog(
ERROR,
"wrong number of partition key expressions");
1349#define PARTITION_CACHED_FIND_THRESHOLD 16
1394 int bound_offset = -1;
1395 int part_index = -1;
1413 switch (
key->strategy)
1458 Datum lastDatum = boundinfo->
datums[last_datum_offset][0];
1463 key->partcollation[0],
1468 return boundinfo->
indexes[last_datum_offset];
1477 if (bound_offset >= 0 &&
equal)
1478 part_index = boundinfo->
indexes[bound_offset];
1485 range_partkey_has_null =
false;
1492 for (
i = 0;
i <
key->partnatts;
i++)
1496 range_partkey_has_null =
true;
1502 if (range_partkey_has_null)
1508 Datum *lastDatums = boundinfo->
datums[last_datum_offset];
1525 return boundinfo->
indexes[last_datum_offset + 1];
1527 if (cmpval < 0 && last_datum_offset + 1 < boundinfo->
ndatums)
1530 lastDatums = boundinfo->
datums[last_datum_offset + 1];
1531 kind = boundinfo->
kind[last_datum_offset + 1];
1540 return boundinfo->
indexes[last_datum_offset + 1];
1558 part_index = boundinfo->
indexes[bound_offset + 1];
1563 elog(
ERROR,
"unexpected partition strategy: %d",
1564 (
int)
key->strategy);
1582 Assert(bound_offset >= 0);
1635 for (
i = 0;
i < partnatts;
i++)
1655 for (
i = 0;
i < partnatts;
i++)
1668 &foutoid, &typisvarlena);
1676 vallen = strlen(
val);
1677 if (vallen <= maxfieldlen)
1728 if (parentattrno <= 0 ||
1729 parentattrno > attrMap->
maplen ||
1730 attrMap->
attnums[parentattrno - 1] == 0)
1731 elog(
ERROR,
"unexpected attno %d in target column list",
1734 attrMap->
attnums[parentattrno - 1]);
1830 &all_leafpart_rtis);
1840 &validsubplan_rtis);
1842 validsubplan_rtis = all_leafpart_rtis;
1873 int n_total_subplans,
1874 int part_prune_index,
1888 elog(
ERROR,
"wrong pruneinfo with relids=%s found at part_prune_index=%d contained in plan node with relids=%s",
1898 Assert(prunestate != NULL);
1908 Assert(n_total_subplans > 0);
1910 n_total_subplans - 1);
1927 *initially_valid_subplans,
1970 int n_part_hierarchies;
1986 Assert(n_part_hierarchies > 0);
2019 int npartrelpruneinfos =
list_length(partrelpruneinfos);
2031 foreach(lc2, partrelpruneinfos)
2076 memcmp(partdesc->
oids, pinfo->relid_map,
2077 sizeof(
int) * partdesc->
nparts) == 0)
2082 sizeof(
int) * pinfo->
nparts);
2103 for (pp_idx = 0; pp_idx < partdesc->
nparts; pp_idx++)
2106 while (pd_idx < pinfo->nparts &&
2111 if (pd_idx < pinfo->nparts &&
2112 pinfo->relid_map[pd_idx] == partdesc->
oids[pp_idx])
2116 pinfo->subplan_map[pd_idx];
2118 pinfo->subpart_map[pd_idx];
2120 pinfo->leafpart_rti_map[pd_idx];
2145 for (
int pd_idx2 = pd_idx + 1; pd_idx2 < pinfo->
nparts; pd_idx2++)
2147 if (pd_idx2 >= pinfo->
nparts)
2149 if (pinfo->relid_map[pd_idx2] == partdesc->
oids[pp_idx])
2180 partdesc, partkey, NULL,
2206 int part_index = -1;
2232 List *pruning_steps,
2262 foreach(lc, pruning_steps)
2274 for (keyno = 0; keyno < partnatts; keyno++)
2297 if (planstate == NULL)
2336 int n_total_subplans)
2339 int *new_subplan_indexes = NULL;
2343 bool fix_subplan_map =
false;
2346 Assert(parent_plan != NULL);
2347 estate = parent_plan->
state;
2355 fix_subplan_map =
true;
2362 new_subplan_indexes = (
int *)
palloc0(
sizeof(
int) * n_total_subplans);
2368 new_subplan_indexes[
i] = newidx++;
2391 int nparts = pprune->
nparts;
2410 partdesc, partkey, parent_plan,
2414 if (!fix_subplan_map)
2421 for (k = 0; k < nparts; k++)
2436 Assert(oldidx < n_total_subplans);
2437 pprune->
subplan_map[k] = new_subplan_indexes[oldidx] - 1;
2439 if (new_subplan_indexes[oldidx] > 0)
2461 if (fix_subplan_map)
2463 new_other_subplans = NULL;
2467 new_subplan_indexes[
i] - 1);
2472 pfree(new_subplan_indexes);
2504 Assert(validsubplan_rtis != NULL || !initial_prune);
2528 &result, validsubplan_rtis);
2546 if (validsubplan_rtis)
2547 *validsubplan_rtis =
bms_copy(*validsubplan_rtis);
2597 if (validsubplan_rtis)
2608 initial_prune, validsubplans,
AclResult pg_attribute_aclcheck(Oid table_oid, AttrNumber attnum, Oid roleid, AclMode mode)
AclResult pg_class_aclcheck(Oid table_oid, Oid roleid, AclMode mode)
AttrMap * build_attrmap_by_name(TupleDesc indesc, TupleDesc outdesc, bool missing_ok)
AttrMap * build_attrmap_by_name_if_req(TupleDesc indesc, TupleDesc outdesc, bool missing_ok)
#define InvalidAttrNumber
bool bms_equal(const Bitmapset *a, const Bitmapset *b)
int bms_next_member(const Bitmapset *a, int prevbit)
Bitmapset * bms_add_range(Bitmapset *a, int lower, int upper)
void bms_free(Bitmapset *a)
int bms_num_members(const Bitmapset *a)
bool bms_is_member(int x, const Bitmapset *a)
Bitmapset * bms_add_member(Bitmapset *a, int x)
Bitmapset * bms_add_members(Bitmapset *a, const Bitmapset *b)
Bitmapset * bms_copy(const Bitmapset *a)
static Datum values[MAXATTR]
#define Assert(condition)
#define FLEXIBLE_ARRAY_MEMBER
#define OidIsValid(objectId)
int errdetail(const char *fmt,...)
int errcode(int sqlerrcode)
int errmsg(const char *fmt,...)
#define ereport(elevel,...)
bool equal(const void *a, const void *b)
ExprState * ExecInitExpr(Expr *node, PlanState *parent)
ProjectionInfo * ExecBuildProjectionInfo(List *targetList, ExprContext *econtext, TupleTableSlot *slot, PlanState *parent, TupleDesc inputDesc)
ExprState * ExecInitQual(List *qual, PlanState *parent)
ExprState * ExecInitExprWithParams(Expr *node, ParamListInfo ext_params)
ProjectionInfo * ExecBuildUpdateProjection(List *targetList, bool evalTargetList, List *targetColnos, TupleDesc relDesc, ExprContext *econtext, TupleTableSlot *slot, PlanState *parent)
List * ExecPrepareExprList(List *nodes, EState *estate)
void ExecCloseIndices(ResultRelInfo *resultRelInfo)
void ExecOpenIndices(ResultRelInfo *resultRelInfo, bool speculative)
void CheckValidResultRel(ResultRelInfo *resultRelInfo, CmdType operation, List *mergeActions)
bool ExecPartitionCheck(ResultRelInfo *resultRelInfo, TupleTableSlot *slot, EState *estate, bool emitError)
void InitResultRelInfo(ResultRelInfo *resultRelInfo, Relation resultRelationDesc, Index resultRelationIndex, ResultRelInfo *partition_root_rri, int instrument_options)
static PartitionDispatch ExecInitPartitionDispatchInfo(EState *estate, PartitionTupleRouting *proute, Oid partoid, PartitionDispatch parent_pd, int partidx, ResultRelInfo *rootResultRelInfo)
void ExecDoInitialPruning(EState *estate)
static ResultRelInfo * ExecInitPartitionInfo(ModifyTableState *mtstate, EState *estate, PartitionTupleRouting *proute, PartitionDispatch dispatch, ResultRelInfo *rootResultRelInfo, int partidx)
PartitionPruneState * ExecInitPartitionExecPruning(PlanState *planstate, int n_total_subplans, int part_prune_index, Bitmapset *relids, Bitmapset **initially_valid_subplans)
static void InitExecPartitionPruneContexts(PartitionPruneState *prunstate, PlanState *parent_plan, Bitmapset *initially_valid_subplans, int n_total_subplans)
Bitmapset * ExecFindMatchingSubPlans(PartitionPruneState *prunestate, bool initial_prune, Bitmapset **validsubplan_rtis)
static void ExecInitRoutingInfo(ModifyTableState *mtstate, EState *estate, PartitionTupleRouting *proute, PartitionDispatch dispatch, ResultRelInfo *partRelInfo, int partidx, bool is_borrowed_rel)
static char * ExecBuildSlotPartitionKeyDescription(Relation rel, Datum *values, bool *isnull, int maxfieldlen)
static void FormPartitionKeyDatum(PartitionDispatch pd, TupleTableSlot *slot, EState *estate, Datum *values, bool *isnull)
#define PARTITION_CACHED_FIND_THRESHOLD
PartitionTupleRouting * ExecSetupPartitionTupleRouting(EState *estate, Relation rel)
static List * adjust_partition_colnos(List *colnos, ResultRelInfo *leaf_part_rri)
static List * adjust_partition_colnos_using_map(List *colnos, AttrMap *attrMap)
ResultRelInfo * ExecFindPartition(ModifyTableState *mtstate, ResultRelInfo *rootResultRelInfo, PartitionTupleRouting *proute, TupleTableSlot *slot, EState *estate)
static void InitPartitionPruneContext(PartitionPruneContext *context, List *pruning_steps, PartitionDesc partdesc, PartitionKey partkey, PlanState *planstate, ExprContext *econtext)
struct PartitionDispatchData PartitionDispatchData
static int get_partition_for_tuple(PartitionDispatch pd, Datum *values, bool *isnull)
static void find_matching_subplans_recurse(PartitionPruningData *prunedata, PartitionedRelPruningData *pprune, bool initial_prune, Bitmapset **validsubplans, Bitmapset **validsubplan_rtis)
static PartitionPruneState * CreatePartitionPruneState(EState *estate, PartitionPruneInfo *pruneinfo, Bitmapset **all_leafpart_rtis)
void ExecCleanupTupleRouting(ModifyTableState *mtstate, PartitionTupleRouting *proute)
struct PartitionDispatchData * PartitionDispatch
struct PartitionedRelPruningData PartitionedRelPruningData
TupleTableSlot * MakeSingleTupleTableSlot(TupleDesc tupdesc, const TupleTableSlotOps *tts_ops)
const TupleTableSlotOps TTSOpsVirtual
void ExecDropSingleTupleTableSlot(TupleTableSlot *slot)
TupleConversionMap * ExecGetRootToChildMap(ResultRelInfo *resultRelInfo, EState *estate)
ExprContext * CreateExprContext(EState *estate)
TupleConversionMap * ExecGetChildToRootMap(ResultRelInfo *resultRelInfo)
Relation ExecGetRangeTableRelation(EState *estate, Index rti)
#define GetPerTupleExprContext(estate)
#define EXEC_FLAG_EXPLAIN_GENERIC
#define ResetExprContext(econtext)
#define GetPerTupleMemoryContext(estate)
static Datum ExecEvalExprSwitchContext(ExprState *state, ExprContext *econtext, bool *isNull)
Datum FunctionCall2Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2)
char * OidOutputFunctionCall(Oid functionId, Datum val)
List * lappend(List *list, void *datum)
List * lappend_int(List *list, int datum)
List * lappend_oid(List *list, Oid datum)
void list_free(List *list)
bool list_member_oid(const List *list, Oid datum)
void getTypeOutputInfo(Oid type, Oid *typOutput, bool *typIsVarlena)
int pg_mbcliplen(const char *mbstr, int len, int limit)
void MemoryContextReset(MemoryContext context)
void * repalloc(void *pointer, Size size)
void pfree(void *pointer)
void * palloc0(Size size)
MemoryContext CurrentMemoryContext
#define AllocSetContextCreate
#define ALLOCSET_DEFAULT_SIZES
#define CHECK_FOR_INTERRUPTS()
ResultRelInfo * ExecLookupResultRelByOid(ModifyTableState *node, Oid resultoid, bool missing_ok, bool update_cache)
void ExecInitMergeTupleSlots(ModifyTableState *mtstate, ResultRelInfo *resultRelInfo)
#define IsA(nodeptr, _type_)
#define castNode(_type_, nodeptr)
char * bmsToString(const Bitmapset *bms)
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
@ PARTITION_STRATEGY_HASH
@ PARTITION_STRATEGY_LIST
@ PARTITION_STRATEGY_RANGE
int32 partition_rbound_datum_cmp(FmgrInfo *partsupfunc, Oid *partcollation, Datum *rb_datums, PartitionRangeDatumKind *rb_kind, Datum *tuple_datums, int n_tuple_datums)
uint64 compute_partition_hash_value(int partnatts, FmgrInfo *partsupfunc, const Oid *partcollation, const Datum *values, const bool *isnull)
int partition_range_datum_bsearch(FmgrInfo *partsupfunc, Oid *partcollation, PartitionBoundInfo boundinfo, int nvalues, Datum *values, bool *is_equal)
int partition_list_bsearch(FmgrInfo *partsupfunc, Oid *partcollation, PartitionBoundInfo boundinfo, Datum value, bool *is_equal)
#define partition_bound_accepts_nulls(bi)
PartitionKey RelationGetPartitionKey(Relation rel)
static int16 get_partition_col_attnum(PartitionKey key, int col)
static int get_partition_natts(PartitionKey key)
static Oid get_partition_col_typid(PartitionKey key, int col)
PartitionDirectory CreatePartitionDirectory(MemoryContext mcxt, bool omit_detached)
PartitionDesc PartitionDirectoryLookup(PartitionDirectory pdir, Relation rel)
List * get_partition_ancestors(Oid relid)
Bitmapset * get_matching_partitions(PartitionPruneContext *context, List *pruning_steps)
#define PruneCxtStateIdx(partnatts, step_id, keyno)
#define PARTITION_MAX_KEYS
#define lfirst_node(type, lc)
static int list_length(const List *l)
static void * list_nth(const List *list, int n)
static ListCell * list_head(const List *l)
#define list_nth_node(type, list, n)
static ListCell * lnext(const List *l, const ListCell *c)
static int32 DatumGetInt32(Datum X)
#define RelationGetForm(relation)
#define RelationGetRelid(relation)
#define RelationGetDescr(relation)
#define RelationGetRelationName(relation)
List * RelationGetIndexList(Relation relation)
int errtable(Relation rel)
Node * map_variable_attnos(Node *node, int target_varno, int sublevels_up, const AttrMap *attno_map, Oid to_rowtype, bool *found_whole_row)
int check_enable_rls(Oid relid, Oid checkAsUser, bool noError)
char * pg_get_partkeydef_columns(Oid relid, bool pretty)
void check_stack_depth(void)
void appendStringInfo(StringInfo str, const char *fmt,...)
void appendBinaryStringInfo(StringInfo str, const void *data, int datalen)
void appendStringInfoString(StringInfo str, const char *s)
void appendStringInfoChar(StringInfo str, char ch)
void initStringInfo(StringInfo str)
List * es_part_prune_infos
List * es_tuple_routing_result_relations
Bitmapset * es_unpruned_relids
List * es_part_prune_states
MemoryContext es_query_cxt
PartitionDirectory es_partition_directory
List * es_part_prune_results
ParamListInfo ecxt_param_list_info
TupleTableSlot * ecxt_scantuple
struct EState * ecxt_estate
EndForeignInsert_function EndForeignInsert
BeginForeignInsert_function BeginForeignInsert
ExecForeignBatchInsert_function ExecForeignBatchInsert
GetForeignModifyBatchSize_function GetForeignModifyBatchSize
ProjectionInfo * mas_proj
ResultRelInfo * resultRelInfo
ResultRelInfo * rootResultRelInfo
List * mergeJoinConditions
List * withCheckOptionLists
OnConflictAction onConflictAction
TupleTableSlot * oc_ProjSlot
TupleTableSlot * oc_Existing
ExprState * oc_WhereClause
ProjectionInfo * oc_ProjInfo
PartitionRangeDatumKind ** kind
int last_found_datum_index
PartitionBoundInfo boundinfo
int last_found_part_index
int indexes[FLEXIBLE_ARRAY_MEMBER]
PartitionStrategy strategy
ExprContext * exprcontext
PartitionBoundInfo boundinfo
Bitmapset * other_subplans
PartitionPruningData * partprunedata[FLEXIBLE_ARRAY_MEMBER]
Bitmapset * other_subplans
MemoryContext prune_context
PartitionedRelPruningData partrelprunedata[FLEXIBLE_ARRAY_MEMBER]
PartitionDispatch * partition_dispatch_info
ResultRelInfo ** partitions
ResultRelInfo ** nonleaf_partitions
Bitmapset * present_parts
List * initial_pruning_steps
List * exec_pruning_steps
List * exec_pruning_steps
PartitionPruneContext exec_context
PartitionPruneContext initial_context
Bitmapset * present_parts
List * initial_pruning_steps
ExprContext * ps_ExprContext
TupleTableSlot * ps_ResultTupleSlot
TupleTableSlot * ri_PartitionTupleSlot
OnConflictSetState * ri_onConflict
List * ri_onConflictArbiterIndexes
struct CopyMultiInsertBuffer * ri_CopyMultiInsertBuffer
struct FdwRoutine * ri_FdwRoutine
void table_close(Relation relation, LOCKMODE lockmode)
Relation table_open(Oid relationId, LOCKMODE lockmode)
TupleTableSlot * table_slot_create(Relation relation, List **reglist)
TupleTableSlot * execute_attr_map_slot(AttrMap *attrMap, TupleTableSlot *in_slot, TupleTableSlot *out_slot)
static Datum slot_getattr(TupleTableSlot *slot, int attnum, bool *isnull)
static TupleTableSlot * ExecClearTuple(TupleTableSlot *slot)
#define IsolationUsesXactSnapshot()