mirror of
https://git.postgresql.org/git/postgresql.git
synced 2024-10-04 10:56:49 +02:00
220db7ccd8
strings. This patch introduces four support functions cstring_to_text, cstring_to_text_with_len, text_to_cstring, and text_to_cstring_buffer, and two macros CStringGetTextDatum and TextDatumGetCString. A number of existing macros that provided variants on these themes were removed. Most of the places that need to make such conversions now require just one function or macro call, in place of the multiple notational layers that used to be needed. There are no longer any direct calls of textout or textin, and we got most of the places that were using handmade conversions via memcpy (there may be a few still lurking, though). This commit doesn't make any serious effort to eliminate transient memory leaks caused by detoasting toasted text objects before they reach text_to_cstring. We changed PG_GETARG_TEXT_P to PG_GETARG_TEXT_PP in a few places where it was easy, but much more could be done. Brendan Jurd and Tom Lane
162 lines
3.7 KiB
C
162 lines
3.7 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* tsginidx.c
|
|
* GIN support functions for tsvector_ops
|
|
*
|
|
* Portions Copyright (c) 1996-2008, PostgreSQL Global Development Group
|
|
*
|
|
*
|
|
* IDENTIFICATION
|
|
* $PostgreSQL: pgsql/src/backend/utils/adt/tsginidx.c,v 1.10 2008/03/25 22:42:44 tgl Exp $
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include "postgres.h"
|
|
|
|
#include "access/skey.h"
|
|
#include "tsearch/ts_type.h"
|
|
#include "tsearch/ts_utils.h"
|
|
#include "utils/builtins.h"
|
|
|
|
|
|
Datum
|
|
gin_extract_tsvector(PG_FUNCTION_ARGS)
|
|
{
|
|
TSVector vector = PG_GETARG_TSVECTOR(0);
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
Datum *entries = NULL;
|
|
|
|
*nentries = vector->size;
|
|
if (vector->size > 0)
|
|
{
|
|
int i;
|
|
WordEntry *we = ARRPTR(vector);
|
|
|
|
entries = (Datum *) palloc(sizeof(Datum) * vector->size);
|
|
|
|
for (i = 0; i < vector->size; i++)
|
|
{
|
|
text *txt;
|
|
|
|
txt = cstring_to_text_with_len(STRPTR(vector) + we->pos, we->len);
|
|
entries[i] = PointerGetDatum(txt);
|
|
|
|
we++;
|
|
}
|
|
}
|
|
|
|
PG_FREE_IF_COPY(vector, 0);
|
|
PG_RETURN_POINTER(entries);
|
|
}
|
|
|
|
Datum
|
|
gin_extract_tsquery(PG_FUNCTION_ARGS)
|
|
{
|
|
TSQuery query = PG_GETARG_TSQUERY(0);
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
StrategyNumber strategy = PG_GETARG_UINT16(2);
|
|
Datum *entries = NULL;
|
|
|
|
*nentries = 0;
|
|
|
|
if (query->size > 0)
|
|
{
|
|
int4 i,
|
|
j = 0,
|
|
len;
|
|
QueryItem *item;
|
|
|
|
item = clean_NOT(GETQUERY(query), &len);
|
|
if (!item)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
errmsg("query requires full scan, which is not supported by GIN indexes")));
|
|
|
|
item = GETQUERY(query);
|
|
|
|
for (i = 0; i < query->size; i++)
|
|
if (item[i].type == QI_VAL)
|
|
(*nentries)++;
|
|
|
|
entries = (Datum *) palloc(sizeof(Datum) * (*nentries));
|
|
|
|
for (i = 0; i < query->size; i++)
|
|
if (item[i].type == QI_VAL)
|
|
{
|
|
text *txt;
|
|
QueryOperand *val = &item[i].operand;
|
|
|
|
txt = cstring_to_text_with_len(GETOPERAND(query) + val->distance,
|
|
val->length);
|
|
entries[j++] = PointerGetDatum(txt);
|
|
|
|
if (strategy != TSearchWithClassStrategyNumber && val->weight != 0)
|
|
ereport(ERROR,
|
|
(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
|
|
errmsg("@@ operator does not support lexeme weight restrictions in GIN index searches"),
|
|
errhint("Use the @@@ operator instead.")));
|
|
}
|
|
}
|
|
else
|
|
*nentries = -1; /* nothing can be found */
|
|
|
|
PG_FREE_IF_COPY(query, 0);
|
|
|
|
PG_RETURN_POINTER(entries);
|
|
}
|
|
|
|
typedef struct
|
|
{
|
|
QueryItem *frst;
|
|
bool *mapped_check;
|
|
} GinChkVal;
|
|
|
|
static bool
|
|
checkcondition_gin(void *checkval, QueryOperand *val)
|
|
{
|
|
GinChkVal *gcv = (GinChkVal *) checkval;
|
|
|
|
return gcv->mapped_check[((QueryItem *) val) - gcv->frst];
|
|
}
|
|
|
|
Datum
|
|
gin_tsquery_consistent(PG_FUNCTION_ARGS)
|
|
{
|
|
bool *check = (bool *) PG_GETARG_POINTER(0);
|
|
/* StrategyNumber strategy = PG_GETARG_UINT16(1); */
|
|
TSQuery query = PG_GETARG_TSQUERY(2);
|
|
bool res = FALSE;
|
|
|
|
if (query->size > 0)
|
|
{
|
|
int i,
|
|
j = 0;
|
|
QueryItem *item;
|
|
GinChkVal gcv;
|
|
|
|
/*
|
|
* check-parameter array has one entry for each value (operand) in the
|
|
* query. We expand that array into mapped_check, so that there's one
|
|
* entry in mapped_check for every node in the query, including
|
|
* operators, to allow quick lookups in checkcondition_gin. Only the
|
|
* entries corresponding operands are actually used.
|
|
*/
|
|
|
|
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 == QI_VAL)
|
|
gcv.mapped_check[i] = check[j++];
|
|
|
|
res = TS_execute(
|
|
GETQUERY(query),
|
|
&gcv,
|
|
true,
|
|
checkcondition_gin
|
|
);
|
|
}
|
|
|
|
PG_RETURN_BOOL(res);
|
|
}
|