#include "postgres.h"
#include "funcapi.h"
#include "catalog/namespace.h"
#include "catalog/pg_type.h"
#include "commands/defrem.h"
#include "tsearch/ts_cache.h"
#include "tsearch/ts_utils.h"
#include "utils/builtins.h"
#include "utils/jsonapi.h"
#include "utils/varlena.h"

Include dependency graph for wparser.c:

Data Structures
struct	TSTokenTypeStorage

struct	HeadlineJsonState

struct	LexemeEntry

struct	PrsStorage

Typedefs
typedef struct HeadlineJsonState	HeadlineJsonState

Functions
static text *	headline_json_value (void _state, char elem_value, int elem_len)

static void	tt_setup_firstcall (FuncCallContext *funcctx, Oid prsid)

static Datum	tt_process_call (FuncCallContext *funcctx)

Datum	ts_token_type_byid (PG_FUNCTION_ARGS)

Datum	ts_token_type_byname (PG_FUNCTION_ARGS)

static void	prs_setup_firstcall (FuncCallContext funcctx, Oid prsid, text txt)

static Datum	prs_process_call (FuncCallContext *funcctx)

Datum	ts_parse_byid (PG_FUNCTION_ARGS)

Datum	ts_parse_byname (PG_FUNCTION_ARGS)

Datum	ts_headline_byid_opt (PG_FUNCTION_ARGS)

Datum	ts_headline_byid (PG_FUNCTION_ARGS)

Datum	ts_headline (PG_FUNCTION_ARGS)

Datum	ts_headline_opt (PG_FUNCTION_ARGS)

Datum	ts_headline_jsonb_byid_opt (PG_FUNCTION_ARGS)

Datum	ts_headline_jsonb (PG_FUNCTION_ARGS)

Datum	ts_headline_jsonb_byid (PG_FUNCTION_ARGS)

Datum	ts_headline_jsonb_opt (PG_FUNCTION_ARGS)

Datum	ts_headline_json_byid_opt (PG_FUNCTION_ARGS)

Datum	ts_headline_json (PG_FUNCTION_ARGS)

Datum	ts_headline_json_byid (PG_FUNCTION_ARGS)

Datum	ts_headline_json_opt (PG_FUNCTION_ARGS)

Typedef Documentation

typedef struct HeadlineJsonState HeadlineJsonState

Function Documentation

static text * headline_json_value	(	void *	_state,
		char *	elem_value,
		int	elem_len
	)

static

Definition at line 539 of file wparser.c.

References HeadlineJsonState::cfg, TSConfigCacheEntry::cfgId, HeadlineParsedText::curwords, FunctionCall3, generateHeadline(), hlparsetext(), PointerGetDatum, HeadlineJsonState::prs, TSParserCacheEntry::prsheadline, HeadlineJsonState::prsobj, HeadlineJsonState::prsoptions, HeadlineJsonState::query, and HeadlineJsonState::transformed.

Referenced by ts_headline_json_byid_opt(), and ts_headline_jsonb_byid_opt().

 {
     HeadlineJsonState *state = (HeadlineJsonState *) _state;
 
     HeadlineParsedText *prs = state->prs;
     TSConfigCacheEntry *cfg = state->cfg;
     TSParserCacheEntry *prsobj = state->prsobj;
     TSQuery     query = state->query;
     List       *prsoptions = state->prsoptions;
 
     prs->curwords = 0;
     hlparsetext(cfg->cfgId, prs, query, elem_value, elem_len);
     FunctionCall3(&(prsobj->prsheadline),
                   PointerGetDatum(prs),
                   PointerGetDatum(prsoptions),
                   PointerGetDatum(query));
 
     state->transformed = true;
     return generateHeadline(prs);
 }

static Datum prs_process_call ( FuncCallContext * funcctx )

static

Definition at line 226 of file wparser.c.

References FuncCallContext::attinmeta, BuildTupleFromCStrings(), PrsStorage::cur, HeapTupleGetDatum, PrsStorage::len, LexemeEntry::lexeme, PrsStorage::list, pfree(), result, LexemeEntry::type, FuncCallContext::user_fctx, and values.

