2008-05-12 02:00:54 +02:00
|
|
|
/*
|
2010-09-20 22:08:53 +02:00
|
|
|
* contrib/hstore/hstore_gin.c
|
2008-05-12 02:00:54 +02:00
|
|
|
*/
|
|
|
|
#include "postgres.h"
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
#include "access/gin.h"
|
Fix GIN to support null keys, empty and null items, and full index scans.
Per my recent proposal(s). Null key datums can now be returned by
extractValue and extractQuery functions, and will be stored in the index.
Also, placeholder entries are made for indexable items that are NULL or
contain no keys according to extractValue. This means that the index is
now always complete, having at least one entry for every indexed heap TID,
and so we can get rid of the prohibition on full-index scans. A full-index
scan is implemented much the same way as partial-match scans were already:
we build a bitmap representing all the TIDs found in the index, and then
drive the results off that.
Also, introduce a concept of a "search mode" that can be requested by
extractQuery when the operator requires matching to empty items (this is
just as cheap as matching to a single key) or requires a full index scan
(which is not so cheap, but it sure beats failing or giving wrong answers).
The behavior remains backward compatible for opclasses that don't return
any null keys or request a non-default search mode.
Using these features, we can now make the GIN index opclass for anyarray
behave in a way that matches the actual anyarray operators for &&, <@, @>,
and = ... which it failed to do before in assorted corner cases.
This commit fixes the core GIN code and ginarrayprocs.c, updates the
documentation, and adds some simple regression test cases for the new
behaviors using the array operators. The tsearch and contrib GIN opclass
support functions still need to be looked over and probably fixed.
Another thing I intend to fix separately is that this is pretty inefficient
for cases where more than one scan condition needs a full-index search:
we'll run duplicate GinScanEntrys, each one of which builds a large bitmap.
There is some existing logic to merge duplicate GinScanEntrys but it needs
refactoring to make it work for entries belonging to different scan keys.
Note that most of gin.h has been split out into a new file gin_private.h,
so that gin.h doesn't export anything that's not supposed to be used by GIN
opclasses or the rest of the backend. I did quite a bit of other code
beautification work as well, mostly fixing comments and choosing more
appropriate names for things.
2011-01-08 01:16:24 +01:00
|
|
|
#include "access/skey.h"
|
2009-09-30 21:50:22 +02:00
|
|
|
#include "catalog/pg_type.h"
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2008-05-12 02:00:54 +02:00
|
|
|
#include "hstore.h"
|
|
|
|
|
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
#define KEYFLAG 'K'
|
|
|
|
#define VALFLAG 'V'
|
|
|
|
#define NULLFLAG 'N'
|
2007-03-14 15:21:53 +01:00
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(gin_extract_hstore);
|
2007-11-15 22:14:46 +01:00
|
|
|
Datum gin_extract_hstore(PG_FUNCTION_ARGS);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
static text *
|
|
|
|
makeitem(char *str, int len)
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2007-11-15 22:14:46 +01:00
|
|
|
text *item;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
item = (text *) palloc(VARHDRSZ + len + 1);
|
2007-03-14 15:21:53 +01:00
|
|
|
SET_VARSIZE(item, VARHDRSZ + len + 1);
|
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
if (str && len > 0)
|
|
|
|
memcpy(VARDATA(item) + 1, str, len);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
|
|
|
return item;
|
|
|
|
}
|
|
|
|
|
|
|
|
Datum
|
|
|
|
gin_extract_hstore(PG_FUNCTION_ARGS)
|
|
|
|
{
|
2007-11-15 22:14:46 +01:00
|
|
|
HStore *hs = PG_GETARG_HS(0);
|
|
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
|
|
Datum *entries = NULL;
|
2010-02-26 03:01:40 +01:00
|
|
|
HEntry *hsent = ARRPTR(hs);
|
|
|
|
char *ptr = STRPTR(hs);
|
|
|
|
int count = HS_COUNT(hs);
|
|
|
|
int i;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2009-09-30 21:50:22 +02:00
|
|
|
*nentries = 2 * count;
|
|
|
|
if (count)
|
|
|
|
entries = (Datum *) palloc(sizeof(Datum) * 2 * count);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2009-09-30 21:50:22 +02:00
|
|
|
for (i = 0; i < count; ++i)
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2009-09-30 21:50:22 +02:00
|
|
|
text *item;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2010-02-26 03:01:40 +01:00
|
|
|
item = makeitem(HS_KEY(hsent, ptr, i), HS_KEYLEN(hsent, i));
|
2009-09-30 21:50:22 +02:00
|
|
|
*VARDATA(item) = KEYFLAG;
|
2010-02-26 03:01:40 +01:00
|
|
|
entries[2 * i] = PointerGetDatum(item);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2010-02-26 03:01:40 +01:00
|
|
|
if (HS_VALISNULL(hsent, i))
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2009-09-30 21:50:22 +02:00
|
|
|
item = makeitem(NULL, 0);
|
|
|
|
*VARDATA(item) = NULLFLAG;
|
2007-03-14 15:21:53 +01:00
|
|
|
}
|
2009-09-30 21:50:22 +02:00
|
|
|
else
|
|
|
|
{
|
2010-02-26 03:01:40 +01:00
|
|
|
item = makeitem(HS_VAL(hsent, ptr, i), HS_VALLEN(hsent, i));
|
2009-09-30 21:50:22 +02:00
|
|
|
*VARDATA(item) = VALFLAG;
|
|
|
|
}
|
2010-02-26 03:01:40 +01:00
|
|
|
entries[2 * i + 1] = PointerGetDatum(item);
|
2007-03-14 15:21:53 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(entries);
|
|
|
|
}
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(gin_extract_hstore_query);
|
2007-11-15 22:14:46 +01:00
|
|
|
Datum gin_extract_hstore_query(PG_FUNCTION_ARGS);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
|
|
|
Datum
|
|
|
|
gin_extract_hstore_query(PG_FUNCTION_ARGS)
|
|
|
|
{
|
|
|
|
StrategyNumber strategy = PG_GETARG_UINT16(2);
|
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
if (strategy == HStoreContainsStrategyNumber)
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2009-09-30 21:50:22 +02:00
|
|
|
PG_RETURN_DATUM(DirectFunctionCall2(gin_extract_hstore,
|
2007-11-15 22:14:46 +01:00
|
|
|
PG_GETARG_DATUM(0),
|
|
|
|
PG_GETARG_DATUM(1)
|
|
|
|
));
|
2007-03-14 15:21:53 +01:00
|
|
|
}
|
2007-11-15 22:14:46 +01:00
|
|
|
else if (strategy == HStoreExistsStrategyNumber)
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2007-11-15 22:14:46 +01:00
|
|
|
text *item,
|
2009-09-30 21:50:22 +02:00
|
|
|
*query = PG_GETARG_TEXT_PP(0);
|
2007-11-15 22:14:46 +01:00
|
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
|
|
Datum *entries = NULL;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
|
|
|
*nentries = 1;
|
2007-11-15 22:14:46 +01:00
|
|
|
entries = (Datum *) palloc(sizeof(Datum));
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2009-09-30 21:50:22 +02:00
|
|
|
item = makeitem(VARDATA_ANY(query), VARSIZE_ANY_EXHDR(query));
|
2007-03-14 15:21:53 +01:00
|
|
|
*VARDATA(item) = KEYFLAG;
|
|
|
|
entries[0] = PointerGetDatum(item);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(entries);
|
|
|
|
}
|
2009-09-30 21:50:22 +02:00
|
|
|
else if (strategy == HStoreExistsAnyStrategyNumber ||
|
|
|
|
strategy == HStoreExistsAllStrategyNumber)
|
|
|
|
{
|
2010-02-26 03:01:40 +01:00
|
|
|
ArrayType *query = PG_GETARG_ARRAYTYPE_P(0);
|
|
|
|
Datum *key_datums;
|
|
|
|
bool *key_nulls;
|
|
|
|
int key_count;
|
|
|
|
int i,
|
|
|
|
j;
|
2009-09-30 21:50:22 +02:00
|
|
|
int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
|
|
|
|
Datum *entries = NULL;
|
2010-02-26 03:01:40 +01:00
|
|
|
text *item;
|
2009-09-30 21:50:22 +02:00
|
|
|
|
|
|
|
deconstruct_array(query,
|
|
|
|
TEXTOID, -1, false, 'i',
|
|
|
|
&key_datums, &key_nulls, &key_count);
|
|
|
|
|
|
|
|
entries = (Datum *) palloc(sizeof(Datum) * key_count);
|
|
|
|
|
|
|
|
for (i = 0, j = 0; i < key_count; ++i)
|
|
|
|
{
|
|
|
|
if (key_nulls[i])
|
|
|
|
continue;
|
|
|
|
item = makeitem(VARDATA(key_datums[i]), VARSIZE(key_datums[i]) - VARHDRSZ);
|
|
|
|
*VARDATA(item) = KEYFLAG;
|
|
|
|
entries[j++] = PointerGetDatum(item);
|
|
|
|
}
|
|
|
|
|
|
|
|
*nentries = j ? j : -1;
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(entries);
|
|
|
|
}
|
2007-03-14 15:21:53 +01:00
|
|
|
else
|
|
|
|
elog(ERROR, "Unsupported strategy number: %d", strategy);
|
|
|
|
|
|
|
|
PG_RETURN_POINTER(NULL);
|
|
|
|
}
|
|
|
|
|
|
|
|
PG_FUNCTION_INFO_V1(gin_consistent_hstore);
|
2007-11-15 22:14:46 +01:00
|
|
|
Datum gin_consistent_hstore(PG_FUNCTION_ARGS);
|
2007-03-14 15:21:53 +01:00
|
|
|
|
|
|
|
Datum
|
|
|
|
gin_consistent_hstore(PG_FUNCTION_ARGS)
|
|
|
|
{
|
2008-04-14 19:05:34 +02:00
|
|
|
bool *check = (bool *) PG_GETARG_POINTER(0);
|
2007-03-14 15:21:53 +01:00
|
|
|
StrategyNumber strategy = PG_GETARG_UINT16(1);
|
2010-02-26 03:01:40 +01:00
|
|
|
|
2009-09-30 21:50:22 +02:00
|
|
|
/* HStore *query = PG_GETARG_HS(2); */
|
|
|
|
int32 nkeys = PG_GETARG_INT32(3);
|
2010-02-26 03:01:40 +01:00
|
|
|
|
2009-03-25 23:19:02 +01:00
|
|
|
/* Pointer *extra_data = (Pointer *) PG_GETARG_POINTER(4); */
|
|
|
|
bool *recheck = (bool *) PG_GETARG_POINTER(5);
|
2007-11-15 22:14:46 +01:00
|
|
|
bool res = true;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2009-09-30 21:50:22 +02:00
|
|
|
*recheck = false;
|
|
|
|
|
2007-11-15 22:14:46 +01:00
|
|
|
if (strategy == HStoreContainsStrategyNumber)
|
2007-03-14 15:21:53 +01:00
|
|
|
{
|
2007-11-15 22:14:46 +01:00
|
|
|
int i;
|
2007-03-14 15:21:53 +01:00
|
|
|
|
2008-04-14 19:05:34 +02:00
|
|
|
/*
|
2009-06-11 16:49:15 +02:00
|
|
|
* Index lost information about correspondence of keys and values, so
|
2009-09-30 21:50:22 +02:00
|
|
|
* we need recheck (pre-8.4 this is handled at SQL level)
|
2008-04-14 19:05:34 +02:00
|
|
|
*/
|
|
|
|
*recheck = true;
|
2009-09-30 21:50:22 +02:00
|
|
|
for (i = 0; res && i < nkeys; i++)
|
2007-11-15 22:14:46 +01:00
|
|
|
if (check[i] == false)
|
2007-03-14 15:21:53 +01:00
|
|
|
res = false;
|
|
|
|
}
|
2007-11-15 22:14:46 +01:00
|
|
|
else if (strategy == HStoreExistsStrategyNumber)
|
2008-04-14 19:05:34 +02:00
|
|
|
{
|
|
|
|
/* Existence of key is guaranteed */
|
2007-03-14 15:21:53 +01:00
|
|
|
res = true;
|
2008-04-14 19:05:34 +02:00
|
|
|
}
|
2009-09-30 21:50:22 +02:00
|
|
|
else if (strategy == HStoreExistsAnyStrategyNumber)
|
|
|
|
{
|
|
|
|
/* Existence of key is guaranteed */
|
|
|
|
res = true;
|
|
|
|
}
|
|
|
|
else if (strategy == HStoreExistsAllStrategyNumber)
|
|
|
|
{
|
2010-02-26 03:01:40 +01:00
|
|
|
int i;
|
2009-09-30 21:50:22 +02:00
|
|
|
|
|
|
|
for (i = 0; res && i < nkeys; ++i)
|
|
|
|
if (!check[i])
|
|
|
|
res = false;
|
|
|
|
}
|
2007-03-14 15:21:53 +01:00
|
|
|
else
|
|
|
|
elog(ERROR, "Unsupported strategy number: %d", strategy);
|
|
|
|
|
|
|
|
PG_RETURN_BOOL(res);
|
|
|
|
}
|