mirror of
https://github.com/hanzoai/postgres.git
synced 2026-10-30 18:33:07 +00:00
Peter Eisentraut
dd2c4f9634
There were many PG_GETARG_* calls, mostly around gin, gist, spgist
code, that were commented out, presumably to indicate that the
argument was unused and to indicate that it wasn't forgotten or
miscounted. But keeping commented-out code updated with refactorings
and style changes is annoying. So this commit changes them to
#ifdef NOT_USED
blocks, which is a style already in use. That way, at least the
indentation and syntax highlighting works correctly, making some of
these blocks much easier to read.
An alternative would be to just delete that code, but there is some
value in making unused arguments explicit, and some of this arguably
serves as example code for index AM APIs.
Reviewed-by: Heikki Linnakangas <[email protected]>
Reviewed-by: David Geier <[email protected]>
Reviewed-by: Chao Li <[email protected]>
Reviewed-by: Dean Rasheed <[email protected]>
Reviewed-by: Tom Lane <[email protected]>
Discussion: https://www.postgresql.org/message-id/flat/328e4371-9a4c-4196-9df9-1f23afc900df%40eisentraut.org
353 lines
8.4 KiB
C
353 lines
8.4 KiB
C
/*
|
|
* contrib/btree_gist/btree_text.c
|
|
*/
|
|
#include "postgres.h"
|
|
|
|
#include "btree_gist.h"
|
|
#include "btree_utils_var.h"
|
|
#include "mb/pg_wchar.h"
|
|
#include "utils/fmgrprotos.h"
|
|
#include "utils/sortsupport.h"
|
|
|
|
/* GiST support functions */
|
|
PG_FUNCTION_INFO_V1(gbt_text_compress);
|
|
PG_FUNCTION_INFO_V1(gbt_bpchar_compress);
|
|
PG_FUNCTION_INFO_V1(gbt_text_union);
|
|
PG_FUNCTION_INFO_V1(gbt_text_picksplit);
|
|
PG_FUNCTION_INFO_V1(gbt_text_consistent);
|
|
PG_FUNCTION_INFO_V1(gbt_bpchar_consistent);
|
|
PG_FUNCTION_INFO_V1(gbt_text_penalty);
|
|
PG_FUNCTION_INFO_V1(gbt_text_same);
|
|
PG_FUNCTION_INFO_V1(gbt_text_sortsupport);
|
|
PG_FUNCTION_INFO_V1(gbt_bpchar_sortsupport);
|
|
|
|
|
|
/* define for comparison */
|
|
|
|
static bool
|
|
gbt_textgt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(text_gt,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_textge(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(text_ge,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_texteq(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(texteq,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_textle(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(text_le,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_textlt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(text_lt,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static int32
|
|
gbt_textcmp(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetInt32(DirectFunctionCall2Coll(bttextcmp,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static gbtree_vinfo tinfo =
|
|
{
|
|
gbt_t_text,
|
|
0,
|
|
false,
|
|
gbt_textgt,
|
|
gbt_textge,
|
|
gbt_texteq,
|
|
gbt_textle,
|
|
gbt_textlt,
|
|
gbt_textcmp,
|
|
NULL
|
|
};
|
|
|
|
/* bpchar needs its own comparison rules */
|
|
|
|
static bool
|
|
gbt_bpchargt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(bpchargt,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_bpcharge(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(bpcharge,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_bpchareq(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(bpchareq,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_bpcharle(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(bpcharle,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static bool
|
|
gbt_bpcharlt(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetBool(DirectFunctionCall2Coll(bpcharlt,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static int32
|
|
gbt_bpcharcmp(const void *a, const void *b, Oid collation, FmgrInfo *flinfo)
|
|
{
|
|
return DatumGetInt32(DirectFunctionCall2Coll(bpcharcmp,
|
|
collation,
|
|
PointerGetDatum(a),
|
|
PointerGetDatum(b)));
|
|
}
|
|
|
|
static gbtree_vinfo bptinfo =
|
|
{
|
|
gbt_t_bpchar,
|
|
0,
|
|
false,
|
|
gbt_bpchargt,
|
|
gbt_bpcharge,
|
|
gbt_bpchareq,
|
|
gbt_bpcharle,
|
|
gbt_bpcharlt,
|
|
gbt_bpcharcmp,
|
|
NULL
|
|
};
|
|
|
|
|
|
/**************************************************
|
|
* GiST support functions
|
|
**************************************************/
|
|
|
|
Datum
|
|
gbt_text_compress(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
|
|
if (tinfo.eml == 0)
|
|
{
|
|
tinfo.eml = pg_database_encoding_max_length();
|
|
}
|
|
|
|
PG_RETURN_POINTER(gbt_var_compress(entry, &tinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_bpchar_compress(PG_FUNCTION_ARGS)
|
|
{
|
|
/* This should never have been distinct from gbt_text_compress */
|
|
return gbt_text_compress(fcinfo);
|
|
}
|
|
|
|
Datum
|
|
gbt_text_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
void *query = DatumGetTextP(PG_GETARG_DATUM(1));
|
|
StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
|
|
#ifdef NOT_USED
|
|
Oid subtype = PG_GETARG_OID(3);
|
|
#endif
|
|
bool *recheck = (bool *) PG_GETARG_POINTER(4);
|
|
bool retval;
|
|
GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
|
|
GBT_VARKEY_R r = gbt_var_key_readable(key);
|
|
|
|
/* All cases served by this function are exact */
|
|
*recheck = false;
|
|
|
|
if (tinfo.eml == 0)
|
|
{
|
|
tinfo.eml = pg_database_encoding_max_length();
|
|
}
|
|
|
|
retval = gbt_var_consistent(&r, query, strategy, PG_GET_COLLATION(),
|
|
GIST_LEAF(entry), &tinfo, fcinfo->flinfo);
|
|
|
|
PG_RETURN_BOOL(retval);
|
|
}
|
|
|
|
Datum
|
|
gbt_bpchar_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
void *query = DatumGetTextP(PG_GETARG_DATUM(1));
|
|
StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
|
|
#ifdef NOT_USED
|
|
Oid subtype = PG_GETARG_OID(3);
|
|
#endif
|
|
bool *recheck = (bool *) PG_GETARG_POINTER(4);
|
|
bool retval;
|
|
GBT_VARKEY *key = (GBT_VARKEY *) DatumGetPointer(entry->key);
|
|
GBT_VARKEY_R r = gbt_var_key_readable(key);
|
|
|
|
/* All cases served by this function are exact */
|
|
*recheck = false;
|
|
|
|
if (bptinfo.eml == 0)
|
|
{
|
|
bptinfo.eml = pg_database_encoding_max_length();
|
|
}
|
|
|
|
retval = gbt_var_consistent(&r, query, strategy, PG_GET_COLLATION(),
|
|
GIST_LEAF(entry), &bptinfo, fcinfo->flinfo);
|
|
PG_RETURN_BOOL(retval);
|
|
}
|
|
|
|
Datum
|
|
gbt_text_union(PG_FUNCTION_ARGS)
|
|
{
|
|
GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
|
|
int32 *size = (int *) PG_GETARG_POINTER(1);
|
|
|
|
PG_RETURN_POINTER(gbt_var_union(entryvec, size, PG_GET_COLLATION(),
|
|
&tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
Datum
|
|
gbt_text_picksplit(PG_FUNCTION_ARGS)
|
|
{
|
|
GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
|
|
GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
|
|
|
|
gbt_var_picksplit(entryvec, v, PG_GET_COLLATION(),
|
|
&tinfo, fcinfo->flinfo);
|
|
PG_RETURN_POINTER(v);
|
|
}
|
|
|
|
Datum
|
|
gbt_text_same(PG_FUNCTION_ARGS)
|
|
{
|
|
Datum d1 = PG_GETARG_DATUM(0);
|
|
Datum d2 = PG_GETARG_DATUM(1);
|
|
bool *result = (bool *) PG_GETARG_POINTER(2);
|
|
|
|
*result = gbt_var_same(d1, d2, PG_GET_COLLATION(), &tinfo, fcinfo->flinfo);
|
|
PG_RETURN_POINTER(result);
|
|
}
|
|
|
|
Datum
|
|
gbt_text_penalty(PG_FUNCTION_ARGS)
|
|
{
|
|
GISTENTRY *o = (GISTENTRY *) PG_GETARG_POINTER(0);
|
|
GISTENTRY *n = (GISTENTRY *) PG_GETARG_POINTER(1);
|
|
float *result = (float *) PG_GETARG_POINTER(2);
|
|
|
|
PG_RETURN_POINTER(gbt_var_penalty(result, o, n, PG_GET_COLLATION(),
|
|
&tinfo, fcinfo->flinfo));
|
|
}
|
|
|
|
static int
|
|
gbt_text_ssup_cmp(Datum x, Datum y, SortSupport ssup)
|
|
{
|
|
GBT_VARKEY *key1 = PG_DETOAST_DATUM(x);
|
|
GBT_VARKEY *key2 = PG_DETOAST_DATUM(y);
|
|
|
|
GBT_VARKEY_R arg1 = gbt_var_key_readable(key1);
|
|
GBT_VARKEY_R arg2 = gbt_var_key_readable(key2);
|
|
Datum result;
|
|
|
|
/* for leaf items we expect lower == upper, so only compare lower */
|
|
result = DirectFunctionCall2Coll(bttextcmp,
|
|
ssup->ssup_collation,
|
|
PointerGetDatum(arg1.lower),
|
|
PointerGetDatum(arg2.lower));
|
|
|
|
GBT_FREE_IF_COPY(key1, x);
|
|
GBT_FREE_IF_COPY(key2, y);
|
|
|
|
return DatumGetInt32(result);
|
|
}
|
|
|
|
Datum
|
|
gbt_text_sortsupport(PG_FUNCTION_ARGS)
|
|
{
|
|
SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
|
|
|
|
ssup->comparator = gbt_text_ssup_cmp;
|
|
ssup->ssup_extra = NULL;
|
|
|
|
PG_RETURN_VOID();
|
|
}
|
|
|
|
static int
|
|
gbt_bpchar_ssup_cmp(Datum x, Datum y, SortSupport ssup)
|
|
{
|
|
GBT_VARKEY *key1 = PG_DETOAST_DATUM(x);
|
|
GBT_VARKEY *key2 = PG_DETOAST_DATUM(y);
|
|
|
|
GBT_VARKEY_R arg1 = gbt_var_key_readable(key1);
|
|
GBT_VARKEY_R arg2 = gbt_var_key_readable(key2);
|
|
Datum result;
|
|
|
|
/* for leaf items we expect lower == upper, so only compare lower */
|
|
result = DirectFunctionCall2Coll(bpcharcmp,
|
|
ssup->ssup_collation,
|
|
PointerGetDatum(arg1.lower),
|
|
PointerGetDatum(arg2.lower));
|
|
|
|
GBT_FREE_IF_COPY(key1, x);
|
|
GBT_FREE_IF_COPY(key2, y);
|
|
|
|
return DatumGetInt32(result);
|
|
}
|
|
|
|
Datum
|
|
gbt_bpchar_sortsupport(PG_FUNCTION_ARGS)
|
|
{
|
|
SortSupport ssup = (SortSupport) PG_GETARG_POINTER(0);
|
|
|
|
ssup->comparator = gbt_bpchar_ssup_cmp;
|
|
ssup->ssup_extra = NULL;
|
|
|
|
PG_RETURN_VOID();
|
|
}
|