Age Owner Branch data TLA 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 : :
7256 teodor@sigaev.ru 10 :CBC 2 : PG_FUNCTION_INFO_V1(ginint4_queryextract);
11 : :
12 : : Datum
7102 bruce@momjian.us 13 : 28 : ginint4_queryextract(PG_FUNCTION_ARGS)
14 : : {
6983 teodor@sigaev.ru 15 : 28 : int32 *nentries = (int32 *) PG_GETARG_POINTER(1);
7102 bruce@momjian.us 16 : 28 : StrategyNumber strategy = PG_GETARG_UINT16(2);
5544 tgl@sss.pgh.pa.us 17 : 28 : int32 *searchMode = (int32 *) PG_GETARG_POINTER(6);
7102 bruce@momjian.us 18 : 28 : Datum *res = NULL;
19 : :
7256 teodor@sigaev.ru 20 : 28 : *nentries = 0;
21 : :
7102 bruce@momjian.us 22 [ + + ]: 28 : if (strategy == BooleanSearchStrategy)
23 : : {
5544 tgl@sss.pgh.pa.us 24 : 14 : QUERYTYPE *query = PG_GETARG_QUERYTYPE_P(0);
7102 bruce@momjian.us 25 : 14 : ITEM *items = GETQUERY(query);
26 : : int i;
27 : :
28 : : /* empty query must fail */
5544 tgl@sss.pgh.pa.us 29 [ - + ]: 14 : if (query->size <= 0)
7256 teodor@sigaev.ru 30 :UBC 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 : : */
5544 tgl@sss.pgh.pa.us 37 [ + + ]:CBC 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 : : */
100 michael@paquier.xyz 45 :GNC 14 : res = palloc_array(Datum, query->size);
7256 teodor@sigaev.ru 46 :CBC 14 : *nentries = 0;
47 : :
7102 bruce@momjian.us 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);
7256 teodor@sigaev.ru 53 : 34 : (*nentries)++;
54 : : }
55 : : }
56 : : }
57 : : else
58 : : {
7102 bruce@momjian.us 59 : 14 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(0);
60 : :
7256 teodor@sigaev.ru 61 [ - + - - : 14 : CHECKARRVALID(query);
- - ]
7102 bruce@momjian.us 62 : 14 : *nentries = ARRNELEMS(query);
63 [ + - ]: 14 : if (*nentries > 0)
64 : : {
65 : : int32 *arr;
66 : : int32 i;
67 : :
100 michael@paquier.xyz 68 :GNC 14 : res = palloc_array(Datum, *nentries);
69 : :
7102 bruce@momjian.us 70 [ - + ]:CBC 14 : arr = ARRPTR(query);
71 [ + + ]: 46 : for (i = 0; i < *nentries; i++)
72 : 32 : res[i] = Int32GetDatum(arr[i]);
73 : : }
74 : :
6695 75 [ + + + + : 14 : switch (strategy)
- ]
76 : : {
6983 teodor@sigaev.ru 77 : 2 : case RTOverlapStrategyNumber:
5544 tgl@sss.pgh.pa.us 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;
6695 bruce@momjian.us 84 : 2 : break;
5544 tgl@sss.pgh.pa.us 85 : 2 : case RTSameStrategyNumber:
86 [ + - ]: 2 : if (*nentries > 0)
87 : 2 : *searchMode = GIN_SEARCH_MODE_DEFAULT;
88 : : else
5544 tgl@sss.pgh.pa.us 89 :UBC 0 : *searchMode = GIN_SEARCH_MODE_INCLUDE_EMPTY;
6695 bruce@momjian.us 90 :CBC 2 : break;
5544 tgl@sss.pgh.pa.us 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 */
5544 tgl@sss.pgh.pa.us 96 :UBC 0 : *searchMode = GIN_SEARCH_MODE_ALL;
5544 tgl@sss.pgh.pa.us 97 :CBC 8 : break;
5544 tgl@sss.pgh.pa.us 98 :UBC 0 : default:
99 [ # # ]: 0 : elog(ERROR, "ginint4_queryextract: unknown strategy number: %d",
100 : : strategy);
101 : : }
102 : : }
103 : :
7102 bruce@momjian.us 104 :CBC 28 : PG_RETURN_POINTER(res);
105 : : }
106 : :
7256 teodor@sigaev.ru 107 : 2 : PG_FUNCTION_INFO_V1(ginint4_consistent);
108 : :
109 : : Datum
7102 bruce@momjian.us 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);
5544 tgl@sss.pgh.pa.us 114 : 16976 : int32 nkeys = PG_GETARG_INT32(3);
115 : : #ifdef NOT_USED
116 : : Pointer *extra_data = (Pointer *) PG_GETARG_POINTER(4);
117 : : #endif
6199 118 : 16976 : bool *recheck = (bool *) PG_GETARG_POINTER(5);
3133 peter_e@gmx.net 119 : 16976 : bool res = false;
120 : : int32 i;
121 : :
7102 bruce@momjian.us 122 [ + + + + : 16976 : switch (strategy)
+ - ]
123 : : {
124 : 405 : case RTOverlapStrategyNumber:
125 : : /* result is not lossy */
6544 tgl@sss.pgh.pa.us 126 : 405 : *recheck = false;
127 : : /* at least one element in check[] is true, so result = true */
3133 peter_e@gmx.net 128 : 405 : res = true;
6544 tgl@sss.pgh.pa.us 129 : 405 : break;
7102 bruce@momjian.us 130 : 632 : case RTContainedByStrategyNumber:
131 : : case RTOldContainedByStrategyNumber:
132 : : /* we will need recheck */
6544 tgl@sss.pgh.pa.us 133 : 632 : *recheck = true;
134 : : /* at least one element in check[] is true, so result = true */
3133 peter_e@gmx.net 135 : 632 : res = true;
7102 bruce@momjian.us 136 : 632 : break;
137 : 205 : case RTSameStrategyNumber:
138 : : /* we will need recheck */
5544 tgl@sss.pgh.pa.us 139 : 205 : *recheck = true;
140 : : /* Must have all elements in check[] true */
3133 peter_e@gmx.net 141 : 205 : res = true;
5544 tgl@sss.pgh.pa.us 142 [ + + ]: 413 : for (i = 0; i < nkeys; i++)
143 : : {
144 [ + + ]: 412 : if (!check[i])
145 : : {
3133 peter_e@gmx.net 146 : 204 : res = false;
5544 tgl@sss.pgh.pa.us 147 : 204 : break;
148 : : }
149 : : }
6544 150 : 205 : break;
7102 bruce@momjian.us 151 : 815 : case RTContainsStrategyNumber:
152 : : case RTOldContainsStrategyNumber:
153 : : /* result is not lossy */
5544 tgl@sss.pgh.pa.us 154 : 815 : *recheck = false;
155 : : /* Must have all elements in check[] true */
3133 peter_e@gmx.net 156 : 815 : res = true;
5544 tgl@sss.pgh.pa.us 157 [ + + ]: 909 : for (i = 0; i < nkeys; i++)
158 : : {
159 [ + + ]: 864 : if (!check[i])
160 : : {
3133 peter_e@gmx.net 161 : 770 : res = false;
5544 tgl@sss.pgh.pa.us 162 : 770 : break;
163 : : }
164 : : }
7256 teodor@sigaev.ru 165 : 815 : break;
7102 bruce@momjian.us 166 : 14919 : case BooleanSearchStrategy:
167 : : {
5544 tgl@sss.pgh.pa.us 168 : 14919 : QUERYTYPE *query = PG_GETARG_QUERYTYPE_P(2);
169 : :
170 : : /* result is not lossy */
6544 171 : 14919 : *recheck = false;
5544 172 : 14919 : res = gin_bool_consistent(query, check);
173 : : }
7102 bruce@momjian.us 174 : 14919 : break;
7102 bruce@momjian.us 175 :UBC 0 : default:
6544 tgl@sss.pgh.pa.us 176 [ # # ]: 0 : elog(ERROR, "ginint4_consistent: unknown strategy number: %d",
177 : : strategy);
178 : : }
179 : :
7102 bruce@momjian.us 180 :CBC 16976 : PG_RETURN_BOOL(res);
181 : : }
|