Referenced by ts_parse_byid(), and ts_parse_byname().

 {
     PrsStorage *st;
 
     st = (PrsStorage *) funcctx->user_fctx;
     if (st->cur < st->len)
     {
         Datum       result;
         char       *values[2];
         char        tid[16];
         HeapTuple   tuple;
 
         values[0] = tid;
         sprintf(tid, "%d", st->list[st->cur].type);
         values[1] = st->list[st->cur].lexeme;
         tuple = BuildTupleFromCStrings(funcctx->attinmeta, values);
         result = HeapTupleGetDatum(tuple);
 
         pfree(values[1]);
         st->cur++;
         return result;
     }
     else
     {
         if (st->list)
             pfree(st->list);
         pfree(st);
     }
     return (Datum) 0;
 }

static void prs_setup_firstcall	(	FuncCallContext *	funcctx,
		Oid	prsid,
		text *	txt
	)

static

Definition at line 170 of file wparser.c.

References FuncCallContext::attinmeta, CreateTemplateTupleDesc(), PrsStorage::cur, DatumGetInt32, DatumGetPointer, FunctionCall1, FunctionCall2, FunctionCall3, Int32GetDatum, INT4OID, PrsStorage::len, LexemeEntry::lexeme, PrsStorage::list, lookup_ts_parser_cache(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, NULL, palloc(), PointerGetDatum, TSParserCacheEntry::prsend, TSParserCacheEntry::prsstart, TSParserCacheEntry::prstoken, repalloc(), TEXTOID, TupleDescGetAttInMetadata(), TupleDescInitEntry(), LexemeEntry::type, FuncCallContext::user_fctx, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

Referenced by ts_parse_byid(), and ts_parse_byname().

 {
     TupleDesc   tupdesc;
     MemoryContext oldcontext;
     PrsStorage *st;
     TSParserCacheEntry *prs = lookup_ts_parser_cache(prsid);
     char       *lex = NULL;
     int         llen = 0,
                 type = 0;
     void       *prsdata;
 
     oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
 
     st = (PrsStorage *) palloc(sizeof(PrsStorage));
     st->cur = 0;
     st->len = 16;
     st->list = (LexemeEntry *) palloc(sizeof(LexemeEntry) * st->len);
 
     prsdata = (void *) DatumGetPointer(FunctionCall2(&prs->prsstart,
                                                      PointerGetDatum(VARDATA_ANY(txt)),
                                                      Int32GetDatum(VARSIZE_ANY_EXHDR(txt))));
 
     while ((type = DatumGetInt32(FunctionCall3(&prs->prstoken,
                                                PointerGetDatum(prsdata),
                                                PointerGetDatum(&lex),
                                                PointerGetDatum(&llen)))) != 0)
     {
         if (st->cur >= st->len)
         {
             st->len = 2 * st->len;
             st->list = (LexemeEntry *) repalloc(st->list, sizeof(LexemeEntry) * st->len);
         }
         st->list[st->cur].lexeme = palloc(llen + 1);
         memcpy(st->list[st->cur].lexeme, lex, llen);
         st->list[st->cur].lexeme[llen] = '\0';
         st->list[st->cur].type = type;
         st->cur++;
     }
 
     FunctionCall1(&prs->prsend, PointerGetDatum(prsdata));
 
     st->len = st->cur;
     st->cur = 0;
 
     funcctx->user_fctx = (void *) st;
     tupdesc = CreateTemplateTupleDesc(2, false);
     TupleDescInitEntry(tupdesc, (AttrNumber) 1, "tokid",
                        INT4OID, -1, 0);
     TupleDescInitEntry(tupdesc, (AttrNumber) 2, "token",
                        TEXTOID, -1, 0);
 
     funcctx->attinmeta = TupleDescGetAttInMetadata(tupdesc);
     MemoryContextSwitchTo(oldcontext);
 }

Datum ts_headline ( PG_FUNCTION_ARGS )

Definition at line 364 of file wparser.c.

References DirectFunctionCall3, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1)));
 }

Datum ts_headline_byid ( PG_FUNCTION_ARGS )

Definition at line 355 of file wparser.c.

References DirectFunctionCall3, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_byid_opt,
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_headline_byid_opt ( PG_FUNCTION_ARGS )

Definition at line 304 of file wparser.c.

