Age Owner Branch data TLA 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-2024, 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 "commands/defrem.h"
17 : : #include "tsearch/ts_locale.h"
18 : : #include "tsearch/ts_public.h"
19 : : #include "utils/fmgrprotos.h"
20 : :
21 : :
22 : : typedef struct
23 : : {
24 : : StopList stoplist;
25 : : bool accept;
26 : : } DictSimple;
27 : :
28 : :
29 : : Datum
6081 tgl@sss.pgh.pa.us 30 :CBC 35 : dsimple_init(PG_FUNCTION_ARGS)
31 : : {
6080 32 : 35 : List *dictoptions = (List *) PG_GETARG_POINTER(0);
6077 33 : 35 : DictSimple *d = (DictSimple *) palloc0(sizeof(DictSimple));
5996 34 : 35 : bool stoploaded = false,
35 : 35 : acceptloaded = false;
36 : : ListCell *l;
37 : :
38 : 35 : d->accept = true; /* default */
39 : :
6080 40 [ - + - - : 35 : foreach(l, dictoptions)
- + ]
41 : : {
6080 tgl@sss.pgh.pa.us 42 :UBC 0 : DefElem *defel = (DefElem *) lfirst(l);
43 : :
2270 44 [ # # ]: 0 : if (strcmp(defel->defname, "stopwords") == 0)
45 : : {
6080 46 [ # # ]: 0 : if (stoploaded)
47 [ # # ]: 0 : ereport(ERROR,
48 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
49 : : errmsg("multiple StopWords parameters")));
6077 50 : 0 : readstoplist(defGetString(defel), &d->stoplist, lowerstr);
6080 51 : 0 : stoploaded = true;
52 : : }
2270 53 [ # # ]: 0 : else if (strcmp(defel->defname, "accept") == 0)
54 : : {
5996 55 [ # # ]: 0 : if (acceptloaded)
56 [ # # ]: 0 : ereport(ERROR,
57 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
58 : : errmsg("multiple Accept parameters")));
59 : 0 : d->accept = defGetBoolean(defel);
60 : 0 : acceptloaded = true;
61 : : }
62 : : else
63 : : {
6080 64 [ # # ]: 0 : ereport(ERROR,
65 : : (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
66 : : errmsg("unrecognized simple dictionary parameter: \"%s\"",
67 : : defel->defname)));
68 : : }
69 : : }
70 : :
6081 tgl@sss.pgh.pa.us 71 :CBC 35 : PG_RETURN_POINTER(d);
72 : : }
73 : :
74 : : Datum
75 : 1837 : dsimple_lexize(PG_FUNCTION_ARGS)
76 : : {
6077 77 : 1837 : DictSimple *d = (DictSimple *) PG_GETARG_POINTER(0);
6081 78 : 1837 : char *in = (char *) PG_GETARG_POINTER(1);
5995 bruce@momjian.us 79 : 1837 : int32 len = PG_GETARG_INT32(2);
80 : : char *txt;
81 : : TSLexeme *res;
82 : :
6077 tgl@sss.pgh.pa.us 83 : 1837 : txt = lowerstr_with_len(in, len);
84 : :
6081 85 [ + - - + ]: 1837 : if (*txt == '\0' || searchstoplist(&(d->stoplist), txt))
86 : : {
87 : : /* reject as stopword */
6081 tgl@sss.pgh.pa.us 88 :UBC 0 : pfree(txt);
5996 89 : 0 : res = palloc0(sizeof(TSLexeme) * 2);
90 : 0 : PG_RETURN_POINTER(res);
91 : : }
5996 tgl@sss.pgh.pa.us 92 [ + - ]:CBC 1837 : else if (d->accept)
93 : : {
94 : : /* accept */
95 : 1837 : res = palloc0(sizeof(TSLexeme) * 2);
6081 96 : 1837 : res[0].lexeme = txt;
5996 97 : 1837 : PG_RETURN_POINTER(res);
98 : : }
99 : : else
100 : : {
101 : : /* report as unrecognized */
5996 tgl@sss.pgh.pa.us 102 :UBC 0 : pfree(txt);
103 : 0 : PG_RETURN_POINTER(NULL);
104 : : }
105 : : }
|
