PostgreSQL Source Code  git master
partbounds.c
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * partbounds.c
4  * Support routines for manipulating partition bounds
5  *
6  * Portions Copyright (c) 1996-2021, PostgreSQL Global Development Group
7  * Portions Copyright (c) 1994, Regents of the University of California
8  *
9  * IDENTIFICATION
10  * src/backend/partitioning/partbounds.c
11  *
12  *-------------------------------------------------------------------------
13  */
14 
15 #include "postgres.h"
16 
17 #include "access/relation.h"
18 #include "access/table.h"
19 #include "access/tableam.h"
20 #include "catalog/partition.h"
21 #include "catalog/pg_inherits.h"
22 #include "catalog/pg_type.h"
23 #include "commands/tablecmds.h"
24 #include "common/hashfn.h"
25 #include "executor/executor.h"
26 #include "miscadmin.h"
27 #include "nodes/makefuncs.h"
28 #include "nodes/nodeFuncs.h"
29 #include "nodes/pathnodes.h"
30 #include "parser/parse_coerce.h"
31 #include "partitioning/partbounds.h"
32 #include "partitioning/partdesc.h"
33 #include "partitioning/partprune.h"
34 #include "utils/builtins.h"
35 #include "utils/datum.h"
36 #include "utils/fmgroids.h"
37 #include "utils/lsyscache.h"
38 #include "utils/partcache.h"
39 #include "utils/ruleutils.h"
40 #include "utils/snapmgr.h"
41 #include "utils/syscache.h"
42 
43 /*
44  * When qsort'ing partition bounds after reading from the catalog, each bound
45  * is represented with one of the following structs.
46  */
47 
48 /* One bound of a hash partition */
49 typedef struct PartitionHashBound
50 {
51  int modulus;
52  int remainder;
53  int index;
54 } PartitionHashBound;
55 
56 /* One value coming from some (index'th) list partition */
57 typedef struct PartitionListValue
58 {
59  int index;
60  Datum value;
61 } PartitionListValue;
62 
63 /* One bound of a range partition */
64 typedef struct PartitionRangeBound
65 {
66  int index;
67  Datum *datums; /* range bound datums */
68  PartitionRangeDatumKind *kind; /* the kind of each datum */
69  bool lower; /* this is the lower (vs upper) bound */
70 } PartitionRangeBound;
71 
72 /*
73  * Mapping from partitions of a joining relation to partitions of a join
74  * relation being computed (a.k.a merged partitions)
75  */
76 typedef struct PartitionMap
77 {
78  int nparts; /* number of partitions */
79  int *merged_indexes; /* indexes of merged partitions */
80  bool *merged; /* flags to indicate whether partitions are
81  * merged with non-dummy partitions */
82  bool did_remapping; /* did we re-map partitions? */
83  int *old_indexes; /* old indexes of merged partitions if
84  * did_remapping */
85 } PartitionMap;
86 
87 /* Macro for comparing two range bounds */
88 #define compare_range_bounds(partnatts, partsupfunc, partcollations, \
89  bound1, bound2) \
90  (partition_rbound_cmp(partnatts, partsupfunc, partcollations, \
91  (bound1)->datums, (bound1)->kind, (bound1)->lower, \
92  bound2))
93 
94 static int32 qsort_partition_hbound_cmp(const void *a, const void *b);
95 static int32 qsort_partition_list_value_cmp(const void *a, const void *b,
96  void *arg);
97 static int32 qsort_partition_rbound_cmp(const void *a, const void *b,
98  void *arg);
99 static PartitionBoundInfo create_hash_bounds(PartitionBoundSpec **boundspecs,
100  int nparts, PartitionKey key, int **mapping);
101 static PartitionBoundInfo create_list_bounds(PartitionBoundSpec **boundspecs,
102  int nparts, PartitionKey key, int **mapping);
103 static PartitionBoundInfo create_range_bounds(PartitionBoundSpec **boundspecs,
104  int nparts, PartitionKey key, int **mapping);
105 static PartitionBoundInfo merge_list_bounds(FmgrInfo *partsupfunc,
106  Oid *collations,
107  RelOptInfo *outer_rel,
108  RelOptInfo *inner_rel,
109  JoinType jointype,
110  List **outer_parts,
111  List **inner_parts);
112 static PartitionBoundInfo merge_range_bounds(int partnatts,
113  FmgrInfo *partsupfuncs,
114  Oid *partcollations,
115  RelOptInfo *outer_rel,
116  RelOptInfo *inner_rel,
117  JoinType jointype,
118  List **outer_parts,
119  List **inner_parts);
120 static void init_partition_map(RelOptInfo *rel, PartitionMap *map);
121 static void free_partition_map(PartitionMap *map);
122 static bool is_dummy_partition(RelOptInfo *rel, int part_index);
123 static int merge_matching_partitions(PartitionMap *outer_map,
124  PartitionMap *inner_map,
125  int outer_part,
126  int inner_part,
127  int *next_index);
128 static int process_outer_partition(PartitionMap *outer_map,
129  PartitionMap *inner_map,
130  bool outer_has_default,
131  bool inner_has_default,
132  int outer_index,
133  int inner_default,
134  JoinType jointype,
135  int *next_index,
136  int *default_index);
137 static int process_inner_partition(PartitionMap *outer_map,
138  PartitionMap *inner_map,
139  bool outer_has_default,
140  bool inner_has_default,
141  int inner_index,
142  int outer_default,
143  JoinType jointype,
144  int *next_index,
145  int *default_index);
146 static void merge_null_partitions(PartitionMap *outer_map,
147  PartitionMap *inner_map,
148  bool outer_has_null,
149  bool inner_has_null,
150  int outer_null,
151  int inner_null,
152  JoinType jointype,
153  int *next_index,
154  int *null_index);
155 static void merge_default_partitions(PartitionMap *outer_map,
156  PartitionMap *inner_map,
157  bool outer_has_default,
158  bool inner_has_default,
159  int outer_default,
160  int inner_default,
161  JoinType jointype,
162  int *next_index,
163  int *default_index);
164 static int merge_partition_with_dummy(PartitionMap *map, int index,
165  int *next_index);
166 static void fix_merged_indexes(PartitionMap *outer_map,
167  PartitionMap *inner_map,
168  int nmerged, List *merged_indexes);
169 static void generate_matching_part_pairs(RelOptInfo *outer_rel,
170  RelOptInfo *inner_rel,
171  PartitionMap *outer_map,
172  PartitionMap *inner_map,
173  int nmerged,
174  List **outer_parts,
175  List **inner_parts);
176 static PartitionBoundInfo build_merged_partition_bounds(char strategy,
177  List *merged_datums,
178  List *merged_kinds,
179  List *merged_indexes,
180  int null_index,
181  int default_index);
182 static int get_range_partition(RelOptInfo *rel,
183  PartitionBoundInfo bi,
184  int *lb_pos,
185  PartitionRangeBound *lb,
186  PartitionRangeBound *ub);
187 static int get_range_partition_internal(PartitionBoundInfo bi,
188  int *lb_pos,
189  PartitionRangeBound *lb,
190  PartitionRangeBound *ub);
191 static bool compare_range_partitions(int partnatts, FmgrInfo *partsupfuncs,
192  Oid *partcollations,
193  PartitionRangeBound *outer_lb,
194  PartitionRangeBound *outer_ub,
195  PartitionRangeBound *inner_lb,
196  PartitionRangeBound *inner_ub,
197  int *lb_cmpval, int *ub_cmpval);
198 static void get_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs,
199  Oid *partcollations, JoinType jointype,
200  PartitionRangeBound *outer_lb,
201  PartitionRangeBound *outer_ub,
202  PartitionRangeBound *inner_lb,
203  PartitionRangeBound *inner_ub,
204  int lb_cmpval, int ub_cmpval,
205  PartitionRangeBound *merged_lb,
206  PartitionRangeBound *merged_ub);
207 static void add_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs,
208  Oid *partcollations,
209  PartitionRangeBound *merged_lb,
210  PartitionRangeBound *merged_ub,
211  int merged_index,
212  List **merged_datums,
213  List **merged_kinds,
214  List **merged_indexes);
215 static PartitionRangeBound *make_one_partition_rbound(PartitionKey key, int index,
216  List *datums, bool lower);
217 static int32 partition_hbound_cmp(int modulus1, int remainder1, int modulus2,
218  int remainder2);
219 static int32 partition_rbound_cmp(int partnatts, FmgrInfo *partsupfunc,
220  Oid *partcollation, Datum *datums1,
221  PartitionRangeDatumKind *kind1, bool lower1,
222  PartitionRangeBound *b2);
223 static int partition_range_bsearch(int partnatts, FmgrInfo *partsupfunc,
224  Oid *partcollation,
225  PartitionBoundInfo boundinfo,
226  PartitionRangeBound *probe, int32 *cmpval);
227 static Expr *make_partition_op_expr(PartitionKey key, int keynum,
228  uint16 strategy, Expr *arg1, Expr *arg2);
229 static Oid get_partition_operator(PartitionKey key, int col,
230  StrategyNumber strategy, bool *need_relabel);
231 static List *get_qual_for_hash(Relation parent, PartitionBoundSpec *spec);
232 static List *get_qual_for_list(Relation parent, PartitionBoundSpec *spec);
233 static List *get_qual_for_range(Relation parent, PartitionBoundSpec *spec,
234  bool for_default);
235 static void get_range_key_properties(PartitionKey key, int keynum,
236  PartitionRangeDatum *ldatum,
237  PartitionRangeDatum *udatum,
238  ListCell **partexprs_item,
239  Expr **keyCol,
240  Const **lower_val, Const **upper_val);
241 static List *get_range_nulltest(PartitionKey key);
242 
243 /*
244  * get_qual_from_partbound
245  * Given a parser node for partition bound, return the list of executable
246  * expressions as partition constraint
247  */
248 List *
249 get_qual_from_partbound(Relation rel, Relation parent,
250  PartitionBoundSpec *spec)
251 {
252  PartitionKey key = RelationGetPartitionKey(parent);
253  List *my_qual = NIL;
254 
255  Assert(key != NULL);
256 
257  switch (key->strategy)
258  {
259  case PARTITION_STRATEGY_HASH:
260  Assert(spec->strategy == PARTITION_STRATEGY_HASH);
261  my_qual = get_qual_for_hash(parent, spec);
262  break;
263 
264  case PARTITION_STRATEGY_LIST:
265  Assert(spec->strategy == PARTITION_STRATEGY_LIST);
266  my_qual = get_qual_for_list(parent, spec);
267  break;
268 
269  case PARTITION_STRATEGY_RANGE:
270  Assert(spec->strategy == PARTITION_STRATEGY_RANGE);
271  my_qual = get_qual_for_range(parent, spec, false);
272  break;
273 
274  default:
275  elog(ERROR, "unexpected partition strategy: %d",
276  (int) key->strategy);
277  }
278 
279  return my_qual;
280 }
281 
282 /*
283  * partition_bounds_create
284  * Build a PartitionBoundInfo struct from a list of PartitionBoundSpec
285  * nodes
286  *
287  * This function creates a PartitionBoundInfo and fills the values of its
288  * various members based on the input list. Importantly, 'datums' array will
289  * contain Datum representation of individual bounds (possibly after
290  * de-duplication as in case of range bounds), sorted in a canonical order
291  * defined by qsort_partition_* functions of respective partitioning methods.
292  * 'indexes' array will contain as many elements as there are bounds (specific
293  * exceptions to this rule are listed in the function body), which represent
294  * the 0-based canonical positions of partitions.
295  *
296  * Upon return from this function, *mapping is set to an array of
297  * list_length(boundspecs) elements, each of which maps the original index of
298  * a partition to its canonical index.
299  *
300  * Note: The objects returned by this function are wholly allocated in the
301  * current memory context.
302  */
303 PartitionBoundInfo
304 partition_bounds_create(PartitionBoundSpec **boundspecs, int nparts,
305  PartitionKey key, int **mapping)
306 {
307  int i;
308 
309  Assert(nparts > 0);
310 
311  /*
312  * For each partitioning method, we first convert the partition bounds
313  * from their parser node representation to the internal representation,
314  * along with any additional preprocessing (such as de-duplicating range
315  * bounds). Resulting bound datums are then added to the 'datums' array
316  * in PartitionBoundInfo. For each datum added, an integer indicating the
317  * canonical partition index is added to the 'indexes' array.
318  *
319  * For each bound, we remember its partition's position (0-based) in the
320  * original list to later map it to the canonical index.
321  */
322 
323  /*
324  * Initialize mapping array with invalid values, this is filled within
325  * each sub-routine below depending on the bound type.
326  */
327  *mapping = (int *) palloc(sizeof(int) * nparts);
328  for (i = 0; i < nparts; i++)
329  (*mapping)[i] = -1;
330 
331  switch (key->strategy)
332  {
333  case PARTITION_STRATEGY_HASH:
334  return create_hash_bounds(boundspecs, nparts, key, mapping);
335 
336  case PARTITION_STRATEGY_LIST:
337  return create_list_bounds(boundspecs, nparts, key, mapping);
338 
339  case PARTITION_STRATEGY_RANGE:
340  return create_range_bounds(boundspecs, nparts, key, mapping);
341 
342  default:
343  elog(ERROR, "unexpected partition strategy: %d",
344  (int) key->strategy);
345  break;
346  }
347 
348  Assert(false);
349  return NULL; /* keep compiler quiet */
350 }
351 
352 /*
353  * create_hash_bounds
354  * Create a PartitionBoundInfo for a hash partitioned table
355  */
356 static PartitionBoundInfo
357 create_hash_bounds(PartitionBoundSpec **boundspecs, int nparts,
358  PartitionKey key, int **mapping)
359 {
360  PartitionBoundInfo boundinfo;
361  PartitionHashBound **hbounds = NULL;
362  int i;
363  int ndatums = 0;
364  int greatest_modulus;
365 
366  boundinfo = (PartitionBoundInfoData *)
367  palloc0(sizeof(PartitionBoundInfoData));
368  boundinfo->strategy = key->strategy;
369  /* No special hash partitions. */
370  boundinfo->null_index = -1;
371  boundinfo->default_index = -1;
372 
373  ndatums = nparts;
374  hbounds = (PartitionHashBound **)
375  palloc(nparts * sizeof(PartitionHashBound *));
376 
377  /* Convert from node to the internal representation */
378  for (i = 0; i < nparts; i++)
379  {
380  PartitionBoundSpec *spec = boundspecs[i];
381 
382  if (spec->strategy != PARTITION_STRATEGY_HASH)
383  elog(ERROR, "invalid strategy in partition bound spec");
384 
385  hbounds[i] = (PartitionHashBound *) palloc(sizeof(PartitionHashBound));
386  hbounds[i]->modulus = spec->modulus;
387  hbounds[i]->remainder = spec->remainder;
388  hbounds[i]->index = i;
389  }
390 
391  /* Sort all the bounds in ascending order */
392  qsort(hbounds, nparts, sizeof(PartitionHashBound *),
393  qsort_partition_hbound_cmp);
394 
395  /* After sorting, moduli are now stored in ascending order. */
396  greatest_modulus = hbounds[ndatums - 1]->modulus;
397 
398  boundinfo->ndatums = ndatums;
399  boundinfo->datums = (Datum **) palloc0(ndatums * sizeof(Datum *));
400  boundinfo->nindexes = greatest_modulus;
401  boundinfo->indexes = (int *) palloc(greatest_modulus * sizeof(int));
402  for (i = 0; i < greatest_modulus; i++)
403  boundinfo->indexes[i] = -1;
404 
405  /*
406  * For hash partitioning, there are as many datums (modulus and remainder
407  * pairs) as there are partitions. Indexes are simply values ranging from
408  * 0 to (nparts - 1).
409  */
410  for (i = 0; i < nparts; i++)
411  {
412  int modulus = hbounds[i]->modulus;
413  int remainder = hbounds[i]->remainder;
414 
415  boundinfo->datums[i] = (Datum *) palloc(2 * sizeof(Datum));
416  boundinfo->datums[i][0] = Int32GetDatum(modulus);
417  boundinfo->datums[i][1] = Int32GetDatum(remainder);
418 
419  while (remainder < greatest_modulus)
420  {
421  /* overlap? */
422  Assert(boundinfo->indexes[remainder] == -1);
423  boundinfo->indexes[remainder] = i;
424  remainder += modulus;
425  }
426 
427  (*mapping)[hbounds[i]->index] = i;
428  pfree(hbounds[i]);
429  }
430  pfree(hbounds);
431 
432  return boundinfo;
433 }
434 
435 /*
436  * create_list_bounds
437  * Create a PartitionBoundInfo for a list partitioned table
438  */
439 static PartitionBoundInfo
440 create_list_bounds(PartitionBoundSpec **boundspecs, int nparts,
441  PartitionKey key, int **mapping)
442 {
443  PartitionBoundInfo boundinfo;
444  PartitionListValue **all_values = NULL;
445  ListCell *cell;
446  int i = 0;
447  int ndatums = 0;
448  int next_index = 0;
449  int default_index = -1;
450  int null_index = -1;
451  List *non_null_values = NIL;
452 
453  boundinfo = (PartitionBoundInfoData *)
454  palloc0(sizeof(PartitionBoundInfoData));
455  boundinfo->strategy = key->strategy;
456  /* Will be set correctly below. */
457  boundinfo->null_index = -1;
458  boundinfo->default_index = -1;
459 
460  /* Create a unified list of non-null values across all partitions. */
461  for (i = 0; i < nparts; i++)
462  {
463  PartitionBoundSpec *spec = boundspecs[i];
464  ListCell *c;
465 
466  if (spec->strategy != PARTITION_STRATEGY_LIST)
467  elog(ERROR, "invalid strategy in partition bound spec");
468 
469  /*
470  * Note the index of the partition bound spec for the default
471  * partition. There's no datum to add to the list on non-null datums
472  * for this partition.
473  */
474  if (spec->is_default)
475  {
476  default_index = i;
477  continue;
478  }
479 
480  foreach(c, spec->listdatums)
481  {
482  Const *val = castNode(Const, lfirst(c));
483  PartitionListValue *list_value = NULL;
484 
485  if (!val->constisnull)
486  {
487  list_value = (PartitionListValue *)
488  palloc0(sizeof(PartitionListValue));
489  list_value->index = i;
490  list_value->value = val->constvalue;
491  }
492  else
493  {
494  /*
495  * Never put a null into the values array; save the index of
496  * the partition that stores nulls, instead.
497  */
498  if (null_index != -1)
499  elog(ERROR, "found null more than once");
500  null_index = i;
501  }
502 
503  if (list_value)
504  non_null_values = lappend(non_null_values, list_value);
505  }
506  }
507 
508  ndatums = list_length(non_null_values);
509 
510  /*
511  * Collect all list values in one array. Alongside the value, we also save
512  * the index of partition the value comes from.
513  */
514  all_values = (PartitionListValue **)
515  palloc(ndatums * sizeof(PartitionListValue *));
516  i = 0;
517  foreach(cell, non_null_values)
518  {
519  PartitionListValue *src = lfirst(cell);
520 
521  all_values[i] = (PartitionListValue *)
522  palloc(sizeof(PartitionListValue));
523  all_values[i]->value = src->value;
524  all_values[i]->index = src->index;
525  i++;
526  }
527 
528  qsort_arg(all_values, ndatums, sizeof(PartitionListValue *),
529  qsort_partition_list_value_cmp, (void *) key);
530 
531  boundinfo->ndatums = ndatums;
532  boundinfo->datums = (Datum **) palloc0(ndatums * sizeof(Datum *));
533  boundinfo->nindexes = ndatums;
534  boundinfo->indexes = (int *) palloc(ndatums * sizeof(int));
535 
536  /*
537  * Copy values. Canonical indexes are values ranging from 0 to (nparts -
538  * 1) assigned to each partition such that all datums of a given partition
539  * receive the same value. The value for a given partition is the index of
540  * that partition's smallest datum in the all_values[] array.
541  */
542  for (i = 0; i < ndatums; i++)
543  {
544  int orig_index = all_values[i]->index;
545 
546  boundinfo->datums[i] = (Datum *) palloc(sizeof(Datum));
547  boundinfo->datums[i][0] = datumCopy(all_values[i]->value,
548  key->parttypbyval[0],
549  key->parttyplen[0]);
550 
551  /* If the old index has no mapping, assign one */
552  if ((*mapping)[orig_index] == -1)
553  (*mapping)[orig_index] = next_index++;
554 
555  boundinfo->indexes[i] = (*mapping)[orig_index];
556  }
557 
558  /*
559  * Set the canonical value for null_index, if any.
560  *
561  * It is possible that the null-accepting partition has not been assigned
562  * an index yet, which could happen if such partition accepts only null
563  * and hence not handled in the above loop which only looked at non-null
564  * values.
565  */
566  if (null_index != -1)
567  {
568  Assert(null_index >= 0);
569  if ((*mapping)[null_index] == -1)
570  (*mapping)[null_index] = next_index++;
571  boundinfo->null_index = (*mapping)[null_index];
572  }
573 
574  /* Set the canonical value for default_index, if any. */
575  if (default_index != -1)
576  {
577  /*
578  * The default partition accepts any value not specified in the lists
579  * of other partitions, hence it should not get mapped index while
580  * assigning those for non-null datums.
581  */
582  Assert(default_index >= 0);
583  Assert((*mapping)[default_index] == -1);
584  (*mapping)[default_index] = next_index++;
585  boundinfo->default_index = (*mapping)[default_index];
586  }
587 
588  /* All partitions must now have been assigned canonical indexes. */
589  Assert(next_index == nparts);
590  return boundinfo;
591 }
592 
593 /*
594  * create_range_bounds
595  * Create a PartitionBoundInfo for a range partitioned table
596  */
597 static PartitionBoundInfo
598 create_range_bounds(PartitionBoundSpec **boundspecs, int nparts,
599  PartitionKey key, int **mapping)
600 {
601  PartitionBoundInfo boundinfo;
602  PartitionRangeBound **rbounds = NULL;
603  PartitionRangeBound **all_bounds,
604  *prev;
605  int i,
606  k;
607  int ndatums = 0;
608  int default_index = -1;
609  int next_index = 0;
610 
611  boundinfo = (PartitionBoundInfoData *)
612  palloc0(sizeof(PartitionBoundInfoData));
613  boundinfo->strategy = key->strategy;
614  /* There is no special null-accepting range partition. */
615  boundinfo->null_index = -1;
616  /* Will be set correctly below. */
617  boundinfo->default_index = -1;
618 
619  all_bounds = (PartitionRangeBound **)
620  palloc0(2 * nparts * sizeof(PartitionRangeBound *));
621 
622  /* Create a unified list of range bounds across all the partitions. */
623  ndatums = 0;
624  for (i = 0; i < nparts; i++)
625  {
626  PartitionBoundSpec *spec = boundspecs[i];
627  PartitionRangeBound *lower,
628  *upper;
629 
630  if (spec->strategy != PARTITION_STRATEGY_RANGE)
631  elog(ERROR, "invalid strategy in partition bound spec");
632 
633  /*
634  * Note the index of the partition bound spec for the default
635  * partition. There's no datum to add to the all_bounds array for
636  * this partition.
637  */
638  if (spec->is_default)
639  {
640  default_index = i;
641  continue;
642  }
643 
644  lower = make_one_partition_rbound(key, i, spec->lowerdatums, true);
645  upper = make_one_partition_rbound(key, i, spec->upperdatums, false);
646  all_bounds[ndatums++] = lower;
647  all_bounds[ndatums++] = upper;
648  }
649 
650  Assert(ndatums == nparts * 2 ||
651  (default_index != -1 && ndatums == (nparts - 1) * 2));
652 
653  /* Sort all the bounds in ascending order */
654  qsort_arg(all_bounds, ndatums,
655  sizeof(PartitionRangeBound *),
656  qsort_partition_rbound_cmp,
657  (void *) key);
658 
659  /* Save distinct bounds from all_bounds into rbounds. */
660  rbounds = (PartitionRangeBound **)
661  palloc(ndatums * sizeof(PartitionRangeBound *));
662  k = 0;
663  prev = NULL;
664  for (i = 0; i < ndatums; i++)
665  {
666  PartitionRangeBound *cur = all_bounds[i];
667  bool is_distinct = false;
668  int j;
669 
670  /* Is the current bound distinct from the previous one? */
671  for (j = 0; j < key->partnatts; j++)
672  {
673  Datum cmpval;
674 
675  if (prev == NULL || cur->kind[j] != prev->kind[j])
676  {
677  is_distinct = true;
678  break;
679  }
680 
681  /*
682  * If the bounds are both MINVALUE or MAXVALUE, stop now and treat
683  * them as equal, since any values after this point must be
684  * ignored.
685  */
686  if (cur->kind[j] != PARTITION_RANGE_DATUM_VALUE)
687  break;
688 
689  cmpval = FunctionCall2Coll(&key->partsupfunc[j],
690  key->partcollation[j],
691  cur->datums[j],
692  prev->datums[j]);
693  if (DatumGetInt32(cmpval) != 0)
694  {
695  is_distinct = true;
696  break;
697  }
698  }
699 
700  /*
701  * Only if the bound is distinct save it into a temporary array, i.e,
702  * rbounds which is later copied into boundinfo datums array.
703  */
704  if (is_distinct)
705  rbounds[k++] = all_bounds[i];
706 
707  prev = cur;
708  }
709 
710  /* Update ndatums to hold the count of distinct datums. */
711  ndatums = k;
712 
713  /*
714  * Add datums to boundinfo. Canonical indexes are values ranging from 0
715  * to nparts - 1, assigned in that order to each partition's upper bound.
716  * For 'datums' elements that are lower bounds, there is -1 in the
717  * 'indexes' array to signify that no partition exists for the values less
718  * than such a bound and greater than or equal to the previous upper
719  * bound.
720  */
721  boundinfo->ndatums = ndatums;
722  boundinfo->datums = (Datum **) palloc0(ndatums * sizeof(Datum *));
723  boundinfo->kind = (PartitionRangeDatumKind **)
724  palloc(ndatums *
725  sizeof(PartitionRangeDatumKind *));
726 
727  /*
728  * For range partitioning, an additional value of -1 is stored as the last
729  * element of the indexes[] array.
730  */
731  boundinfo->nindexes = ndatums + 1;
732  boundinfo->indexes = (int *) palloc((ndatums + 1) * sizeof(int));
733 
734  for (i = 0; i < ndatums; i++)
735  {
736  int j;
737 
738  boundinfo->datums[i] = (Datum *) palloc(key->partnatts *
739  sizeof(Datum));
740  boundinfo->kind[i] = (PartitionRangeDatumKind *)
741  palloc(key->partnatts *
742  sizeof(PartitionRangeDatumKind));
743  for (j = 0; j < key->partnatts; j++)
744  {
745  if (rbounds[i]->kind[j] == PARTITION_RANGE_DATUM_VALUE)
746  boundinfo->datums[i][j] =
747  datumCopy(rbounds[i]->datums[j],
748  key->parttypbyval[j],
749  key->parttyplen[j]);
750  boundinfo->kind[i][j] = rbounds[i]->kind[j];
751  }
752 
753  /*
754  * There is no mapping for invalid indexes.
755  *
756  * Any lower bounds in the rbounds array have invalid indexes
757  * assigned, because the values between the previous bound (if there
758  * is one) and this (lower) bound are not part of the range of any
759  * existing partition.
760  */
761  if (rbounds[i]->lower)
762  boundinfo->indexes[i] = -1;
763  else
764  {
765  int orig_index = rbounds[i]->index;
766 
767  /* If the old index has no mapping, assign one */
768  if ((*mapping)[orig_index] == -1)
769  (*mapping)[orig_index] = next_index++;
770 
771  boundinfo->indexes[i] = (*mapping)[orig_index];
772  }
773  }
774 
775  /* Set the canonical value for default_index, if any. */
776  if (default_index != -1)
777  {
778  Assert(default_index >= 0 && (*mapping)[default_index] == -1);
779  (*mapping)[default_index] = next_index++;
780  boundinfo->default_index = (*mapping)[default_index];
781  }
782 
783  /* The extra -1 element. */
784  Assert(i == ndatums);
785  boundinfo->indexes[i] = -1;
786 
787  /* All partitions must now have been assigned canonical indexes. */
788  Assert(next_index == nparts);
789  return boundinfo;
790 }
791 
792 /*
793  * Are two partition bound collections logically equal?
794  *
795  * Used in the keep logic of relcache.c (ie, in RelationClearRelation()).
796  * This is also useful when b1 and b2 are bound collections of two separate
797  * relations, respectively, because PartitionBoundInfo is a canonical
798  * representation of partition bounds.
799  */
800 bool
801 partition_bounds_equal(int partnatts, int16 *parttyplen, bool *parttypbyval,
802  PartitionBoundInfo b1, PartitionBoundInfo b2)
803 {
804  int i;
805 
806  if (b1->strategy != b2->strategy)
807  return false;
808 
809  if (b1->ndatums != b2->ndatums)
810  return false;
811 
812  if (b1->nindexes != b2->nindexes)
813  return false;
814 
815  if (b1->null_index != b2->null_index)
816  return false;
817 
818  if (b1->default_index != b2->default_index)
819  return false;
820 
821  /* For all partition strategies, the indexes[] arrays have to match */
822  for (i = 0; i < b1->nindexes; i++)
823  {
824  if (b1->indexes[i] != b2->indexes[i])
825  return false;
826  }
827 
828  /* Finally, compare the datums[] arrays */
829  if (b1->strategy == PARTITION_STRATEGY_HASH)
830  {
831  /*
832  * We arrange the partitions in the ascending order of their moduli
833  * and remainders. Also every modulus is factor of next larger
834  * modulus. Therefore we can safely store index of a given partition
835  * in indexes array at remainder of that partition. Also entries at
836  * (remainder + N * modulus) positions in indexes array are all same
837  * for (modulus, remainder) specification for any partition. Thus the
838  * datums arrays from the given bounds are the same, if and only if
839  * their indexes arrays are the same. So, it suffices to compare the
840  * indexes arrays.
841  *
842  * Nonetheless make sure that the bounds are indeed the same when the
843  * indexes match. Hash partition bound stores modulus and remainder
844  * at b1->datums[i][0] and b1->datums[i][1] position respectively.
845  */
846 #ifdef USE_ASSERT_CHECKING
847  for (i = 0; i < b1->ndatums; i++)
848  Assert((b1->datums[i][0] == b2->datums[i][0] &&
849  b1->datums[i][1] == b2->datums[i][1]));
850 #endif
851  }
852  else
853  {
854  for (i = 0; i < b1->ndatums; i++)
855  {
856  int j;
857 
858  for (j = 0; j < partnatts; j++)
859  {
860  /* For range partitions, the bounds might not be finite. */
861  if (b1->kind != NULL)
862  {
863  /* The different kinds of bound all differ from each other */
864  if (b1->kind[i][j] != b2->kind[i][j])
865  return false;
866 
867  /*
868  * Non-finite bounds are equal without further
869  * examination.
870  */
871  if (b1->kind[i][j] != PARTITION_RANGE_DATUM_VALUE)
872  continue;
873  }
874 
875  /*
876  * Compare the actual values. Note that it would be both
877  * incorrect and unsafe to invoke the comparison operator
878  * derived from the partitioning specification here. It would
879  * be incorrect because we want the relcache entry to be
880  * updated for ANY change to the partition bounds, not just
881  * those that the partitioning operator thinks are
882  * significant. It would be unsafe because we might reach
883  * this code in the context of an aborted transaction, and an
884  * arbitrary partitioning operator might not be safe in that
885  * context. datumIsEqual() should be simple enough to be
886  * safe.
887  */
888  if (!datumIsEqual(b1->datums[i][j], b2->datums[i][j],
889  parttypbyval[j], parttyplen[j]))
890  return false;
891  }
892  }
893  }
894  return true;
895 }
896 
897 /*
898  * Return a copy of given PartitionBoundInfo structure. The data types of bounds
899  * are described by given partition key specification.
900  *
901  * Note: it's important that this function and its callees not do any catalog
902  * access, nor anything else that would result in allocating memory other than
903  * the returned data structure. Since this is called in a long-lived context,
904  * that would result in unwanted memory leaks.
905  */
906 PartitionBoundInfo
907 partition_bounds_copy(PartitionBoundInfo src,
908  PartitionKey key)
909 {
910  PartitionBoundInfo dest;
911  int i;
912  int ndatums;
913  int nindexes;
914  int partnatts;
915  bool hash_part;
916  int natts;
917 
918  dest = (PartitionBoundInfo) palloc(sizeof(PartitionBoundInfoData));
919 
920  dest->strategy = src->strategy;
921  ndatums = dest->ndatums = src->ndatums;
922  nindexes = dest->nindexes = src->nindexes;
923  partnatts = key->partnatts;
924 
925  /* List partitioned tables have only a single partition key. */
926  Assert(key->strategy != PARTITION_STRATEGY_LIST || partnatts == 1);
927 
928  dest->datums = (Datum **) palloc(sizeof(Datum *) * ndatums);
929 
930  if (src->kind != NULL)
931  {
932  dest->kind = (PartitionRangeDatumKind **) palloc(ndatums *
933  sizeof(PartitionRangeDatumKind *));
934  for (i = 0; i < ndatums; i++)
935  {
936  dest->kind[i] = (PartitionRangeDatumKind *) palloc(partnatts *
937  sizeof(PartitionRangeDatumKind));
938 
939  memcpy(dest->kind[i], src->kind[i],
940  sizeof(PartitionRangeDatumKind) * key->partnatts);
941  }
942  }
943  else
944  dest->kind = NULL;
945 
946  /*
947  * For hash partitioning, datums array will have two elements - modulus
948  * and remainder.
949  */
950  hash_part = (key->strategy == PARTITION_STRATEGY_HASH);
951  natts = hash_part ? 2 : partnatts;
952 
953  for (i = 0; i < ndatums; i++)
954  {
955  int j;
956 
957  dest->datums[i] = (Datum *) palloc(sizeof(Datum) * natts);
958 
959  for (j = 0; j < natts; j++)
960  {
961  bool byval;
962  int typlen;
963 
964  if (hash_part)
965  {
966  typlen = sizeof(int32); /* Always int4 */
967  byval = true; /* int4 is pass-by-value */
968  }
969  else
970  {
971  byval = key->parttypbyval[j];
972  typlen = key->parttyplen[j];
973  }
974 
975  if (dest->kind == NULL ||
976  dest->kind[i][j] == PARTITION_RANGE_DATUM_VALUE)
977  dest->datums[i][j] = datumCopy(src->datums[i][j],
978  byval, typlen);
979  }
980  }
981 
982  dest->indexes = (int *) palloc(sizeof(int) * nindexes);
983  memcpy(dest->indexes, src->indexes, sizeof(int) * nindexes);
984 
985  dest->null_index = src->null_index;
986  dest->default_index = src->default_index;
987 
988  return dest;
989 }
990 
991 /*
992  * partition_bounds_merge
993  * Check to see whether every partition of 'outer_rel' matches/overlaps
994  * one partition of 'inner_rel' at most, and vice versa; and if so, build
995  * and return the partition bounds for a join relation between the rels,
996  * generating two lists of the matching/overlapping partitions, which are
997  * returned to *outer_parts and *inner_parts respectively.
998  *
999  * The lists contain the same number of partitions, and the partitions at the
1000  * same positions in the lists indicate join pairs used for partitioned join.
1001  * If a partition on one side matches/overlaps multiple partitions on the other
1002  * side, this function returns NULL, setting *outer_parts and *inner_parts to
1003  * NIL.
1004  */
1005 PartitionBoundInfo
1006 partition_bounds_merge(int partnatts,
1007  FmgrInfo *partsupfunc, Oid *partcollation,
1008  RelOptInfo *outer_rel, RelOptInfo *inner_rel,
1009  JoinType jointype,
1010  List **outer_parts, List **inner_parts)
1011 {
1012  /*
1013  * Currently, this function is called only from try_partitionwise_join(),
1014  * so the join type should be INNER, LEFT, FULL, SEMI, or ANTI.
1015  */
1016  Assert(jointype == JOIN_INNER || jointype == JOIN_LEFT ||
1017  jointype == JOIN_FULL || jointype == JOIN_SEMI ||
1018  jointype == JOIN_ANTI);
1019 
1020  /* The partitioning strategies should be the same. */
1021  Assert(outer_rel->boundinfo->strategy == inner_rel->boundinfo->strategy);
1022 
1023  *outer_parts = *inner_parts = NIL;
1024  switch (outer_rel->boundinfo->strategy)
1025  {
1026  case PARTITION_STRATEGY_HASH:
1027 
1028  /*
1029  * For hash partitioned tables, we currently support partitioned
1030  * join only when they have exactly the same partition bounds.
1031  *
1032  * XXX: it might be possible to relax the restriction to support
1033  * cases where hash partitioned tables have missing partitions
1034  * and/or different moduli, but it's not clear if it would be
1035  * useful to support the former case since it's unusual to have
1036  * missing partitions. On the other hand, it would be useful to
1037  * support the latter case, but in that case, there is a high
1038  * probability that a partition on one side will match multiple
1039  * partitions on the other side, which is the scenario the current
1040  * implementation of partitioned join can't handle.
1041  */
1042  return NULL;
1043 
1044  case PARTITION_STRATEGY_LIST:
1045  return merge_list_bounds(partsupfunc,
1046  partcollation,
1047  outer_rel,
1048  inner_rel,
1049  jointype,
1050  outer_parts,
1051  inner_parts);
1052 
1053  case PARTITION_STRATEGY_RANGE:
1054  return merge_range_bounds(partnatts,
1055  partsupfunc,
1056  partcollation,
1057  outer_rel,
1058  inner_rel,
1059  jointype,
1060  outer_parts,
1061  inner_parts);
1062 
1063  default:
1064  elog(ERROR, "unexpected partition strategy: %d",
1065  (int) outer_rel->boundinfo->strategy);
1066  return NULL; /* keep compiler quiet */
1067  }
1068 }
1069 
1070 /*
1071  * merge_list_bounds
1072  * Create the partition bounds for a join relation between list
1073  * partitioned tables, if possible
1074  *
1075  * In this function we try to find sets of matching partitions from both sides
1076  * by comparing list values stored in their partition bounds. Since the list
1077  * values appear in the ascending order, an algorithm similar to merge join is
1078  * used for that. If a partition on one side doesn't have a matching
1079  * partition on the other side, the algorithm tries to match it with the
1080  * default partition on the other side if any; if not, the algorithm tries to
1081  * match it with a dummy partition on the other side if it's on the
1082  * non-nullable side of an outer join. Also, if both sides have the default
1083  * partitions, the algorithm tries to match them with each other. We give up
1084  * if the algorithm finds a partition matching multiple partitions on the
1085  * other side, which is the scenario the current implementation of partitioned
1086  * join can't handle.
1087  */
1088 static PartitionBoundInfo
1089 merge_list_bounds(FmgrInfo *partsupfunc, Oid *partcollation,
1090  RelOptInfo *outer_rel, RelOptInfo *inner_rel,
1091  JoinType jointype,
1092  List **outer_parts, List **inner_parts)
1093 {
1094  PartitionBoundInfo merged_bounds = NULL;
1095  PartitionBoundInfo outer_bi = outer_rel->boundinfo;
1096  PartitionBoundInfo inner_bi = inner_rel->boundinfo;
1097  bool outer_has_default = partition_bound_has_default(outer_bi);
1098  bool inner_has_default = partition_bound_has_default(inner_bi);
1099  int outer_default = outer_bi->default_index;
1100  int inner_default = inner_bi->default_index;
1101  bool outer_has_null = partition_bound_accepts_nulls(outer_bi);
1102  bool inner_has_null = partition_bound_accepts_nulls(inner_bi);
1103  PartitionMap outer_map;
1104  PartitionMap inner_map;
1105  int outer_pos;
1106  int inner_pos;
1107  int next_index = 0;
1108  int null_index = -1;
1109  int default_index = -1;
1110  List *merged_datums = NIL;
1111  List *merged_indexes = NIL;
1112 
1113  Assert(*outer_parts == NIL);
1114  Assert(*inner_parts == NIL);
1115  Assert(outer_bi->strategy == inner_bi->strategy &&
1116  outer_bi->strategy == PARTITION_STRATEGY_LIST);
1117  /* List partitioning doesn't require kinds. */
1118  Assert(!outer_bi->kind && !inner_bi->kind);
1119 
1120  init_partition_map(outer_rel, &outer_map);
1121  init_partition_map(inner_rel, &inner_map);
1122 
1123  /*
1124  * If the default partitions (if any) have been proven empty, deem them
1125  * non-existent.
1126  */
1127  if (outer_has_default && is_dummy_partition(outer_rel, outer_default))
1128  outer_has_default = false;
1129  if (inner_has_default && is_dummy_partition(inner_rel, inner_default))
1130  inner_has_default = false;
1131 
1132  /*
1133  * Merge partitions from both sides. In each iteration we compare a pair
1134  * of list values, one from each side, and decide whether the
1135  * corresponding partitions match or not. If the two values match
1136  * exactly, move to the next pair of list values, otherwise move to the
1137  * next list value on the side with a smaller list value.
1138  */
1139  outer_pos = inner_pos = 0;
1140  while (outer_pos < outer_bi->ndatums || inner_pos < inner_bi->ndatums)
1141  {
1142  int outer_index = -1;
1143  int inner_index = -1;
1144  Datum *outer_datums;
1145  Datum *inner_datums;
1146  int cmpval;
1147  Datum *merged_datum = NULL;
1148  int merged_index = -1;
1149 
1150  if (outer_pos < outer_bi->ndatums)
1151  {
1152  /*
1153  * If the partition on the outer side has been proven empty,
1154  * ignore it and move to the next datum on the outer side.
1155  */
1156  outer_index = outer_bi->indexes[outer_pos];
1157  if (is_dummy_partition(outer_rel, outer_index))
1158  {
1159  outer_pos++;
1160  continue;
1161  }
1162  }
1163  if (inner_pos < inner_bi->ndatums)
1164  {
1165  /*
1166  * If the partition on the inner side has been proven empty,
1167  * ignore it and move to the next datum on the inner side.
1168  */
1169  inner_index = inner_bi->indexes[inner_pos];
1170  if (is_dummy_partition(inner_rel, inner_index))
1171  {
1172  inner_pos++;
1173  continue;
1174  }
1175  }
1176 
1177  /* Get the list values. */
1178  outer_datums = outer_pos < outer_bi->ndatums ?
1179  outer_bi->datums[outer_pos] : NULL;
1180  inner_datums = inner_pos < inner_bi->ndatums ?
1181  inner_bi->datums[inner_pos] : NULL;
1182 
1183  /*
1184  * We run this loop till both sides finish. This allows us to avoid
1185  * duplicating code to handle the remaining values on the side which
1186  * finishes later. For that we set the comparison parameter cmpval in
1187  * such a way that it appears as if the side which finishes earlier
1188  * has an extra value higher than any other value on the unfinished
1189  * side. That way we advance the values on the unfinished side till
1190  * all of its values are exhausted.
1191  */
1192  if (outer_pos >= outer_bi->ndatums)
1193  cmpval = 1;
1194  else if (inner_pos >= inner_bi->ndatums)
1195  cmpval = -1;
1196  else
1197  {
1198  Assert(outer_datums != NULL && inner_datums != NULL);
1199  cmpval = DatumGetInt32(FunctionCall2Coll(&partsupfunc[0],
1200  partcollation[0],
1201  outer_datums[0],
1202  inner_datums[0]));
1203  }
1204 
1205  if (cmpval == 0)
1206  {
1207  /* Two list values match exactly. */
1208  Assert(outer_pos < outer_bi->ndatums);
1209  Assert(inner_pos < inner_bi->ndatums);
1210  Assert(outer_index >= 0);
1211  Assert(inner_index >= 0);
1212 
1213  /*
1214  * Try merging both partitions. If successful, add the list value
1215  * and index of the merged partition below.
1216  */
1217  merged_index = merge_matching_partitions(&outer_map, &inner_map,
1218  outer_index, inner_index,
1219  &next_index);
1220  if (merged_index == -1)
1221  goto cleanup;
1222 
1223  merged_datum = outer_datums;
1224 
1225  /* Move to the next pair of list values. */
1226  outer_pos++;
1227  inner_pos++;
1228  }
1229  else if (cmpval < 0)
1230  {
1231  /* A list value missing from the inner side. */
1232  Assert(outer_pos < outer_bi->ndatums);
1233 
1234  /*
1235  * If the inner side has the default partition, or this is an
1236  * outer join, try to assign a merged partition to the outer
1237  * partition (see process_outer_partition()). Otherwise, the
1238  * outer partition will not contribute to the result.
1239  */
1240  if (inner_has_default || IS_OUTER_JOIN(jointype))
1241  {
1242  /* Get the outer partition. */
1243  outer_index = outer_bi->indexes[outer_pos];
1244  Assert(outer_index >= 0);
1245  merged_index = process_outer_partition(&outer_map,
1246  &inner_map,
1247  outer_has_default,
1248  inner_has_default,
1249  outer_index,
1250  inner_default,
1251  jointype,
1252  &next_index,
1253  &default_index);
1254  if (merged_index == -1)
1255  goto cleanup;
1256  merged_datum = outer_datums;
1257  }
1258 
1259  /* Move to the next list value on the outer side. */
1260  outer_pos++;
1261  }
1262  else
1263  {
1264  /* A list value missing from the outer side. */
1265  Assert(cmpval > 0);
1266  Assert(inner_pos < inner_bi->ndatums);
1267 
1268  /*
1269  * If the outer side has the default partition, or this is a FULL
1270  * join, try to assign a merged partition to the inner partition
1271  * (see process_inner_partition()). Otherwise, the inner
1272  * partition will not contribute to the result.
1273  */
1274  if (outer_has_default || jointype == JOIN_FULL)
1275  {
1276  /* Get the inner partition. */
1277  inner_index = inner_bi->indexes[inner_pos];
1278  Assert(inner_index >= 0);
1279  merged_index = process_inner_partition(&outer_map,
1280  &inner_map,
1281  outer_has_default,
1282  inner_has_default,
1283  inner_index,
1284  outer_default,
1285  jointype,
1286  &next_index,
1287  &default_index);
1288  if (merged_index == -1)
1289  goto cleanup;
1290  merged_datum = inner_datums;
1291  }
1292 
1293  /* Move to the next list value on the inner side. */
1294  inner_pos++;
1295  }
1296 
1297  /*
1298  * If we assigned a merged partition, add the list value and index of
1299  * the merged partition if appropriate.
1300  */
1301  if (merged_index >= 0 && merged_index != default_index)
1302  {
1303  merged_datums = lappend(merged_datums, merged_datum);
1304  merged_indexes = lappend_int(merged_indexes, merged_index);
1305  }
1306  }
1307 
1308  /*
1309  * If the NULL partitions (if any) have been proven empty, deem them
1310  * non-existent.
1311  */
1312  if (outer_has_null &&
1313  is_dummy_partition(outer_rel, outer_bi->null_index))
1314  outer_has_null = false;
1315  if (inner_has_null &&
1316  is_dummy_partition(inner_rel, inner_bi->null_index))
1317  inner_has_null = false;
1318 
1319  /* Merge the NULL partitions if any. */
1320  if (outer_has_null || inner_has_null)
1321  merge_null_partitions(&outer_map, &inner_map,
1322  outer_has_null, inner_has_null,
1323  outer_bi->null_index, inner_bi->null_index,
1324  jointype, &next_index, &null_index);
1325  else
1326  Assert(null_index == -1);
1327 
1328  /* Merge the default partitions if any. */
1329  if (outer_has_default || inner_has_default)
1330  merge_default_partitions(&outer_map, &inner_map,
1331  outer_has_default, inner_has_default,
1332  outer_default, inner_default,
1333  jointype, &next_index, &default_index);
1334  else
1335  Assert(default_index == -1);
1336 
1337  /* If we have merged partitions, create the partition bounds. */
1338  if (next_index > 0)
1339  {
1340  /* Fix the merged_indexes list if necessary. */
1341  if (outer_map.did_remapping || inner_map.did_remapping)
1342  {
1343  Assert(jointype == JOIN_FULL);
1344  fix_merged_indexes(&outer_map, &inner_map,
1345  next_index, merged_indexes);
1346  }
1347 
1348  /* Use maps to match partitions from inputs. */
1349  generate_matching_part_pairs(outer_rel, inner_rel,
1350  &outer_map, &inner_map,
1351  next_index,
1352  outer_parts, inner_parts);
1353  Assert(*outer_parts != NIL);
1354  Assert(*inner_parts != NIL);
1355  Assert(list_length(*outer_parts) == list_length(*inner_parts));
1356  Assert(list_length(*outer_parts) <= next_index);
1357 
1358  /* Make a PartitionBoundInfo struct to return. */
1359  merged_bounds = build_merged_partition_bounds(outer_bi->strategy,
1360  merged_datums,
1361  NIL,
1362  merged_indexes,
1363  null_index,
1364  default_index);
1365  Assert(merged_bounds);
1366  }
1367 
1368 cleanup:
1369  /* Free local memory before returning. */
1370  list_free(merged_datums);
1371  list_free(merged_indexes);
1372  free_partition_map(&outer_map);
1373  free_partition_map(&inner_map);
1374 
1375  return merged_bounds;
1376 }
1377 
1378 /*
1379  * merge_range_bounds
1380  * Create the partition bounds for a join relation between range
1381  * partitioned tables, if possible
1382  *
1383  * In this function we try to find sets of overlapping partitions from both
1384  * sides by comparing ranges stored in their partition bounds. Since the
1385  * ranges appear in the ascending order, an algorithm similar to merge join is
1386  * used for that. If a partition on one side doesn't have an overlapping
1387  * partition on the other side, the algorithm tries to match it with the
1388  * default partition on the other side if any; if not, the algorithm tries to
1389  * match it with a dummy partition on the other side if it's on the
1390  * non-nullable side of an outer join. Also, if both sides have the default
1391  * partitions, the algorithm tries to match them with each other. We give up
1392  * if the algorithm finds a partition overlapping multiple partitions on the
1393  * other side, which is the scenario the current implementation of partitioned
1394  * join can't handle.
1395  */
1396 static PartitionBoundInfo
1397 merge_range_bounds(int partnatts, FmgrInfo *partsupfuncs,
1398  Oid *partcollations,
1399  RelOptInfo *outer_rel, RelOptInfo *inner_rel,
1400  JoinType jointype,
1401  List **outer_parts, List **inner_parts)
1402 {
1403  PartitionBoundInfo merged_bounds = NULL;
1404  PartitionBoundInfo outer_bi = outer_rel->boundinfo;
1405  PartitionBoundInfo inner_bi = inner_rel->boundinfo;
1406  bool outer_has_default = partition_bound_has_default(outer_bi);
1407  bool inner_has_default = partition_bound_has_default(inner_bi);
1408  int outer_default = outer_bi->default_index;
1409  int inner_default = inner_bi->default_index;
1410  PartitionMap outer_map;
1411  PartitionMap inner_map;
1412  int outer_index;
1413  int inner_index;
1414  int outer_lb_pos;
1415  int inner_lb_pos;
1416  PartitionRangeBound outer_lb;
1417  PartitionRangeBound outer_ub;
1418  PartitionRangeBound inner_lb;
1419  PartitionRangeBound inner_ub;
1420  int next_index = 0;
1421  int default_index = -1;
1422  List *merged_datums = NIL;
1423  List *merged_kinds = NIL;
1424  List *merged_indexes = NIL;
1425 
1426  Assert(*outer_parts == NIL);
1427  Assert(*inner_parts == NIL);
1428  Assert(outer_bi->strategy == inner_bi->strategy &&
1429  outer_bi->strategy == PARTITION_STRATEGY_RANGE);
1430 
1431  init_partition_map(outer_rel, &outer_map);
1432  init_partition_map(inner_rel, &inner_map);
1433 
1434  /*
1435  * If the default partitions (if any) have been proven empty, deem them
1436  * non-existent.
1437  */
1438  if (outer_has_default && is_dummy_partition(outer_rel, outer_default))
1439  outer_has_default = false;
1440  if (inner_has_default && is_dummy_partition(inner_rel, inner_default))
1441  inner_has_default = false;
1442 
1443  /*
1444  * Merge partitions from both sides. In each iteration we compare a pair
1445  * of ranges, one from each side, and decide whether the corresponding
1446  * partitions match or not. If the two ranges overlap, move to the next
1447  * pair of ranges, otherwise move to the next range on the side with a
1448  * lower range. outer_lb_pos/inner_lb_pos keep track of the positions of
1449  * lower bounds in the datums arrays in the outer/inner
1450  * PartitionBoundInfos respectively.
1451  */
1452  outer_lb_pos = inner_lb_pos = 0;
1453  outer_index = get_range_partition(outer_rel, outer_bi, &outer_lb_pos,
1454  &outer_lb, &outer_ub);
1455  inner_index = get_range_partition(inner_rel, inner_bi, &inner_lb_pos,
1456  &inner_lb, &inner_ub);
1457  while (outer_index >= 0 || inner_index >= 0)
1458  {
1459  bool overlap;
1460  int ub_cmpval;
1461  int lb_cmpval;
1462  PartitionRangeBound merged_lb = {-1, NULL, NULL, true};
1463  PartitionRangeBound merged_ub = {-1, NULL, NULL, false};
1464  int merged_index = -1;
1465 
1466  /*
1467  * We run this loop till both sides finish. This allows us to avoid
1468  * duplicating code to handle the remaining ranges on the side which
1469  * finishes later. For that we set the comparison parameter cmpval in
1470  * such a way that it appears as if the side which finishes earlier
1471  * has an extra range higher than any other range on the unfinished
1472  * side. That way we advance the ranges on the unfinished side till
1473  * all of its ranges are exhausted.
1474  */
1475  if (outer_index == -1)
1476  {
1477  overlap = false;
1478  lb_cmpval = 1;
1479  ub_cmpval = 1;
1480  }
1481  else if (inner_index == -1)
1482  {
1483  overlap = false;
1484  lb_cmpval = -1;
1485  ub_cmpval = -1;
1486  }
1487  else
1488  overlap = compare_range_partitions(partnatts, partsupfuncs,
1489  partcollations,
1490  &outer_lb, &outer_ub,
1491  &inner_lb, &inner_ub,
1492  &lb_cmpval, &ub_cmpval);
1493 
1494  if (overlap)
1495  {
1496  /* Two ranges overlap; form a join pair. */
1497 
1498  PartitionRangeBound save_outer_ub;
1499  PartitionRangeBound save_inner_ub;
1500 
1501  /* Both partitions should not have been merged yet. */
1502  Assert(outer_index >= 0);
1503  Assert(outer_map.merged_indexes[outer_index] == -1 &&
1504  outer_map.merged[outer_index] == false);
1505  Assert(inner_index >= 0);
1506  Assert(inner_map.merged_indexes[inner_index] == -1 &&
1507  inner_map.merged[inner_index] == false);
1508 
1509  /*
1510  * Get the index of the merged partition. Both partitions aren't
1511  * merged yet, so the partitions should be merged successfully.
1512  */
1513  merged_index = merge_matching_partitions(&outer_map, &inner_map,
1514  outer_index, inner_index,
1515  &next_index);
1516  Assert(merged_index >= 0);
1517 
1518  /* Get the range bounds of the merged partition. */
1519  get_merged_range_bounds(partnatts, partsupfuncs,
1520  partcollations, jointype,
1521  &outer_lb, &outer_ub,
1522  &inner_lb, &inner_ub,
1523  lb_cmpval, ub_cmpval,
1524  &merged_lb, &merged_ub);
1525 
1526  /* Save the upper bounds of both partitions for use below. */
1527  save_outer_ub = outer_ub;
1528  save_inner_ub = inner_ub;
1529 
1530  /* Move to the next pair of ranges. */
1531  outer_index = get_range_partition(outer_rel, outer_bi, &outer_lb_pos,
1532  &outer_lb, &outer_ub);
1533  inner_index = get_range_partition(inner_rel, inner_bi, &inner_lb_pos,
1534  &inner_lb, &inner_ub);
1535 
1536  /*
1537  * If the range of a partition on one side overlaps the range of
1538  * the next partition on the other side, that will cause the
1539  * partition on one side to match at least two partitions on the
1540  * other side, which is the case that we currently don't support
1541  * partitioned join for; give up.
1542  */
1543  if (ub_cmpval > 0 && inner_index >= 0 &&
1544  compare_range_bounds(partnatts, partsupfuncs, partcollations,
1545  &save_outer_ub, &inner_lb) > 0)
1546  goto cleanup;
1547  if (ub_cmpval < 0 && outer_index >= 0 &&
1548  compare_range_bounds(partnatts, partsupfuncs, partcollations,
1549  &outer_lb, &save_inner_ub) < 0)
1550  goto cleanup;
1551 
1552  /*
1553  * A row from a non-overlapping portion (if any) of a partition on
1554  * one side might find its join partner in the default partition
1555  * (if any) on the other side, causing the same situation as
1556  * above; give up in that case.
1557  */
1558  if ((outer_has_default && (lb_cmpval > 0 || ub_cmpval < 0)) ||
1559  (inner_has_default && (lb_cmpval < 0 || ub_cmpval > 0)))
1560  goto cleanup;
1561  }
1562  else if (ub_cmpval < 0)
1563  {
1564  /* A non-overlapping outer range. */
1565 
1566  /* The outer partition should not have been merged yet. */
1567  Assert(outer_index >= 0);
1568  Assert(outer_map.merged_indexes[outer_index] == -1 &&
1569  outer_map.merged[outer_index] == false);
1570 
1571  /*
1572  * If the inner side has the default partition, or this is an
1573  * outer join, try to assign a merged partition to the outer
1574  * partition (see process_outer_partition()). Otherwise, the
1575  * outer partition will not contribute to the result.
1576  */
1577  if (inner_has_default || IS_OUTER_JOIN(jointype))
1578  {
1579  merged_index = process_outer_partition(&outer_map,
1580  &inner_map,
1581  outer_has_default,
1582  inner_has_default,
1583  outer_index,
1584  inner_default,
1585  jointype,
1586  &next_index,
1587  &default_index);
1588  if (merged_index == -1)
1589  goto cleanup;
1590  merged_lb = outer_lb;
1591  merged_ub = outer_ub;
1592  }
1593 
1594  /* Move to the next range on the outer side. */
1595  outer_index = get_range_partition(outer_rel, outer_bi, &outer_lb_pos,
1596  &outer_lb, &outer_ub);
1597  }
1598  else
1599  {
1600  /* A non-overlapping inner range. */
1601  Assert(ub_cmpval > 0);
1602 
1603  /* The inner partition should not have been merged yet. */
1604  Assert(inner_index >= 0);
1605  Assert(inner_map.merged_indexes[inner_index] == -1 &&
1606  inner_map.merged[inner_index] == false);
1607 
1608  /*
1609  * If the outer side has the default partition, or this is a FULL
1610  * join, try to assign a merged partition to the inner partition
1611  * (see process_inner_partition()). Otherwise, the inner
1612  * partition will not contribute to the result.
1613  */
1614  if (outer_has_default || jointype == JOIN_FULL)
1615  {
1616  merged_index = process_inner_partition(&outer_map,
1617  &inner_map,
1618  outer_has_default,
1619  inner_has_default,
1620  inner_index,
1621  outer_default,
1622  jointype,
1623  &next_index,
1624  &default_index);
1625  if (merged_index == -1)
1626  goto cleanup;
1627  merged_lb = inner_lb;
1628  merged_ub = inner_ub;
1629  }
1630 
1631  /* Move to the next range on the inner side. */
1632  inner_index = get_range_partition(inner_rel, inner_bi, &inner_lb_pos,
1633  &inner_lb, &inner_ub);
1634  }
1635 
1636  /*
1637  * If we assigned a merged partition, add the range bounds and index
1638  * of the merged partition if appropriate.
1639  */
1640  if (merged_index >= 0 && merged_index != default_index)
1641  add_merged_range_bounds(partnatts, partsupfuncs, partcollations,
1642  &merged_lb, &merged_ub, merged_index,
1643  &merged_datums, &merged_kinds,
1644  &merged_indexes);
1645  }
1646 
1647  /* Merge the default partitions if any. */
1648  if (outer_has_default || inner_has_default)
1649  merge_default_partitions(&outer_map, &inner_map,
1650  outer_has_default, inner_has_default,
1651  outer_default, inner_default,
1652  jointype, &next_index, &default_index);
1653  else
1654  Assert(default_index == -1);
1655 
1656  /* If we have merged partitions, create the partition bounds. */
1657  if (next_index > 0)
1658  {
1659  /*
1660  * Unlike the case of list partitioning, we wouldn't have re-merged
1661  * partitions, so did_remapping should be left alone.
1662  */
1663  Assert(!outer_map.did_remapping);
1664  Assert(!inner_map.did_remapping);
1665 
1666  /* Use maps to match partitions from inputs. */
1667  generate_matching_part_pairs(outer_rel, inner_rel,
1668  &outer_map, &inner_map,
1669  next_index,
1670  outer_parts, inner_parts);
1671  Assert(*outer_parts != NIL);
1672  Assert(*inner_parts != NIL);
1673  Assert(list_length(*outer_parts) == list_length(*inner_parts));
1674  Assert(list_length(*outer_parts) == next_index);
1675 
1676  /* Make a PartitionBoundInfo struct to return. */
1677  merged_bounds = build_merged_partition_bounds(outer_bi->strategy,
1678  merged_datums,
1679  merged_kinds,
1680  merged_indexes,
1681  -1,
1682  default_index);
1683  Assert(merged_bounds);
1684  }
1685 
1686 cleanup:
1687  /* Free local memory before returning. */
1688  list_free(merged_datums);
1689  list_free(merged_kinds);
1690  list_free(merged_indexes);
1691  free_partition_map(&outer_map);
1692  free_partition_map(&inner_map);
1693 
1694  return merged_bounds;
1695 }
1696 
1697 /*
1698  * init_partition_map
1699  * Initialize a PartitionMap struct for given relation
1700  */
1701 static void
1702 init_partition_map(RelOptInfo *rel, PartitionMap *map)
1703 {
1704  int nparts = rel->nparts;
1705  int i;
1706 
1707  map->nparts = nparts;
1708  map->merged_indexes = (int *) palloc(sizeof(int) * nparts);
1709  map->merged = (bool *) palloc(sizeof(bool) * nparts);
1710  map->did_remapping = false;
1711  map->old_indexes = (int *) palloc(sizeof(int) * nparts);
1712  for (i = 0; i < nparts; i++)
1713  {
1714  map->merged_indexes[i] = map->old_indexes[i] = -1;
1715  map->merged[i] = false;
1716  }
1717 }
1718 
1719 /*
1720  * free_partition_map
1721  */
1722 static void
1723 free_partition_map(PartitionMap *map)
1724 {
1725  pfree(map->merged_indexes);
1726  pfree(map->merged);
1727  pfree(map->old_indexes);
1728 }
1729 
1730 /*
1731  * is_dummy_partition --- has partition been proven empty?
1732  */
1733 static bool
1734 is_dummy_partition(RelOptInfo *rel, int part_index)
1735 {
1736  RelOptInfo *part_rel;
1737 
1738  Assert(part_index >= 0);
1739  part_rel = rel->part_rels[part_index];
1740  if (part_rel == NULL || IS_DUMMY_REL(part_rel))
1741  return true;
1742  return false;
1743 }
1744 
1745 /*
1746  * merge_matching_partitions
1747  * Try to merge given outer/inner partitions, and return the index of a
1748  * merged partition produced from them if successful, -1 otherwise
1749  *
1750  * If the merged partition is newly created, *next_index is incremented.
1751  */
1752 static int
1753 merge_matching_partitions(PartitionMap *outer_map, PartitionMap *inner_map,
1754  int outer_index, int inner_index, int *next_index)
1755 {
1756  int outer_merged_index;
1757  int inner_merged_index;
1758  bool outer_merged;
1759  bool inner_merged;
1760 
1761  Assert(outer_index >= 0 && outer_index < outer_map->nparts);
1762  outer_merged_index = outer_map->merged_indexes[outer_index];
1763  outer_merged = outer_map->merged[outer_index];
1764  Assert(inner_index >= 0 && inner_index < inner_map->nparts);
1765  inner_merged_index = inner_map->merged_indexes[inner_index];
1766  inner_merged = inner_map->merged[inner_index];
1767 
1768  /*
1769  * Handle cases where we have already assigned a merged partition to each
1770  * of the given partitions.
1771  */
1772  if (outer_merged_index >= 0 && inner_merged_index >= 0)
1773  {
1774  /*
1775  * If the merged partitions are the same, no need to do anything;
1776  * return the index of the merged partitions. Otherwise, if each of
1777  * the given partitions has been merged with a dummy partition on the
1778  * other side, re-map them to either of the two merged partitions.
1779  * Otherwise, they can't be merged, so return -1.
1780  */
1781  if (outer_merged_index == inner_merged_index)
1782  {
1783  Assert(outer_merged);
1784  Assert(inner_merged);
1785  return outer_merged_index;
1786  }
1787  if (!outer_merged && !inner_merged)
1788  {
1789  /*
1790  * This can only happen for a list-partitioning case. We re-map
1791  * them to the merged partition with the smaller of the two merged
1792  * indexes to preserve the property that the canonical order of
1793  * list partitions is determined by the indexes assigned to the
1794  * smallest list value of each partition.
1795  */
1796  if (outer_merged_index < inner_merged_index)
1797  {
1798  outer_map->merged[outer_index] = true;
1799  inner_map->merged_indexes[inner_index] = outer_merged_index;
1800  inner_map->merged[inner_index] = true;
1801  inner_map->did_remapping = true;
1802  inner_map->old_indexes[inner_index] = inner_merged_index;
1803  return outer_merged_index;
1804  }
1805  else
1806  {
1807  inner_map->merged[inner_index] = true;
1808  outer_map->merged_indexes[outer_index] = inner_merged_index;
1809  outer_map->merged[outer_index] = true;
1810  outer_map->did_remapping = true;
1811  outer_map->old_indexes[outer_index] = outer_merged_index;
1812  return inner_merged_index;
1813  }
1814  }
1815  return -1;
1816  }
1817 
1818  /* At least one of the given partitions should not have yet been merged. */
1819  Assert(outer_merged_index == -1 || inner_merged_index == -1);
1820 
1821  /*
1822  * If neither of them has been merged, merge them. Otherwise, if one has
1823  * been merged with a dummy partition on the other side (and the other
1824  * hasn't yet been merged with anything), re-merge them. Otherwise, they
1825  * can't be merged, so return -1.
1826  */
1827  if (outer_merged_index == -1 && inner_merged_index == -1)
1828  {
1829  int merged_index = *next_index;
1830 
1831  Assert(!outer_merged);
1832  Assert(!inner_merged);
1833  outer_map->merged_indexes[outer_index] = merged_index;
1834  outer_map->merged[outer_index] = true;
1835  inner_map->merged_indexes[inner_index] = merged_index;
1836  inner_map->merged[inner_index] = true;
1837  *next_index = *next_index + 1;
1838  return merged_index;
1839  }
1840  if (outer_merged_index >= 0 && !outer_map->merged[outer_index])
1841  {
1842  Assert(inner_merged_index == -1);
1843  Assert(!inner_merged);
1844  inner_map->merged_indexes[inner_index] = outer_merged_index;
1845  inner_map->merged[inner_index] = true;
1846  outer_map->merged[outer_index] = true;
1847  return outer_merged_index;
1848  }
1849  if (inner_merged_index >= 0 && !inner_map->merged[inner_index])
1850  {
1851  Assert(outer_merged_index == -1);
1852  Assert(!outer_merged);
1853  outer_map->merged_indexes[outer_index] = inner_merged_index;
1854  outer_map->merged[outer_index] = true;
1855  inner_map->merged[inner_index] = true;
1856  return inner_merged_index;
1857  }
1858  return -1;
1859 }
1860 
1861 /*
1862  * process_outer_partition
1863  * Try to assign given outer partition a merged partition, and return the
1864  * index of the merged partition if successful, -1 otherwise
1865  *
1866  * If the partition is newly created, *next_index is incremented. Also, if it
1867  * is the default partition of the join relation, *default_index is set to the
1868  * index if not already done.
1869  */
1870 static int
1871 process_outer_partition(PartitionMap *outer_map,
1872  PartitionMap *inner_map,
1873  bool outer_has_default,
1874  bool inner_has_default,
1875  int outer_index,
1876  int inner_default,
1877  JoinType jointype,
1878  int *next_index,
1879  int *default_index)
1880 {
1881  int merged_index = -1;
1882 
1883  Assert(outer_index >= 0);
1884 
1885  /*
1886  * If the inner side has the default partition, a row from the outer
1887  * partition might find its join partner in the default partition; try
1888  * merging the outer partition with the default partition. Otherwise,
1889  * this should be an outer join, in which case the outer partition has to
1890  * be scanned all the way anyway; merge the outer partition with a dummy
1891  * partition on the other side.
1892  */
1893  if (inner_has_default)
1894  {
1895  Assert(inner_default >= 0);
1896 
1897  /*
1898  * If the outer side has the default partition as well, the default
1899  * partition on the inner side will have two matching partitions on
1900  * the other side: the outer partition and the default partition on
1901  * the outer side. Partitionwise join doesn't handle this scenario
1902  * yet.
1903  */
1904  if (outer_has_default)
1905  return -1;
1906 
1907  merged_index = merge_matching_partitions(outer_map, inner_map,
1908  outer_index, inner_default,
1909  next_index);
1910  if (merged_index == -1)
1911  return -1;
1912 
1913  /*
1914  * If this is a FULL join, the default partition on the inner side has
1915  * to be scanned all the way anyway, so the resulting partition will
1916  * contain all key values from the default partition, which any other
1917  * partition of the join relation will not contain. Thus the
1918  * resulting partition will act as the default partition of the join
1919  * relation; record the index in *default_index if not already done.
1920  */
1921  if (jointype == JOIN_FULL)
1922  {
1923  if (*default_index == -1)
1924  *default_index = merged_index;
1925  else
1926  Assert(*default_index == merged_index);
1927  }
1928  }
1929  else
1930  {
1931  Assert(IS_OUTER_JOIN(jointype));
1932  Assert(jointype != JOIN_RIGHT);
1933 
1934  /* If we have already assigned a partition, no need to do anything. */
1935  merged_index = outer_map->merged_indexes[outer_index];
1936  if (merged_index == -1)
1937  merged_index = merge_partition_with_dummy(outer_map, outer_index,
1938  next_index);
1939  }
1940  return merged_index;
1941 }
1942 
1943 /*
1944  * process_inner_partition
1945  * Try to assign given inner partition a merged partition, and return the
1946  * index of the merged partition if successful, -1 otherwise
1947  *
1948  * If the partition is newly created, *next_index is incremented. Also, if it
1949  * is the default partition of the join relation, *default_index is set to the
1950  * index if not already done.
1951  */
1952 static int
1953 process_inner_partition(PartitionMap *outer_map,
1954  PartitionMap *inner_map,
1955  bool outer_has_default,
1956  bool inner_has_default,
1957  int inner_index,
1958  int outer_default,
1959  JoinType jointype,
1960  int *next_index,
1961  int *default_index)
1962 {
1963  int merged_index = -1;
1964 
1965  Assert(inner_index >= 0);
1966 
1967  /*
1968  * If the outer side has the default partition, a row from the inner
1969  * partition might find its join partner in the default partition; try
1970  * merging the inner partition with the default partition. Otherwise,
1971  * this should be a FULL join, in which case the inner partition has to be
1972  * scanned all the way anyway; merge the inner partition with a dummy
1973  * partition on the other side.
1974  */
1975  if (outer_has_default)
1976  {
1977  Assert(outer_default >= 0);
1978 
1979  /*
1980  * If the inner side has the default partition as well, the default
1981  * partition on the outer side will have two matching partitions on
1982  * the other side: the inner partition and the default partition on
1983  * the inner side. Partitionwise join doesn't handle this scenario
1984  * yet.
1985  */
1986  if (inner_has_default)
1987  return -1;
1988 
1989  merged_index = merge_matching_partitions(outer_map, inner_map,
1990  outer_default, inner_index,
1991  next_index);
1992  if (merged_index == -1)
1993  return -1;
1994 
1995  /*
1996  * If this is an outer join, the default partition on the outer side
1997  * has to be scanned all the way anyway, so the resulting partition
1998  * will contain all key values from the default partition, which any
1999  * other partition of the join relation will not contain. Thus the
2000  * resulting partition will act as the default partition of the join
2001  * relation; record the index in *default_index if not already done.
2002  */
2003  if (IS_OUTER_JOIN(jointype))
2004  {
2005  Assert(jointype != JOIN_RIGHT);
2006  if (*default_index == -1)
2007  *default_index = merged_index;
2008  else
2009  Assert(*default_index == merged_index);
2010  }
2011  }
2012  else
2013  {
2014  Assert(jointype == JOIN_FULL);
2015 
2016  /* If we have already assigned a partition, no need to do anything. */
2017  merged_index = inner_map->merged_indexes[inner_index];
2018  if (merged_index == -1)
2019  merged_index = merge_partition_with_dummy(inner_map, inner_index,
2020  next_index);
2021  }
2022  return merged_index;
2023 }
2024 
2025 /*
2026  * merge_null_partitions
2027  * Merge the NULL partitions from a join's outer and inner sides.
2028  *
2029  * If the merged partition produced from them is the NULL partition of the join
2030  * relation, *null_index is set to the index of the merged partition.
2031  *
2032  * Note: We assume here that the join clause for a partitioned join is strict
2033  * because have_partkey_equi_join() requires that the corresponding operator
2034  * be mergejoinable, and we currently assume that mergejoinable operators are
2035  * strict (see MJEvalOuterValues()/MJEvalInnerValues()).
2036  */
2037 static void
2038 merge_null_partitions(PartitionMap *outer_map,
2039  PartitionMap *inner_map,
2040  bool outer_has_null,
2041  bool inner_has_null,
2042  int outer_null,
2043  int inner_null,
2044  JoinType jointype,
2045  int *next_index,
2046  int *null_index)
2047 {
2048  bool consider_outer_null = false;
2049  bool consider_inner_null = false;
2050 
2051  Assert(outer_has_null || inner_has_null);
2052  Assert(*null_index == -1);
2053 
2054  /*
2055  * Check whether the NULL partitions have already been merged and if so,
2056  * set the consider_outer_null/consider_inner_null flags.
2057  */
2058  if (outer_has_null)
2059  {
2060  Assert(outer_null >= 0 && outer_null < outer_map->nparts);
2061  if (outer_map->merged_indexes[outer_null] == -1)
2062  consider_outer_null = true;
2063  }
2064  if (inner_has_null)
2065  {
2066  Assert(inner_null >= 0 && inner_null < inner_map->nparts);
2067  if (inner_map->merged_indexes[inner_null] == -1)
2068  consider_inner_null = true;
2069  }
2070 
2071  /* If both flags are set false, we don't need to do anything. */
2072  if (!consider_outer_null && !consider_inner_null)
2073  return;
2074 
2075  if (consider_outer_null && !consider_inner_null)
2076  {
2077  Assert(outer_has_null);
2078 
2079  /*
2080  * If this is an outer join, the NULL partition on the outer side has
2081  * to be scanned all the way anyway; merge the NULL partition with a
2082  * dummy partition on the other side. In that case
2083  * consider_outer_null means that the NULL partition only contains
2084  * NULL values as the key values, so the merged partition will do so;
2085  * treat it as the NULL partition of the join relation.
2086  */
2087  if (IS_OUTER_JOIN(jointype))
2088  {
2089  Assert(jointype != JOIN_RIGHT);
2090  *null_index = merge_partition_with_dummy(outer_map, outer_null,
2091  next_index);
2092  }
2093  }
2094  else if (!consider_outer_null && consider_inner_null)
2095  {
2096  Assert(inner_has_null);
2097 
2098  /*
2099  * If this is a FULL join, the NULL partition on the inner side has to
2100  * be scanned all the way anyway; merge the NULL partition with a
2101  * dummy partition on the other side. In that case
2102  * consider_inner_null means that the NULL partition only contains
2103  * NULL values as the key values, so the merged partition will do so;
2104  * treat it as the NULL partition of the join relation.
2105  */
2106  if (jointype == JOIN_FULL)
2107  *null_index = merge_partition_with_dummy(inner_map, inner_null,
2108  next_index);
2109  }
2110  else
2111  {
2112  Assert(consider_outer_null && consider_inner_null);
2113  Assert(outer_has_null);
2114  Assert(inner_has_null);
2115 
2116  /*
2117  * If this is an outer join, the NULL partition on the outer side (and
2118  * that on the inner side if this is a FULL join) have to be scanned
2119  * all the way anyway, so merge them. Note that each of the NULL
2120  * partitions isn't merged yet, so they should be merged successfully.
2121  * Like the above, each of the NULL partitions only contains NULL
2122  * values as the key values, so the merged partition will do so; treat
2123  * it as the NULL partition of the join relation.
2124  *
2125  * Note: if this an INNER/SEMI join, the join clause will never be
2126  * satisfied by two NULL values (see comments above), so both the NULL
2127  * partitions can be eliminated.
2128  */
2129  if (IS_OUTER_JOIN(jointype))
2130  {
2131  Assert(jointype != JOIN_RIGHT);
2132  *null_index = merge_matching_partitions(outer_map, inner_map,
2133  outer_null, inner_null,
2134  next_index);
2135  Assert(*null_index >= 0);
2136  }
2137  }
2138 }
2139 
2140 /*
2141  * merge_default_partitions
2142  * Merge the default partitions from a join's outer and inner sides.
2143  *
2144  * If the merged partition produced from them is the default partition of the
2145  * join relation, *default_index is set to the index of the merged partition.
2146  */
2147 static void
2148 merge_default_partitions(PartitionMap *outer_map,
2149  PartitionMap *inner_map,
2150  bool outer_has_default,
2151  bool inner_has_default,
2152  int outer_default,
2153  int inner_default,
2154  JoinType jointype,
2155  int *next_index,
2156  int *default_index)
2157 {
2158  int outer_merged_index = -1;
2159  int inner_merged_index = -1;
2160 
2161  Assert(outer_has_default || inner_has_default);
2162 
2163  /* Get the merged partition indexes for the default partitions. */
2164  if (outer_has_default)
2165  {
2166  Assert(outer_default >= 0 && outer_default < outer_map->nparts);
2167  outer_merged_index = outer_map->merged_indexes[outer_default];
2168  }
2169  if (inner_has_default)
2170  {
2171  Assert(inner_default >= 0 && inner_default < inner_map->nparts);
2172  inner_merged_index = inner_map->merged_indexes[inner_default];
2173  }
2174 
2175  if (outer_has_default && !inner_has_default)
2176  {
2177  /*
2178  * If this is an outer join, the default partition on the outer side
2179  * has to be scanned all the way anyway; if we have not yet assigned a
2180  * partition, merge the default partition with a dummy partition on
2181  * the other side. The merged partition will act as the default
2182  * partition of the join relation (see comments in
2183  * process_inner_partition()).
2184  */
2185  if (IS_OUTER_JOIN(jointype))
2186  {
2187  Assert(jointype != JOIN_RIGHT);
2188  if (outer_merged_index == -1)
2189  {
2190  Assert(*default_index == -1);
2191  *default_index = merge_partition_with_dummy(outer_map,
2192  outer_default,
2193  next_index);
2194  }
2195  else
2196  Assert(*default_index == outer_merged_index);
2197  }
2198  else
2199  Assert(*default_index == -1);
2200  }
2201  else if (!outer_has_default && inner_has_default)
2202  {
2203  /*
2204  * If this is a FULL join, the default partition on the inner side has
2205  * to be scanned all the way anyway; if we have not yet assigned a
2206  * partition, merge the default partition with a dummy partition on
2207  * the other side. The merged partition will act as the default
2208  * partition of the join relation (see comments in
2209  * process_outer_partition()).
2210  */
2211  if (jointype == JOIN_FULL)
2212  {
2213  if (inner_merged_index == -1)
2214  {
2215  Assert(*default_index == -1);
2216  *default_index = merge_partition_with_dummy(inner_map,
2217  inner_default,
2218  next_index);
2219  }
2220  else
2221  Assert(*default_index == inner_merged_index);
2222  }
2223  else
2224  Assert(*default_index == -1);
2225  }
2226  else
2227  {
2228  Assert(outer_has_default && inner_has_default);
2229 
2230  /*
2231  * The default partitions have to be joined with each other, so merge
2232  * them. Note that each of the default partitions isn't merged yet
2233  * (see, process_outer_partition()/process_innerer_partition()), so
2234  * they should be merged successfully. The merged partition will act
2235  * as the default partition of the join relation.
2236  */
2237  Assert(outer_merged_index == -1);
2238  Assert(inner_merged_index == -1);
2239  Assert(*default_index == -1);
2240  *default_index = merge_matching_partitions(outer_map,
2241  inner_map,
2242  outer_default,
2243  inner_default,
2244  next_index);
2245  Assert(*default_index >= 0);
2246  }
2247 }
2248 
2249 /*
2250  * merge_partition_with_dummy
2251  * Assign given partition a new partition of a join relation
2252  *
2253  * Note: The caller assumes that the given partition doesn't have a non-dummy
2254  * matching partition on the other side, but if the given partition finds the
2255  * matching partition later, we will adjust the assignment.
2256  */
2257 static int
2258 merge_partition_with_dummy(PartitionMap *map, int index, int *next_index)
2259 {
2260  int merged_index = *next_index;
2261 
2262  Assert(index >= 0 && index < map->nparts);
2263  Assert(map->merged_indexes[index] == -1);
2264  Assert(!map->merged[index]);
2265  map->merged_indexes[index] = merged_index;
2266  /* Leave the merged flag alone! */
2267  *next_index = *next_index + 1;
2268  return merged_index;
2269 }
2270 
2271 /*
2272  * fix_merged_indexes
2273  * Adjust merged indexes of re-merged partitions
2274  */
2275 static void
2276 fix_merged_indexes(PartitionMap *outer_map, PartitionMap *inner_map,
2277  int nmerged, List *merged_indexes)
2278 {
2279  int *new_indexes;
2280  int merged_index;
2281  int i;
2282  ListCell *lc;
2283 
2284  Assert(nmerged > 0);
2285 
2286  new_indexes = (int *) palloc(sizeof(int) * nmerged);
2287  for (i = 0; i < nmerged; i++)
2288  new_indexes[i] = -1;
2289 
2290  /* Build the mapping of old merged indexes to new merged indexes. */
2291  if (outer_map->did_remapping)
2292  {
2293  for (i = 0; i < outer_map->nparts; i++)
2294  {
2295  merged_index = outer_map->old_indexes[i];
2296  if (merged_index >= 0)
2297  new_indexes[merged_index] = outer_map->merged_indexes[i];
2298  }
2299  }
2300  if (inner_map->did_remapping)
2301  {
2302  for (i = 0; i < inner_map->nparts; i++)
2303  {
2304  merged_index = inner_map->old_indexes[i];
2305  if (merged_index >= 0)
2306  new_indexes[merged_index] = inner_map->merged_indexes[i];
2307  }
2308  }
2309 
2310  /* Fix the merged_indexes list using the mapping. */
2311  foreach(lc, merged_indexes)
2312  {
2313  merged_index = lfirst_int(lc);
2314  Assert(merged_index >= 0);
2315  if (new_indexes[merged_index] >= 0)
2316  lfirst_int(lc) = new_indexes[merged_index];
2317  }
2318 
2319  pfree(new_indexes);
2320 }
2321 
2322 /*
2323  * generate_matching_part_pairs
2324  * Generate a pair of lists of partitions that produce merged partitions
2325  *
2326  * The lists of partitions are built in the order of merged partition indexes,
2327  * and returned in *outer_parts and *inner_parts.
2328  */
2329 static void
2330 generate_matching_part_pairs(RelOptInfo *outer_rel, RelOptInfo *inner_rel,
2331  PartitionMap *outer_map, PartitionMap *inner_map,
2332  int nmerged,
2333  List **outer_parts, List **inner_parts)
2334 {
2335  int outer_nparts = outer_map->nparts;
2336  int inner_nparts = inner_map->nparts;
2337  int *outer_indexes;
2338  int *inner_indexes;
2339  int max_nparts;
2340  int i;
2341 
2342  Assert(nmerged > 0);
2343  Assert(*outer_parts == NIL);
2344  Assert(*inner_parts == NIL);
2345 
2346  outer_indexes = (int *) palloc(sizeof(int) * nmerged);
2347  inner_indexes = (int *) palloc(sizeof(int) * nmerged);
2348  for (i = 0; i < nmerged; i++)
2349  outer_indexes[i] = inner_indexes[i] = -1;
2350 
2351  /* Set pairs of matching partitions. */
2352  Assert(outer_nparts == outer_rel->nparts);
2353  Assert(inner_nparts == inner_rel->nparts);
2354  max_nparts = Max(outer_nparts, inner_nparts);
2355  for (i = 0; i < max_nparts; i++)
2356  {
2357  if (i < outer_nparts)
2358  {
2359  int merged_index = outer_map->merged_indexes[i];
2360 
2361  if (merged_index >= 0)
2362  {
2363  Assert(merged_index < nmerged);
2364  outer_indexes[merged_index] = i;
2365  }
2366  }
2367  if (i < inner_nparts)
2368  {
2369  int merged_index = inner_map->merged_indexes[i];
2370 
2371  if (merged_index >= 0)
2372  {
2373  Assert(merged_index < nmerged);
2374  inner_indexes[merged_index] = i;
2375  }
2376  }
2377  }
2378 
2379  /* Build the list pairs. */
2380  for (i = 0; i < nmerged; i++)
2381  {
2382  int outer_index = outer_indexes[i];
2383  int inner_index = inner_indexes[i];
2384 
2385  /*
2386  * If both partitions are dummy, it means the merged partition that
2387  * had been assigned to the outer/inner partition was removed when
2388  * re-merging the outer/inner partition in
2389  * merge_matching_partitions(); ignore the merged partition.
2390  */
2391  if (outer_index == -1 && inner_index == -1)
2392  continue;
2393 
2394  *outer_parts = lappend(*outer_parts, outer_index >= 0 ?
2395  outer_rel->part_rels[outer_index] : NULL);
2396  *inner_parts = lappend(*inner_parts, inner_index >= 0 ?
2397  inner_rel->part_rels[inner_index] : NULL);
2398  }
2399 
2400  pfree(outer_indexes);
2401  pfree(inner_indexes);
2402 }
2403 
2404 /*
2405  * build_merged_partition_bounds
2406  * Create a PartitionBoundInfo struct from merged partition bounds
2407  */
2408 static PartitionBoundInfo
2409 build_merged_partition_bounds(char strategy, List *merged_datums,
2410  List *merged_kinds, List *merged_indexes,
2411  int null_index, int default_index)
2412 {
2413  PartitionBoundInfo merged_bounds;
2414  int ndatums = list_length(merged_datums);
2415  int pos;
2416  ListCell *lc;
2417 
2418  merged_bounds = (PartitionBoundInfo) palloc(sizeof(PartitionBoundInfoData));
2419  merged_bounds->strategy = strategy;
2420  merged_bounds->ndatums = ndatums;
2421 
2422  merged_bounds->datums = (Datum **) palloc(sizeof(Datum *) * ndatums);
2423  pos = 0;
2424  foreach(lc, merged_datums)
2425  merged_bounds->datums[pos++] = (Datum *) lfirst(lc);
2426 
2427  if (strategy == PARTITION_STRATEGY_RANGE)
2428  {
2429  Assert(list_length(merged_kinds) == ndatums);
2430  merged_bounds->kind = (PartitionRangeDatumKind **)
2431  palloc(sizeof(PartitionRangeDatumKind *) * ndatums);
2432  pos = 0;
2433  foreach(lc, merged_kinds)
2434  merged_bounds->kind[pos++] = (PartitionRangeDatumKind *) lfirst(lc);
2435 
2436  /* There are ndatums+1 indexes in the case of range partitioning. */
2437  merged_indexes = lappend_int(merged_indexes, -1);
2438  ndatums++;
2439  }
2440  else
2441  {
2442  Assert(strategy == PARTITION_STRATEGY_LIST);
2443  Assert(merged_kinds == NIL);
2444  merged_bounds->kind = NULL;
2445  }
2446 
2447  Assert(list_length(merged_indexes) == ndatums);
2448  merged_bounds->nindexes = ndatums;
2449  merged_bounds->indexes = (int *) palloc(sizeof(int) * ndatums);
2450  pos = 0;
2451  foreach(lc, merged_indexes)
2452  merged_bounds->indexes[pos++] = lfirst_int(lc);
2453 
2454  merged_bounds->null_index = null_index;
2455  merged_bounds->default_index = default_index;
2456 
2457  return merged_bounds;
2458 }
2459 
2460 /*
2461  * get_range_partition
2462  * Get the next non-dummy partition of a range-partitioned relation,
2463  * returning the index of that partition
2464  *
2465  * *lb and *ub are set to the lower and upper bounds of that partition
2466  * respectively, and *lb_pos is advanced to the next lower bound, if any.
2467  */
2468 static int
2469 get_range_partition(RelOptInfo *rel,
2470  PartitionBoundInfo bi,
2471  int *lb_pos,
2472  PartitionRangeBound *lb,
2473  PartitionRangeBound *ub)
2474 {
2475  int part_index;
2476 
2477  Assert(bi->strategy == PARTITION_STRATEGY_RANGE);
2478 
2479  do
2480  {
2481  part_index = get_range_partition_internal(bi, lb_pos, lb, ub);
2482  if (part_index == -1)
2483  return -1;
2484  } while (is_dummy_partition(rel, part_index));
2485 
2486  return part_index;
2487 }
2488 
2489 static int
2490 get_range_partition_internal(PartitionBoundInfo bi,
2491  int *lb_pos,
2492  PartitionRangeBound *lb,
2493  PartitionRangeBound *ub)
2494 {
2495  /* Return the index as -1 if we've exhausted all lower bounds. */
2496  if (*lb_pos >= bi->ndatums)
2497  return -1;
2498 
2499  /* A lower bound should have at least one more bound after it. */
2500  Assert(*lb_pos + 1 < bi->ndatums);
2501 
2502  /* Set the lower bound. */
2503  lb->index = bi->indexes[*lb_pos];
2504  lb->datums = bi->datums[*lb_pos];
2505  lb->kind = bi->kind[*lb_pos];
2506  lb->lower = true;
2507  /* Set the upper bound. */
2508  ub->index = bi->indexes[*lb_pos + 1];
2509  ub->datums = bi->datums[*lb_pos + 1];
2510  ub->kind = bi->kind[*lb_pos + 1];
2511  ub->lower = false;
2512 
2513  /* The index assigned to an upper bound should be valid. */
2514  Assert(ub->index >= 0);
2515 
2516  /*
2517  * Advance the position to the next lower bound. If there are no bounds
2518  * left beyond the upper bound, we have reached the last lower bound.
2519  */
2520  if (*lb_pos + 2 >= bi->ndatums)
2521  *lb_pos = bi->ndatums;
2522  else
2523  {
2524  /*
2525  * If the index assigned to the bound next to the upper bound isn't
2526  * valid, that is the next lower bound; else, the upper bound is also
2527  * the lower bound of the next range partition.
2528  */
2529  if (bi->indexes[*lb_pos + 2] < 0)
2530  *lb_pos = *lb_pos + 2;
2531  else
2532  *lb_pos = *lb_pos + 1;
2533  }
2534 
2535  return ub->index;
2536 }
2537 
2538 /*
2539  * compare_range_partitions
2540  * Compare the bounds of two range partitions, and return true if the
2541  * two partitions overlap, false otherwise
2542  *
2543  * *lb_cmpval is set to -1, 0, or 1 if the outer partition's lower bound is
2544  * lower than, equal to, or higher than the inner partition's lower bound
2545  * respectively. Likewise, *ub_cmpval is set to -1, 0, or 1 if the outer
2546  * partition's upper bound is lower than, equal to, or higher than the inner
2547  * partition's upper bound respectively.
2548  */
2549 static bool
2550 compare_range_partitions(int partnatts, FmgrInfo *partsupfuncs,
2551  Oid *partcollations,
2552  PartitionRangeBound *outer_lb,
2553  PartitionRangeBound *outer_ub,
2554  PartitionRangeBound *inner_lb,
2555  PartitionRangeBound *inner_ub,
2556  int *lb_cmpval, int *ub_cmpval)
2557 {
2558  /*
2559  * Check if the outer partition's upper bound is lower than the inner
2560  * partition's lower bound; if so the partitions aren't overlapping.
2561  */
2562  if (compare_range_bounds(partnatts, partsupfuncs, partcollations,
2563  outer_ub, inner_lb) < 0)
2564  {
2565  *lb_cmpval = -1;
2566  *ub_cmpval = -1;
2567  return false;
2568  }
2569 
2570  /*
2571  * Check if the outer partition's lower bound is higher than the inner
2572  * partition's upper bound; if so the partitions aren't overlapping.
2573  */
2574  if (compare_range_bounds(partnatts, partsupfuncs, partcollations,
2575  outer_lb, inner_ub) > 0)
2576  {
2577  *lb_cmpval = 1;
2578  *ub_cmpval = 1;
2579  return false;
2580  }
2581 
2582  /* All other cases indicate overlapping partitions. */
2583  *lb_cmpval = compare_range_bounds(partnatts, partsupfuncs, partcollations,
2584  outer_lb, inner_lb);
2585  *ub_cmpval = compare_range_bounds(partnatts, partsupfuncs, partcollations,
2586  outer_ub, inner_ub);
2587  return true;
2588 }
2589 
2590 /*
2591  * get_merged_range_bounds
2592  * Given the bounds of range partitions to be joined, determine the bounds
2593  * of a merged partition produced from the range partitions
2594  *
2595  * *merged_lb and *merged_ub are set to the lower and upper bounds of the
2596  * merged partition.
2597  */
2598 static void
2599 get_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs,
2600  Oid *partcollations, JoinType jointype,
2601  PartitionRangeBound *outer_lb,
2602  PartitionRangeBound *outer_ub,
2603  PartitionRangeBound *inner_lb,
2604  PartitionRangeBound *inner_ub,
2605  int lb_cmpval, int ub_cmpval,
2606  PartitionRangeBound *merged_lb,
2607  PartitionRangeBound *merged_ub)
2608 {
2609  Assert(compare_range_bounds(partnatts, partsupfuncs, partcollations,
2610  outer_lb, inner_lb) == lb_cmpval);
2611  Assert(compare_range_bounds(partnatts, partsupfuncs, partcollations,
2612  outer_ub, inner_ub) == ub_cmpval);
2613 
2614  switch (jointype)
2615  {
2616  case JOIN_INNER:
2617  case JOIN_SEMI:
2618 
2619  /*
2620  * An INNER/SEMI join will have the rows that fit both sides, so
2621  * the lower bound of the merged partition will be the higher of
2622  * the two lower bounds, and the upper bound of the merged
2623  * partition will be the lower of the two upper bounds.
2624  */
2625  *merged_lb = (lb_cmpval > 0) ? *outer_lb : *inner_lb;
2626  *merged_ub = (ub_cmpval < 0) ? *outer_ub : *inner_ub;
2627  break;
2628 
2629  case JOIN_LEFT:
2630  case JOIN_ANTI:
2631 
2632  /*
2633  * A LEFT/ANTI join will have all the rows from the outer side, so
2634  * the bounds of the merged partition will be the same as the
2635  * outer bounds.
2636  */
2637  *merged_lb = *outer_lb;
2638  *merged_ub = *outer_ub;
2639  break;
2640 
2641  case JOIN_FULL:
2642 
2643  /*
2644  * A FULL join will have all the rows from both sides, so the
2645  * lower bound of the merged partition will be the lower of the
2646  * two lower bounds, and the upper bound of the merged partition
2647  * will be the higher of the two upper bounds.
2648  */
2649  *merged_lb = (lb_cmpval < 0) ? *outer_lb : *inner_lb;
2650  *merged_ub = (ub_cmpval > 0) ? *outer_ub : *inner_ub;
2651  break;
2652 
2653  default:
2654  elog(ERROR, "unrecognized join type: %d", (int) jointype);
2655  }
2656 }
2657 
2658 /*
2659  * add_merged_range_bounds
2660  * Add the bounds of a merged partition to the lists of range bounds
2661  */
2662 static void
2663 add_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs,
2664  Oid *partcollations,
2665  PartitionRangeBound *merged_lb,
2666  PartitionRangeBound *merged_ub,
2667  int merged_index,
2668  List **merged_datums,
2669  List **merged_kinds,
2670  List **merged_indexes)
2671 {
2672  int cmpval;
2673 
2674  if (!*merged_datums)
2675  {
2676  /* First merged partition */
2677  Assert(!*merged_kinds);
2678  Assert(!*merged_indexes);
2679  cmpval = 1;
2680  }
2681  else
2682  {
2683  PartitionRangeBound prev_ub;
2684 
2685  Assert(*merged_datums);
2686  Assert(*merged_kinds);
2687  Assert(*merged_indexes);
2688 
2689  /* Get the last upper bound. */
2690  prev_ub.index = llast_int(*merged_indexes);
2691  prev_ub.datums = (Datum *) llast(*merged_datums);
2692  prev_ub.kind = (PartitionRangeDatumKind *) llast(*merged_kinds);
2693  prev_ub.lower = false;
2694 
2695  /*
2696  * We pass lower1 = false to partition_rbound_cmp() to prevent it from
2697  * considering the last upper bound to be smaller than the lower bound
2698  * of the merged partition when the values of the two range bounds
2699  * compare equal.
2700  */
2701  cmpval = partition_rbound_cmp(partnatts, partsupfuncs, partcollations,
2702  merged_lb->datums, merged_lb->kind,
2703  false, &prev_ub);
2704  Assert(cmpval >= 0);
2705  }
2706 
2707  /*
2708  * If the lower bound is higher than the last upper bound, add the lower
2709  * bound with the index as -1 indicating that that is a lower bound; else,
2710  * the last upper bound will be reused as the lower bound of the merged
2711  * partition, so skip this.
2712  */
2713  if (cmpval > 0)
2714  {
2715  *merged_datums = lappend(*merged_datums, merged_lb->datums);
2716  *merged_kinds = lappend(*merged_kinds, merged_lb->kind);
2717  *merged_indexes = lappend_int(*merged_indexes, -1);
2718  }
2719 
2720  /* Add the upper bound and index of the merged partition. */
2721  *merged_datums = lappend(*merged_datums, merged_ub->datums);
2722  *merged_kinds = lappend(*merged_kinds, merged_ub->kind);
2723  *merged_indexes = lappend_int(*merged_indexes, merged_index);
2724 }
2725 
2726 /*
2727  * partitions_are_ordered
2728  * Determine whether the partitions described by 'boundinfo' are ordered,
2729  * that is partitions appearing earlier in the PartitionDesc sequence
2730  * contain partition keys strictly less than those appearing later.
2731  * Also, if NULL values are possible, they must come in the last
2732  * partition defined in the PartitionDesc.
2733  *
2734  * If out of order, or there is insufficient info to know the order,
2735  * then we return false.
2736  */
2737 bool
2738 partitions_are_ordered(PartitionBoundInfo boundinfo, int nparts)
2739 {
2740  Assert(boundinfo != NULL);
2741 
2742  switch (boundinfo->strategy)
2743  {
2744  case PARTITION_STRATEGY_RANGE:
2745 
2746  /*
2747  * RANGE-type partitioning guarantees that the partitions can be
2748  * scanned in the order that they're defined in the PartitionDesc
2749  * to provide sequential, non-overlapping ranges of tuples.
2750  * However, if a DEFAULT partition exists then it doesn't work, as
2751  * that could contain tuples from either below or above the
2752  * defined range, or tuples belonging to gaps between partitions.
2753  */
2754  if (!partition_bound_has_default(boundinfo))
2755  return true;
2756  break;
2757 
2758  case PARTITION_STRATEGY_LIST:
2759 
2760  /*
2761  * LIST partitioning can also guarantee ordering, but only if the
2762  * partitions don't accept interleaved values. We could likely
2763  * check for this by looping over the PartitionBound's indexes
2764  * array to check that the indexes are in order. For now, let's
2765  * just keep it simple and just accept LIST partitioning when
2766  * there's no DEFAULT partition, exactly one value per partition,
2767  * and optionally a NULL partition that does not accept any other
2768  * values. Such a NULL partition will come last in the
2769  * PartitionDesc, and the other partitions will be properly
2770  * ordered. This is a cheap test to make as it does not require
2771  * any per-partition processing. Maybe we'd like to handle more
2772  * complex cases in the future.
2773  */
2774  if (partition_bound_has_default(boundinfo))
2775  return false;
2776 
2777  if (boundinfo->ndatums + partition_bound_accepts_nulls(boundinfo)
2778  == nparts)
2779  return true;
2780  break;
2781 
2782  default:
2783  /* HASH, or some other strategy */
2784  break;
2785  }
2786 
2787  return false;
2788 }
2789 
2790 /*
2791  * check_new_partition_bound
2792  *
2793  * Checks if the new partition's bound overlaps any of the existing partitions
2794  * of parent. Also performs additional checks as necessary per strategy.
2795  */
2796 void
2797 check_new_partition_bound(char *relname, Relation parent,
2798  PartitionBoundSpec *spec, ParseState *pstate)
2799 {
2800  PartitionKey key = RelationGetPartitionKey(parent);
2801  PartitionDesc partdesc = RelationGetPartitionDesc(parent, false);
2802  PartitionBoundInfo boundinfo = partdesc->boundinfo;
2803  int with = -1;
2804  bool overlap = false;
2805  int overlap_location = -1;
2806 
2807  if (spec->is_default)
2808  {
2809  /*
2810  * The default partition bound never conflicts with any other
2811  * partition's; if that's what we're attaching, the only possible
2812  * problem is that one already exists, so check for that and we're
2813  * done.
2814  */
2815  if (boundinfo == NULL || !partition_bound_has_default(boundinfo))
2816  return;
2817 
2818  /* Default partition already exists, error out. */
2819  ereport(ERROR,
2820  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2821  errmsg("partition \"%s\" conflicts with existing default partition \"%s\"",
2822  relname, get_rel_name(partdesc->oids[boundinfo->default_index])),
2823  parser_errposition(pstate, spec->location)));
2824  }
2825 
2826  switch (key->strategy)
2827  {
2828  case PARTITION_STRATEGY_HASH:
2829  {
2830  Assert(spec->strategy == PARTITION_STRATEGY_HASH);
2831  Assert(spec->remainder >= 0 && spec->remainder < spec->modulus);
2832 
2833  if (partdesc->nparts > 0)
2834  {
2835  int greatest_modulus;
2836  int remainder;
2837  int offset;
2838 
2839  /*
2840  * Check rule that every modulus must be a factor of the
2841  * next larger modulus. For example, if you have a bunch
2842  * of partitions that all have modulus 5, you can add a
2843  * new partition with modulus 10 or a new partition with
2844  * modulus 15, but you cannot add both a partition with
2845  * modulus 10 and a partition with modulus 15, because 10
2846  * is not a factor of 15.
2847  */
2848 
2849  /*
2850  * Get the greatest (modulus, remainder) pair contained in
2851  * boundinfo->datums that is less than or equal to the
2852  * (spec->modulus, spec->remainder) pair.
2853  */
2854  offset = partition_hash_bsearch(boundinfo,
2855  spec->modulus,
2856  spec->remainder);
2857  if (offset < 0)
2858  {
2859  int next_modulus;
2860 
2861  /*
2862  * All existing moduli are greater or equal, so the
2863  * new one must be a factor of the smallest one, which
2864  * is first in the boundinfo.
2865  */
2866  next_modulus = DatumGetInt32(boundinfo->datums[0][0]);
2867  if (next_modulus % spec->modulus != 0)
2868  ereport(ERROR,
2869  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2870  errmsg("every hash partition modulus must be a factor of the next larger modulus"),
2871  errdetail("The new modulus %d is not a factor of %d, the modulus of existing partition \"%s\".",
2872  spec->modulus, next_modulus,
2873  get_rel_name(partdesc->oids[boundinfo->indexes[0]]))));
2874  }
2875  else
2876  {
2877  int prev_modulus;
2878 
2879  /* We found the largest modulus less than or equal to ours. */
2880  prev_modulus = DatumGetInt32(boundinfo->datums[offset][0]);
2881 
2882  if (spec->modulus % prev_modulus != 0)
2883  ereport(ERROR,
2884  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2885  errmsg("every hash partition modulus must be a factor of the next larger modulus"),
2886  errdetail("The new modulus %d is not divisible by %d, the modulus of existing partition \"%s\".",
2887  spec->modulus,
2888  prev_modulus,
2889  get_rel_name(partdesc->oids[boundinfo->indexes[offset]]))));
2890 
2891  if (offset + 1 < boundinfo->ndatums)
2892  {
2893  int next_modulus;
2894 
2895  /* Look at the next higher modulus */
2896  next_modulus = DatumGetInt32(boundinfo->datums[offset + 1][0]);
2897 
2898  if (next_modulus % spec->modulus != 0)
2899  ereport(ERROR,
2900  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
2901  errmsg("every hash partition modulus must be a factor of the next larger modulus"),
2902  errdetail("The new modulus %d is not a factor of %d, the modulus of existing partition \"%s\".",
2903  spec->modulus, next_modulus,
2904  get_rel_name(partdesc->oids[boundinfo->indexes[offset + 1]]))));
2905  }
2906  }
2907 
2908  greatest_modulus = boundinfo->nindexes;
2909  remainder = spec->remainder;
2910 
2911  /*
2912  * Normally, the lowest remainder that could conflict with
2913  * the new partition is equal to the remainder specified
2914  * for the new partition, but when the new partition has a
2915  * modulus higher than any used so far, we need to adjust.
2916  */
2917  if (remainder >= greatest_modulus)
2918  remainder = remainder % greatest_modulus;
2919 
2920  /* Check every potentially-conflicting remainder. */
2921  do
2922  {
2923  if (boundinfo->indexes[remainder] != -1)
2924  {
2925  overlap = true;
2926  overlap_location = spec->location;
2927  with = boundinfo->indexes[remainder];
2928  break;
2929  }
2930  remainder += spec->modulus;
2931  } while (remainder < greatest_modulus);
2932  }
2933 
2934  break;
2935  }
2936 
2937  case PARTITION_STRATEGY_LIST:
2938  {
2939  Assert(spec->strategy == PARTITION_STRATEGY_LIST);
2940 
2941  if (partdesc->nparts > 0)
2942  {
2943  ListCell *cell;
2944 
2945  Assert(boundinfo &&
2946  boundinfo->strategy == PARTITION_STRATEGY_LIST &&
2947  (boundinfo->ndatums > 0 ||
2948  partition_bound_accepts_nulls(boundinfo) ||
2949  partition_bound_has_default(boundinfo)));
2950 
2951  foreach(cell, spec->listdatums)
2952  {
2953  Const *val = castNode(Const, lfirst(cell));
2954 
2955  overlap_location = val->location;
2956  if (!val->constisnull)
2957  {
2958  int offset;
2959  bool equal;
2960 
2961  offset = partition_list_bsearch(&key->partsupfunc[0],
2962  key->partcollation,
2963  boundinfo,
2964  val->constvalue,
2965  &equal);
2966  if (offset >= 0 && equal)
2967  {
2968  overlap = true;
2969  with = boundinfo->indexes[offset];
2970  break;
2971  }
2972  }
2973  else if (partition_bound_accepts_nulls(boundinfo))
2974  {
2975  overlap = true;
2976  with = boundinfo->null_index;
2977  break;
2978  }
2979  }
2980  }
2981 
2982  break;
2983  }
2984 
2985  case PARTITION_STRATEGY_RANGE:
2986  {
2987  PartitionRangeBound *lower,
2988  *upper;
2989  int cmpval;
2990 
2991  Assert(spec->strategy == PARTITION_STRATEGY_RANGE);
2992  lower = make_one_partition_rbound(key, -1, spec->lowerdatums, true);
2993  upper = make_one_partition_rbound(key, -1, spec->upperdatums, false);
2994 
2995  /*
2996  * First check if the resulting range would be empty with
2997  * specified lower and upper bounds. partition_rbound_cmp
2998  * cannot return zero here, since the lower-bound flags are
2999  * different.
3000  */
3001  cmpval = partition_rbound_cmp(key->partnatts,
3002  key->partsupfunc,
3003  key->partcollation,
3004  lower->datums, lower->kind,
3005  true, upper);
3006  Assert(cmpval != 0);
3007  if (cmpval > 0)
3008  {
3009  /* Point to problematic key in the lower datums list. */
3010  PartitionRangeDatum *datum = list_nth(spec->lowerdatums,
3011  cmpval - 1);
3012 
3013  ereport(ERROR,
3014  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3015  errmsg("empty range bound specified for partition \"%s\"",
3016  relname),
3017  errdetail("Specified lower bound %s is greater than or equal to upper bound %s.",
3018  get_range_partbound_string(spec->lowerdatums),
3019  get_range_partbound_string(spec->upperdatums)),
3020  parser_errposition(pstate, datum->location)));
3021  }
3022 
3023  if (partdesc->nparts > 0)
3024  {
3025  int offset;
3026 
3027  Assert(boundinfo &&
3028  boundinfo->strategy == PARTITION_STRATEGY_RANGE &&
3029  (boundinfo->ndatums > 0 ||
3030  partition_bound_has_default(boundinfo)));
3031 
3032  /*
3033  * Test whether the new lower bound (which is treated
3034  * inclusively as part of the new partition) lies inside
3035  * an existing partition, or in a gap.
3036  *
3037  * If it's inside an existing partition, the bound at
3038  * offset + 1 will be the upper bound of that partition,
3039  * and its index will be >= 0.
3040  *
3041  * If it's in a gap, the bound at offset + 1 will be the
3042  * lower bound of the next partition, and its index will
3043  * be -1. This is also true if there is no next partition,
3044  * since the index array is initialised with an extra -1
3045  * at the end.
3046  */
3047  offset = partition_range_bsearch(key->partnatts,
3048  key->partsupfunc,
3049  key->partcollation,
3050  boundinfo, lower,
3051  &cmpval);
3052 
3053  if (boundinfo->indexes[offset + 1] < 0)
3054  {
3055  /*
3056  * Check that the new partition will fit in the gap.
3057  * For it to fit, the new upper bound must be less
3058  * than or equal to the lower bound of the next
3059  * partition, if there is one.
3060  */
3061  if (offset + 1 < boundinfo->ndatums)
3062  {
3063  Datum *datums;
3064  PartitionRangeDatumKind *kind;
3065  bool is_lower;
3066 
3067  datums = boundinfo->datums[offset + 1];
3068  kind = boundinfo->kind[offset + 1];
3069  is_lower = (boundinfo->indexes[offset + 1] == -1);
3070 
3071  cmpval = partition_rbound_cmp(key->partnatts,
3072  key->partsupfunc,
3073  key->partcollation,
3074  datums, kind,
3075  is_lower, upper);
3076  if (cmpval < 0)
3077  {
3078  /*
3079  * Point to problematic key in the upper
3080  * datums list.
3081  */
3082  PartitionRangeDatum *datum =
3083  list_nth(spec->upperdatums, Abs(cmpval) - 1);
3084 
3085  /*
3086  * The new partition overlaps with the
3087  * existing partition between offset + 1 and
3088  * offset + 2.
3089  */
3090  overlap = true;
3091  overlap_location = datum->location;
3092  with = boundinfo->indexes[offset + 2];
3093  }
3094  }
3095  }
3096  else
3097  {
3098  /*
3099  * The new partition overlaps with the existing
3100  * partition between offset and offset + 1.
3101  */
3102  PartitionRangeDatum *datum;
3103 
3104  /*
3105  * Point to problematic key in the lower datums list;
3106  * if we have equality, point to the first one.
3107  */
3108  datum = cmpval == 0 ? linitial(spec->lowerdatums) :
3109  list_nth(spec->lowerdatums, Abs(cmpval) - 1);
3110  overlap = true;
3111  overlap_location = datum->location;
3112  with = boundinfo->indexes[offset + 1];
3113  }
3114  }
3115 
3116  break;
3117  }
3118 
3119  default:
3120  elog(ERROR, "unexpected partition strategy: %d",
3121  (int) key->strategy);
3122  }
3123 
3124  if (overlap)
3125  {
3126  Assert(with >= 0);
3127  ereport(ERROR,
3128  (errcode(ERRCODE_INVALID_OBJECT_DEFINITION),
3129  errmsg("partition \"%s\" would overlap partition \"%s\"",
3130  relname, get_rel_name(partdesc->oids[with])),
3131  parser_errposition(pstate, overlap_location)));
3132  }
3133 }
3134 
3135 /*
3136  * check_default_partition_contents
3137  *
3138  * This function checks if there exists a row in the default partition that
3139  * would properly belong to the new partition being added. If it finds one,
3140  * it throws an error.
3141  */
3142 void
3143 check_default_partition_contents(Relation parent, Relation default_rel,
3144  PartitionBoundSpec *new_spec)
3145 {
3146  List *new_part_constraints;
3147  List *def_part_constraints;
3148  List *all_parts;
3149  ListCell *lc;
3150 
3151  new_part_constraints = (new_spec->strategy == PARTITION_STRATEGY_LIST)
3152  ? get_qual_for_list(parent, new_spec)
3153  : get_qual_for_range(parent, new_spec, false);
3154  def_part_constraints =
3155  get_proposed_default_constraint(new_part_constraints);
3156 
3157  /*
3158  * Map the Vars in the constraint expression from parent's attnos to
3159  * default_rel's.
3160  */
3161  def_part_constraints =
3162  map_partition_varattnos(def_part_constraints, 1, default_rel,
3163  parent);
3164 
3165  /*
3166  * If the existing constraints on the default partition imply that it will
3167  * not contain any row that would belong to the new partition, we can
3168  * avoid scanning the default partition.
3169  */
3170  if (PartConstraintImpliedByRelConstraint(default_rel, def_part_constraints))
3171  {
3172  ereport(DEBUG1,
3173  (errmsg_internal("updated partition constraint for default partition \"%s\" is implied by existing constraints",
3174  RelationGetRelationName(default_rel))));
3175  return;
3176  }
3177 
3178  /*
3179  * Scan the default partition and its subpartitions, and check for rows
3180  * that do not satisfy the revised partition constraints.
3181  */
3182  if (default_rel->rd_rel->relkind == RELKIND_PARTITIONED_TABLE)
3183  all_parts = find_all_inheritors(RelationGetRelid(default_rel),
3184  AccessExclusiveLock, NULL);
3185  else
3186  all_parts = list_make1_oid(RelationGetRelid(default_rel));
3187 
3188  foreach(lc, all_parts)
3189  {
3190  Oid part_relid = lfirst_oid(lc);
3191  Relation part_rel;
3192  Expr *partition_constraint;
3193  EState *estate;
3194  ExprState *partqualstate = NULL;
3195  Snapshot snapshot;
3196  ExprContext *econtext;
3197  TableScanDesc scan;
3198  MemoryContext oldCxt;
3199  TupleTableSlot *tupslot;
3200 
3201  /* Lock already taken above. */
3202  if (part_relid != RelationGetRelid(default_rel))
3203  {
3204  part_rel = table_open(part_relid, NoLock);
3205 
3206  /*
3207  * Map the Vars in the constraint expression from default_rel's
3208  * the sub-partition's.
3209  */
3210  partition_constraint = make_ands_explicit(def_part_constraints);
3211  partition_constraint = (Expr *)
3212  map_partition_varattnos((List *) partition_constraint, 1,
3213  part_rel, default_rel);
3214 
3215  /*
3216  * If the partition constraints on default partition child imply
3217  * that it will not contain any row that would belong to the new
3218  * partition, we can avoid scanning the child table.
3219  */
3220  if (PartConstraintImpliedByRelConstraint(part_rel,
3221  def_part_constraints))
3222  {
3223  ereport(DEBUG1,
3224  (errmsg_internal("updated partition constraint for default partition \"%s\" is implied by existing constraints",
3225  RelationGetRelationName(part_rel))));
3226 
3227  table_close(part_rel, NoLock);
3228  continue;
3229  }
3230  }
3231  else
3232  {
3233  part_rel = default_rel;
3234  partition_constraint = make_ands_explicit(def_part_constraints);
3235  }
3236 
3237  /*
3238  * Only RELKIND_RELATION relations (i.e. leaf partitions) need to be
3239  * scanned.
3240  */
3241  if (part_rel->rd_rel->relkind != RELKIND_RELATION)
3242  {
3243  if (part_rel->rd_rel->relkind == RELKIND_FOREIGN_TABLE)
3244  ereport(WARNING,
3245  (errcode(ERRCODE_CHECK_VIOLATION),
3246  errmsg("skipped scanning foreign table \"%s\" which is a partition of default partition \"%s\"",
3247  RelationGetRelationName(part_rel),
3248  RelationGetRelationName(default_rel))));
3249 
3250  if (RelationGetRelid(default_rel) != RelationGetRelid(part_rel))
3251  table_close(part_rel, NoLock);
3252 
3253  continue;
3254  }
3255 
3256  estate = CreateExecutorState();
3257 
3258  /* Build expression execution states for partition check quals */
3259  partqualstate = ExecPrepareExpr(partition_constraint, estate);
3260 
3261  econtext = GetPerTupleExprContext(estate);
3262  snapshot = RegisterSnapshot(GetLatestSnapshot());
3263  tupslot = table_slot_create(part_rel, &estate->es_tupleTable);
3264  scan = table_beginscan(part_rel, snapshot, 0, NULL);
3265 
3266  /*
3267  * Switch to per-tuple memory context and reset it for each tuple
3268  * produced, so we don't leak memory.
3269  */
3270  oldCxt = MemoryContextSwitchTo(GetPerTupleMemoryContext(estate));
3271 
3272  while (table_scan_getnextslot(scan, ForwardScanDirection, tupslot))
3273  {
3274  econtext->ecxt_scantuple = tupslot;
3275 
3276  if (!ExecCheck(partqualstate, econtext))
3277  ereport(ERROR,
3278  (errcode(ERRCODE_CHECK_VIOLATION),
3279  errmsg("updated partition constraint for default partition \"%s\" would be violated by some row",
3280  RelationGetRelationName(default_rel)),
3281  errtable(default_rel)));
3282 
3283  ResetExprContext(econtext);
3284  CHECK_FOR_INTERRUPTS();
3285  }
3286 
3287  MemoryContextSwitchTo(oldCxt);
3288  table_endscan(scan);
3289  UnregisterSnapshot(snapshot);
3290  ExecDropSingleTupleTableSlot(tupslot);
3291  FreeExecutorState(estate);
3292 
3293  if (RelationGetRelid(default_rel) != RelationGetRelid(part_rel))
3294  table_close(part_rel, NoLock); /* keep the lock until commit */
3295  }
3296 }
3297 
3298 /*
3299  * get_hash_partition_greatest_modulus
3300  *
3301  * Returns the greatest modulus of the hash partition bound.
3302  * This is no longer used in the core code, but we keep it around
3303  * in case external modules are using it.
3304  */
3305 int
3306 get_hash_partition_greatest_modulus(PartitionBoundInfo bound)
3307 {
3308  Assert(bound && bound->strategy == PARTITION_STRATEGY_HASH);
3309  return bound->nindexes;
3310 }
3311 
3312 /*
3313  * make_one_partition_rbound
3314  *
3315  * Return a PartitionRangeBound given a list of PartitionRangeDatum elements
3316  * and a flag telling whether the bound is lower or not. Made into a function
3317  * because there are multiple sites that want to use this facility.
3318  */
3319 static PartitionRangeBound *
3320 make_one_partition_rbound(PartitionKey key, int index, List *datums, bool lower)
3321 {
3322  PartitionRangeBound *bound;
3323  ListCell *lc;
3324  int i;
3325 
3326  Assert(datums != NIL);
3327 
3328  bound = (PartitionRangeBound *) palloc0(sizeof(PartitionRangeBound));
3329  bound->index = index;
3330  bound->datums = (Datum *) palloc0(key->partnatts * sizeof(Datum));
3331  bound->kind = (PartitionRangeDatumKind *) palloc0(key->partnatts *
3332  sizeof(PartitionRangeDatumKind));
3333  bound->lower = lower;
3334 
3335  i = 0;
3336  foreach(lc, datums)
3337  {
3338  PartitionRangeDatum *datum = castNode(PartitionRangeDatum, lfirst(lc));
3339 
3340  /* What's contained in this range datum? */
3341  bound->kind[i] = datum->kind;
3342 
3343  if (datum->kind == PARTITION_RANGE_DATUM_VALUE)
3344  {
3345  Const *val = castNode(Const, datum->value);
3346 
3347  if (val->constisnull)
3348  elog(ERROR, "invalid range bound datum");
3349  bound->datums[i] = val->constvalue;
3350  }
3351 
3352  i++;
3353  }
3354 
3355  return bound;
3356 }
3357 
3358 /*
3359  * partition_rbound_cmp
3360  *
3361  * For two range bounds this decides whether the 1st one (specified by
3362  * datums1, kind1, and lower1) is <, =, or > the bound specified in *b2.
3363  *
3364  * 0 is returned if they are equal, otherwise a non-zero integer whose sign
3365  * indicates the ordering, and whose absolute value gives the 1-based
3366  * partition key number of the first mismatching column.
3367  *
3368  * partnatts, partsupfunc and partcollation give the number of attributes in the
3369  * bounds to be compared, comparison function to be used and the collations of
3370  * attributes, respectively.
3371  *
3372  * Note that if the values of the two range bounds compare equal, then we take
3373  * into account whether they are upper or lower bounds, and an upper bound is
3374  * considered to be smaller than a lower bound. This is important to the way
3375  * that RelationBuildPartitionDesc() builds the PartitionBoundInfoData
3376  * structure, which only stores the upper bound of a common boundary between
3377  * two contiguous partitions.
3378  */
3379 static int32
3380 partition_rbound_cmp(int partnatts, FmgrInfo *partsupfunc,
3381  Oid *partcollation,
3382  Datum *datums1, PartitionRangeDatumKind *kind1,
3383  bool lower1, PartitionRangeBound *b2)
3384 {
3385  int32 colnum = 0;
3386  int32 cmpval = 0; /* placate compiler */
3387  int i;
3388  Datum *datums2 = b2->datums;
3389  PartitionRangeDatumKind *kind2 = b2->kind;
3390  bool lower2 = b2->lower;
3391 
3392  for (i = 0; i < partnatts; i++)
3393  {
3394  /* Track column number in case we need it for result */
3395  colnum++;
3396 
3397  /*
3398  * First, handle cases where the column is unbounded, which should not
3399  * invoke the comparison procedure, and should not consider any later
3400  * columns. Note that the PartitionRangeDatumKind enum elements
3401  * compare the same way as the values they represent.
3402  */
3403  if (kind1[i] < kind2[i])
3404  return -colnum;
3405  else if (kind1[i] > kind2[i])
3406  return colnum;
3407  else if (kind1[i] != PARTITION_RANGE_DATUM_VALUE)
3408  {
3409  /*
3410  * The column bounds are both MINVALUE or both MAXVALUE. No later
3411  * columns should be considered, but we still need to compare
3412  * whether they are upper or lower bounds.
3413  */
3414  break;
3415  }
3416 
3417  cmpval = DatumGetInt32(FunctionCall2Coll(&partsupfunc[i],
3418  partcollation[i],
3419  datums1[i],
3420  datums2[i]));
3421  if (cmpval != 0)
3422  break;
3423  }
3424 
3425  /*
3426  * If the comparison is anything other than equal, we're done. If they
3427  * compare equal though, we still have to consider whether the boundaries
3428  * are inclusive or exclusive. Exclusive one is considered smaller of the
3429  * two.
3430  */
3431  if (cmpval == 0 && lower1 != lower2)
3432  cmpval = lower1 ? 1 : -1;
3433 
3434  return cmpval == 0 ? 0 : (cmpval < 0 ? -colnum : colnum);
3435 }
3436 
3437 /*
3438  * partition_rbound_datum_cmp
3439  *
3440  * Return whether range bound (specified in rb_datums and rb_kind)
3441  * is <, =, or > partition key of tuple (tuple_datums)
3442  *
3443  * n_tuple_datums, partsupfunc and partcollation give number of attributes in
3444  * the bounds to be compared, comparison function to be used and the collations
3445  * of attributes resp.
3446  */
3447 int32
3448 partition_rbound_datum_cmp(FmgrInfo *partsupfunc, Oid *partcollation,
3449  Datum *rb_datums, PartitionRangeDatumKind *rb_kind,
3450  Datum *tuple_datums, int n_tuple_datums)
3451 {
3452  int i;
3453  int32 cmpval = -1;
3454 
3455  for (i = 0; i < n_tuple_datums; i++)
3456  {
3457  if (rb_kind[i] == PARTITION_RANGE_DATUM_MINVALUE)
3458  return -1;
3459  else if (rb_kind[i] == PARTITION_RANGE_DATUM_MAXVALUE)
3460  return 1;
3461 
3462  cmpval = DatumGetInt32(FunctionCall2Coll(&partsupfunc[i],
3463  partcollation[i],
3464  rb_datums[i],
3465  tuple_datums[i]));
3466  if (cmpval != 0)
3467  break;
3468  }
3469 
3470  return cmpval;
3471 }
3472 
3473 /*
3474  * partition_hbound_cmp
3475  *
3476  * Compares modulus first, then remainder if modulus is equal.
3477  */
3478 static int32
3479 partition_hbound_cmp(int modulus1, int remainder1, int modulus2, int remainder2)
3480 {
3481  if (modulus1 < modulus2)
3482  return -1;
3483  if (modulus1 > modulus2)
3484  return 1;
3485  if (modulus1 == modulus2 && remainder1 != remainder2)
3486  return (remainder1 > remainder2) ? 1 : -1;
3487  return 0;
3488 }
3489 
3490 /*
3491  * partition_list_bsearch
3492  * Returns the index of the greatest bound datum that is less than equal
3493  * to the given value or -1 if all of the bound datums are greater
3494  *
3495  * *is_equal is set to true if the bound datum at the returned index is equal
3496  * to the input value.
3497  */
3498 int
3499 partition_list_bsearch(FmgrInfo *partsupfunc, Oid *partcollation,
3500  PartitionBoundInfo boundinfo,
3501  Datum value, bool *is_equal)
3502 {
3503  int lo,
3504  hi,
3505  mid;
3506 
3507  lo = -1;
3508  hi = boundinfo->ndatums - 1;
3509  while (lo < hi)
3510  {
3511  int32 cmpval;
3512 
3513  mid = (lo + hi + 1) / 2;
3514  cmpval = DatumGetInt32(FunctionCall2Coll(&partsupfunc[0],
3515  partcollation[0],
3516  boundinfo->datums[mid][0],
3517  value));
3518  if (cmpval <= 0)
3519  {
3520  lo = mid;
3521  *is_equal = (cmpval == 0);
3522  if (*is_equal)
3523  break;
3524  }
3525  else
3526  hi = mid - 1;
3527  }
3528 
3529  return lo;
3530 }
3531 
3532 /*
3533  * partition_range_bsearch
3534  * Returns the index of the greatest range bound that is less than or
3535  * equal to the given range bound or -1 if all of the range bounds are
3536  * greater
3537  *
3538  * Upon return from this function, *cmpval is set to 0 if the bound at the
3539  * returned index matches the input range bound exactly, otherwise a
3540  * non-zero integer whose sign indicates the ordering, and whose absolute
3541  * value gives the 1-based partition key number of the first mismatching
3542  * column.
3543  */
3544 static int
3545 partition_range_bsearch(int partnatts, FmgrInfo *partsupfunc,
3546  Oid *partcollation,
3547  PartitionBoundInfo boundinfo,
3548  PartitionRangeBound *probe, int32 *cmpval)
3549 {
3550  int lo,
3551  hi,
3552  mid;
3553 
3554  lo = -1;
3555  hi = boundinfo->ndatums - 1;
3556  while (lo < hi)
3557  {
3558  mid = (lo + hi + 1) / 2;
3559  *cmpval = partition_rbound_cmp(partnatts, partsupfunc,
3560  partcollation,
3561  boundinfo->datums[mid],
3562  boundinfo->kind[mid],
3563  (boundinfo->indexes[mid] == -1),
3564  probe);
3565  if (*cmpval <= 0)
3566  {
3567  lo = mid;
3568  if (*cmpval == 0)
3569  break;
3570  }
3571  else
3572  hi = mid - 1;
3573  }
3574 
3575  return lo;
3576 }
3577 
3578 /*
3579  * partition_range_datum_bsearch
3580  * Returns the index of the greatest range bound that is less than or
3581  * equal to the given tuple or -1 if all of the range bounds are greater
3582  *
3583  * *is_equal is set to true if the range bound at the returned index is equal
3584  * to the input tuple.
3585  */
3586 int
3587 partition_range_datum_bsearch(FmgrInfo *partsupfunc, Oid *partcollation,
3588  PartitionBoundInfo boundinfo,
3589  int nvalues, Datum *values, bool *is_equal)
3590 {
3591  int lo,
3592  hi,
3593  mid;
3594 
3595  lo = -1;
3596  hi = boundinfo->ndatums - 1;
3597  while (lo < hi)
3598  {
3599  int32 cmpval;
3600 
3601  mid = (lo + hi + 1) / 2;
3602  cmpval = partition_rbound_datum_cmp(partsupfunc,
3603  partcollation,
3604  boundinfo->datums[mid],
3605  boundinfo->kind[mid],
3606  values,
3607  nvalues);
3608  if (cmpval <= 0)
3609  {
3610  lo = mid;
3611  *is_equal = (cmpval == 0);
3612 
3613  if (*is_equal)
3614  break;
3615  }
3616  else
3617  hi = mid - 1;
3618  }
3619 
3620  return lo;
3621 }
3622 
3623 /*
3624  * partition_hash_bsearch
3625  * Returns the index of the greatest (modulus, remainder) pair that is
3626  * less than or equal to the given (modulus, remainder) pair or -1 if
3627  * all of them are greater
3628  */
3629 int
3630 partition_hash_bsearch(PartitionBoundInfo boundinfo,
3631  int modulus, int remainder)
3632 {
3633  int lo,
3634  hi,
3635  mid;
3636 
3637  lo = -1;
3638  hi = boundinfo->ndatums - 1;
3639  while (lo < hi)
3640  {
3641  int32 cmpval,
3642  bound_modulus,
3643  bound_remainder;
3644 
3645  mid = (lo + hi + 1) / 2;
3646  bound_modulus = DatumGetInt32(boundinfo->datums[mid][0]);
3647  bound_remainder = DatumGetInt32(boundinfo->datums[mid][1]);
3648  cmpval = partition_hbound_cmp(bound_modulus, bound_remainder,
3649  modulus, remainder);
3650  if (cmpval <= 0)
3651  {
3652  lo = mid;
3653 
3654  if (cmpval == 0)
3655  break;
3656  }
3657  else
3658  hi = mid - 1;
3659  }
3660 
3661  return lo;
3662 }
3663 
3664 /*
3665  * qsort_partition_hbound_cmp
3666  *
3667  * Hash bounds are sorted by modulus, then by remainder.
3668  */
3669 static int32
3670 qsort_partition_hbound_cmp(const void *a, const void *b)
3671 {
3672  PartitionHashBound *h1 = (*(PartitionHashBound *const *) a);
3673  PartitionHashBound *h2 = (*(PartitionHashBound *const *) b);
3674 
3675  return partition_hbound_cmp(h1->modulus, h1->remainder,
3676  h2->modulus, h2->remainder);
3677 }
3678 
3679 /*
3680  * qsort_partition_list_value_cmp
3681  *
3682  * Compare two list partition bound datums.
3683  */
3684 static int32
3685 qsort_partition_list_value_cmp(const void *a, const void *b, void *arg)
3686 {
3687  Datum val1 = (*(PartitionListValue *const *) a)->value,
3688  val2 = (*(PartitionListValue *const *) b)->value;
3689  PartitionKey key = (PartitionKey) arg;
3690 
3691  return DatumGetInt32(FunctionCall2Coll(&key->partsupfunc[0],
3692  key->partcollation[0],
3693  val1, val2));
3694 }
3695 
3696 /*
3697  * qsort_partition_rbound_cmp
3698  *
3699  * Used when sorting range bounds across all range partitions.
3700  */
3701 static int32
3702 qsort_partition_rbound_cmp(const void *a, const void *b, void *arg)
3703 {
3704  PartitionRangeBound *b1 = (*(PartitionRangeBound *const *) a);
3705  PartitionRangeBound *b2 = (*(PartitionRangeBound *const *) b);
3706  PartitionKey key = (PartitionKey) arg;
3707 
3708  return compare_range_bounds(key->partnatts, key->partsupfunc,
3709  key->partcollation,
3710  b1, b2);
3711 }
3712 
3713 /*
3714  * get_partition_operator
3715  *
3716  * Return oid of the operator of the given strategy for the given partition
3717  * key column. It is assumed that the partitioning key is of the same type as
3718  * the chosen partitioning opclass, or at least binary-compatible. In the
3719  * latter case, *need_relabel is set to true if the opclass is not of a
3720  * polymorphic type (indicating a RelabelType node needed on top), otherwise
3721  * false.
3722  */
3723 static Oid
3724 get_partition_operator(PartitionKey key, int col, StrategyNumber strategy,
3725  bool *need_relabel)
3726 {
3727  Oid operoid;
3728 
3729  /*
3730  * Get the operator in the partitioning opfamily using the opclass'
3731  * declared input type as both left- and righttype.
3732  */
3733  operoid = get_opfamily_member(key->partopfamily[col],
3734  key->partopcintype[col],
3735  key->partopcintype[col],
3736  strategy);
3737  if (!OidIsValid(operoid))
3738  elog(ERROR, "missing operator %d(%u,%u) in partition opfamily %u",
3739  strategy, key->partopcintype[col], key->partopcintype[col],
3740  key->partopfamily[col]);
3741 
3742  /*
3743  * If the partition key column is not of the same type as the operator
3744  * class and not polymorphic, tell caller to wrap the non-Const expression
3745  * in a RelabelType. This matches what parse_coerce.c does.
3746  */
3747  *need_relabel = (key->parttypid[col] != key->partopcintype[col] &&
3748  key->partopcintype[col] != RECORDOID &&
3749  !IsPolymorphicType(key->partopcintype[col]));
3750 
3751  return operoid;
3752 }
3753 
3754 /*
3755  * make_partition_op_expr
3756  * Returns an Expr for the given partition key column with arg1 and
3757  * arg2 as its leftop and rightop, respectively
3758  */
3759 static Expr *
3760 make_partition_op_expr(PartitionKey key, int keynum,
3761  uint16 strategy, Expr *arg1, Expr *arg2)
3762 {
3763  Oid operoid;
3764  bool need_relabel = false;
3765  Expr *result = NULL;
3766 
3767  /* Get the correct btree operator for this partitioning column */
3768  operoid = get_partition_operator(key, keynum, strategy, &need_relabel);
3769 
3770  /*
3771  * Chosen operator may be such that the non-Const operand needs to be
3772  * coerced, so apply the same; see the comment in
3773  * get_partition_operator().
3774  */
3775  if (!IsA(arg1, Const) &&
3776  (need_relabel ||
3777  key->partcollation[keynum] != key->parttypcoll[keynum]))
3778  arg1 = (Expr *) makeRelabelType(arg1,
3779  key->partopcintype[keynum],
3780  -1,
3781  key->partcollation[keynum],
3782  COERCE_EXPLICIT_CAST);
3783 
3784  /* Generate the actual expression */
3785  switch (key->strategy)
3786  {
3787  case PARTITION_STRATEGY_LIST:
3788  {
3789  List *elems = (List *) arg2;
3790  int nelems = list_length(elems);
3791 
3792  Assert(nelems >= 1);
3793  Assert(keynum == 0);
3794 
3795  if (nelems > 1 &&
3796  !type_is_array(key->parttypid[keynum]))
3797  {
3798  ArrayExpr *arrexpr;
3799  ScalarArrayOpExpr *saopexpr;
3800 
3801  /* Construct an ArrayExpr for the right-hand inputs */
3802  arrexpr = makeNode(ArrayExpr);
3803  arrexpr->array_typeid =
3804  get_array_type(key->parttypid[keynum]);
3805  arrexpr->array_collid = key->parttypcoll[keynum];
3806  arrexpr->element_typeid = key->parttypid[keynum];
3807  arrexpr->elements = elems;
3808  arrexpr->multidims = false;
3809  arrexpr->location = -1;
3810 
3811  /* Build leftop = ANY (rightop) */
3812  saopexpr = makeNode(ScalarArrayOpExpr);
3813  saopexpr->opno = operoid;
3814  saopexpr->opfuncid = get_opcode(operoid);
3815  saopexpr->hashfuncid = InvalidOid;
3816  saopexpr->useOr = true;
3817  saopexpr->inputcollid = key->partcollation[keynum];
3818  saopexpr->args = list_make2(arg1, arrexpr);
3819  saopexpr->location = -1;
3820 
3821  result = (Expr *) saopexpr;
3822  }
3823  else
3824  {
3825  List *elemops = NIL;
3826  ListCell *lc;
3827 
3828  foreach(lc, elems)
3829  {
3830  Expr *elem = lfirst(lc),
3831  *elemop;
3832 
3833  elemop = make_opclause(operoid,
3834  BOOLOID,
3835  false,
3836  arg1, elem,
3837  InvalidOid,
3838  key->partcollation[keynum]);
3839  elemops = lappend(elemops, elemop);
3840  }
3841 
3842  result = nelems > 1 ? makeBoolExpr(OR_EXPR, elemops, -1) : linitial(elemops);
3843  }
3844  break;
3845  }
3846 
3847  case PARTITION_STRATEGY_RANGE:
3848  result = make_opclause(operoid,
3849  BOOLOID,
3850  false,
3851  arg1, arg2,
3852  InvalidOid,
3853  key->partcollation[keynum]);
3854  break;
3855 
3856  default:
3857  elog(ERROR, "invalid partitioning strategy");
3858  break;
3859  }
3860 
3861  return result;
3862 }
3863 
3864 /*
3865  * get_qual_for_hash
3866  *
3867  * Returns a CHECK constraint expression to use as a hash partition's
3868  * constraint, given the parent relation and partition bound structure.
3869  *
3870  * The partition constraint for a hash partition is always a call to the
3871  * built-in function satisfies_hash_partition().
3872  */
3873 static List *
3874 get_qual_for_hash(Relation parent, PartitionBoundSpec *spec)
3875 {
3876  PartitionKey key = RelationGetPartitionKey(parent);
3877  FuncExpr *fexpr;
3878  Node *relidConst;
3879  Node *modulusConst;
3880  Node *remainderConst;
3881  List *args;
3882  ListCell *partexprs_item;
3883  int i;
3884 
3885  /* Fixed arguments. */
3886  relidConst = (Node *) makeConst(OIDOID,
3887  -1,
3888  InvalidOid,
3889  sizeof(Oid),
3890  ObjectIdGetDatum(RelationGetRelid(parent)),
3891  false,
3892  true);
3893 
3894  modulusConst = (Node *) makeConst(INT4OID,
3895  -1,
3896  InvalidOid,
3897  sizeof(int32),
3898  Int32GetDatum(spec->modulus),
3899  false,
3900  true);
3901 
3902  remainderConst = (Node *) makeConst(INT4OID,
3903  -1,
3904  InvalidOid,
3905  sizeof(int32),
3906  Int32GetDatum(spec->remainder),
3907  false,
3908  true);
3909 
3910  args = list_make3(relidConst, modulusConst, remainderConst);
3911  partexprs_item = list_head(key->partexprs);
3912 
3913  /* Add an argument for each key column. */
3914  for (i = 0; i < key->partnatts; i++)
3915  {
3916  Node *keyCol;
3917 
3918  /* Left operand */
3919  if (key->partattrs[i] != 0)
3920  {
3921  keyCol = (Node *) makeVar(1,
3922  key->partattrs[i],
3923  key->parttypid[i],
3924  key->parttypmod[i],
3925  key->parttypcoll[i],
3926  0);
3927  }
3928  else
3929  {
3930  keyCol = (Node *) copyObject(lfirst(partexprs_item));
3931  partexprs_item = lnext(key->partexprs, partexprs_item);
3932  }
3933 
3934  args = lappend(args, keyCol);
3935  }
3936 
3937  fexpr = makeFuncExpr(F_SATISFIES_HASH_PARTITION,
3938  BOOLOID,
3939  args,
3940  InvalidOid,
3941  InvalidOid,
3942  COERCE_EXPLICIT_CALL);
3943 
3944  return list_make1(fexpr);
3945 }
3946 
3947 /*
3948  * get_qual_for_list
3949  *
3950  * Returns an implicit-AND list of expressions to use as a list partition's
3951  * constraint, given the parent relation and partition bound structure.
3952  *
3953  * The function returns NIL for a default partition when it's the only
3954  * partition since in that case there is no constraint.
3955  */
3956 static List *
3957 get_qual_for_list(Relation parent, PartitionBoundSpec *spec)
3958 {
3959  PartitionKey key = RelationGetPartitionKey(parent);
3960  List *result;
3961  Expr *keyCol;
3962  Expr *opexpr;
3963  NullTest *nulltest;
3964  ListCell *cell;
3965  List *elems = NIL;
3966  bool list_has_null = false;
3967 
3968  /*
3969  * Only single-column list partitioning is supported, so we are worried
3970  * only about the partition key with index 0.
3971  */
3972  Assert(key->partnatts == 1);
3973 
3974  /* Construct Var or expression representing the partition column */
3975  if (key->partattrs[0] != 0)
3976  keyCol = (Expr *) makeVar(1,
3977  key->partattrs[0],
3978  key->parttypid[0],
3979  key->parttypmod[0],
3980  key->parttypcoll[0],
3981  0);
3982  else
3983  keyCol = (Expr *) copyObject(linitial(key->partexprs));
3984 
3985  /*
3986  * For default list partition, collect datums for all the partitions. The
3987  * default partition constraint should check that the partition key is
3988  * equal to none of those.
3989  */
3990  if (spec->is_default)
3991  {
3992  int i;
3993  int ndatums = 0;
3994  PartitionDesc pdesc = RelationGetPartitionDesc(parent, false);
3995  PartitionBoundInfo boundinfo = pdesc->boundinfo;
3996 
3997  if (boundinfo)
3998  {
3999  ndatums = boundinfo->ndatums;
4000 
4001  if (partition_bound_accepts_nulls(boundinfo))
4002  list_has_null = true;
4003  }
4004 
4005  /*
4006  * If default is the only partition, there need not be any partition
4007  * constraint on it.
4008  */
4009  if (ndatums == 0 && !list_has_null)
4010  return NIL;
4011 
4012  for (i = 0; i < ndatums; i++)
4013  {
4014  Const *val;
4015 
4016  /*
4017  * Construct Const from known-not-null datum. We must be careful
4018  * to copy the value, because our result has to be able to outlive
4019  * the relcache entry we're copying from.
4020  */
4021  val = makeConst(key->parttypid[0],
4022  key->parttypmod[0],
4023  key->parttypcoll[0],
4024  key->parttyplen[0],
4025  datumCopy(*boundinfo->datums[i],
4026  key->parttypbyval[0],
4027  key->parttyplen[0]),
4028  false, /* isnull */
4029  key->parttypbyval[0]);
4030 
4031  elems = lappend(elems, val);
4032  }
4033  }
4034  else
4035  {
4036  /*
4037  * Create list of Consts for the allowed values, excluding any nulls.
4038  */
4039  foreach(cell, spec->listdatums)
4040  {
4041  Const *val = castNode(Const, lfirst(cell));
4042 
4043  if (val->constisnull)
4044  list_has_null = true;
4045  else
4046  elems = lappend(elems, copyObject(val));
4047  }
4048  }
4049 
4050  if (elems)
4051  {
4052  /*
4053  * Generate the operator expression from the non-null partition
4054  * values.
4055  */
4056  opexpr = make_partition_op_expr(key, 0, BTEqualStrategyNumber,
4057  keyCol, (Expr *) elems);
4058  }
4059  else
4060  {
4061  /*
4062  * If there are no partition values, we don't need an operator
4063  * expression.
4064  */
4065  opexpr = NULL;
4066  }
4067 
4068  if (!list_has_null)
4069  {
4070  /*
4071  * Gin up a "col IS NOT NULL" test that will be ANDed with the main
4072  * expression. This might seem redundant, but the partition routing
4073  * machinery needs it.
4074  */
4075  nulltest = makeNode(NullTest);
4076  nulltest->arg = keyCol;
4077  nulltest->nulltesttype = IS_NOT_NULL;
4078  nulltest->argisrow = false;
4079  nulltest->location = -1;
4080 
4081  result = opexpr ? list_make2(nulltest, opexpr) : list_make1(nulltest);
4082  }
4083  else
4084  {
4085  /*
4086  * Gin up a "col IS NULL" test that will be OR'd with the main
4087  * expression.
4088  */
4089  nulltest = makeNode(NullTest);
4090  nulltest->arg = keyCol;
4091  nulltest->nulltesttype = IS_NULL;
4092  nulltest->argisrow = false;
4093  nulltest->location = -1;
4094 
4095  if (opexpr)
4096  {
4097  Expr *or;
4098 
4099  or = makeBoolExpr(OR_EXPR, list_make2(nulltest, opexpr), -1);
4100  result = list_make1(or);
4101  }
4102  else
4103  result = list_make1(nulltest);
4104  }
4105 
4106  /*
4107  * Note that, in general, applying NOT to a constraint expression doesn't
4108  * necessarily invert the set of rows it accepts, because NOT (NULL) is
4109  * NULL. However, the partition constraints we construct here never
4110  * evaluate to NULL, so applying NOT works as intended.
4111  */
4112  if (spec->is_default)
4113  {
4114  result = list_make1(make_ands_explicit(result));
4115  result = list_make1(makeBoolExpr(NOT_EXPR, result, -1));
4116  }
4117 
4118  return result;
4119 }
4120 
4121 /*
4122  * get_qual_for_range
4123  *
4124  * Returns an implicit-AND list of expressions to use as a range partition's
4125  * constraint, given the parent relation and partition bound structure.
4126  *
4127  * For a multi-column range partition key, say (a, b, c), with (al, bl, cl)
4128  * as the lower bound tuple and (au, bu, cu) as the upper bound tuple, we
4129  * generate an expression tree of the following form:
4130  *
4131  * (a IS NOT NULL) and (b IS NOT NULL) and (c IS NOT NULL)
4132  * AND
4133  * (a > al OR (a = al AND b > bl) OR (a = al AND b = bl AND c >= cl))
4134  * AND
4135  * (a < au OR (a = au AND b < bu) OR (a = au AND b = bu AND c < cu))
4136  *
4137  * It is often the case that a prefix of lower and upper bound tuples contains
4138  * the same values, for example, (al = au), in which case, we will emit an
4139  * expression tree of the following form:
4140  *
4141  * (a IS NOT NULL) and (b IS NOT NULL) and (c IS NOT NULL)
4142  * AND
4143  * (a = al)
4144  * AND
4145  * (b > bl OR (b = bl AND c >= cl))
4146  * AND
4147  * (b < bu OR (b = bu AND c < cu))
4148  *
4149  * If a bound datum is either MINVALUE or MAXVALUE, these expressions are
4150  * simplified using the fact that any value is greater than MINVALUE and less
4151  * than MAXVALUE. So, for example, if cu = MAXVALUE, c < cu is automatically
4152  * true, and we need not emit any expression for it, and the last line becomes
4153  *
4154  * (b < bu) OR (b = bu), which is simplified to (b <= bu)
4155  *
4156  * In most common cases with only one partition column, say a, the following
4157  * expression tree will be generated: a IS NOT NULL AND a >= al AND a < au
4158  *
4159  * For default partition, it returns the negation of the constraints of all
4160  * the other partitions.
4161  *
4162  * External callers should pass for_default as false; we set it to true only
4163  * when recursing.
4164  */
4165 static List *
4166 get_qual_for_range(Relation parent, PartitionBoundSpec *spec,
4167  bool for_default)
4168 {
4169  List *result = NIL;
4170  ListCell *cell1,
4171  *cell2,
4172  *partexprs_item,
4173  *partexprs_item_saved;
4174  int i,
4175  j;
4176  PartitionRangeDatum *ldatum,
4177  *udatum;
4178  PartitionKey key = RelationGetPartitionKey(parent);
4179  Expr *keyCol;
4180  Const *lower_val,
4181  *upper_val;
4182  List *lower_or_arms,
4183  *upper_or_arms;
4184  int num_or_arms,
4185  current_or_arm;
4186  ListCell *lower_or_start_datum,
4187  *upper_or_start_datum;
4188  bool need_next_lower_arm,
4189  need_next_upper_arm;
4190 
4191  if (spec->is_default)
4192  {
4193  List *or_expr_args = NIL;
4194  PartitionDesc pdesc = RelationGetPartitionDesc(parent, false);
4195  Oid *inhoids = pdesc->oids;
4196  int nparts = pdesc->nparts,
4197  i;
4198 
4199  for (i = 0; i < nparts; i++)
4200  {
4201  Oid inhrelid = inhoids[i];
4202  HeapTuple tuple;
4203  Datum datum;
4204  bool isnull;
4205  PartitionBoundSpec *bspec;
4206 
4207  tuple = SearchSysCache1(RELOID, inhrelid);
4208  if (!HeapTupleIsValid(tuple))
4209  elog(ERROR, "cache lookup failed for relation %u", inhrelid);
4210 
4211  datum = SysCacheGetAttr(RELOID, tuple,
4212  Anum_pg_class_relpartbound,
4213  &isnull);
4214  if (isnull)
4215  elog(ERROR, "null relpartbound for relation %u", inhrelid);
4216 
4217  bspec = (PartitionBoundSpec *)
4218  stringToNode(TextDatumGetCString(datum));
4219  if (!IsA(bspec, PartitionBoundSpec))
4220  elog(ERROR, "expected PartitionBoundSpec");
4221 
4222  if (!bspec->is_default)
4223  {
4224  List *part_qual;
4225 
4226  part_qual = get_qual_for_range(parent, bspec, true);
4227 
4228  /*
4229  * AND the constraints of the partition and add to
4230  * or_expr_args
4231  */
4232  or_expr_args = lappend(or_expr_args, list_length(part_qual) > 1
4233  ? makeBoolExpr(AND_EXPR, part_qual, -1)
4234  : linitial(part_qual));
4235  }
4236  ReleaseSysCache(tuple);
4237  }
4238 
4239  if (or_expr_args != NIL)
4240  {
4241  Expr *other_parts_constr;
4242 
4243  /*
4244  * Combine the constraints obtained for non-default partitions
4245  * using OR. As requested, each of the OR's args doesn't include
4246  * the NOT NULL test for partition keys (which is to avoid its
4247  * useless repetition). Add the same now.
4248  */
4249  other_parts_constr =
4250  makeBoolExpr(AND_EXPR,
4251  lappend(get_range_nulltest(key),
4252  list_length(or_expr_args) > 1
4253  ? makeBoolExpr(OR_EXPR, or_expr_args,
4254  -1)
4255  : linitial(or_expr_args)),
4256  -1);
4257 
4258  /*
4259  * Finally, the default partition contains everything *NOT*
4260  * contained in the non-default partitions.
4261  */
4262  result = list_make1(makeBoolExpr(NOT_EXPR,
4263  list_make1(other_parts_constr), -1));
4264  }
4265 
4266  return result;
4267  }
4268 
4269  /*
4270  * If it is the recursive call for default, we skip the get_range_nulltest
4271  * to avoid accumulating the NullTest on the same keys for each partition.
4272  */
4273  if (!for_default)
4274  result = get_range_nulltest(key);
4275 
4276  /*
4277  * Iterate over the key columns and check if the corresponding lower and
4278  * upper datums are equal using the btree equality operator for the
4279  * column's type. If equal, we emit single keyCol = common_value
4280  * expression. Starting from the first column for which the corresponding
4281  * lower and upper bound datums are not equal, we generate OR expressions
4282  * as shown in the function's header comment.
4283  */
4284  i = 0;
4285  partexprs_item = list_head(key->partexprs);
4286  partexprs_item_saved = partexprs_item; /* placate compiler */
4287  forboth(cell1, spec->lowerdatums, cell2, spec->upperdatums)
4288  {
4289  EState *estate;
4290  MemoryContext oldcxt;
4291  Expr *test_expr;
4292  ExprState *test_exprstate;
4293  Datum test_result;
4294  bool isNull;
4295 
4296  ldatum = castNode(PartitionRangeDatum, lfirst(cell1));
4297  udatum = castNode(PartitionRangeDatum, lfirst(cell2));
4298 
4299  /*
4300  * Since get_range_key_properties() modifies partexprs_item, and we
4301  * might need to start over from the previous expression in the later
4302  * part of this function, save away the current value.
4303  */
4304  partexprs_item_saved = partexprs_item;
4305 
4306  get_range_key_properties(key, i, ldatum, udatum,
4307  &partexprs_item,
4308  &keyCol,
4309  &lower_val, &upper_val);
4310 
4311  /*
4312  * If either value is NULL, the corresponding partition bound is
4313  * either MINVALUE or MAXVALUE, and we treat them as unequal, because
4314  * even if they're the same, there is no common value to equate the
4315  * key column with.
4316  */
4317  if (!lower_val || !upper_val)
4318  break;
4319 
4320  /* Create the test expression */
4321  estate = CreateExecutorState();
4322  oldcxt = MemoryContextSwitchTo(estate->es_query_cxt);
4323  test_expr = make_partition_op_expr(key, i, BTEqualStrategyNumber,
4324  (Expr *) lower_val,
4325  (Expr *) upper_val);
4326  fix_opfuncids((Node *) test_expr);
4327  test_exprstate = ExecInitExpr(test_expr, NULL);
4328  test_result = ExecEvalExprSwitchContext(test_exprstate,
4329  GetPerTupleExprContext(estate),
4330  &isNull);
4331  MemoryContextSwitchTo(oldcxt);
4332  FreeExecutorState(estate);
4333 
4334  /* If not equal, go generate the OR expressions */
4335  if (!DatumGetBool(test_result))
4336  break;
4337 
4338  /*
4339  * The bounds for the last key column can't be equal, because such a
4340  * range partition would never be allowed to be defined (it would have
4341  * an empty range otherwise).
4342  */
4343  if (i == key->partnatts - 1)
4344  elog(ERROR, "invalid range bound specification");
4345 
4346  /* Equal, so generate keyCol = lower_val expression */
4347  result = lappend(result,
4348  make_partition_op_expr(key, i, BTEqualStrategyNumber,
4349  keyCol, (Expr *) lower_val));
4350 
4351  i++;
4352  }
4353 
4354  /* First pair of lower_val and upper_val that are not equal. */
4355  lower_or_start_datum = cell1;
4356  upper_or_start_datum = cell2;
4357 
4358  /* OR will have as many arms as there are key columns left. */
4359  num_or_arms = key->partnatts - i;
4360  current_or_arm = 0;
4361  lower_or_arms = upper_or_arms = NIL;
4362  need_next_lower_arm = need_next_upper_arm = true;
4363  while (current_or_arm < num_or_arms)
4364  {
4365  List *lower_or_arm_args = NIL,
4366  *upper_or_arm_args = NIL;
4367 
4368  /* Restart scan of columns from the i'th one */
4369  j = i;
4370  partexprs_item = partexprs_item_saved;
4371 
4372  for_both_cell(cell1, spec->lowerdatums, lower_or_start_datum,
4373  cell2, spec->upperdatums, upper_or_start_datum)
4374  {
4375  PartitionRangeDatum *ldatum_next = NULL,
4376  *udatum_next = NULL;
4377 
4378  ldatum = castNode(PartitionRangeDatum, lfirst(cell1));
4379  if (lnext(spec->lowerdatums, cell1))
4380  ldatum_next = castNode(PartitionRangeDatum,
4381  lfirst(lnext(spec->lowerdatums, cell1)));
4382  udatum = castNode(PartitionRangeDatum, lfirst(cell2));
4383  if (lnext(spec->upperdatums, cell2))
4384  udatum_next = castNode(PartitionRangeDatum,
4385  lfirst(lnext(spec->upperdatums, cell2)));
4386  get_range_key_properties(key, j, ldatum, udatum,
4387  &partexprs_item,
4388  &keyCol,
4389  &lower_val, &upper_val);
4390 
4391  if (need_next_lower_arm && lower_val)
4392  {
4393  uint16 strategy;
4394 
4395  /*
4396  * For the non-last columns of this arm, use the EQ operator.
4397  * For the last column of this arm, use GT, unless this is the
4398  * last column of the whole bound check, or the next bound
4399  * datum is MINVALUE, in which case use GE.
4400  */
4401  if (j - i < current_or_arm)
4402  strategy = BTEqualStrategyNumber;
4403  else if (j == key->partnatts - 1 ||
4404  (ldatum_next &&
4405  ldatum_next->kind == PARTITION_RANGE_DATUM_MINVALUE))
4406  strategy = BTGreaterEqualStrategyNumber;
4407  else
4408  strategy = BTGreaterStrategyNumber;
4409 
4410  lower_or_arm_args = lappend(lower_or_arm_args,
4411  make_partition_op_expr(key, j,
4412  strategy,
4413  keyCol,
4414  (Expr *) lower_val));
4415  }
4416 
4417  if (need_next_upper_arm && upper_val)
4418  {
4419  uint16 strategy;
4420 
4421  /*
4422  * For the non-last columns of this arm, use the EQ operator.
4423  * For the last column of this arm, use LT, unless the next
4424  * bound datum is MAXVALUE, in which case use LE.
4425  */
4426  if (j - i < current_or_arm)
4427  strategy = BTEqualStrategyNumber;
4428  else if (udatum_next &&
4429  udatum_next->kind == PARTITION_RANGE_DATUM_MAXVALUE)
4430  strategy = BTLessEqualStrategyNumber;
4431  else
4432  strategy = BTLessStrategyNumber;
4433 
4434  upper_or_arm_args = lappend(upper_or_arm_args,
4435  make_partition_op_expr(key, j,
4436  strategy,
4437  keyCol,
4438  (Expr *) upper_val));
4439  }
4440 
4441  /*
4442  * Did we generate enough of OR's arguments? First arm considers
4443  * the first of the remaining columns, second arm considers first
4444  * two of the remaining columns, and so on.
4445  */
4446  ++j;
4447  if (j - i > current_or_arm)
4448  {
4449  /*
4450  * We must not emit any more arms if the new column that will
4451  * be considered is unbounded, or this one was.
4452  */
4453  if (!lower_val || !ldatum_next ||
4454  ldatum_next->kind != PARTITION_RANGE_DATUM_VALUE)
4455  need_next_lower_arm = false;
4456  if (!upper_val || !udatum_next ||
4457  udatum_next->kind != PARTITION_RANGE_DATUM_VALUE)
4458  need_next_upper_arm = false;
4459  break;
4460  }
4461  }
4462 
4463  if (lower_or_arm_args != NIL)
4464  lower_or_arms = lappend(lower_or_arms,
4465  list_length(lower_or_arm_args) > 1
4466  ? makeBoolExpr(AND_EXPR, lower_or_arm_args, -1)
4467  : linitial(lower_or_arm_args));
4468 
4469  if (upper_or_arm_args != NIL)
4470  upper_or_arms = lappend(upper_or_arms,
4471  list_length(upper_or_arm_args) > 1
4472  ? makeBoolExpr(AND_EXPR, upper_or_arm_args, -1)
4473  : linitial(upper_or_arm_args));
4474 
4475  /* If no work to do in the next iteration, break away. */
4476  if (!need_next_lower_arm && !need_next_upper_arm)
4477  break;
4478 
4479  ++current_or_arm;
4480  }
4481 
4482  /*
4483  * Generate the OR expressions for each of lower and upper bounds (if
4484  * required), and append to the list of implicitly ANDed list of
4485  * expressions.
4486  */
4487  if (lower_or_arms != NIL)
4488  result = lappend(result,
4489  list_length(lower_or_arms) > 1
4490  ? makeBoolExpr(OR_EXPR, lower_or_arms, -1)
4491  : linitial(lower_or_arms));
4492  if (upper_or_arms != NIL)
4493  result = lappend(result,
4494  list_length(upper_or_arms) > 1
4495  ? makeBoolExpr(OR_EXPR, upper_or_arms, -1)
4496  : linitial(upper_or_arms));
4497 
4498  /*
4499  * As noted above, for non-default, we return list with constant TRUE. If
4500  * the result is NIL during the recursive call for default, it implies
4501  * this is the only other partition which can hold every value of the key
4502  * except NULL. Hence we return the NullTest result skipped earlier.
4503  */
4504  if (result == NIL)
4505  result = for_default
4506  ? get_range_nulltest(key)
4507  : list_make1(makeBoolConst(true, false));
4508 
4509  return result;
4510 }
4511 
4512 /*
4513  * get_range_key_properties
4514  * Returns range partition key information for a given column
4515  *
4516  * This is a subroutine for get_qual_for_range, and its API is pretty
4517  * specialized to that caller.
4518  *
4519  * Constructs an Expr for the key column (returned in *keyCol) and Consts
4520  * for the lower and upper range limits (returned in *lower_val and
4521  * *upper_val). For MINVALUE/MAXVALUE limits, NULL is returned instead of
4522  * a Const. All of these structures are freshly palloc'd.
4523  *
4524  * *partexprs_item points to the cell containing the next expression in
4525  * the key->partexprs list, or NULL. It may be advanced upon return.
4526  */
4527 static void
4528 get_range_key_properties(PartitionKey key, int keynum,
4529  PartitionRangeDatum *ldatum,
4530  PartitionRangeDatum *udatum,
4531  ListCell **partexprs_item,
4532  Expr **keyCol,
4533  Const **lower_val, Const **upper_val)
4534 {
4535  /* Get partition key expression for this column */
4536  if (key->partattrs[keynum] != 0)
4537  {
4538  *keyCol = (Expr *) makeVar(1,
4539  key->partattrs[keynum],
4540  key->parttypid[keynum],
4541  key->parttypmod[keynum],
4542  key->parttypcoll[keynum],
4543  0);
4544  }
4545  else
4546  {
4547  if (*partexprs_item == NULL)
4548  elog(ERROR, "wrong number of partition key expressions");
4549  *keyCol = copyObject(lfirst(*partexprs_item));
4550  *partexprs_item = lnext(key->partexprs, *partexprs_item);
4551  }
4552 
4553  /* Get appropriate Const nodes for the bounds */
4554  if (ldatum->kind == PARTITION_RANGE_DATUM_VALUE)
4555  *lower_val = castNode(Const, copyObject(ldatum->value));
4556  else
4557  *lower_val = NULL;
4558 
4559  if (udatum->kind == PARTITION_RANGE_DATUM_VALUE)
4560  *upper_val = castNode(Const, copyObject(udatum->value));
4561  else
4562  *upper_val = NULL;
4563 }
4564 
4565 /*
4566  * get_range_nulltest
4567  *
4568  * A non-default range partition table does not currently allow partition
4569  * keys to be null, so emit an IS NOT NULL expression for each key column.
4570  */
4571 static List *
4572 get_range_nulltest(PartitionKey key)
4573 {
4574  List *result = NIL;
4575  NullTest *nulltest;
4576  ListCell *partexprs_item;
4577  int i;
4578 
4579  partexprs_item = list_head(key->partexprs);
4580  for (i = 0; i < key->partnatts; i++)
4581  {
4582  Expr *keyCol;
4583 
4584  if (key->partattrs[i] != 0)
4585  {
4586  keyCol = (Expr *) makeVar(1,
4587  key->partattrs[i],
4588  key->parttypid[i],
4589  key->parttypmod[i],
4590  key->parttypcoll[i],
4591  0);
4592  }
4593  else
4594  {
4595  if (partexprs_item == NULL)
4596  elog(ERROR, "wrong number of partition key expressions");
4597  keyCol = copyObject(lfirst(partexprs_item));
4598  partexprs_item = lnext(key->partexprs, partexprs_item);
4599  }
4600 
4601  nulltest = makeNode(NullTest);
4602  nulltest->arg = keyCol;
4603  nulltest->nulltesttype = IS_NOT_NULL;
4604  nulltest->argisrow = false;
4605  nulltest->location = -1;
4606  result = lappend(result, nulltest);
4607  }
4608 
4609  return result;
4610 }
4611 
4612 /*
4613  * compute_partition_hash_value
4614  *
4615  * Compute the hash value for given partition key values.
4616  */
4617 uint64
4618 compute_partition_hash_value(int partnatts, FmgrInfo *partsupfunc, Oid *partcollation,
4619  Datum *values, bool *isnull)
4620 {
4621  int i;
4622  uint64 rowHash = 0;
4623  Datum seed = UInt64GetDatum(HASH_PARTITION_SEED);
4624 
4625  for (i = 0; i < partnatts; i++)
4626  {
4627  /* Nulls are just ignored */
4628  if (!isnull[i])
4629  {
4630  Datum hash;
4631 
4632  Assert(OidIsValid(partsupfunc[i].fn_oid));
4633 
4634  /*
4635  * Compute hash for each datum value by calling respective
4636  * datatype-specific hash functions of each partition key
4637  * attribute.
4638  */
4639  hash = FunctionCall2Coll(&partsupfunc[i], partcollation[i],
4640  values[i], seed);
4641 
4642  /* Form a single 64-bit hash value */
4643  rowHash = hash_combine64(rowHash, DatumGetUInt64(hash));
4644  }
4645  }
4646 
4647  return rowHash;
4648 }
4649 
4650 /*
4651  * satisfies_hash_partition
4652  *
4653  * This is an SQL-callable function for use in hash partition constraints.
4654  * The first three arguments are the parent table OID, modulus, and remainder.
4655  * The remaining arguments are the value of the partitioning columns (or
4656  * expressions); these are hashed and the results are combined into a single
4657  * hash value by calling hash_combine64.
4658  *
4659  * Returns true if remainder produced when this computed single hash value is
4660  * divided by the given modulus is equal to given remainder, otherwise false.
4661  * NB: it's important that this never return null, as the constraint machinery
4662  * would consider that to be a "pass".
4663  *
4664  * See get_qual_for_hash() for usage.
4665  */
4666 Datum
4667 satisfies_hash_partition(PG_FUNCTION_ARGS)
4668 {
4669  typedef struct ColumnsHashData
4670  {
4671  Oid relid;
4672  int nkeys;
4673  Oid variadic_type;
4674  int16 variadic_typlen;
4675  bool variadic_typbyval;
4676  char variadic_typalign;
4677  Oid partcollid[PARTITION_MAX_KEYS];
4678  FmgrInfo partsupfunc[FLEXIBLE_ARRAY_MEMBER];
4679  } ColumnsHashData;
4680  Oid parentId;
4681  int modulus;
4682  int remainder;
4683  Datum seed = UInt64GetDatum(HASH_PARTITION_SEED);
4684  ColumnsHashData *my_extra;
4685  uint64 rowHash = 0;
4686 
4687  /* Return false if the parent OID, modulus, or remainder is NULL. */
4688  if (PG_ARGISNULL(0) || PG_ARGISNULL(1) || PG_ARGISNULL(2))
4689  PG_RETURN_BOOL(false);
4690  parentId = PG_GETARG_OID(0);
4691  modulus = PG_GETARG_INT32(1);
4692  remainder = PG_GETARG_INT32(2);
4693 
4694  /* Sanity check modulus and remainder. */
4695  if (modulus <= 0)
4696  ereport(ERROR,
4697  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4698  errmsg("modulus for hash partition must be a positive integer")));
4699  if (remainder < 0)
4700  ereport(ERROR,
4701  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4702  errmsg("remainder for hash partition must be a non-negative integer")));
4703  if (remainder >= modulus)
4704  ereport(ERROR,
4705  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4706  errmsg("remainder for hash partition must be less than modulus")));
4707 
4708  /*
4709  * Cache hash function information.
4710  */
4711  my_extra = (ColumnsHashData *) fcinfo->flinfo->fn_extra;
4712  if (my_extra == NULL || my_extra->relid != parentId)
4713  {
4714  Relation parent;
4715  PartitionKey key;
4716  int j;
4717 
4718  /* Open parent relation and fetch partition key info */
4719  parent = relation_open(parentId, AccessShareLock);
4720  key = RelationGetPartitionKey(parent);
4721 
4722  /* Reject parent table that is not hash-partitioned. */
4723  if (key == NULL || key->strategy != PARTITION_STRATEGY_HASH)
4724  ereport(ERROR,
4725  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4726  errmsg("\"%s\" is not a hash partitioned table",
4727  get_rel_name(parentId))));
4728 
4729  if (!get_fn_expr_variadic(fcinfo->flinfo))
4730  {
4731  int nargs = PG_NARGS() - 3;
4732 
4733  /* complain if wrong number of column values */
4734  if (key->partnatts != nargs)
4735  ereport(ERROR,
4736  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4737  errmsg("number of partitioning columns (%d) does not match number of partition keys provided (%d)",
4738  key->partnatts, nargs)));
4739 
4740  /* allocate space for our cache */
4741  fcinfo->flinfo->fn_extra =
4742  MemoryContextAllocZero(fcinfo->flinfo->fn_mcxt,
4743  offsetof(ColumnsHashData, partsupfunc) +
4744  sizeof(FmgrInfo) * nargs);
4745  my_extra = (ColumnsHashData *) fcinfo->flinfo->fn_extra;
4746  my_extra->relid = parentId;
4747  my_extra->nkeys = key->partnatts;
4748  memcpy(my_extra->partcollid, key->partcollation,
4749  key->partnatts * sizeof(Oid));
4750 
4751  /* check argument types and save fmgr_infos */
4752  for (j = 0; j < key->partnatts; ++j)
4753  {
4754  Oid argtype = get_fn_expr_argtype(fcinfo->flinfo, j + 3);
4755 
4756  if (argtype != key->parttypid[j] && !IsBinaryCoercible(argtype, key->parttypid[j]))
4757  ereport(ERROR,
4758  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4759  errmsg("column %d of the partition key has type %s, but supplied value is of type %s",
4760  j + 1, format_type_be(key->parttypid[j]), format_type_be(argtype))));
4761 
4762  fmgr_info_copy(&my_extra->partsupfunc[j],
4763  &key->partsupfunc[j],
4764  fcinfo->flinfo->fn_mcxt);
4765  }
4766  }
4767  else
4768  {
4769  ArrayType *variadic_array = PG_GETARG_ARRAYTYPE_P(3);
4770 
4771  /* allocate space for our cache -- just one FmgrInfo in this case */
4772  fcinfo->flinfo->fn_extra =
4773  MemoryContextAllocZero(fcinfo->flinfo->fn_mcxt,
4774  offsetof(ColumnsHashData, partsupfunc) +
4775  sizeof(FmgrInfo));
4776  my_extra = (ColumnsHashData *) fcinfo->flinfo->fn_extra;
4777  my_extra->relid = parentId;
4778  my_extra->nkeys = key->partnatts;
4779  my_extra->variadic_type = ARR_ELEMTYPE(variadic_array);
4780  get_typlenbyvalalign(my_extra->variadic_type,
4781  &my_extra->variadic_typlen,
4782  &my_extra->variadic_typbyval,
4783  &my_extra->variadic_typalign);
4784  my_extra->partcollid[0] = key->partcollation[0];
4785 
4786  /* check argument types */
4787  for (j = 0; j < key->partnatts; ++j)
4788  if (key->parttypid[j] != my_extra->variadic_type)
4789  ereport(ERROR,
4790  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4791  errmsg("column %d of the partition key has type \"%s\", but supplied value is of type \"%s\"",
4792  j + 1,
4793  format_type_be(key->parttypid[j]),
4794  format_type_be(my_extra->variadic_type))));
4795 
4796  fmgr_info_copy(&my_extra->partsupfunc[0],
4797  &key->partsupfunc[0],
4798  fcinfo->flinfo->fn_mcxt);
4799  }
4800 
4801  /* Hold lock until commit */
4802  relation_close(parent, NoLock);
4803  }
4804 
4805  if (!OidIsValid(my_extra->variadic_type))
4806  {
4807  int nkeys = my_extra->nkeys;
4808  int i;
4809 
4810  /*
4811  * For a non-variadic call, neither the number of arguments nor their
4812  * types can change across calls, so avoid the expense of rechecking
4813  * here.
4814  */
4815 
4816  for (i = 0; i < nkeys; i++)
4817  {
4818  Datum hash;
4819 
4820  /* keys start from fourth argument of function. */
4821  int argno = i + 3;
4822 
4823  if (PG_ARGISNULL(argno))
4824  continue;
4825 
4826  hash = FunctionCall2Coll(&my_extra->partsupfunc[i],
4827  my_extra->partcollid[i],
4828  PG_GETARG_DATUM(argno),
4829  seed);
4830 
4831  /* Form a single 64-bit hash value */
4832  rowHash = hash_combine64(rowHash, DatumGetUInt64(hash));
4833  }
4834  }
4835  else
4836  {
4837  ArrayType *variadic_array = PG_GETARG_ARRAYTYPE_P(3);
4838  int i;
4839  int nelems;
4840  Datum *datum;
4841  bool *isnull;
4842 
4843  deconstruct_array(variadic_array,
4844  my_extra->variadic_type,
4845  my_extra->variadic_typlen,
4846  my_extra->variadic_typbyval,
4847  my_extra->variadic_typalign,
4848  &datum, &isnull, &nelems);
4849 
4850  /* complain if wrong number of column values */
4851  if (nelems != my_extra->nkeys)
4852  ereport(ERROR,
4853  (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
4854  errmsg("number of partitioning columns (%d) does not match number of partition keys provided (%d)",
4855  my_extra->nkeys, nelems)));
4856 
4857  for (i = 0; i < nelems; i++)
4858  {
4859  Datum hash;
4860 
4861  if (isnull[i])
4862  continue;
4863 
4864  hash = FunctionCall2Coll(&my_extra->partsupfunc[0],
4865  my_extra->partcollid[0],
4866  datum[i],
4867  seed);
4868 
4869  /* Form a single 64-bit hash value */
4870  rowHash = hash_combine64(rowHash, DatumGetUInt64(hash));
4871  }
4872  }
4873 
4874  PG_RETURN_BOOL(rowHash % modulus == remainder);
4875 }
Const::constvalue
Datum constvalue
Definition: primnodes.h:219
table_slot_create
TupleTableSlot * table_slot_create(Relation relation, List **reglist)
Definition: tableam.c:91
list_make2
#define list_make2(x1, x2)
Definition: pg_list.h:208
list_make3
#define list_make3(x1, x2, x3)
Definition: pg_list.h:210
int16
signed short int16
Definition: c.h:428
PartitionMap::did_remapping
bool did_remapping
Definition: partbounds.c:82
generate_matching_part_pairs
static void generate_matching_part_pairs(RelOptInfo *outer_rel, RelOptInfo *inner_rel, PartitionMap *outer_map, PartitionMap *inner_map, int nmerged, List **outer_parts, List **inner_parts)
Definition: partbounds.c:2330
ArrayExpr::multidims
bool multidims
Definition: primnodes.h:1028
NIL
#define NIL
Definition: pg_list.h:65
get_merged_range_bounds
static void get_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs, Oid *partcollations, JoinType jointype, PartitionRangeBound *outer_lb, PartitionRangeBound *outer_ub, PartitionRangeBound *inner_lb, PartitionRangeBound *inner_ub, int lb_cmpval, int ub_cmpval, PartitionRangeBound *merged_lb, PartitionRangeBound *merged_ub)
Definition: partbounds.c:2599
PG_GETARG_INT32
#define PG_GETARG_INT32(n)
Definition: fmgr.h:269
FmgrInfo
Definition: fmgr.h:56
merge_list_bounds
static PartitionBoundInfo merge_list_bounds(FmgrInfo *partsupfunc, Oid *collations, RelOptInfo *outer_rel, RelOptInfo *inner_rel, JoinType jointype, List **outer_parts, List **inner_parts)
Definition: partbounds.c:1089
PartitionBoundInfoData::kind
PartitionRangeDatumKind ** kind
Definition: partbounds.h:70
IsA
#define IsA(nodeptr, _type_)
Definition: nodes.h:590
PartitionRangeBound::kind
PartitionRangeDatumKind * kind
Definition: partbounds.c:68
ExecEvalExprSwitchContext
static Datum ExecEvalExprSwitchContext(ExprState *state, ExprContext *econtext, bool *isNull)
Definition: executor.h:330
DEBUG1
#define DEBUG1
Definition: elog.h:25
table_close
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:167
BTGreaterStrategyNumber
#define BTGreaterStrategyNumber
Definition: stratnum.h:33
forboth
#define forboth(cell1, list1, cell2, list2)
Definition: pg_list.h:446
FmgrInfo::fn_mcxt
MemoryContext fn_mcxt
Definition: fmgr.h:65
qsort_partition_hbound_cmp
static int32 qsort_partition_hbound_cmp(const void *a, const void *b)
Definition: partbounds.c:3670
value
static struct @142 value
build_merged_partition_bounds
static PartitionBoundInfo build_merged_partition_bounds(char strategy, List *merged_datums, List *merged_kinds, List *merged_indexes, int null_index, int default_index)
Definition: partbounds.c:2409
RegisterSnapshot
Snapshot RegisterSnapshot(Snapshot snapshot)
Definition: snapmgr.c:810
lnext
static ListCell * lnext(const List *l, const ListCell *c)
Definition: pg_list.h:322
DatumGetInt32
#define DatumGetInt32(X)
Definition: postgres.h:516
equal
bool equal(const void *a, const void *b)
Definition: equalfuncs.c:3113
PartitionKeyData::partopfamily
Oid * partopfamily
Definition: partcache.h:33
lower
Datum lower(PG_FUNCTION_ARGS)
Definition: oracle_compat.c:46
PartitionKeyData::partsupfunc
FmgrInfo * partsupfunc
Definition: partcache.h:35
castNode
#define castNode(_type_, nodeptr)
Definition: nodes.h:608
get_qual_for_hash
static List * get_qual_for_hash(Relation parent, PartitionBoundSpec *spec)
Definition: partbounds.c:3874
PartitionRangeDatumKind
PartitionRangeDatumKind
Definition: parsenodes.h:850
get_typlenbyvalalign
void get_typlenbyvalalign(Oid typid, int16 *typlen, bool *typbyval, char *typalign)
Definition: lsyscache.c:2218
merge_matching_partitions
static int merge_matching_partitions(PartitionMap *outer_map, PartitionMap *inner_map, int outer_part, int inner_part, int *next_index)
Definition: partbounds.c:1753
PG_GETARG_DATUM
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:268
get_array_type
Oid get_array_type(Oid typid)
Definition: lsyscache.c:2734
fix_opfuncids
void fix_opfuncids(Node *node)
Definition: nodeFuncs.c:1652
UInt64GetDatum
#define UInt64GetDatum(X)
Definition: postgres.h:692
datumIsEqual
bool datumIsEqual(Datum value1, Datum value2, bool typByVal, int typLen)
Definition: datum.c:222
process_inner_partition
static int process_inner_partition(PartitionMap *outer_map, PartitionMap *inner_map, bool outer_has_default, bool inner_has_default, int inner_index, int outer_default, JoinType jointype, int *next_index, int *default_index)
Definition: partbounds.c:1953
llast
#define llast(l)
Definition: pg_list.h:194
get_qual_from_partbound
List * get_qual_from_partbound(Relation rel, Relation parent, PartitionBoundSpec *spec)
Definition: partbounds.c:249
MemoryContextSwitchTo
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
IS_OUTER_JOIN
#define IS_OUTER_JOIN(jointype)
Definition: nodes.h:755
PartitionRangeDatum::kind
PartitionRangeDatumKind kind
Definition: parsenodes.h:861
AccessShareLock
#define AccessShareLock
Definition: lockdefs.h:36
FLEXIBLE_ARRAY_MEMBER
#define FLEXIBLE_ARRAY_MEMBER
Definition: c.h:350
Node
Definition: nodes.h:539
cur
struct cursor * cur
Definition: ecpg.c:28
get_fn_expr_variadic
bool get_fn_expr_variadic(FmgrInfo *flinfo)
Definition: fmgr.c:1934
StrategyNumber
uint16 StrategyNumber
Definition: stratnum.h:22
errcode
int errcode(int sqlerrcode)
Definition: elog.c:698
compare_range_partitions
static bool compare_range_partitions(int partnatts, FmgrInfo *partsupfuncs, Oid *partcollations, PartitionRangeBound *outer_lb, PartitionRangeBound *outer_ub, PartitionRangeBound *inner_lb, PartitionRangeBound *inner_ub, int *lb_cmpval, int *ub_cmpval)
Definition: partbounds.c:2550
get_range_nulltest
static List * get_range_nulltest(PartitionKey key)
Definition: partbounds.c:4572
stringToNode
void * stringToNode(const char *str)
Definition: read.c:89
qsort_partition_rbound_cmp
static int32 qsort_partition_rbound_cmp(const void *a, const void *b, void *arg)
Definition: partbounds.c:3702
PARTITION_MAX_KEYS
#define PARTITION_MAX_KEYS
Definition: pg_config_manual.h:57
generate_unaccent_rules.dest
dest
Definition: generate_unaccent_rules.py:282
ArrayExpr::array_typeid
Oid array_typeid
Definition: primnodes.h:1024
format_type_be
char * format_type_be(Oid type_oid)
Definition: format_type.c:339
upper
Datum upper(PG_FUNCTION_ARGS)
Definition: oracle_compat.c:77
PartitionKeyData::partexprs
List * partexprs
Definition: partcache.h:30
FunctionCall2Coll
Datum FunctionCall2Coll(FmgrInfo *flinfo, Oid collation, Datum arg1, Datum arg2)
Definition: fmgr.c:1148
table_scan_getnextslot
static bool table_scan_getnextslot(TableScanDesc sscan, ScanDirection direction, TupleTableSlot *slot)
Definition: tableam.h:1032
RelationGetPartitionKey
PartitionKey RelationGetPartitionKey(Relation rel)
Definition: partcache.c:54
PartitionRangeBound
struct PartitionRangeBound PartitionRangeBound
RelationData::rd_rel
Form_pg_class rd_rel
Definition: rel.h:109
relname
NameData relname
Definition: pg_class.h:38
Oid
unsigned int Oid
Definition: postgres_ext.h:31
pg_type.h
merge_range_bounds
static PartitionBoundInfo merge_range_bounds(int partnatts, FmgrInfo *partsupfuncs, Oid *partcollations, RelOptInfo *outer_rel, RelOptInfo *inner_rel, JoinType jointype, List **outer_parts, List **inner_parts)
Definition: partbounds.c:1397
makeConst
Const * makeConst(Oid consttype, int32 consttypmod, Oid constcollid, int constlen, Datum constvalue, bool constisnull, bool constbyval)
Definition: makefuncs.c:299
hash_combine64
static uint64 hash_combine64(uint64 a, uint64 b)
Definition: hashfn.h:80
OidIsValid
#define OidIsValid(objectId)
Definition: c.h:710
free_partition_map
static void free_partition_map(PartitionMap *map)
Definition: partbounds.c:1723
ExecPrepareExpr
ExprState * ExecPrepareExpr(Expr *node, EState *estate)
Definition: execExpr.c:692
make_opclause
Expr * make_opclause(Oid opno, Oid opresulttype, bool opretset, Expr *leftop, Expr *rightop, Oid opcollid, Oid inputcollid)
Definition: makefuncs.c:610
parse_coerce.h
int32
signed int int32
Definition: c.h:429
makeBoolExpr
Expr * makeBoolExpr(BoolExprType boolop, List *args, int location)
Definition: makefuncs.c:369
PartitionDescData::boundinfo
PartitionBoundInfo boundinfo
Definition: partdesc.h:38
JoinType
JoinType
Definition: nodes.h:706
BTLessEqualStrategyNumber
#define BTLessEqualStrategyNumber
Definition: stratnum.h:30
get_range_partition_internal
static int get_range_partition_internal(PartitionBoundInfo bi, int *lb_pos, PartitionRangeBound *lb, PartitionRangeBound *ub)
Definition: partbounds.c:2490
Abs
#define Abs(x)
Definition: c.h:992
index
Definition: type.h:89
list_make1
#define list_make1(x1)
Definition: pg_list.h:206
add_merged_range_bounds
static void add_merged_range_bounds(int partnatts, FmgrInfo *partsupfuncs, Oid *partcollations, PartitionRangeBound *merged_lb, PartitionRangeBound *merged_ub, int merged_index, List **merged_datums, List **merged_kinds, List **merged_indexes)
Definition: partbounds.c:2663
FreeExecutorState
void FreeExecutorState(EState *estate)
Definition: execUtils.c:186
make_partition_op_expr
static Expr * make_partition_op_expr(PartitionKey key, int keynum, uint16 strategy, Expr *arg1, Expr *arg2)
Definition: partbounds.c:3760
get_range_key_properties
static void get_range_key_properties(PartitionKey key, int keynum, PartitionRangeDatum *ldatum, PartitionRangeDatum *udatum, ListCell **partexprs_item, Expr **keyCol, Const **lower_val, Const **upper_val)
Definition: partbounds.c:4528
GetPerTupleExprContext
#define GetPerTupleExprContext(estate)
Definition: executor.h:532
make_one_partition_rbound
static PartitionRangeBound * make_one_partition_rbound(PartitionKey key, int index, List *datums, bool lower)
Definition: partbounds.c:3320
PG_GETARG_ARRAYTYPE_P
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:256
partition_range_datum_bsearch
int partition_range_datum_bsearch(FmgrInfo *partsupfunc, Oid *partcollation, PartitionBoundInfo boundinfo, int nvalues, Datum *values, bool *is_equal)
Definition: partbounds.c:3587
check_new_partition_bound
void check_new_partition_bound(char *relname, Relation parent, PartitionBoundSpec *spec, ParseState *pstate)
Definition: partbounds.c:2797
uint16
unsigned short uint16
Definition: c.h:440
pfree
void pfree(void *pointer)
Definition: mcxt.c:1169
EState::es_query_cxt
MemoryContext es_query_cxt
Definition: execnodes.h:598
PartitionKeyData::parttypcoll
Oid * parttypcoll
Definition: partcache.h:46
linitial
#define linitial(l)
Definition: pg_list.h:174
ObjectIdGetDatum
#define ObjectIdGetDatum(X)
Definition: postgres.h:551
ERROR
#define ERROR
Definition: elog.h:46
table_beginscan
static TableScanDesc table_beginscan(Relation rel, Snapshot snapshot, int nkeys, struct ScanKeyData *key)
Definition: tableam.h:883
lfirst_int
#define lfirst_int(lc)
Definition: pg_list.h:170
list_nth
static void * list_nth(const List *list, int n)
Definition: pg_list.h:278
get_fn_expr_argtype
Oid get_fn_expr_argtype(FmgrInfo *flinfo, int argnum)
Definition: fmgr.c:1800
relation_open
Relation relation_open(Oid relationId, LOCKMODE lockmode)
Definition: relation.c:48
fix_merged_indexes
static void fix_merged_indexes(PartitionMap *outer_map, PartitionMap *inner_map, int nmerged, List *merged_indexes)
Definition: partbounds.c:2276
makeBoolConst
Node * makeBoolConst(bool value, bool isnull)
Definition: makefuncs.c:357
PartitionMap
struct PartitionMap PartitionMap
NullTest::arg
Expr * arg
Definition: primnodes.h:1255
IS_DUMMY_REL
#define IS_DUMMY_REL(r)
Definition: pathnodes.h:1458
check_default_partition_contents
void check_default_partition_contents(Relation parent, Relation default_rel, PartitionBoundSpec *new_spec)
Definition: partbounds.c:3143
create_hash_bounds
static PartitionBoundInfo create_hash_bounds(PartitionBoundSpec **boundspecs, int nparts, PartitionKey key, int **mapping)
Definition: partbounds.c:357
fmgr_info_copy
void fmgr_info_copy(FmgrInfo *dstinfo, FmgrInfo *srcinfo, MemoryContext destcxt)
Definition: fmgr.c:608
get_hash_partition_greatest_modulus
int get_hash_partition_greatest_modulus(PartitionBoundInfo bound)
Definition: partbounds.c:3306
c
char * c
Definition: preproc-cursor.c:31
create_list_bounds
static PartitionBoundInfo create_list_bounds(PartitionBoundSpec **boundspecs, int nparts, PartitionKey key, int **mapping)
Definition: partbounds.c:440
NoLock
#define NoLock
Definition: lockdefs.h:34
partition_rbound_datum_cmp
int32 partition_rbound_datum_cmp(FmgrInfo *partsupfunc, Oid *partcollation, Datum *rb_datums, PartitionRangeDatumKind *rb_kind, Datum *tuple_datums, int n_tuple_datums)
Definition: partbounds.c:3448
get_proposed_default_constraint
List * get_proposed_default_constraint(List *new_part_constraints)
Definition: partition.c:368
makeRelabelType
RelabelType * makeRelabelType(Expr *arg, Oid rtype, int32 rtypmod, Oid rcollid, CoercionForm rformat)
Definition: makefuncs.c:402
PG_GETARG_OID
#define PG_GETARG_OID(n)
Definition: fmgr.h:275
ExecDropSingleTupleTableSlot
void ExecDropSingleTupleTableSlot(TupleTableSlot *slot)
Definition: execTuples.c:1254
RelationGetPartitionDesc
PartitionDesc RelationGetPartitionDesc(Relation rel, bool omit_detached)
Definition: partdesc.c:72
errdetail
int errdetail(const char *fmt,...)
Definition: elog.c:1042
RelOptInfo::nparts
int nparts
Definition: pathnodes.h:756
generate_unaccent_rules.args
args
Definition: generate_unaccent_rules.py:285
get_opfamily_member
Oid get_opfamily_member(Oid opfamily, Oid lefttype, Oid righttype, int16 strategy)
Definition: lsyscache.c:164
DatumGetBool
#define DatumGetBool(X)
Definition: postgres.h:437
partition_hbound_cmp
static int32 partition_hbound_cmp(int modulus1, int remainder1, int modulus2, int remainder2)
Definition: partbounds.c:3479
merge_partition_with_dummy
static int merge_partition_with_dummy(PartitionMap *map, int index, int *next_index)
Definition: partbounds.c:2258
PartitionMap::merged
bool * merged
Definition: partbounds.c:80
is_dummy_partition
static bool is_dummy_partition(RelOptInfo *rel, int part_index)
Definition: partbounds.c:1734
RelationGetRelationName
#define RelationGetRelationName(relation)
Definition: rel.h:503
partition_bound_has_default
#define partition_bound_has_default(bi)
Definition: partbounds.h:82
list_head
static ListCell * list_head(const List *l)
Definition: pg_list.h:125
partition_hash_bsearch
int partition_hash_bsearch(PartitionBoundInfo boundinfo, int modulus, int remainder)
Definition: partbounds.c:3630
ArrayExpr::elements
List * elements
Definition: primnodes.h:1027
makeVar
Var * makeVar(Index varno, AttrNumber varattno, Oid vartype, int32 vartypmod, Oid varcollid, Index varlevelsup)
Definition: makefuncs.c:66
datumCopy
Datum datumCopy(Datum value, bool typByVal, int typLen)
Definition: datum.c:131
CreateExecutorState
EState * CreateExecutorState(void)
Definition: execUtils.c:90
lappend_int
List * lappend_int(List *list, int datum)
Definition: list.c:354
get_range_partbound_string
char * get_range_partbound_string(List *bound_datums)
Definition: ruleutils.c:11969
UnregisterSnapshot
void UnregisterSnapshot(Snapshot snapshot)
Definition: snapmgr.c:852
IsBinaryCoercible
bool IsBinaryCoercible(Oid srctype, Oid targettype)
Definition: parse_coerce.c:2877
lappend
List * lappend(List *list, void *datum)
Definition: list.c:336
qsort_arg
void qsort_arg(void *base, size_t nel, size_t elsize, qsort_arg_comparator cmp, void *arg)
WARNING
#define WARNING
Definition: elog.h:40
SearchSysCache1
HeapTuple SearchSysCache1(int cacheId, Datum key1)
Definition: syscache.c:1127
PartitionKeyData::partcollation
Oid * partcollation
Definition: partcache.h:38
TextDatumGetCString
#define TextDatumGetCString(d)
Definition: builtins.h:83
partition_bounds_copy
PartitionBoundInfo partition_bounds_copy(PartitionBoundInfo src, PartitionKey key)
Definition: partbounds.c:907
Const::location
int location
Definition: primnodes.h:226
EState::es_tupleTable
List * es_tupleTable
Definition: execnodes.h:600
palloc0
void * palloc0(Size size)
Definition: mcxt.c:1093
ArrayExpr::location
int location
Definition: primnodes.h:1029
PartitionKeyData::partattrs
AttrNumber * partattrs
Definition: partcache.h:28
PG_RETURN_BOOL
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359
Datum
uintptr_t Datum
Definition: postgres.h:411
ReleaseSysCache
void ReleaseSysCache(HeapTuple tuple)
Definition: syscache.c:1175
for_both_cell
#define for_both_cell(cell1, list1, initcell1, cell2, list2, initcell2)
Definition: pg_list.h:468
make_ands_explicit
Expr * make_ands_explicit(List *andclauses)
Definition: makefuncs.c:708
SysCacheGetAttr
Datum SysCacheGetAttr(int cacheId, HeapTuple tup, AttrNumber attributeNumber, bool *isNull)
Definition: syscache.c:1388
list_make1_oid
#define list_make1_oid(x1)
Definition: pg_list.h:236
cleanup
static void cleanup(void)
Definition: bootstrap.c:872
RelOptInfo::boundinfo
struct PartitionBoundInfoData * boundinfo
Definition: pathnodes.h:759
get_partition_operator
static Oid get_partition_operator(PartitionKey key, int col, StrategyNumber strategy, bool *need_relabel)
Definition: partbounds.c:3724
PARTITION_STRATEGY_HASH
#define PARTITION_STRATEGY_HASH
Definition: parsenodes.h:814
NullTest::nulltesttype
NullTestType nulltesttype
Definition: primnodes.h:1256
partition_bound_accepts_nulls
#define partition_bound_accepts_nulls(bi)
Definition: partbounds.h:81
MemoryContextAllocZero
void * MemoryContextAllocZero(MemoryContext context, Size size)
Definition: mcxt.c:906
init_partition_map
static void init_partition_map(RelOptInfo *rel, PartitionMap *map)
Definition: partbounds.c:1702
PartitionListValue
struct PartitionListValue PartitionListValue
PartitionKeyData::parttypmod
int32 * parttypmod
Definition: partcache.h:42
partition_list_bsearch
int partition_list_bsearch(FmgrInfo *partsupfunc, Oid *partcollation, PartitionBoundInfo boundinfo, Datum value, bool *is_equal)
Definition: partbounds.c:3499
compute_partition_hash_value
uint64 compute_partition_hash_value(int partnatts, FmgrInfo *partsupfunc, Oid *partcollation, Datum *values, bool *isnull)
Definition: partbounds.c:4618
InvalidOid
#define InvalidOid
Definition: postgres_ext.h:36
get_opcode
RegProcedure get_opcode(Oid opno)
Definition: lsyscache.c:1256
ereport
#define ereport(elevel,...)
Definition: elog.h:157
partition_rbound_cmp
static int32 partition_rbound_cmp(int partnatts, FmgrInfo *partsupfunc, Oid *partcollation, Datum *datums1, PartitionRangeDatumKind *kind1, bool lower1, PartitionRangeBound *b2)
Definition: partbounds.c:3380
errmsg_internal
int errmsg_internal(const char *fmt,...)
Definition: elog.c:996
get_qual_for_range
static List * get_qual_for_range(Relation parent, PartitionBoundSpec *spec, bool for_default)
Definition: partbounds.c:4166
DatumGetUInt64
#define DatumGetUInt64(X)
Definition: postgres.h:678
Max
#define Max(x, y)
Definition: c.h:980
makeNode
#define makeNode(_type_)
Definition: nodes.h:587
partitions_are_ordered
bool partitions_are_ordered(PartitionBoundInfo boundinfo, int nparts)
Definition: partbounds.c:2738
PartitionKeyData::parttypbyval
bool * parttypbyval
Definition: partcache.h:44
PG_ARGISNULL
#define PG_ARGISNULL(n)
Definition: fmgr.h:209
HeapTupleIsValid
#define HeapTupleIsValid(tuple)
Definition: htup.h:78
relation_close
void relation_close(Relation relation, LOCKMODE lockmode)
Definition: relation.c:206
Assert
#define Assert(condition)
Definition: c.h:804
lfirst
#define lfirst(lc)
Definition: pg_list.h:169
PartitionKeyData::parttyplen
int16 * parttyplen
Definition: partcache.h:43
partition_range_bsearch
static int partition_range_bsearch(int partnatts, FmgrInfo *partsupfunc, Oid *partcollation, PartitionBoundInfo boundinfo, PartitionRangeBound *probe, int32 *cmpval)
Definition: partbounds.c:3545
ArrayExpr::array_collid
Oid array_collid
Definition: primnodes.h:1025
RelOptInfo::part_rels
struct RelOptInfo ** part_rels
Definition: pathnodes.h:763
NullTest::location
int location
Definition: primnodes.h:1258
list_length
static int list_length(const List *l)
Definition: pg_list.h:149
parser_errposition
int parser_errposition(ParseState *pstate, int location)
Definition: parse_node.c:111
merge_default_partitions
static void merge_default_partitions(PartitionMap *outer_map, PartitionMap *inner_map, bool outer_has_default, bool inner_has_default, int outer_default, int inner_default, JoinType jointype, int *next_index, int *default_index)
Definition: partbounds.c:2148
ExprContext::ecxt_scantuple
TupleTableSlot * ecxt_scantuple
Definition: execnodes.h:226
type_is_array
#define type_is_array(typid)
Definition: lsyscache.h:202
PG_NARGS
#define PG_NARGS()
Definition: fmgr.h:203
HASH_PARTITION_SEED
#define HASH_PARTITION_SEED
Definition: partition.h:20
PARTITION_STRATEGY_LIST
#define PARTITION_STRATEGY_LIST
Definition: parsenodes.h:815
GetLatestSnapshot
Snapshot GetLatestSnapshot(void)
Definition: snapmgr.c:325
GetPerTupleMemoryContext
#define GetPerTupleMemoryContext(estate)
Definition: executor.h:537
ArrayExpr::element_typeid
Oid element_typeid
Definition: primnodes.h:1026
process_outer_partition
static int process_outer_partition(PartitionMap *outer_map, PartitionMap *inner_map, bool outer_has_default, bool inner_has_default, int outer_index, int inner_default, JoinType jointype, int *next_index, int *default_index)
Definition: partbounds.c:1871
deconstruct_array
void deconstruct_array(ArrayType *array, Oid elmtype, int elmlen, bool elmbyval, char elmalign, Datum **elemsp, bool **nullsp, int *nelemsp)
Definition: arrayfuncs.c:3488
table_endscan
static void table_endscan(TableScanDesc scan)
Definition: tableam.h:991
values
static Datum values[MAXATTR]
Definition: bootstrap.c:166
PARTITION_STRATEGY_RANGE
#define PARTITION_STRATEGY_RANGE
Definition: parsenodes.h:816
AccessExclusiveLock
#define AccessExclusiveLock
Definition: lockdefs.h:45
find_all_inheritors
List * find_all_inheritors(Oid parentrelId, LOCKMODE lockmode, List **numparents)
Definition: pg_inherits.c:256
Int32GetDatum
#define Int32GetDatum(X)
Definition: postgres.h:523
pg_inherits.h
create_range_bounds
static PartitionBoundInfo create_range_bounds(PartitionBoundSpec **boundspecs, int nparts, PartitionKey key, int **mapping)
Definition: partbounds.c:598
palloc
void * palloc(Size size)
Definition: mcxt.c:1062
errmsg
int errmsg(const char *fmt,...)
Definition: elog.c:909
PartitionMap::old_indexes
int * old_indexes
Definition: partbounds.c:83
partition_bounds_equal
bool partition_bounds_equal(int partnatts, int16 *parttyplen, bool *parttypbyval, PartitionBoundInfo b1, PartitionBoundInfo b2)
Definition: partbounds.c:801
PartitionKeyData::partopcintype
Oid * partopcintype
Definition: partcache.h:34
list_free
void list_free(List *list)
Definition: list.c:1391
elog
#define elog(elevel,...)
Definition: elog.h:232
i
int i
Definition: preproc-comment.c:23
arg
void * arg
Definition: pg_backup_utils.c:27
ExecCheck
bool ExecCheck(ExprState *state, ExprContext *econtext)
Definition: execExpr.c:799
compare_range_bounds
#define compare_range_bounds(partnatts, partsupfunc, partcollations, bound1, bound2)
Definition: partbounds.c:88
NullTest::argisrow
bool argisrow
Definition: primnodes.h:1257
satisfies_hash_partition
Datum satisfies_hash_partition(PG_FUNCTION_ARGS)
Definition: partbounds.c:4667
PG_FUNCTION_ARGS
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
ExecInitExpr
ExprState * ExecInitExpr(Expr *node, PlanState *parent)
Definition: execExpr.c:123
CHECK_FOR_INTERRUPTS
#define CHECK_FOR_INTERRUPTS()
Definition: miscadmin.h:102
PartitionBoundInfo
struct PartitionBoundInfoData * PartitionBoundInfo
Definition: partdefs.h:16
qsort
#define qsort(a, b, c, d)
Definition: port.h:504
copyObject
#define copyObject(obj)
Definition: nodes.h:655
partition_bounds_merge
PartitionBoundInfo partition_bounds_merge(int partnatts, FmgrInfo *partsupfunc, Oid *partcollation, RelOptInfo *outer_rel, RelOptInfo *inner_rel, JoinType jointype, List **outer_parts, List **inner_parts)
Definition: partbounds.c:1006
BTLessStrategyNumber
#define BTLessStrategyNumber
Definition: stratnum.h:29
table_open
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:39
qsort_partition_list_value_cmp
static int32 qsort_partition_list_value_cmp(const void *a, const void *b, void *arg)
Definition: partbounds.c:3685
List
Definition: pg_list.h:50
hash
static unsigned hash(unsigned *uv, int n)
Definition: rege_dfa.c:719
get_rel_name
char * get_rel_name(Oid relid)
Definition: lsyscache.c:1899
errtable
int errtable(Relation rel)
Definition: relcache.c:5628
ARR_ELEMTYPE
#define ARR_ELEMTYPE(a)
Definition: array.h:285
PartitionMap::merged_indexes
int * merged_indexes
Definition: partbounds.c:79
map_partition_varattnos
List * map_partition_varattnos(List *expr, int fromrel_varno, Relation to_rel, Relation from_rel)
Definition: partition.c:221
RelationGetRelid
#define RelationGetRelid(relation)
Definition: rel.h:469
partition_bounds_create
PartitionBoundInfo partition_bounds_create(PartitionBoundSpec **boundspecs, int nparts, PartitionKey key, int **mapping)
Definition: partbounds.c:304
val
long val
Definition: informix.c:664
get_qual_for_list
static List * get_qual_for_list(Relation parent, PartitionBoundSpec *spec)
Definition: partbounds.c:3957
Const::constisnull
bool constisnull
Definition: primnodes.h:220
BTEqualStrategyNumber
#define BTEqualStrategyNumber
Definition: stratnum.h:31
offsetof
#define offsetof(type, field)
Definition: c.h:727
BTGreaterEqualStrategyNumber
#define BTGreaterEqualStrategyNumber
Definition: stratnum.h:32
ResetExprContext
#define ResetExprContext(econtext)
Definition: executor.h:526
lfirst_oid
#define lfirst_oid(lc)
Definition: pg_list.h:171
PartConstraintImpliedByRelConstraint
bool PartConstraintImpliedByRelConstraint(Relation scanrel, List *partConstraint)
Definition: tablecmds.c:16890
makeFuncExpr
FuncExpr * makeFuncExpr(Oid funcid, Oid rettype, List *args, Oid funccollid, Oid inputcollid, CoercionForm fformat)
Definition: makefuncs.c:519
PartitionKey
struct PartitionKeyData * PartitionKey
Definition: partdefs.h:18
llast_int
#define llast_int(l)
Definition: pg_list.h:195
merge_null_partitions
static void merge_null_partitions(PartitionMap *outer_map, PartitionMap *inner_map, bool outer_has_null, bool inner_has_null, int outer_null, int inner_null, JoinType jointype, int *next_index, int *null_index)
Definition: partbounds.c:2038
get_range_partition
static int get_range_partition(RelOptInfo *rel, PartitionBoundInfo bi, int *lb_pos, PartitionRangeBound *lb, PartitionRangeBound *ub)
Definition: partbounds.c:2469
PartitionHashBound
struct PartitionHashBound PartitionHashBound