References TSConfigCacheEntry::cfgId, deserialize_deflist(), ereport, errcode(), errmsg(), ERROR, FunctionCall3, generateHeadline(), TSParserCacheEntry::headlineOid, hlparsetext(), HeadlineParsedText::lenwords, lookup_ts_config_cache(), lookup_ts_parser_cache(), NIL, NULL, OidIsValid, palloc(), pfree(), PG_FREE_IF_COPY, PG_GETARG_OID, PG_GETARG_POINTER, PG_GETARG_TEXT_PP, PG_GETARG_TSQUERY, PG_NARGS, PG_RETURN_POINTER, PointerGetDatum, TSParserCacheEntry::prsheadline, TSConfigCacheEntry::prsId, HeadlineParsedText::startsel, HeadlineParsedText::stopsel, VARDATA_ANY, VARSIZE_ANY_EXHDR, and HeadlineParsedText::words.

Referenced by ts_headline(), ts_headline_byid(), and ts_headline_opt().

 {
     Oid         tsconfig = PG_GETARG_OID(0);
     text       *in = PG_GETARG_TEXT_PP(1);
     TSQuery     query = PG_GETARG_TSQUERY(2);
     text       *opt = (PG_NARGS() > 3 && PG_GETARG_POINTER(3)) ? PG_GETARG_TEXT_PP(3) : NULL;
     HeadlineParsedText prs;
     List       *prsoptions;
     text       *out;
     TSConfigCacheEntry *cfg;
     TSParserCacheEntry *prsobj;
 
     cfg = lookup_ts_config_cache(tsconfig);
     prsobj = lookup_ts_parser_cache(cfg->prsId);
 
     if (!OidIsValid(prsobj->headlineOid))
         ereport(ERROR,
                 (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                  errmsg("text search parser does not support headline creation")));
 
     memset(&prs, 0, sizeof(HeadlineParsedText));
     prs.lenwords = 32;
     prs.words = (HeadlineWordEntry *) palloc(sizeof(HeadlineWordEntry) * prs.lenwords);
 
     hlparsetext(cfg->cfgId, &prs, query,
                 VARDATA_ANY(in), VARSIZE_ANY_EXHDR(in));
 
     if (opt)
         prsoptions = deserialize_deflist(PointerGetDatum(opt));
     else
         prsoptions = NIL;
 
     FunctionCall3(&(prsobj->prsheadline),
                   PointerGetDatum(&prs),
                   PointerGetDatum(prsoptions),
                   PointerGetDatum(query));
 
     out = generateHeadline(&prs);
 
     PG_FREE_IF_COPY(in, 1);
     PG_FREE_IF_COPY(query, 2);
     if (opt)
         PG_FREE_IF_COPY(opt, 3);
     pfree(prs.words);
     pfree(prs.startsel);
     pfree(prs.stopsel);
 
     PG_RETURN_POINTER(out);
 }

Datum ts_headline_json ( PG_FUNCTION_ARGS )

Definition at line 507 of file wparser.c.

References DirectFunctionCall3, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_json_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_json_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1)));
 }

Datum ts_headline_json_byid ( PG_FUNCTION_ARGS )

Definition at line 516 of file wparser.c.

References DirectFunctionCall3, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_json_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_json_byid_opt,
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_headline_json_byid_opt ( PG_FUNCTION_ARGS )

Definition at line 459 of file wparser.c.

References HeadlineJsonState::cfg, deserialize_deflist(), ereport, errcode(), errmsg(), ERROR, headline_json_value(), TSParserCacheEntry::headlineOid, HeadlineParsedText::lenwords, lookup_ts_config_cache(), lookup_ts_parser_cache(), NIL, NULL, OidIsValid, palloc(), palloc0(), pfree(), PG_FREE_IF_COPY, PG_GETARG_OID, PG_GETARG_POINTER, PG_GETARG_TEXT_P, PG_GETARG_TSQUERY, PG_NARGS, PG_RETURN_TEXT_P, PointerGetDatum, HeadlineJsonState::prs, TSConfigCacheEntry::prsId, HeadlineJsonState::prsobj, HeadlineJsonState::prsoptions, HeadlineJsonState::query, HeadlineParsedText::startsel, HeadlineParsedText::stopsel, transform_json_string_values(), HeadlineJsonState::transformed, and HeadlineParsedText::words.

