wparser.c File Reference

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

Include dependency graph for wparser.c:

Go to the source code of this file.

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, FunctionCallInfo fcinfo, 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, FunctionCallInfo fcinfo, 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

HeadlineJsonState

typedef struct HeadlineJsonState HeadlineJsonState

Function Documentation

headline_json_value()

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

static

Definition at line 525 of file wparser.c.

{
    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);
}

References TSConfigCacheEntry::cfgId, HeadlineParsedText::curwords, FunctionCall3, generateHeadline(), hlparsetext(), PointerGetDatum(), and TSParserCacheEntry::prsheadline.

Referenced by ts_headline_json_byid_opt(), and ts_headline_jsonb_byid_opt().

prs_process_call()

static Datum prs_process_call ( FuncCallContext *  funcctx )

static

Definition at line 218 of file wparser.c.

{
    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;
    }
    return (Datum) 0;
}

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

Referenced by ts_parse_byid(), and ts_parse_byname().

prs_setup_firstcall()

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

static

Definition at line 164 of file wparser.c.

{
    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;
    if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
        elog(ERROR, "return type must be a row type");
    funcctx->tuple_desc = tupdesc;
    funcctx->attinmeta = TupleDescGetAttInMetadata(tupdesc);
    MemoryContextSwitchTo(oldcontext);
}

References FuncCallContext::attinmeta, PrsStorage::cur, DatumGetInt32(), DatumGetPointer(), elog(), ERROR, FunctionCall1, FunctionCall2, FunctionCall3, get_call_result_type(), Int32GetDatum(), PrsStorage::len, LexemeEntry::lexeme, PrsStorage::list, lookup_ts_parser_cache(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, palloc(), PointerGetDatum(), TSParserCacheEntry::prsend, TSParserCacheEntry::prsstart, TSParserCacheEntry::prstoken, repalloc(), FuncCallContext::tuple_desc, TupleDescGetAttInMetadata(), generate_unaccent_rules::type, LexemeEntry::type, TYPEFUNC_COMPOSITE, FuncCallContext::user_fctx, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

Referenced by ts_parse_byid(), and ts_parse_byname().

ts_headline()

Datum ts_headline

(

PG_FUNCTION_ARGS

)

Definition at line 350 of file wparser.c.

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

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

ts_headline_byid()

Datum ts_headline_byid

(

PG_FUNCTION_ARGS

)

Definition at line 341 of file wparser.c.

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

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

ts_headline_byid_opt()

Datum ts_headline_byid_opt

(

PG_FUNCTION_ARGS

)

Definition at line 290 of file wparser.c.

{
    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);
}

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, 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().

ts_headline_json()

Datum ts_headline_json

(

PG_FUNCTION_ARGS

)

Definition at line 493 of file wparser.c.

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

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

ts_headline_json_byid()

Datum ts_headline_json_byid

(

PG_FUNCTION_ARGS

)

Definition at line 502 of file wparser.c.

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

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

ts_headline_json_byid_opt()

Datum ts_headline_json_byid_opt

(

PG_FUNCTION_ARGS

)

Definition at line 445 of file wparser.c.

{
    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);
}

References generate_unaccent_rules::action, deserialize_deflist(), ereport, errcode(), errmsg(), ERROR, headline_json_value(), HeadlineParsedText::lenwords, lookup_ts_config_cache(), lookup_ts_parser_cache(), NIL, 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(), HeadlineParsedText::startsel, HeadlineParsedText::stopsel, transform_json_string_values(), and HeadlineParsedText::words.

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

ts_headline_json_opt()

Datum ts_headline_json_opt

(

PG_FUNCTION_ARGS

)

Definition at line 511 of file wparser.c.

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

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

ts_headline_jsonb()

Datum ts_headline_jsonb

(

PG_FUNCTION_ARGS

)

Definition at line 417 of file wparser.c.

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

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

ts_headline_jsonb_byid()

Datum ts_headline_jsonb_byid

(

PG_FUNCTION_ARGS

)

Definition at line 426 of file wparser.c.

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

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

ts_headline_jsonb_byid_opt()

Datum ts_headline_jsonb_byid_opt

(

PG_FUNCTION_ARGS

)

Definition at line 369 of file wparser.c.

