Age Owner Branch data TLA Line data Source code
1 : : /*
2 : : * txtquery operations with ltree
3 : : * Teodor Sigaev <teodor@stack.net>
4 : : * contrib/ltree/ltxtquery_op.c
5 : : */
6 : : #include "postgres.h"
7 : :
8 : : #include <ctype.h>
9 : :
10 : : #include "ltree.h"
11 : : #include "miscadmin.h"
12 : :
8736 bruce@momjian.us 13 :CBC 3 : PG_FUNCTION_INFO_V1(ltxtq_exec);
14 : 2 : PG_FUNCTION_INFO_V1(ltxtq_rexec);
15 : :
16 : : /*
17 : : * check for boolean condition
18 : : */
19 : : bool
6228 20 : 25311 : ltree_execute(ITEM *curitem, void *checkval, bool calcnot, bool (*chkcond) (void *checkval, ITEM *val))
21 : : {
22 : : /* since this function recurses, it could be driven to stack overflow */
3921 noah@leadboat.com 23 : 25311 : check_stack_depth();
24 : :
8736 bruce@momjian.us 25 [ + + ]: 25311 : if (curitem->type == VAL)
26 : 14505 : return (*chkcond) (checkval, curitem);
5118 peter_e@gmx.net 27 [ + + ]: 10806 : else if (curitem->val == (int32) '!')
28 : : {
29 : : return calcnot ?
8725 bruce@momjian.us 30 : 3 : ((ltree_execute(curitem + 1, checkval, calcnot, chkcond)) ? false : true)
8736 31 [ + - + + ]: 6 : : true;
32 : : }
5118 peter_e@gmx.net 33 [ + + ]: 10803 : else if (curitem->val == (int32) '&')
34 : : {
8700 bruce@momjian.us 35 [ + + ]: 10801 : if (ltree_execute(curitem + curitem->left, checkval, calcnot, chkcond))
8725 36 : 3701 : return ltree_execute(curitem + 1, checkval, calcnot, chkcond);
37 : : else
8736 38 : 7100 : return false;
39 : : }
40 : : else
41 : : { /* |-operator */
8725 42 [ + + ]: 2 : if (ltree_execute(curitem + curitem->left, checkval, calcnot, chkcond))
8736 43 : 1 : return true;
44 : : else
8725 45 : 1 : return ltree_execute(curitem + 1, checkval, calcnot, chkcond);
46 : : }
47 : : }
48 : :
49 : : typedef struct
50 : : {
51 : : ltree *node;
52 : : char *operand;
53 : : } CHKVAL;
54 : :
55 : : static bool
6228 56 : 10763 : checkcondition_str(void *checkval, ITEM *val)
57 : : {
8700 58 : 10763 : ltree_level *level = LTREE_FIRST(((CHKVAL *) checkval)->node);
59 : 10763 : int tlen = ((CHKVAL *) checkval)->node->numlevel;
60 : 10763 : char *op = ((CHKVAL *) checkval)->operand + val->distance;
124 jdavis@postgresql.or 61 : 10763 : bool prefix = (val->flag & LVAR_ANYEND);
62 : 10763 : bool ci = (val->flag & LVAR_INCASE);
63 : :
8700 bruce@momjian.us 64 [ + + ]: 72305 : while (tlen > 0)
65 : : {
7426 neilc@samurai.com 66 [ + + ]: 64192 : if (val->flag & LVAR_SUBLEXEME)
67 : : {
124 jdavis@postgresql.or 68 [ + + ]: 4 : if (compare_subnode(level, op, val->length, prefix, ci))
8736 bruce@momjian.us 69 : 1 : return true;
70 : : }
124 jdavis@postgresql.or 71 [ + + ]: 64188 : else if (ltree_label_match(op, val->length, level->name, level->len,
72 : : prefix, ci))
8736 bruce@momjian.us 73 : 2649 : return true;
74 : :
75 : 61542 : tlen--;
8700 76 : 61542 : level = LEVEL_NEXT(level);
77 : : }
78 : :
8736 79 : 8113 : return false;
80 : : }
81 : :
82 : : Datum
8700 83 : 8680 : ltxtq_exec(PG_FUNCTION_ARGS)
84 : : {
3207 tgl@sss.pgh.pa.us 85 : 8680 : ltree *val = PG_GETARG_LTREE_P(0);
86 : 8680 : ltxtquery *query = PG_GETARG_LTXTQUERY_P(1);
87 : : CHKVAL chkval;
88 : : bool result;
89 : :
8736 bruce@momjian.us 90 : 8680 : chkval.node = val;
91 : 8680 : chkval.operand = GETOPERAND(query);
92 : :
2343 alvherre@alvh.no-ip. 93 : 8680 : result = ltree_execute(GETQUERY(query),
94 : : &chkval,
95 : : true,
96 : : checkcondition_str);
97 : :
8736 bruce@momjian.us 98 [ + + ]: 8680 : PG_FREE_IF_COPY(val, 0);
99 [ - + ]: 8680 : PG_FREE_IF_COPY(query, 1);
100 : 8680 : PG_RETURN_BOOL(result);
101 : : }
102 : :
103 : : Datum
8700 bruce@momjian.us 104 :UBC 0 : ltxtq_rexec(PG_FUNCTION_ARGS)
105 : : {
106 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(ltxtq_exec,
107 : : PG_GETARG_DATUM(1),
108 : : PG_GETARG_DATUM(0)
109 : : ));
110 : : }
|
