Branch data Line data Source code
1 : : /*
2 : : * contrib/ltree/_ltree_op.c
3 : : *
4 : : *
5 : : * op function for ltree[]
6 : : * Teodor Sigaev <teodor@stack.net>
7 : : */
8 : : #include "postgres.h"
9 : :
10 : : #include <ctype.h>
11 : :
12 : : #include "ltree.h"
13 : : #include "utils/array.h"
14 : :
15 : 3 : PG_FUNCTION_INFO_V1(_ltree_isparent);
16 : 2 : PG_FUNCTION_INFO_V1(_ltree_r_isparent);
17 : 3 : PG_FUNCTION_INFO_V1(_ltree_risparent);
18 : 2 : PG_FUNCTION_INFO_V1(_ltree_r_risparent);
19 : 3 : PG_FUNCTION_INFO_V1(_ltq_regex);
20 : 2 : PG_FUNCTION_INFO_V1(_ltq_rregex);
21 : 3 : PG_FUNCTION_INFO_V1(_lt_q_regex);
22 : 2 : PG_FUNCTION_INFO_V1(_lt_q_rregex);
23 : 3 : PG_FUNCTION_INFO_V1(_ltxtq_exec);
24 : 2 : PG_FUNCTION_INFO_V1(_ltxtq_rexec);
25 : :
26 : 3 : PG_FUNCTION_INFO_V1(_ltree_extract_isparent);
27 : 3 : PG_FUNCTION_INFO_V1(_ltree_extract_risparent);
28 : 3 : PG_FUNCTION_INFO_V1(_ltq_extract_regex);
29 : 3 : PG_FUNCTION_INFO_V1(_ltxtq_extract_exec);
30 : :
31 : 3 : PG_FUNCTION_INFO_V1(_lca);
32 : :
33 : : typedef Datum (*PGCALL2) (PG_FUNCTION_ARGS);
34 : :
35 : : #define NEXTVAL(x) ( (ltree*)( (char*)(x) + INTALIGN( VARSIZE(x) ) ) )
36 : :
37 : : static bool
38 : 20323 : array_iterator(ArrayType *la, PGCALL2 callback, void *param, ltree **found)
39 : : {
40 : 20323 : int num = ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la));
41 [ - + ]: 20323 : ltree *item = (ltree *) ARR_DATA_PTR(la);
42 : :
43 [ - + ]: 20323 : if (ARR_NDIM(la) > 1)
44 [ # # ]: 0 : ereport(ERROR,
45 : : (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
46 : : errmsg("array must be one-dimensional")));
47 [ - + ]: 20323 : if (array_contains_nulls(la))
48 [ # # ]: 0 : ereport(ERROR,
49 : : (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
50 : : errmsg("array must not contain nulls")));
51 : :
52 [ + + ]: 20323 : if (found)
53 : 6 : *found = NULL;
54 [ + + ]: 91782 : while (num > 0)
55 : : {
56 [ + + ]: 72507 : if (DatumGetBool(DirectFunctionCall2(callback,
57 : : PointerGetDatum(item), PointerGetDatum(param))))
58 : : {
59 : :
60 [ + + ]: 1048 : if (found)
61 : 4 : *found = item;
62 : 1048 : return true;
63 : : }
64 : 71459 : num--;
65 : 71459 : item = NEXTVAL(item);
66 : : }
67 : :
68 : 19275 : return false;
69 : : }
70 : :
71 : : Datum
72 : 3012 : _ltree_isparent(PG_FUNCTION_ARGS)
73 : : {
74 : 3012 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
75 : 3012 : ltree *query = PG_GETARG_LTREE_P(1);
76 : 3012 : bool res = array_iterator(la, ltree_isparent, query, NULL);
77 : :
78 [ + + ]: 3012 : PG_FREE_IF_COPY(la, 0);
79 [ - + ]: 3012 : PG_FREE_IF_COPY(query, 1);
80 : 3012 : PG_RETURN_BOOL(res);
81 : : }
82 : :
83 : : Datum
84 : 0 : _ltree_r_isparent(PG_FUNCTION_ARGS)
85 : : {
86 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(_ltree_isparent,
87 : : PG_GETARG_DATUM(1),
88 : : PG_GETARG_DATUM(0)
89 : : ));
90 : : }
91 : :
92 : : Datum
93 : 2310 : _ltree_risparent(PG_FUNCTION_ARGS)
94 : : {
95 : 2310 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
96 : 2310 : ltree *query = PG_GETARG_LTREE_P(1);
97 : 2310 : bool res = array_iterator(la, ltree_risparent, query, NULL);
98 : :
99 [ + + ]: 2310 : PG_FREE_IF_COPY(la, 0);
100 [ - + ]: 2310 : PG_FREE_IF_COPY(query, 1);
101 : 2310 : PG_RETURN_BOOL(res);
102 : : }
103 : :
104 : : Datum
105 : 0 : _ltree_r_risparent(PG_FUNCTION_ARGS)
106 : : {
107 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(_ltree_risparent,
108 : : PG_GETARG_DATUM(1),
109 : : PG_GETARG_DATUM(0)
110 : : ));
111 : : }
112 : :
113 : : Datum
114 : 9575 : _ltq_regex(PG_FUNCTION_ARGS)
115 : : {
116 : 9575 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
117 : 9575 : lquery *query = PG_GETARG_LQUERY_P(1);
118 : 9575 : bool res = array_iterator(la, ltq_regex, query, NULL);
119 : :
120 [ + + ]: 9575 : PG_FREE_IF_COPY(la, 0);
121 [ - + ]: 9575 : PG_FREE_IF_COPY(query, 1);
122 : 9575 : PG_RETURN_BOOL(res);
123 : : }
124 : :
125 : : Datum
126 : 0 : _ltq_rregex(PG_FUNCTION_ARGS)
127 : : {
128 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(_ltq_regex,
129 : : PG_GETARG_DATUM(1),
130 : : PG_GETARG_DATUM(0)
131 : : ));
132 : : }
133 : :
134 : : Datum
135 : 1868 : _lt_q_regex(PG_FUNCTION_ARGS)
136 : : {
137 : 1868 : ArrayType *_tree = PG_GETARG_ARRAYTYPE_P(0);
138 : 1868 : ArrayType *_query = PG_GETARG_ARRAYTYPE_P(1);
139 [ - + ]: 1868 : lquery *query = (lquery *) ARR_DATA_PTR(_query);
140 : 1868 : bool res = false;
141 : 1868 : int num = ArrayGetNItems(ARR_NDIM(_query), ARR_DIMS(_query));
142 : :
143 [ - + ]: 1868 : if (ARR_NDIM(_query) > 1)
144 [ # # ]: 0 : ereport(ERROR,
145 : : (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
146 : : errmsg("array must be one-dimensional")));
147 [ - + ]: 1868 : if (array_contains_nulls(_query))
148 [ # # ]: 0 : ereport(ERROR,
149 : : (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
150 : : errmsg("array must not contain nulls")));
151 : :
152 [ + + ]: 5524 : while (num > 0)
153 : : {
154 [ + + ]: 3702 : if (array_iterator(_tree, ltq_regex, query, NULL))
155 : : {
156 : 46 : res = true;
157 : 46 : break;
158 : : }
159 : 3656 : num--;
160 : 3656 : query = (lquery *) NEXTVAL(query);
161 : : }
162 : :
163 [ + + ]: 1868 : PG_FREE_IF_COPY(_tree, 0);
164 [ - + ]: 1868 : PG_FREE_IF_COPY(_query, 1);
165 : 1868 : PG_RETURN_BOOL(res);
166 : : }
167 : :
168 : : Datum
169 : 0 : _lt_q_rregex(PG_FUNCTION_ARGS)
170 : : {
171 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(_lt_q_regex,
172 : : PG_GETARG_DATUM(1),
173 : : PG_GETARG_DATUM(0)
174 : : ));
175 : : }
176 : :
177 : :
178 : : Datum
179 : 1718 : _ltxtq_exec(PG_FUNCTION_ARGS)
180 : : {
181 : 1718 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
182 : 1718 : ltxtquery *query = PG_GETARG_LTXTQUERY_P(1);
183 : 1718 : bool res = array_iterator(la, ltxtq_exec, query, NULL);
184 : :
185 [ + + ]: 1718 : PG_FREE_IF_COPY(la, 0);
186 [ - + ]: 1718 : PG_FREE_IF_COPY(query, 1);
187 : 1718 : PG_RETURN_BOOL(res);
188 : : }
189 : :
190 : : Datum
191 : 0 : _ltxtq_rexec(PG_FUNCTION_ARGS)
192 : : {
193 : 0 : PG_RETURN_DATUM(DirectFunctionCall2(_ltxtq_exec,
194 : : PG_GETARG_DATUM(1),
195 : : PG_GETARG_DATUM(0)
196 : : ));
197 : : }
198 : :
199 : :
200 : : Datum
201 : 2 : _ltree_extract_isparent(PG_FUNCTION_ARGS)
202 : : {
203 : 2 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
204 : 2 : ltree *query = PG_GETARG_LTREE_P(1);
205 : : ltree *found,
206 : : *item;
207 : :
208 [ + + ]: 2 : if (!array_iterator(la, ltree_isparent, query, &found))
209 : : {
210 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
211 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
212 : 1 : PG_RETURN_NULL();
213 : : }
214 : :
215 : 1 : item = (ltree *) palloc0(VARSIZE(found));
216 : 1 : memcpy(item, found, VARSIZE(found));
217 : :
218 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
219 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
220 : 1 : PG_RETURN_POINTER(item);
221 : : }
222 : :
223 : : Datum
224 : 2 : _ltree_extract_risparent(PG_FUNCTION_ARGS)
225 : : {
226 : 2 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
227 : 2 : ltree *query = PG_GETARG_LTREE_P(1);
228 : : ltree *found,
229 : : *item;
230 : :
231 [ + + ]: 2 : if (!array_iterator(la, ltree_risparent, query, &found))
232 : : {
233 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
234 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
235 : 1 : PG_RETURN_NULL();
236 : : }
237 : :
238 : 1 : item = (ltree *) palloc0(VARSIZE(found));
239 : 1 : memcpy(item, found, VARSIZE(found));
240 : :
241 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
242 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
243 : 1 : PG_RETURN_POINTER(item);
244 : : }
245 : :
246 : : Datum
247 : 1 : _ltq_extract_regex(PG_FUNCTION_ARGS)
248 : : {
249 : 1 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
250 : 1 : lquery *query = PG_GETARG_LQUERY_P(1);
251 : : ltree *found,
252 : : *item;
253 : :
254 [ - + ]: 1 : if (!array_iterator(la, ltq_regex, query, &found))
255 : : {
256 [ # # ]: 0 : PG_FREE_IF_COPY(la, 0);
257 [ # # ]: 0 : PG_FREE_IF_COPY(query, 1);
258 : 0 : PG_RETURN_NULL();
259 : : }
260 : :
261 : 1 : item = (ltree *) palloc0(VARSIZE(found));
262 : 1 : memcpy(item, found, VARSIZE(found));
263 : :
264 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
265 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
266 : 1 : PG_RETURN_POINTER(item);
267 : : }
268 : :
269 : : Datum
270 : 1 : _ltxtq_extract_exec(PG_FUNCTION_ARGS)
271 : : {
272 : 1 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
273 : 1 : ltxtquery *query = PG_GETARG_LTXTQUERY_P(1);
274 : : ltree *found,
275 : : *item;
276 : :
277 [ - + ]: 1 : if (!array_iterator(la, ltxtq_exec, query, &found))
278 : : {
279 [ # # ]: 0 : PG_FREE_IF_COPY(la, 0);
280 [ # # ]: 0 : PG_FREE_IF_COPY(query, 1);
281 : 0 : PG_RETURN_NULL();
282 : : }
283 : :
284 : 1 : item = (ltree *) palloc0(VARSIZE(found));
285 : 1 : memcpy(item, found, VARSIZE(found));
286 : :
287 [ - + ]: 1 : PG_FREE_IF_COPY(la, 0);
288 [ - + ]: 1 : PG_FREE_IF_COPY(query, 1);
289 : 1 : PG_RETURN_POINTER(item);
290 : : }
291 : :
292 : : Datum
293 : 8 : _lca(PG_FUNCTION_ARGS)
294 : : {
295 : 8 : ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
296 : 8 : int num = ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la));
297 [ - + ]: 8 : ltree *item = (ltree *) ARR_DATA_PTR(la);
298 : : ltree **a,
299 : : *res;
300 : :
301 [ - + ]: 8 : if (ARR_NDIM(la) > 1)
302 [ # # ]: 0 : ereport(ERROR,
303 : : (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
304 : : errmsg("array must be one-dimensional")));
305 [ - + ]: 8 : if (array_contains_nulls(la))
306 [ # # ]: 0 : ereport(ERROR,
307 : : (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
308 : : errmsg("array must not contain nulls")));
309 : :
310 : 8 : a = palloc_array(ltree *, num);
311 [ + + ]: 20 : while (num > 0)
312 : : {
313 : 12 : num--;
314 : 12 : a[num] = item;
315 : 12 : item = NEXTVAL(item);
316 : : }
317 : 8 : res = lca_inner(a, ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la)));
318 : 8 : pfree(a);
319 : :
320 [ - + ]: 8 : PG_FREE_IF_COPY(la, 0);
321 : :
322 [ + + ]: 8 : if (res)
323 : 6 : PG_RETURN_POINTER(res);
324 : : else
325 : 2 : PG_RETURN_NULL();
326 : : }
|