Referenced by ts_headline_json(), ts_headline_json_byid(), and ts_headline_json_opt().

 {
     Oid         tsconfig = PG_GETARG_OID(0);
     text       *json = PG_GETARG_TEXT_P(1);
     TSQuery     query = PG_GETARG_TSQUERY(2);
     text       *opt = (PG_NARGS() > 3 && PG_GETARG_POINTER(3)) ? PG_GETARG_TEXT_P(3) : NULL;
     text       *out;
     JsonTransformStringValuesAction action = (JsonTransformStringValuesAction) headline_json_value;
 
     HeadlineParsedText prs;
     HeadlineJsonState *state = palloc0(sizeof(HeadlineJsonState));
 
     memset(&prs, 0, sizeof(HeadlineParsedText));
     prs.lenwords = 32;
     prs.words = (HeadlineWordEntry *) palloc(sizeof(HeadlineWordEntry) * prs.lenwords);
 
     state->prs = &prs;
     state->cfg = lookup_ts_config_cache(tsconfig);
     state->prsobj = lookup_ts_parser_cache(state->cfg->prsId);
     state->query = query;
     if (opt)
         state->prsoptions = deserialize_deflist(PointerGetDatum(opt));
     else
         state->prsoptions = NIL;
 
     if (!OidIsValid(state->prsobj->headlineOid))
         ereport(ERROR,
                 (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                  errmsg("text search parser does not support headline creation")));
 
     out = transform_json_string_values(json, state, action);
 
     PG_FREE_IF_COPY(json, 1);
     PG_FREE_IF_COPY(query, 2);
     if (opt)
         PG_FREE_IF_COPY(opt, 3);
     pfree(prs.words);
 
     if (state->transformed)
     {
         pfree(prs.startsel);
         pfree(prs.stopsel);
     }
 
     PG_RETURN_TEXT_P(out);
 }

Datum ts_headline_json_opt ( PG_FUNCTION_ARGS )

Definition at line 525 of file wparser.c.

References DirectFunctionCall4, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_json_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall4(ts_headline_json_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_headline_jsonb ( PG_FUNCTION_ARGS )

Definition at line 431 of file wparser.c.

References DirectFunctionCall3, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_jsonb_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_jsonb_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1)));
 }

Datum ts_headline_jsonb_byid ( PG_FUNCTION_ARGS )

Definition at line 440 of file wparser.c.

