Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * dict.c
       4                 :             :  *              Standard interface to dictionary
       5                 :             :  *
       6                 :             :  * Portions Copyright (c) 1996-2026, 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_public.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                 :         127 : ts_lexize(PG_FUNCTION_ARGS)
      28                 :             : {
      29                 :         127 :         Oid                     dictId = PG_GETARG_OID(0);
      30                 :         127 :         text       *in = PG_GETARG_TEXT_PP(1);
      31                 :         127 :         ArrayType  *a;
      32                 :         127 :         TSDictionaryCacheEntry *dict;
      33                 :         127 :         TSLexeme   *res,
      34                 :             :                            *ptr;
      35                 :         127 :         Datum      *da;
      36                 :         127 :         DictSubState dstate = {false, false, NULL};
      37                 :             : 
      38                 :         127 :         dict = lookup_ts_dictionary_cache(dictId);
      39                 :             : 
      40                 :         127 :         res = (TSLexeme *) DatumGetPointer(FunctionCall4(&dict->lexize,
      41                 :             :                                                                                                          PointerGetDatum(dict->dictData),
      42                 :             :                                                                                                          PointerGetDatum(VARDATA_ANY(in)),
      43                 :             :                                                                                                          Int32GetDatum(VARSIZE_ANY_EXHDR(in)),
      44                 :             :                                                                                                          PointerGetDatum(&dstate)));
      45                 :             : 
      46         [ +  + ]:         127 :         if (dstate.getnext)
      47                 :             :         {
      48                 :           1 :                 dstate.isend = true;
      49                 :           1 :                 ptr = (TSLexeme *) DatumGetPointer(FunctionCall4(&dict->lexize,
      50                 :             :                                                                                                                  PointerGetDatum(dict->dictData),
      51                 :             :                                                                                                                  PointerGetDatum(VARDATA_ANY(in)),
      52                 :             :                                                                                                                  Int32GetDatum(VARSIZE_ANY_EXHDR(in)),
      53                 :             :                                                                                                                  PointerGetDatum(&dstate)));
      54         [ +  - ]:           1 :                 if (ptr != NULL)
      55                 :           0 :                         res = ptr;
      56                 :           1 :         }
      57                 :             : 
      58         [ +  + ]:         127 :         if (!res)
      59                 :           5 :                 PG_RETURN_NULL();
      60                 :             : 
      61                 :         122 :         ptr = res;
      62         [ +  + ]:         295 :         while (ptr->lexeme)
      63                 :         173 :                 ptr++;
      64                 :         122 :         da = (Datum *) palloc_array(Datum, ptr - res);
      65                 :         122 :         ptr = res;
      66         [ +  + ]:         295 :         while (ptr->lexeme)
      67                 :             :         {
      68                 :         173 :                 da[ptr - res] = CStringGetTextDatum(ptr->lexeme);
      69                 :         173 :                 ptr++;
      70                 :             :         }
      71                 :             : 
      72                 :         122 :         a = construct_array_builtin(da, ptr - res, TEXTOID);
      73                 :             : 
      74                 :         122 :         ptr = res;
      75         [ +  + ]:         295 :         while (ptr->lexeme)
      76                 :             :         {
      77                 :         173 :                 pfree(DatumGetPointer(da[ptr - res]));
      78                 :         173 :                 pfree(ptr->lexeme);
      79                 :         173 :                 ptr++;
      80                 :             :         }
      81                 :         122 :         pfree(res);
      82                 :         122 :         pfree(da);
      83                 :             : 
      84                 :         122 :         PG_RETURN_POINTER(a);
      85                 :         127 : }