{
    Oid         tsconfig = PG_GETARG_OID(0);
    Jsonb      *jb = PG_GETARG_JSONB_P(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_P(out);
}

References generate_unaccent_rules::action, deserialize_deflist(), ereport, errcode(), errmsg(), ERROR, headline_json_value(), HeadlineParsedText::lenwords, lookup_ts_config_cache(), lookup_ts_parser_cache(), NIL, OidIsValid, palloc(), palloc0(), pfree(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_GETARG_OID, PG_GETARG_POINTER, PG_GETARG_TEXT_P, PG_GETARG_TSQUERY, PG_NARGS, PG_RETURN_JSONB_P, PointerGetDatum(), HeadlineParsedText::startsel, HeadlineParsedText::stopsel, transform_jsonb_string_values(), and HeadlineParsedText::words.

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

ts_headline_jsonb_opt()

Datum ts_headline_jsonb_opt

(

PG_FUNCTION_ARGS

)

Definition at line 435 of file wparser.c.

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

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

ts_headline_opt()

Datum ts_headline_opt

(

PG_FUNCTION_ARGS

)

Definition at line 359 of file wparser.c.

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

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

ts_parse_byid()

Datum ts_parse_byid

(

PG_FUNCTION_ARGS

)

Definition at line 244 of file wparser.c.

{
    FuncCallContext *funcctx;
    Datum       result;
 
    if (SRF_IS_FIRSTCALL())
    {
        text       *txt = PG_GETARG_TEXT_PP(1);
 
        funcctx = SRF_FIRSTCALL_INIT();
        prs_setup_firstcall(funcctx, fcinfo, 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);
}

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

ts_parse_byname()

Datum ts_parse_byname

(

PG_FUNCTION_ARGS

)

Definition at line 266 of file wparser.c.

{
    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, fcinfo, prsId, txt);
    }
 
    funcctx = SRF_PERCALL_SETUP();
 
    if ((result = prs_process_call(funcctx)) != (Datum) 0)
        SRF_RETURN_NEXT(funcctx, result);
    SRF_RETURN_DONE(funcctx);
}

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

ts_token_type_byid()

Datum ts_token_type_byid

(

PG_FUNCTION_ARGS

)

Definition at line 108 of file wparser.c.

{
    FuncCallContext *funcctx;
    Datum       result;
 
    if (SRF_IS_FIRSTCALL())
    {
        funcctx = SRF_FIRSTCALL_INIT();
        tt_setup_firstcall(funcctx, fcinfo, 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);
}

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

ts_token_type_byname()

Datum ts_token_type_byname

(

PG_FUNCTION_ARGS

)

Definition at line 127 of file wparser.c.

{
    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, fcinfo, prsId);
    }
 
    funcctx = SRF_PERCALL_SETUP();
 
    if ((result = tt_process_call(funcctx)) != (Datum) 0)
        SRF_RETURN_NEXT(funcctx, result);
    SRF_RETURN_DONE(funcctx);
}

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

tt_process_call()

static Datum tt_process_call ( FuncCallContext *  funcctx )

static

Definition at line 79 of file wparser.c.

{
    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;
    }
    return (Datum) 0;
}

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

Referenced by ts_token_type_byid(), and ts_token_type_byname().

tt_setup_firstcall()

static void tt_setup_firstcall ( FuncCallContext *  funcctx, FunctionCallInfo  fcinfo, Oid  prsid  )

static

Definition at line 49 of file wparser.c.

{
    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;
 
    if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
        elog(ERROR, "return type must be a row type");
    funcctx->tuple_desc = tupdesc;
    funcctx->attinmeta = TupleDescGetAttInMetadata(tupdesc);
 
    MemoryContextSwitchTo(oldcontext);
}

References FuncCallContext::attinmeta, TSTokenTypeStorage::cur, DatumGetPointer(), elog(), ERROR, get_call_result_type(), TSParserCacheEntry::lextypeOid, TSTokenTypeStorage::list, lookup_ts_parser_cache(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, OidFunctionCall1, OidIsValid, palloc(), FuncCallContext::tuple_desc, TupleDescGetAttInMetadata(), TYPEFUNC_COMPOSITE, and FuncCallContext::user_fctx.

Referenced by ts_token_type_byid(), and ts_token_type_byname().