Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * dict_int.c
       4                 :             :  *    Text search dictionary for integers
       5                 :             :  *
       6                 :             :  * Copyright (c) 2007-2026, PostgreSQL Global Development Group
       7                 :             :  *
       8                 :             :  * IDENTIFICATION
       9                 :             :  *    contrib/dict_int/dict_int.c
      10                 :             :  *
      11                 :             :  *-------------------------------------------------------------------------
      12                 :             :  */
      13                 :             : #include "postgres.h"
      14                 :             : 
      15                 :             : #include "commands/defrem.h"
      16                 :             : #include "tsearch/ts_public.h"
      17                 :             : 
      18                 :           1 : PG_MODULE_MAGIC_EXT(
      19                 :             :                     .name = "dict_int",
      20                 :             :                     .version = PG_VERSION
      21                 :             : );
      22                 :             : 
      23                 :             : typedef struct
      24                 :             : {
      25                 :             :     int         maxlen;
      26                 :             :     bool        rejectlong;
      27                 :             :     bool        absval;
      28                 :             : } DictInt;
      29                 :             : 
      30                 :             : 
      31                 :           2 : PG_FUNCTION_INFO_V1(dintdict_init);
      32                 :           2 : PG_FUNCTION_INFO_V1(dintdict_lexize);
      33                 :             : 
      34                 :             : Datum
      35                 :           9 : dintdict_init(PG_FUNCTION_ARGS)
      36                 :             : {
      37                 :           9 :     List       *dictoptions = (List *) PG_GETARG_POINTER(0);
      38                 :             :     DictInt    *d;
      39                 :             :     ListCell   *l;
      40                 :             : 
      41                 :           9 :     d = palloc0_object(DictInt);
      42                 :           9 :     d->maxlen = 6;
      43                 :           9 :     d->rejectlong = false;
      44                 :           9 :     d->absval = false;
      45                 :             : 
      46   [ +  +  +  +  :          21 :     foreach(l, dictoptions)
                   +  + ]
      47                 :             :     {
      48                 :          13 :         DefElem    *defel = (DefElem *) lfirst(l);
      49                 :             : 
      50         [ +  + ]:          13 :         if (strcmp(defel->defname, "maxlen") == 0)
      51                 :             :         {
      52                 :           7 :             d->maxlen = atoi(defGetString(defel));
      53                 :             : 
      54         [ +  + ]:           7 :             if (d->maxlen < 1)
      55         [ +  - ]:           1 :                 ereport(ERROR,
      56                 :             :                         (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      57                 :             :                          errmsg("maxlen value has to be >= 1")));
      58                 :             :         }
      59         [ +  + ]:           6 :         else if (strcmp(defel->defname, "rejectlong") == 0)
      60                 :             :         {
      61                 :           2 :             d->rejectlong = defGetBoolean(defel);
      62                 :             :         }
      63         [ +  - ]:           4 :         else if (strcmp(defel->defname, "absval") == 0)
      64                 :             :         {
      65                 :           4 :             d->absval = defGetBoolean(defel);
      66                 :             :         }
      67                 :             :         else
      68                 :             :         {
      69         [ #  # ]:           0 :             ereport(ERROR,
      70                 :             :                     (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      71                 :             :                      errmsg("unrecognized intdict parameter: \"%s\"",
      72                 :             :                             defel->defname)));
      73                 :             :         }
      74                 :             :     }
      75                 :             : 
      76                 :           8 :     PG_RETURN_POINTER(d);
      77                 :             : }
      78                 :             : 
      79                 :             : Datum
      80                 :          57 : dintdict_lexize(PG_FUNCTION_ARGS)
      81                 :             : {
      82                 :          57 :     DictInt    *d = (DictInt *) PG_GETARG_POINTER(0);
      83                 :          57 :     char       *in = (char *) PG_GETARG_POINTER(1);
      84                 :          57 :     int         len = PG_GETARG_INT32(2);
      85                 :             :     char       *txt;
      86                 :          57 :     TSLexeme   *res = palloc0_array(TSLexeme, 2);
      87                 :             : 
      88                 :          57 :     res[1].lexeme = NULL;
      89                 :             : 
      90   [ +  +  +  +  :          57 :     if (d->absval && (in[0] == '+' || in[0] == '-'))
                   +  - ]
      91                 :             :     {
      92                 :           5 :         len--;
      93                 :           5 :         txt = pnstrdup(in + 1, len);
      94                 :             :     }
      95                 :             :     else
      96                 :          52 :         txt = pnstrdup(in, len);
      97                 :             : 
      98         [ +  + ]:          57 :     if (len > d->maxlen)
      99                 :             :     {
     100         [ +  + ]:          38 :         if (d->rejectlong)
     101                 :             :         {
     102                 :             :             /* reject by returning void array */
     103                 :           2 :             pfree(txt);
     104                 :           2 :             res[0].lexeme = NULL;
     105                 :             :         }
     106                 :             :         else
     107                 :             :         {
     108                 :             :             /* trim integer */
     109                 :          36 :             txt[d->maxlen] = '\0';
     110                 :          36 :             res[0].lexeme = txt;
     111                 :             :         }
     112                 :             :     }
     113                 :             :     else
     114                 :             :     {
     115                 :          19 :         res[0].lexeme = txt;
     116                 :             :     }
     117                 :             : 
     118                 :          57 :     PG_RETURN_POINTER(res);
     119                 :             : }