Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * dict_simple.c
       4                 :             :  *              Simple dictionary: just lowercase and check for stopword
       5                 :             :  *
       6                 :             :  * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
       7                 :             :  *
       8                 :             :  *
       9                 :             :  * IDENTIFICATION
      10                 :             :  *        src/backend/tsearch/dict_simple.c
      11                 :             :  *
      12                 :             :  *-------------------------------------------------------------------------
      13                 :             :  */
      14                 :             : #include "postgres.h"
      15                 :             : 
      16                 :             : #include "catalog/pg_collation_d.h"
      17                 :             : #include "commands/defrem.h"
      18                 :             : #include "tsearch/ts_public.h"
      19                 :             : #include "utils/fmgrprotos.h"
      20                 :             : #include "utils/formatting.h"
      21                 :             : 
      22                 :             : 
      23                 :             : typedef struct
      24                 :             : {
      25                 :             :         StopList        stoplist;
      26                 :             :         bool            accept;
      27                 :             : } DictSimple;
      28                 :             : 
      29                 :             : 
      30                 :             : Datum
      31                 :          10 : dsimple_init(PG_FUNCTION_ARGS)
      32                 :             : {
      33                 :          10 :         List       *dictoptions = (List *) PG_GETARG_POINTER(0);
      34                 :          10 :         DictSimple *d = palloc0_object(DictSimple);
      35                 :          10 :         bool            stoploaded = false,
      36                 :          10 :                                 acceptloaded = false;
      37                 :          10 :         ListCell   *l;
      38                 :             : 
      39                 :          10 :         d->accept = true;                    /* default */
      40                 :             : 
      41   [ -  +  #  #  :          10 :         foreach(l, dictoptions)
                   +  - ]
      42                 :             :         {
      43                 :           0 :                 DefElem    *defel = (DefElem *) lfirst(l);
      44                 :             : 
      45         [ #  # ]:           0 :                 if (strcmp(defel->defname, "stopwords") == 0)
      46                 :             :                 {
      47         [ #  # ]:           0 :                         if (stoploaded)
      48   [ #  #  #  # ]:           0 :                                 ereport(ERROR,
      49                 :             :                                                 (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      50                 :             :                                                  errmsg("multiple StopWords parameters")));
      51                 :           0 :                         readstoplist(defGetString(defel), &d->stoplist, str_tolower);
      52                 :           0 :                         stoploaded = true;
      53                 :           0 :                 }
      54         [ #  # ]:           0 :                 else if (strcmp(defel->defname, "accept") == 0)
      55                 :             :                 {
      56         [ #  # ]:           0 :                         if (acceptloaded)
      57   [ #  #  #  # ]:           0 :                                 ereport(ERROR,
      58                 :             :                                                 (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      59                 :             :                                                  errmsg("multiple Accept parameters")));
      60                 :           0 :                         d->accept = defGetBoolean(defel);
      61                 :           0 :                         acceptloaded = true;
      62                 :           0 :                 }
      63                 :             :                 else
      64                 :             :                 {
      65   [ #  #  #  # ]:           0 :                         ereport(ERROR,
      66                 :             :                                         (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      67                 :             :                                          errmsg("unrecognized simple dictionary parameter: \"%s\"",
      68                 :             :                                                         defel->defname)));
      69                 :             :                 }
      70                 :           0 :         }
      71                 :             : 
      72                 :          20 :         PG_RETURN_POINTER(d);
      73                 :          10 : }
      74                 :             : 
      75                 :             : Datum
      76                 :         610 : dsimple_lexize(PG_FUNCTION_ARGS)
      77                 :             : {
      78                 :         610 :         DictSimple *d = (DictSimple *) PG_GETARG_POINTER(0);
      79                 :         610 :         char       *in = (char *) PG_GETARG_POINTER(1);
      80                 :         610 :         int32           len = PG_GETARG_INT32(2);
      81                 :         610 :         char       *txt;
      82                 :         610 :         TSLexeme   *res;
      83                 :             : 
      84                 :         610 :         txt = str_tolower(in, len, DEFAULT_COLLATION_OID);
      85                 :             : 
      86   [ +  -  -  + ]:         610 :         if (*txt == '\0' || searchstoplist(&(d->stoplist), txt))
      87                 :             :         {
      88                 :             :                 /* reject as stopword */
      89                 :           0 :                 pfree(txt);
      90                 :           0 :                 res = palloc0_array(TSLexeme, 2);
      91                 :           0 :                 PG_RETURN_POINTER(res);
      92                 :             :         }
      93         [ +  - ]:         610 :         else if (d->accept)
      94                 :             :         {
      95                 :             :                 /* accept */
      96                 :         610 :                 res = palloc0_array(TSLexeme, 2);
      97                 :         610 :                 res[0].lexeme = txt;
      98                 :         610 :                 PG_RETURN_POINTER(res);
      99                 :             :         }
     100                 :             :         else
     101                 :             :         {
     102                 :             :                 /* report as unrecognized */
     103                 :           0 :                 pfree(txt);
     104                 :           0 :                 PG_RETURN_POINTER(NULL);
     105                 :             :         }
     106                 :         610 : }