References DirectFunctionCall3, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_jsonb_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall3(ts_headline_jsonb_byid_opt,
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_headline_jsonb_byid_opt ( PG_FUNCTION_ARGS )

Definition at line 383 of file wparser.c.

References HeadlineJsonState::cfg, deserialize_deflist(), ereport, errcode(), errmsg(), ERROR, headline_json_value(), TSParserCacheEntry::headlineOid, HeadlineParsedText::lenwords, lookup_ts_config_cache(), lookup_ts_parser_cache(), NIL, NULL, OidIsValid, palloc(), palloc0(), pfree(), PG_FREE_IF_COPY, PG_GETARG_JSONB, PG_GETARG_OID, PG_GETARG_POINTER, PG_GETARG_TEXT_P, PG_GETARG_TSQUERY, PG_NARGS, PG_RETURN_JSONB, PointerGetDatum, HeadlineJsonState::prs, TSConfigCacheEntry::prsId, HeadlineJsonState::prsobj, HeadlineJsonState::prsoptions, HeadlineJsonState::query, HeadlineParsedText::startsel, HeadlineParsedText::stopsel, transform_jsonb_string_values(), HeadlineJsonState::transformed, and HeadlineParsedText::words.

Referenced by ts_headline_jsonb(), ts_headline_jsonb_byid(), and ts_headline_jsonb_opt().

 {
     Oid         tsconfig = PG_GETARG_OID(0);
     Jsonb      *jb = PG_GETARG_JSONB(1);
     TSQuery     query = PG_GETARG_TSQUERY(2);
     text       *opt = (PG_NARGS() > 3 && PG_GETARG_POINTER(3)) ? PG_GETARG_TEXT_P(3) : NULL;
     Jsonb      *out;
     JsonTransformStringValuesAction action = (JsonTransformStringValuesAction) headline_json_value;
     HeadlineParsedText prs;
     HeadlineJsonState *state = palloc0(sizeof(HeadlineJsonState));
 
     memset(&prs, 0, sizeof(HeadlineParsedText));
     prs.lenwords = 32;
     prs.words = (HeadlineWordEntry *) palloc(sizeof(HeadlineWordEntry) * prs.lenwords);
 
     state->prs = &prs;
     state->cfg = lookup_ts_config_cache(tsconfig);
     state->prsobj = lookup_ts_parser_cache(state->cfg->prsId);
     state->query = query;
     if (opt)
         state->prsoptions = deserialize_deflist(PointerGetDatum(opt));
     else
         state->prsoptions = NIL;
 
     if (!OidIsValid(state->prsobj->headlineOid))
         ereport(ERROR,
                 (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                  errmsg("text search parser does not support headline creation")));
 
     out = transform_jsonb_string_values(jb, state, action);
 
     PG_FREE_IF_COPY(jb, 1);
     PG_FREE_IF_COPY(query, 2);
     if (opt)
         PG_FREE_IF_COPY(opt, 3);
 
     pfree(prs.words);
 
     if (state->transformed)
     {
         pfree(prs.startsel);
         pfree(prs.stopsel);
     }
 
     PG_RETURN_JSONB(out);
 }

Datum ts_headline_jsonb_opt ( PG_FUNCTION_ARGS )

Definition at line 449 of file wparser.c.

References DirectFunctionCall4, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_jsonb_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall4(ts_headline_jsonb_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_headline_opt ( PG_FUNCTION_ARGS )

Definition at line 373 of file wparser.c.

References DirectFunctionCall4, getTSCurrentConfig(), ObjectIdGetDatum, PG_GETARG_DATUM, PG_RETURN_DATUM, and ts_headline_byid_opt().

 {
     PG_RETURN_DATUM(DirectFunctionCall4(ts_headline_byid_opt,
                                         ObjectIdGetDatum(getTSCurrentConfig(true)),
                                         PG_GETARG_DATUM(0),
                                         PG_GETARG_DATUM(1),
                                         PG_GETARG_DATUM(2)));
 }

Datum ts_parse_byid ( PG_FUNCTION_ARGS )

Definition at line 258 of file wparser.c.

References PG_FREE_IF_COPY, PG_GETARG_OID, PG_GETARG_TEXT_PP, prs_process_call(), prs_setup_firstcall(), result, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, and SRF_RETURN_NEXT.

 {
     FuncCallContext *funcctx;
     Datum       result;
 
     if (SRF_IS_FIRSTCALL())
     {
         text       *txt = PG_GETARG_TEXT_PP(1);
 
         funcctx = SRF_FIRSTCALL_INIT();
         prs_setup_firstcall(funcctx, PG_GETARG_OID(0), txt);
         PG_FREE_IF_COPY(txt, 1);
     }
 
     funcctx = SRF_PERCALL_SETUP();
 
     if ((result = prs_process_call(funcctx)) != (Datum) 0)
         SRF_RETURN_NEXT(funcctx, result);
     SRF_RETURN_DONE(funcctx);
 }

Datum ts_parse_byname ( PG_FUNCTION_ARGS )

Definition at line 280 of file wparser.c.

References get_ts_parser_oid(), PG_GETARG_TEXT_PP, prs_process_call(), prs_setup_firstcall(), result, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, SRF_RETURN_NEXT, and textToQualifiedNameList().

 {
     FuncCallContext *funcctx;
     Datum       result;
 
     if (SRF_IS_FIRSTCALL())
     {
         text       *prsname = PG_GETARG_TEXT_PP(0);
         text       *txt = PG_GETARG_TEXT_PP(1);
         Oid         prsId;
 
         funcctx = SRF_FIRSTCALL_INIT();
         prsId = get_ts_parser_oid(textToQualifiedNameList(prsname), false);
         prs_setup_firstcall(funcctx, prsId, txt);
     }
 
     funcctx = SRF_PERCALL_SETUP();
 
     if ((result = prs_process_call(funcctx)) != (Datum) 0)
         SRF_RETURN_NEXT(funcctx, result);
     SRF_RETURN_DONE(funcctx);
 }

Datum ts_token_type_byid ( PG_FUNCTION_ARGS )

Definition at line 114 of file wparser.c.

References PG_GETARG_OID, result, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, SRF_RETURN_NEXT, tt_process_call(), and tt_setup_firstcall().

 {
     FuncCallContext *funcctx;
     Datum       result;
 
     if (SRF_IS_FIRSTCALL())
     {
         funcctx = SRF_FIRSTCALL_INIT();
         tt_setup_firstcall(funcctx, PG_GETARG_OID(0));
     }
 
     funcctx = SRF_PERCALL_SETUP();
 
     if ((result = tt_process_call(funcctx)) != (Datum) 0)
         SRF_RETURN_NEXT(funcctx, result);
     SRF_RETURN_DONE(funcctx);
 }

Datum ts_token_type_byname ( PG_FUNCTION_ARGS )

Definition at line 133 of file wparser.c.

References get_ts_parser_oid(), PG_GETARG_TEXT_PP, result, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, SRF_RETURN_NEXT, textToQualifiedNameList(), tt_process_call(), and tt_setup_firstcall().

 {
     FuncCallContext *funcctx;
     Datum       result;
 
     if (SRF_IS_FIRSTCALL())
     {
         text       *prsname = PG_GETARG_TEXT_PP(0);
         Oid         prsId;
 
         funcctx = SRF_FIRSTCALL_INIT();
         prsId = get_ts_parser_oid(textToQualifiedNameList(prsname), false);
         tt_setup_firstcall(funcctx, prsId);
     }
 
     funcctx = SRF_PERCALL_SETUP();
 
     if ((result = tt_process_call(funcctx)) != (Datum) 0)
         SRF_RETURN_NEXT(funcctx, result);
     SRF_RETURN_DONE(funcctx);
 }

static Datum tt_process_call ( FuncCallContext * funcctx )

static

Definition at line 82 of file wparser.c.

References LexDescr::alias, FuncCallContext::attinmeta, BuildTupleFromCStrings(), TSTokenTypeStorage::cur, LexDescr::descr, HeapTupleGetDatum, LexDescr::lexid, TSTokenTypeStorage::list, pfree(), result, FuncCallContext::user_fctx, and values.

Referenced by ts_token_type_byid(), and ts_token_type_byname().

 {
     TSTokenTypeStorage *st;
 
     st = (TSTokenTypeStorage *) funcctx->user_fctx;
     if (st->list && st->list[st->cur].lexid)
     {
         Datum       result;
         char       *values[3];
         char        txtid[16];
         HeapTuple   tuple;
 
         sprintf(txtid, "%d", st->list[st->cur].lexid);
         values[0] = txtid;
         values[1] = st->list[st->cur].alias;
         values[2] = st->list[st->cur].descr;
 
         tuple = BuildTupleFromCStrings(funcctx->attinmeta, values);
         result = HeapTupleGetDatum(tuple);
 
         pfree(values[1]);
         pfree(values[2]);
         st->cur++;
         return result;
     }
     if (st->list)
         pfree(st->list);
     pfree(st);
     return (Datum) 0;
 }

static void tt_setup_firstcall	(	FuncCallContext *	funcctx,
		Oid	prsid
	)

static

Definition at line 49 of file wparser.c.

References FuncCallContext::attinmeta, CreateTemplateTupleDesc(), TSTokenTypeStorage::cur, DatumGetPointer, elog, ERROR, INT4OID, TSParserCacheEntry::lextypeOid, TSTokenTypeStorage::list, lookup_ts_parser_cache(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, OidFunctionCall1, OidIsValid, palloc(), TEXTOID, TupleDescGetAttInMetadata(), TupleDescInitEntry(), and FuncCallContext::user_fctx.

Referenced by ts_token_type_byid(), and ts_token_type_byname().

 {
     TupleDesc   tupdesc;
     MemoryContext oldcontext;
     TSTokenTypeStorage *st;
     TSParserCacheEntry *prs = lookup_ts_parser_cache(prsid);
 
     if (!OidIsValid(prs->lextypeOid))
         elog(ERROR, "method lextype isn't defined for text search parser %u",
              prsid);
 
     oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
 
     st = (TSTokenTypeStorage *) palloc(sizeof(TSTokenTypeStorage));
     st->cur = 0;
     /* lextype takes one dummy argument */
     st->list = (LexDescr *) DatumGetPointer(OidFunctionCall1(prs->lextypeOid,
                                                              (Datum) 0));
     funcctx->user_fctx = (void *) st;
 
     tupdesc = CreateTemplateTupleDesc(3, false);
     TupleDescInitEntry(tupdesc, (AttrNumber) 1, "tokid",
                        INT4OID, -1, 0);
     TupleDescInitEntry(tupdesc, (AttrNumber) 2, "alias",
                        TEXTOID, -1, 0);
     TupleDescInitEntry(tupdesc, (AttrNumber) 3, "description",
                        TEXTOID, -1, 0);
 
     funcctx->attinmeta = TupleDescGetAttInMetadata(tupdesc);
     MemoryContextSwitchTo(oldcontext);
 }

Data Structures

Typedefs

Functions

Typedef Documentation

Function Documentation