LCOV - code coverage report
Current view: top level - contrib/dict_int - dict_int.c (source / functions) Hit Total Coverage
Test: PostgreSQL 13devel Lines: 22 30 73.3 %
Date: 2019-11-13 23:06:49 Functions: 5 5 100.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*-------------------------------------------------------------------------
       2             :  *
       3             :  * dict_int.c
       4             :  *    Text search dictionary for integers
       5             :  *
       6             :  * Copyright (c) 2007-2019, 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           2 : PG_MODULE_MAGIC;
      19             : 
      20             : typedef struct
      21             : {
      22             :     int         maxlen;
      23             :     bool        rejectlong;
      24             : } DictInt;
      25             : 
      26             : 
      27           4 : PG_FUNCTION_INFO_V1(dintdict_init);
      28           4 : PG_FUNCTION_INFO_V1(dintdict_lexize);
      29             : 
      30             : Datum
      31           4 : dintdict_init(PG_FUNCTION_ARGS)
      32             : {
      33           4 :     List       *dictoptions = (List *) PG_GETARG_POINTER(0);
      34             :     DictInt    *d;
      35             :     ListCell   *l;
      36             : 
      37           4 :     d = (DictInt *) palloc0(sizeof(DictInt));
      38           4 :     d->maxlen = 6;
      39           4 :     d->rejectlong = false;
      40             : 
      41           4 :     foreach(l, dictoptions)
      42             :     {
      43           0 :         DefElem    *defel = (DefElem *) lfirst(l);
      44             : 
      45           0 :         if (strcmp(defel->defname, "maxlen") == 0)
      46             :         {
      47           0 :             d->maxlen = atoi(defGetString(defel));
      48             :         }
      49           0 :         else if (strcmp(defel->defname, "rejectlong") == 0)
      50             :         {
      51           0 :             d->rejectlong = defGetBoolean(defel);
      52             :         }
      53             :         else
      54             :         {
      55           0 :             ereport(ERROR,
      56             :                     (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
      57             :                      errmsg("unrecognized intdict parameter: \"%s\"",
      58             :                             defel->defname)));
      59             :         }
      60             :     }
      61             : 
      62           4 :     PG_RETURN_POINTER(d);
      63             : }
      64             : 
      65             : Datum
      66         100 : dintdict_lexize(PG_FUNCTION_ARGS)
      67             : {
      68         100 :     DictInt    *d = (DictInt *) PG_GETARG_POINTER(0);
      69         100 :     char       *in = (char *) PG_GETARG_POINTER(1);
      70         100 :     char       *txt = pnstrdup(in, PG_GETARG_INT32(2));
      71         100 :     TSLexeme   *res = palloc0(sizeof(TSLexeme) * 2);
      72             : 
      73         100 :     res[1].lexeme = NULL;
      74         100 :     if (PG_GETARG_INT32(2) > d->maxlen)
      75             :     {
      76          64 :         if (d->rejectlong)
      77             :         {
      78             :             /* reject by returning void array */
      79           0 :             pfree(txt);
      80           0 :             res[0].lexeme = NULL;
      81             :         }
      82             :         else
      83             :         {
      84             :             /* trim integer */
      85          64 :             txt[d->maxlen] = '\0';
      86          64 :             res[0].lexeme = txt;
      87             :         }
      88             :     }
      89             :     else
      90             :     {
      91          36 :         res[0].lexeme = txt;
      92             :     }
      93             : 
      94         100 :     PG_RETURN_POINTER(res);
      95             : }

Generated by: LCOV version 1.13