1
0
mirror of https://github.com/postgres/postgres.git synced 2025-07-27 12:41:57 +03:00

Add sortsupport for gist_btree opclasses, for faster index builds.

Commit 16fa9b2b30 introduced a faster way to build GiST indexes, by
sorting all the data. This commit adds the sortsupport functions needed
to make use of that feature for btree_gist.

Author: Andrey Borodin
Discussion: https://www.postgresql.org/message-id/2F3F7265-0D22-44DB-AD71-8554C743D943@yandex-team.ru
This commit is contained in:
Heikki Linnakangas
2021-04-07 13:22:05 +03:00
parent dd13ad9d39
commit 9f984ba6d2
77 changed files with 1530 additions and 3 deletions

View File

@ -24,6 +24,7 @@ PG_FUNCTION_INFO_V1(gbt_int8_consistent);
PG_FUNCTION_INFO_V1(gbt_int8_distance);
PG_FUNCTION_INFO_V1(gbt_int8_penalty);
PG_FUNCTION_INFO_V1(gbt_int8_same);
PG_FUNCTION_INFO_V1(gbt_int8_sortsupport);
static bool
@ -215,3 +216,82 @@ gbt_int8_same(PG_FUNCTION_ARGS)
*result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
PG_RETURN_POINTER(result);
}
static int
gbt_int8_sort_build_cmp(Datum a, Datum b, SortSupport ssup)
{
int64KEY *ia = (int64KEY *) DatumGetPointer(a);
int64KEY *ib = (int64KEY *) DatumGetPointer(b);
/* for leaf items we expect lower == upper */
Assert(ia->lower == ia->upper);
Assert(ib->lower == ib->upper);
if (ia->lower == ib->lower)
return 0;
return (ia->lower > ib->lower) ? 1 : -1;
}
static Datum
gbt_int8_abbrev_convert(Datum original, SortSupport ssup)
{
int64KEY *b1 = (int64KEY *) DatumGetPointer(original);
int64 z = b1->lower;
#if SIZEOF_DATUM == 8
return Int64GetDatum(z);
#else
return Int32GetDatum(z >> 32);
#endif
}
static int
gbt_int8_cmp_abbrev(Datum z1, Datum z2, SortSupport ssup)
{
#if SIZEOF_DATUM == 8
int64 a = DatumGetInt64(z1);
int64 b = DatumGetInt64(z2);
#else
int32 a = DatumGetInt32(z1);
int32 b = DatumGetInt32(z2);
#endif
if (a > b)
return 1;
else if (a < b)
return -1;
else
return 0;
}
/*
* We never consider aborting the abbreviation.
*/
static bool
gbt_int8_abbrev_abort(int memtupcount, SortSupport ssup)
{
return false;
}
/*
* Sort support routine for fast GiST index build by sorting.
*/
Datum
gbt_int8_sortsupport(PG_FUNCTION_ARGS)
{
SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
if (ssup->abbreviate)
{
ssup->comparator = gbt_int8_cmp_abbrev;
ssup->abbrev_converter = gbt_int8_abbrev_convert;
ssup->abbrev_abort = gbt_int8_abbrev_abort;
ssup->abbrev_full_comparator = gbt_int8_sort_build_cmp;
}
else
{
ssup->comparator = gbt_int8_sort_build_cmp;
}
PG_RETURN_VOID();
}