database/postgres
database/postgres
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-04-21 12:05:57 +03:00
Code

Add caching of query to GIN/GiST consistent function.

Browse Source 
Per performance gripe from nomao.com
This commit is contained in:
Teodor Sigaev 2008-07-11 11:56:48 +00:00
parent e3afbb3504
commit 2d6599f471
2 changed files with 48 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
contrib/pg_trgm/trgm_gin.c
View File

@ -1,5 +1,5 @@
/* /*
* $PostgreSQL: pgsql/contrib/pg_trgm/trgm_gin.c,v 1.4 2008/05/17 01:28:21 adunstan Exp $ * $PostgreSQL: pgsql/contrib/pg_trgm/trgm_gin.c,v 1.5 2008/07/11 11:56:48 teodor Exp $
*/ */
#include "trgm.h" #include "trgm.h"
@ -52,6 +52,16 @@ gin_extract_trgm(PG_FUNCTION_ARGS)
PG_RETURN_POINTER(entries); PG_RETURN_POINTER(entries);
} }
/*
* Per call strage for consistent functions to
* cache computed value from query
*/
typedef struct PerCallConsistentStorage {
int trglen;
text data[1]; /* query */
} PerCallConsistentStorage;
#define PCCSHDR_SZ offsetof(PerCallConsistentStorage, data)
Datum Datum
gin_trgm_consistent(PG_FUNCTION_ARGS) gin_trgm_consistent(PG_FUNCTION_ARGS)
{ {
@ -60,16 +70,30 @@ gin_trgm_consistent(PG_FUNCTION_ARGS)
text *query = PG_GETARG_TEXT_P(2); text *query = PG_GETARG_TEXT_P(2);
bool *recheck = (bool *) PG_GETARG_POINTER(3); bool *recheck = (bool *) PG_GETARG_POINTER(3);
bool res = FALSE; bool res = FALSE;
TRGM *trg;
int4 i, int4 i,
trglen, trglen,
ntrue = 0; ntrue = 0;
PerCallConsistentStorage *pccs = (PerCallConsistentStorage*) fcinfo->flinfo->fn_extra;
/* All cases served by this function are inexact */ /* All cases served by this function are inexact */
*recheck = true; *recheck = true;
trg = generate_trgm(VARDATA(query), VARSIZE(query) - VARHDRSZ); if ( pccs == NULL || VARSIZE(pccs->data) != VARSIZE(query) || memcmp( pccs->data, query, VARSIZE(query) ) !=0 )
trglen = ARRNELEM(trg); {
TRGM *trg = generate_trgm(VARDATA(query), VARSIZE(query) - VARHDRSZ);
if ( pccs )
pfree(pccs);
fcinfo->flinfo->fn_extra = MemoryContextAlloc(fcinfo->flinfo->fn_mcxt,
VARSIZE(query) + PCCSHDR_SZ);
pccs = (PerCallConsistentStorage*) fcinfo->flinfo->fn_extra;
pccs->trglen = ARRNELEM(trg);
memcpy( pccs->data, query, VARSIZE(query) );
}
trglen = pccs->trglen;
for (i = 0; i < trglen; i++) for (i = 0; i < trglen; i++)
if (check[i]) if (check[i])

22
contrib/pg_trgm/trgm_gist.c
View File

@ -1,5 +1,5 @@
/* /*
* $PostgreSQL: pgsql/contrib/pg_trgm/trgm_gist.c,v 1.14 2008/05/17 01:28:21 adunstan Exp $ * $PostgreSQL: pgsql/contrib/pg_trgm/trgm_gist.c,v 1.15 2008/07/11 11:56:48 teodor Exp $
*/ */
#include "trgm.h" #include "trgm.h"
@ -168,12 +168,30 @@ gtrgm_consistent(PG_FUNCTION_ARGS)
/* Oid subtype = PG_GETARG_OID(3); */ /* Oid subtype = PG_GETARG_OID(3); */
bool *recheck = (bool *) PG_GETARG_POINTER(4); bool *recheck = (bool *) PG_GETARG_POINTER(4);
TRGM *key = (TRGM *) DatumGetPointer(entry->key); TRGM *key = (TRGM *) DatumGetPointer(entry->key);
TRGM *qtrg = generate_trgm(VARDATA(query), VARSIZE(query) - VARHDRSZ); TRGM *qtrg;
bool res = false; bool res = false;
char *cache = (char*) fcinfo->flinfo->fn_extra;
/* All cases served by this function are exact */ /* All cases served by this function are exact */
*recheck = false; *recheck = false;
if ( cache == NULL || VARSIZE(cache) != VARSIZE(query) || memcmp( cache, query, VARSIZE(query) ) !=0 )
{
qtrg = generate_trgm(VARDATA(query), VARSIZE(query) - VARHDRSZ);
if (cache)
pfree(cache);
fcinfo->flinfo->fn_extra = MemoryContextAlloc(fcinfo->flinfo->fn_mcxt,
MAXALIGN(VARSIZE(query)) + VARSIZE(qtrg) );
cache = (char*) fcinfo->flinfo->fn_extra;
memcpy( cache, query, VARSIZE(query) );
memcpy( cache + MAXALIGN(VARSIZE(query)), qtrg, VARSIZE(qtrg) );
}
qtrg = (TRGM*)( cache + MAXALIGN(VARSIZE(query)) );
if (GIST_LEAF(entry)) if (GIST_LEAF(entry))
{ /* all leafs contains orig trgm */ { /* all leafs contains orig trgm */
float4 tmpsml = cnt_sml(key, qtrg); float4 tmpsml = cnt_sml(key, qtrg);