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:
@ -25,6 +25,7 @@ PG_FUNCTION_INFO_V1(gbt_cash_consistent);
|
||||
PG_FUNCTION_INFO_V1(gbt_cash_distance);
|
||||
PG_FUNCTION_INFO_V1(gbt_cash_penalty);
|
||||
PG_FUNCTION_INFO_V1(gbt_cash_same);
|
||||
PG_FUNCTION_INFO_V1(gbt_cash_sortsupport);
|
||||
|
||||
static bool
|
||||
gbt_cashgt(const void *a, const void *b, FmgrInfo *flinfo)
|
||||
@ -216,3 +217,82 @@ gbt_cash_same(PG_FUNCTION_ARGS)
|
||||
*result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
|
||||
PG_RETURN_POINTER(result);
|
||||
}
|
||||
|
||||
static int
|
||||
gbt_cash_sort_build_cmp(Datum a, Datum b, SortSupport ssup)
|
||||
{
|
||||
cashKEY *ia = (cashKEY *) DatumGetPointer(a);
|
||||
cashKEY *ib = (cashKEY *) 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_cash_abbrev_convert(Datum original, SortSupport ssup)
|
||||
{
|
||||
cashKEY *b1 = (cashKEY *) DatumGetPointer(original);
|
||||
int64 z = b1->lower;
|
||||
|
||||
#if SIZEOF_DATUM == 8
|
||||
return Int64GetDatum(z);
|
||||
#else
|
||||
return Int32GetDatum(z >> 32);
|
||||
#endif
|
||||
}
|
||||
|
||||
static int
|
||||
gbt_cash_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_cash_abbrev_abort(int memtupcount, SortSupport ssup)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
/*
|
||||
* Sort support routine for fast GiST index build by sorting.
|
||||
*/
|
||||
Datum
|
||||
gbt_cash_sortsupport(PG_FUNCTION_ARGS)
|
||||
{
|
||||
SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
|
||||
|
||||
if (ssup->abbreviate)
|
||||
{
|
||||
ssup->comparator = gbt_cash_cmp_abbrev;
|
||||
ssup->abbrev_converter = gbt_cash_abbrev_convert;
|
||||
ssup->abbrev_abort = gbt_cash_abbrev_abort;
|
||||
ssup->abbrev_full_comparator = gbt_cash_sort_build_cmp;
|
||||
}
|
||||
else
|
||||
{
|
||||
ssup->comparator = gbt_cash_sort_build_cmp;
|
||||
}
|
||||
PG_RETURN_VOID();
|
||||
}
|
||||
|
Reference in New Issue
Block a user