Line data Source code
1 : /*
2 : * contrib/intarray/_int_gin.c
3 : */
4 : #include "postgres.h"
5 :
6 : #include "_int.h"
7 : #include "access/gin.h"
8 : #include "access/stratnum.h"
9 :
10 2 : PG_FUNCTION_INFO_V1(ginint4_queryextract);
11 :
12 : Datum
13 28 : ginint4_queryextract(PG_FUNCTION_ARGS)
14 : {
15 28 : int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
16 28 : StrategyNumber strategy = PG_GETARG_UINT16(2);
17 28 : int32 *searchMode = (int32 *) PG_GETARG_POINTER(6);
18 28 : Datum *res = NULL;
19 :
20 28 : *nentries = 0;
21 :
22 28 : if (strategy == BooleanSearchStrategy)
23 : {
24 14 : QUERYTYPE *query = PG_GETARG_QUERYTYPE_P(0);
25 14 : ITEM *items = GETQUERY(query);
26 : int i;
27 :
28 : /* empty query must fail */
29 14 : if (query->size <= 0)
30 0 : PG_RETURN_POINTER(NULL);
31 :
32 : /*
33 : * If the query doesn't have any required primitive values (for
34 : * instance, it's something like '! 42'), we have to do a full index
35 : * scan.
36 : */
37 14 : if (query_has_required_values(query))
38 10 : *searchMode = GIN_SEARCH_MODE_DEFAULT;
39 : else
40 4 : *searchMode = GIN_SEARCH_MODE_ALL;
41 :
42 : /*
43 : * Extract all the VAL items as things we want GIN to check for.
44 : */
45 14 : res = palloc_array(Datum, query->size);
46 14 : *nentries = 0;
47 :
48 76 : for (i = 0; i < query->size; i++)
49 : {
50 62 : if (items[i].type == VAL)
51 : {
52 34 : res[*nentries] = Int32GetDatum(items[i].val);
53 34 : (*nentries)++;
54 : }
55 : }
56 : }
57 : else
58 : {
59 14 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(0);
60 :
61 14 : CHECKARRVALID(query);
62 14 : *nentries = ARRNELEMS(query);
63 14 : if (*nentries > 0)
64 : {
65 : int32 *arr;
66 : int32 i;
67 :
68 14 : res = palloc_array(Datum, *nentries);
69 :
70 14 : arr = ARRPTR(query);
71 46 : for (i = 0; i < *nentries; i++)
72 32 : res[i] = Int32GetDatum(arr[i]);
73 : }
74 :
75 14 : switch (strategy)
76 : {
77 2 : case RTOverlapStrategyNumber:
78 2 : *searchMode = GIN_SEARCH_MODE_DEFAULT;
79 2 : break;
80 2 : case RTContainedByStrategyNumber:
81 : case RTOldContainedByStrategyNumber:
82 : /* empty set is contained in everything */
83 2 : *searchMode = GIN_SEARCH_MODE_INCLUDE_EMPTY;
84 2 : break;
85 2 : case RTSameStrategyNumber:
86 2 : if (*nentries > 0)
87 2 : *searchMode = GIN_SEARCH_MODE_DEFAULT;
88 : else
89 0 : *searchMode = GIN_SEARCH_MODE_INCLUDE_EMPTY;
90 2 : break;
91 8 : case RTContainsStrategyNumber:
92 : case RTOldContainsStrategyNumber:
93 8 : if (*nentries > 0)
94 8 : *searchMode = GIN_SEARCH_MODE_DEFAULT;
95 : else /* everything contains the empty set */
96 0 : *searchMode = GIN_SEARCH_MODE_ALL;
97 8 : break;
98 0 : default:
99 0 : elog(ERROR, "ginint4_queryextract: unknown strategy number: %d",
100 : strategy);
101 : }
102 : }
103 :
104 28 : PG_RETURN_POINTER(res);
105 : }
106 :
107 2 : PG_FUNCTION_INFO_V1(ginint4_consistent);
108 :
109 : Datum
110 16976 : ginint4_consistent(PG_FUNCTION_ARGS)
111 : {
112 16976 : bool *check = (bool *) PG_GETARG_POINTER(0);
113 16976 : StrategyNumber strategy = PG_GETARG_UINT16(1);
114 16976 : int32 nkeys = PG_GETARG_INT32(3);
115 : #ifdef NOT_USED
116 : Pointer *extra_data = (Pointer *) PG_GETARG_POINTER(4);
117 : #endif
118 16976 : bool *recheck = (bool *) PG_GETARG_POINTER(5);
119 16976 : bool res = false;
120 : int32 i;
121 :
122 16976 : switch (strategy)
123 : {
124 405 : case RTOverlapStrategyNumber:
125 : /* result is not lossy */
126 405 : *recheck = false;
127 : /* at least one element in check[] is true, so result = true */
128 405 : res = true;
129 405 : break;
130 632 : case RTContainedByStrategyNumber:
131 : case RTOldContainedByStrategyNumber:
132 : /* we will need recheck */
133 632 : *recheck = true;
134 : /* at least one element in check[] is true, so result = true */
135 632 : res = true;
136 632 : break;
137 205 : case RTSameStrategyNumber:
138 : /* we will need recheck */
139 205 : *recheck = true;
140 : /* Must have all elements in check[] true */
141 205 : res = true;
142 413 : for (i = 0; i < nkeys; i++)
143 : {
144 412 : if (!check[i])
145 : {
146 204 : res = false;
147 204 : break;
148 : }
149 : }
150 205 : break;
151 815 : case RTContainsStrategyNumber:
152 : case RTOldContainsStrategyNumber:
153 : /* result is not lossy */
154 815 : *recheck = false;
155 : /* Must have all elements in check[] true */
156 815 : res = true;
157 909 : for (i = 0; i < nkeys; i++)
158 : {
159 864 : if (!check[i])
160 : {
161 770 : res = false;
162 770 : break;
163 : }
164 : }
165 815 : break;
166 14919 : case BooleanSearchStrategy:
167 : {
168 14919 : QUERYTYPE *query = PG_GETARG_QUERYTYPE_P(2);
169 :
170 : /* result is not lossy */
171 14919 : *recheck = false;
172 14919 : res = gin_bool_consistent(query, check);
173 : }
174 14919 : break;
175 0 : default:
176 0 : elog(ERROR, "ginint4_consistent: unknown strategy number: %d",
177 : strategy);
178 : }
179 :
180 16976 : PG_RETURN_BOOL(res);
181 : }
|
