mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-12-27 08:39:28 +08:00
234a02b2a8
Get rid of VARATT_SIZE and VARATT_DATA, which were simply redundant with VARSIZE and VARDATA, and as a consequence almost no code was using the longer names. Rename the length fields of struct varlena and various derived structures to catch anyplace that was accessing them directly; and clean up various places so caught. In itself this patch doesn't change any behavior at all, but it is necessary infrastructure if we hope to play any games with the representation of varlena headers. Greg Stark and Tom Lane
160 lines
3.3 KiB
C
160 lines
3.3 KiB
C
#include "postgres.h"
|
|
|
|
#include <float.h>
|
|
|
|
#include "access/gist.h"
|
|
#include "access/itup.h"
|
|
#include "access/skey.h"
|
|
#include "access/tuptoaster.h"
|
|
#include "storage/bufpage.h"
|
|
#include "utils/array.h"
|
|
#include "utils/builtins.h"
|
|
|
|
#include "tsvector.h"
|
|
#include "query.h"
|
|
#include "query_cleanup.h"
|
|
|
|
PG_FUNCTION_INFO_V1(gin_extract_tsvector);
|
|
Datum gin_extract_tsvector(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
gin_extract_tsvector(PG_FUNCTION_ARGS)
|
|
{
|
|
tsvector *vector = (tsvector *) PG_DETOAST_DATUM(PG_GETARG_DATUM(0));
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
Datum *entries = NULL;
|
|
|
|
*nentries = 0;
|
|
if (vector->size > 0)
|
|
{
|
|
int i;
|
|
WordEntry *we = ARRPTR(vector);
|
|
|
|
*nentries = (int32) vector->size;
|
|
entries = (Datum *) palloc(sizeof(Datum) * vector->size);
|
|
|
|
for (i = 0; i < vector->size; i++)
|
|
{
|
|
text *txt = (text *) palloc(VARHDRSZ + we->len);
|
|
|
|
SET_VARSIZE(txt, VARHDRSZ + we->len);
|
|
memcpy(VARDATA(txt), STRPTR(vector) + we->pos, we->len);
|
|
|
|
entries[i] = PointerGetDatum(txt);
|
|
|
|
we++;
|
|
}
|
|
}
|
|
|
|
PG_FREE_IF_COPY(vector, 0);
|
|
PG_RETURN_POINTER(entries);
|
|
}
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(gin_extract_tsquery);
|
|
Datum gin_extract_tsquery(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
gin_extract_tsquery(PG_FUNCTION_ARGS)
|
|
{
|
|
QUERYTYPE *query = (QUERYTYPE *) PG_DETOAST_DATUM(PG_GETARG_DATUM(0));
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
StrategyNumber strategy = DatumGetUInt16(PG_GETARG_DATUM(2));
|
|
Datum *entries = NULL;
|
|
|
|
*nentries = 0;
|
|
if (query->size > 0)
|
|
{
|
|
int4 i,
|
|
j = 0,
|
|
len;
|
|
ITEM *item;
|
|
|
|
item = clean_NOT_v2(GETQUERY(query), &len);
|
|
if (!item)
|
|
elog(ERROR, "Query requires full scan, GIN doesn't support it");
|
|
|
|
item = GETQUERY(query);
|
|
|
|
for (i = 0; i < query->size; i++)
|
|
if (item[i].type == VAL)
|
|
(*nentries)++;
|
|
|
|
entries = (Datum *) palloc(sizeof(Datum) * (*nentries));
|
|
|
|
for (i = 0; i < query->size; i++)
|
|
if (item[i].type == VAL)
|
|
{
|
|
text *txt;
|
|
|
|
txt = (text *) palloc(VARHDRSZ + item[i].length);
|
|
|
|
SET_VARSIZE(txt, VARHDRSZ + item[i].length);
|
|
memcpy(VARDATA(txt), GETOPERAND(query) + item[i].distance, item[i].length);
|
|
|
|
entries[j++] = PointerGetDatum(txt);
|
|
|
|
if (strategy == 1 && item[i].weight != 0)
|
|
elog(ERROR, "With class of lexeme restrictions use @@@ operation");
|
|
}
|
|
|
|
}
|
|
else
|
|
*nentries = -1; /* nothing can be found */
|
|
|
|
PG_FREE_IF_COPY(query, 0);
|
|
PG_RETURN_POINTER(entries);
|
|
}
|
|
|
|
typedef struct
|
|
{
|
|
ITEM *frst;
|
|
bool *mapped_check;
|
|
} GinChkVal;
|
|
|
|
static bool
|
|
checkcondition_gin(void *checkval, ITEM * val)
|
|
{
|
|
GinChkVal *gcv = (GinChkVal *) checkval;
|
|
|
|
return gcv->mapped_check[val - gcv->frst];
|
|
}
|
|
|
|
PG_FUNCTION_INFO_V1(gin_ts_consistent);
|
|
Datum gin_ts_consistent(PG_FUNCTION_ARGS);
|
|
|
|
Datum
|
|
gin_ts_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
bool *check = (bool *) PG_GETARG_POINTER(0);
|
|
QUERYTYPE *query = (QUERYTYPE *) PG_DETOAST_DATUM(PG_GETARG_DATUM(2));
|
|
bool res = FALSE;
|
|
|
|
if (query->size > 0)
|
|
{
|
|
int4 i,
|
|
j = 0;
|
|
ITEM *item;
|
|
GinChkVal gcv;
|
|
|
|
gcv.frst = item = GETQUERY(query);
|
|
gcv.mapped_check = (bool *) palloc(sizeof(bool) * query->size);
|
|
|
|
for (i = 0; i < query->size; i++)
|
|
if (item[i].type == VAL)
|
|
gcv.mapped_check[i] = check[j++];
|
|
|
|
|
|
res = TS_execute(
|
|
GETQUERY(query),
|
|
&gcv,
|
|
true,
|
|
checkcondition_gin
|
|
);
|
|
|
|
}
|
|
|
|
PG_FREE_IF_COPY(query, 2);
|
|
PG_RETURN_BOOL(res);
|
|
}
|