PostgreSQL Source Code  git master
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros
_ltree_op.c
Go to the documentation of this file.
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 
14 PG_FUNCTION_INFO_V1(_ltree_isparent);
15 PG_FUNCTION_INFO_V1(_ltree_r_isparent);
16 PG_FUNCTION_INFO_V1(_ltree_risparent);
17 PG_FUNCTION_INFO_V1(_ltree_r_risparent);
18 PG_FUNCTION_INFO_V1(_ltq_regex);
19 PG_FUNCTION_INFO_V1(_ltq_rregex);
20 PG_FUNCTION_INFO_V1(_lt_q_regex);
21 PG_FUNCTION_INFO_V1(_lt_q_rregex);
22 PG_FUNCTION_INFO_V1(_ltxtq_exec);
23 PG_FUNCTION_INFO_V1(_ltxtq_rexec);
24 
25 PG_FUNCTION_INFO_V1(_ltree_extract_isparent);
26 PG_FUNCTION_INFO_V1(_ltree_extract_risparent);
27 PG_FUNCTION_INFO_V1(_ltq_extract_regex);
28 PG_FUNCTION_INFO_V1(_ltxtq_extract_exec);
29 
30 PG_FUNCTION_INFO_V1(_lca);
31 
32 typedef Datum (*PGCALL2) (PG_FUNCTION_ARGS);
33 
34 #define NEXTVAL(x) ( (ltree*)( (char*)(x) + INTALIGN( VARSIZE(x) ) ) )
35 
36 static bool
37 array_iterator(ArrayType *la, PGCALL2 callback, void *param, ltree **found)
38 {
39  int num = ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la));
40  ltree *item = (ltree *) ARR_DATA_PTR(la);
41 
42  if (ARR_NDIM(la) > 1)
43  ereport(ERROR,
44  (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
45  errmsg("array must be one-dimensional")));
46  if (array_contains_nulls(la))
47  ereport(ERROR,
48  (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
49  errmsg("array must not contain nulls")));
50 
51  if (found)
52  *found = NULL;
53  while (num > 0)
54  {
55  if (DatumGetBool(DirectFunctionCall2(callback,
56  PointerGetDatum(item), PointerGetDatum(param))))
57  {
58 
59  if (found)
60  *found = item;
61  return true;
62  }
63  num--;
64  item = NEXTVAL(item);
65  }
66 
67  return false;
68 }
69 
70 Datum
71 _ltree_isparent(PG_FUNCTION_ARGS)
72 {
73  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
74  ltree *query = PG_GETARG_LTREE(1);
75  bool res = array_iterator(la, ltree_isparent, (void *) query, NULL);
76 
77  PG_FREE_IF_COPY(la, 0);
78  PG_FREE_IF_COPY(query, 1);
79  PG_RETURN_BOOL(res);
80 }
81 
82 Datum
83 _ltree_r_isparent(PG_FUNCTION_ARGS)
84 {
85  PG_RETURN_DATUM(DirectFunctionCall2(_ltree_isparent,
86  PG_GETARG_DATUM(1),
87  PG_GETARG_DATUM(0)
88  ));
89 }
90 
91 Datum
92 _ltree_risparent(PG_FUNCTION_ARGS)
93 {
94  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
95  ltree *query = PG_GETARG_LTREE(1);
96  bool res = array_iterator(la, ltree_risparent, (void *) query, NULL);
97 
98  PG_FREE_IF_COPY(la, 0);
99  PG_FREE_IF_COPY(query, 1);
100  PG_RETURN_BOOL(res);
101 }
102 
103 Datum
104 _ltree_r_risparent(PG_FUNCTION_ARGS)
105 {
106  PG_RETURN_DATUM(DirectFunctionCall2(_ltree_risparent,
107  PG_GETARG_DATUM(1),
108  PG_GETARG_DATUM(0)
109  ));
110 }
111 
112 Datum
113 _ltq_regex(PG_FUNCTION_ARGS)
114 {
115  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
116  lquery *query = PG_GETARG_LQUERY(1);
117  bool res = array_iterator(la, ltq_regex, (void *) query, NULL);
118 
119  PG_FREE_IF_COPY(la, 0);
120  PG_FREE_IF_COPY(query, 1);
121  PG_RETURN_BOOL(res);
122 }
123 
124 Datum
125 _ltq_rregex(PG_FUNCTION_ARGS)
126 {
127  PG_RETURN_DATUM(DirectFunctionCall2(_ltq_regex,
128  PG_GETARG_DATUM(1),
129  PG_GETARG_DATUM(0)
130  ));
131 }
132 
133 Datum
134 _lt_q_regex(PG_FUNCTION_ARGS)
135 {
136  ArrayType *_tree = PG_GETARG_ARRAYTYPE_P(0);
137  ArrayType *_query = PG_GETARG_ARRAYTYPE_P(1);
138  lquery *query = (lquery *) ARR_DATA_PTR(_query);
139  bool res = false;
140  int num = ArrayGetNItems(ARR_NDIM(_query), ARR_DIMS(_query));
141 
142  if (ARR_NDIM(_query) > 1)
143  ereport(ERROR,
144  (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
145  errmsg("array must be one-dimensional")));
146  if (array_contains_nulls(_query))
147  ereport(ERROR,
148  (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
149  errmsg("array must not contain nulls")));
150 
151  while (num > 0)
152  {
153  if (array_iterator(_tree, ltq_regex, (void *) query, NULL))
154  {
155  res = true;
156  break;
157  }
158  num--;
159  query = (lquery *) NEXTVAL(query);
160  }
161 
162  PG_FREE_IF_COPY(_tree, 0);
163  PG_FREE_IF_COPY(_query, 1);
164  PG_RETURN_BOOL(res);
165 }
166 
167 Datum
168 _lt_q_rregex(PG_FUNCTION_ARGS)
169 {
170  PG_RETURN_DATUM(DirectFunctionCall2(_lt_q_regex,
171  PG_GETARG_DATUM(1),
172  PG_GETARG_DATUM(0)
173  ));
174 }
175 
176 
177 Datum
178 _ltxtq_exec(PG_FUNCTION_ARGS)
179 {
180  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
181  ltxtquery *query = PG_GETARG_LTXTQUERY(1);
182  bool res = array_iterator(la, ltxtq_exec, (void *) query, NULL);
183 
184  PG_FREE_IF_COPY(la, 0);
185  PG_FREE_IF_COPY(query, 1);
186  PG_RETURN_BOOL(res);
187 }
188 
189 Datum
190 _ltxtq_rexec(PG_FUNCTION_ARGS)
191 {
192  PG_RETURN_DATUM(DirectFunctionCall2(_ltxtq_exec,
193  PG_GETARG_DATUM(1),
194  PG_GETARG_DATUM(0)
195  ));
196 }
197 
198 
199 Datum
200 _ltree_extract_isparent(PG_FUNCTION_ARGS)
201 {
202  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
203  ltree *query = PG_GETARG_LTREE(1);
204  ltree *found,
205  *item;
206 
207  if (!array_iterator(la, ltree_isparent, (void *) query, &found))
208  {
209  PG_FREE_IF_COPY(la, 0);
210  PG_FREE_IF_COPY(query, 1);
211  PG_RETURN_NULL();
212  }
213 
214  item = (ltree *) palloc0(VARSIZE(found));
215  memcpy(item, found, VARSIZE(found));
216 
217  PG_FREE_IF_COPY(la, 0);
218  PG_FREE_IF_COPY(query, 1);
219  PG_RETURN_POINTER(item);
220 }
221 
222 Datum
223 _ltree_extract_risparent(PG_FUNCTION_ARGS)
224 {
225  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
226  ltree *query = PG_GETARG_LTREE(1);
227  ltree *found,
228  *item;
229 
230  if (!array_iterator(la, ltree_risparent, (void *) query, &found))
231  {
232  PG_FREE_IF_COPY(la, 0);
233  PG_FREE_IF_COPY(query, 1);
234  PG_RETURN_NULL();
235  }
236 
237  item = (ltree *) palloc0(VARSIZE(found));
238  memcpy(item, found, VARSIZE(found));
239 
240  PG_FREE_IF_COPY(la, 0);
241  PG_FREE_IF_COPY(query, 1);
242  PG_RETURN_POINTER(item);
243 }
244 
245 Datum
246 _ltq_extract_regex(PG_FUNCTION_ARGS)
247 {
248  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
249  lquery *query = PG_GETARG_LQUERY(1);
250  ltree *found,
251  *item;
252 
253  if (!array_iterator(la, ltq_regex, (void *) query, &found))
254  {
255  PG_FREE_IF_COPY(la, 0);
256  PG_FREE_IF_COPY(query, 1);
257  PG_RETURN_NULL();
258  }
259 
260  item = (ltree *) palloc0(VARSIZE(found));
261  memcpy(item, found, VARSIZE(found));
262 
263  PG_FREE_IF_COPY(la, 0);
264  PG_FREE_IF_COPY(query, 1);
265  PG_RETURN_POINTER(item);
266 }
267 
268 Datum
269 _ltxtq_extract_exec(PG_FUNCTION_ARGS)
270 {
271  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
272  ltxtquery *query = PG_GETARG_LTXTQUERY(1);
273  ltree *found,
274  *item;
275 
276  if (!array_iterator(la, ltxtq_exec, (void *) query, &found))
277  {
278  PG_FREE_IF_COPY(la, 0);
279  PG_FREE_IF_COPY(query, 1);
280  PG_RETURN_NULL();
281  }
282 
283  item = (ltree *) palloc0(VARSIZE(found));
284  memcpy(item, found, VARSIZE(found));
285 
286  PG_FREE_IF_COPY(la, 0);
287  PG_FREE_IF_COPY(query, 1);
288  PG_RETURN_POINTER(item);
289 }
290 
291 Datum
292 _lca(PG_FUNCTION_ARGS)
293 {
294  ArrayType *la = PG_GETARG_ARRAYTYPE_P(0);
295  int num = ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la));
296  ltree *item = (ltree *) ARR_DATA_PTR(la);
297  ltree **a,
298  *res;
299 
300  if (ARR_NDIM(la) > 1)
301  ereport(ERROR,
302  (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
303  errmsg("array must be one-dimensional")));
304  if (array_contains_nulls(la))
305  ereport(ERROR,
306  (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
307  errmsg("array must not contain nulls")));
308 
309  a = (ltree **) palloc(sizeof(ltree *) * num);
310  while (num > 0)
311  {
312  num--;
313  a[num] = item;
314  item = NEXTVAL(item);
315  }
316  res = lca_inner(a, ArrayGetNItems(ARR_NDIM(la), ARR_DIMS(la)));
317  pfree(a);
318 
319  PG_FREE_IF_COPY(la, 0);
320 
321  if (res)
322  PG_RETURN_POINTER(res);
323  else
324  PG_RETURN_NULL();
325 }
_lt_q_rregex
Datum _lt_q_rregex(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:168
PG_RETURN_POINTER
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:321
_ltxtq_exec
Datum _ltxtq_exec(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:178
_ltree_r_risparent
Datum _ltree_r_risparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:104
PG_GETARG_LTXTQUERY
#define PG_GETARG_LTXTQUERY(x)
Definition: ltree.h:172
VARSIZE
#define VARSIZE(PTR)
Definition: postgres.h:304
PointerGetDatum
#define PointerGetDatum(X)
Definition: postgres.h:562
PG_GETARG_DATUM
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:233
lquery
Definition: ltree.h:69
ArrayGetNItems
int ArrayGetNItems(int ndim, const int *dims)
Definition: arrayutils.c:75
ltree_risparent
Datum ltree_risparent(PG_FUNCTION_ARGS)
Definition: ltree_op.c:172
errcode
int errcode(int sqlerrcode)
Definition: elog.c:575
NEXTVAL
#define NEXTVAL(x)
Definition: _ltree_op.c:34
_ltq_extract_regex
Datum _ltq_extract_regex(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:246
_lt_q_regex
Datum _lt_q_regex(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:134
_ltree_isparent
Datum _ltree_isparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:71
_ltree_extract_isparent
Datum _ltree_extract_isparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:200
PG_GETARG_LTREE
#define PG_GETARG_LTREE(x)
Definition: ltree.h:168
PG_FUNCTION_INFO_V1
PG_FUNCTION_INFO_V1(_ltree_isparent)
PG_GETARG_ARRAYTYPE_P
#define PG_GETARG_ARRAYTYPE_P(n)
Definition: array.h:244
_ltq_regex
Datum _ltq_regex(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:113
pfree
void pfree(void *pointer)
Definition: mcxt.c:950
ERROR
#define ERROR
Definition: elog.h:43
lca_inner
ltree * lca_inner(ltree **a, int len)
Definition: ltree_op.c:406
ARR_DIMS
#define ARR_DIMS(a)
Definition: array.h:275
ltq_regex
Datum ltq_regex(PG_FUNCTION_ARGS)
Definition: lquery_op.c:303
callback
static void callback(struct sockaddr *addr, struct sockaddr *mask, void *unused)
Definition: test_ifaddrs.c:48
ARR_DATA_PTR
#define ARR_DATA_PTR(a)
Definition: array.h:303
array_iterator
static bool array_iterator(ArrayType *la, PGCALL2 callback, void *param, ltree **found)
Definition: _ltree_op.c:37
PG_GETARG_LQUERY
#define PG_GETARG_LQUERY(x)
Definition: ltree.h:170
PGCALL2
Datum(* PGCALL2)(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:32
DatumGetBool
#define DatumGetBool(X)
Definition: postgres.h:399
_ltxtq_rexec
Datum _ltxtq_rexec(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:190
_ltree_extract_risparent
Datum _ltree_extract_risparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:223
ereport
#define ereport(elevel, rest)
Definition: elog.h:122
ltree
Definition: ltree.h:19
ltree_isparent
Datum ltree_isparent(PG_FUNCTION_ARGS)
Definition: ltree_op.c:160
palloc0
void * palloc0(Size size)
Definition: mcxt.c:878
PG_RETURN_BOOL
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:319
Datum
uintptr_t Datum
Definition: postgres.h:372
PG_RETURN_DATUM
#define PG_RETURN_DATUM(x)
Definition: fmgr.h:313
_ltxtq_extract_exec
Datum _ltxtq_extract_exec(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:269
NULL
#define NULL
Definition: c.h:229
ltxtq_exec
Datum ltxtq_exec(PG_FUNCTION_ARGS)
Definition: ltxtquery_op.c:87
_ltree_r_isparent
Datum _ltree_r_isparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:83
PG_FREE_IF_COPY
#define PG_FREE_IF_COPY(ptr, n)
Definition: fmgr.h:225
ARR_NDIM
#define ARR_NDIM(a)
Definition: array.h:271
palloc
void * palloc(Size size)
Definition: mcxt.c:849
errmsg
int errmsg(const char *fmt,...)
Definition: elog.c:797
_ltree_risparent
Datum _ltree_risparent(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:92
_ltq_rregex
Datum _ltq_rregex(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:125
PG_FUNCTION_ARGS
#define PG_FUNCTION_ARGS
Definition: fmgr.h:158
array_contains_nulls
bool array_contains_nulls(ArrayType *array)
Definition: arrayfuncs.c:3542
_lca
Datum _lca(PG_FUNCTION_ARGS)
Definition: _ltree_op.c:292
DirectFunctionCall2
#define DirectFunctionCall2(func, arg1, arg2)
Definition: fmgr.h:586
PG_RETURN_NULL
#define PG_RETURN_NULL()
Definition: fmgr.h:305