TLA  Line data    Source code

       1                 : /*-------------------------------------------------------------------------
       2                 :  *
       3                 :  * dict.c
       4                 :  *      Standard interface to dictionary
       5                 :  *
       6                 :  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
       7                 :  *
       8                 :  *
       9                 :  * IDENTIFICATION
      10                 :  *    src/backend/tsearch/dict.c
      11                 :  *
      12                 :  *-------------------------------------------------------------------------
      13                 :  */
      14                 : #include "postgres.h"
      15                 : 
      16                 : #include "catalog/pg_type.h"
      17                 : #include "tsearch/ts_cache.h"
      18                 : #include "tsearch/ts_utils.h"
      19                 : #include "utils/array.h"
      20                 : #include "utils/builtins.h"
      21                 : 
      22                 : 
      23                 : /*
      24                 :  * Lexize one word by dictionary, mostly debug function
      25                 :  */
      26                 : Datum
      27 GIC         466 : ts_lexize(PG_FUNCTION_ARGS)
      28 ECB             : {
      29 GIC         466 :     Oid         dictId = PG_GETARG_OID(0);
      30 CBC         466 :     text       *in = PG_GETARG_TEXT_PP(1);
      31 ECB             :     ArrayType  *a;
      32                 :     TSDictionaryCacheEntry *dict;
      33                 :     TSLexeme   *res,
      34                 :                *ptr;
      35                 :     Datum      *da;
      36 GIC         466 :     DictSubState dstate = {false, false, NULL};
      37 ECB             : 
      38 GIC         466 :     dict = lookup_ts_dictionary_cache(dictId);
      39 ECB             : 
      40 GIC         466 :     res = (TSLexeme *) DatumGetPointer(FunctionCall4(&dict->lexize,
      41 ECB             :                                                      PointerGetDatum(dict->dictData),
      42                 :                                                      PointerGetDatum(VARDATA_ANY(in)),
      43                 :                                                      Int32GetDatum(VARSIZE_ANY_EXHDR(in)),
      44                 :                                                      PointerGetDatum(&dstate)));
      45                 : 
      46 GIC         466 :     if (dstate.getnext)
      47 ECB             :     {
      48 GIC           3 :         dstate.isend = true;
      49 CBC           3 :         ptr = (TSLexeme *) DatumGetPointer(FunctionCall4(&dict->lexize,
      50 ECB             :                                                          PointerGetDatum(dict->dictData),
      51                 :                                                          PointerGetDatum(VARDATA_ANY(in)),
      52                 :                                                          Int32GetDatum(VARSIZE_ANY_EXHDR(in)),
      53                 :                                                          PointerGetDatum(&dstate)));
      54 GIC           3 :         if (ptr != NULL)
      55 LBC           0 :             res = ptr;
      56 EUB             :     }
      57                 : 
      58 GIC         466 :     if (!res)
      59 CBC          28 :         PG_RETURN_NULL();
      60 ECB             : 
      61 GIC         438 :     ptr = res;
      62 CBC        1040 :     while (ptr->lexeme)
      63             602 :         ptr++;
      64             438 :     da = (Datum *) palloc(sizeof(Datum) * (ptr - res));
      65             438 :     ptr = res;
      66            1040 :     while (ptr->lexeme)
      67 ECB             :     {
      68 GIC         602 :         da[ptr - res] = CStringGetTextDatum(ptr->lexeme);
      69 CBC         602 :         ptr++;
      70 ECB             :     }
      71                 : 
      72 GNC         438 :     a = construct_array_builtin(da, ptr - res, TEXTOID);
      73 ECB             : 
      74 CBC         438 :     ptr = res;
      75            1040 :     while (ptr->lexeme)
      76                 :     {
      77             602 :         pfree(DatumGetPointer(da[ptr - res]));
      78             602 :         pfree(ptr->lexeme);
      79 GIC         602 :         ptr++;
      80 ECB             :     }
      81 GIC         438 :     pfree(res);
      82             438 :     pfree(da);
      83                 : 
      84             438 :     PG_RETURN_POINTER(a);
      85                 : }