Age Owner Branch data TLA Line data Source code
1 : : /*
2 : : * contrib/hstore/hstore_gist.c
3 : : */
4 : : #include "postgres.h"
5 : :
6 : : #include "access/gist.h"
7 : : #include "access/reloptions.h"
8 : : #include "access/stratnum.h"
9 : : #include "catalog/pg_type.h"
10 : : #include "common/int.h"
11 : : #include "hstore.h"
12 : : #include "utils/pg_crc.h"
13 : :
14 : : /* gist_hstore_ops opclass options */
15 : : typedef struct
16 : : {
17 : : int32 vl_len_; /* varlena header (do not touch directly!) */
18 : : int siglen; /* signature length in bytes */
19 : : } GistHstoreOptions;
20 : :
21 : : /* bigint defines */
22 : : #define BITBYTE 8
23 : : #define SIGLEN_DEFAULT (sizeof(int32) * 4)
24 : : #define SIGLEN_MAX GISTMaxIndexKeySize
25 : : #define SIGLENBIT(siglen) ((siglen) * BITBYTE)
26 : : #define GET_SIGLEN() (PG_HAS_OPCLASS_OPTIONS() ? \
27 : : ((GistHstoreOptions *) PG_GET_OPCLASS_OPTIONS())->siglen : \
28 : : SIGLEN_DEFAULT)
29 : :
30 : :
31 : : typedef char *BITVECP;
32 : :
33 : : #define LOOPBYTE(siglen) \
34 : : for (i = 0; i < (siglen); i++)
35 : :
36 : : #define LOOPBIT(siglen) \
37 : : for (i = 0; i < SIGLENBIT(siglen); i++)
38 : :
39 : : /* beware of multiple evaluation of arguments to these macros! */
40 : : #define GETBYTE(x,i) ( *( (BITVECP)(x) + (int)( (i) / BITBYTE ) ) )
41 : : #define GETBITBYTE(x,i) ( (*((char*)(x)) >> (i)) & 0x01 )
42 : : #define CLRBIT(x,i) GETBYTE(x,i) &= ~( 0x01 << ( (i) % BITBYTE ) )
43 : : #define SETBIT(x,i) GETBYTE(x,i) |= ( 0x01 << ( (i) % BITBYTE ) )
44 : : #define GETBIT(x,i) ( (GETBYTE(x,i) >> ( (i) % BITBYTE )) & 0x01 )
45 : : #define HASHVAL(val, siglen) (((unsigned int)(val)) % SIGLENBIT(siglen))
46 : : #define HASH(sign, val, siglen) SETBIT((sign), HASHVAL(val, siglen))
47 : :
48 : : typedef struct
49 : : {
50 : : int32 vl_len_; /* varlena header (do not touch directly!) */
51 : : int32 flag;
52 : : char data[FLEXIBLE_ARRAY_MEMBER];
53 : : } GISTTYPE;
54 : :
55 : : #define ALLISTRUE 0x04
56 : :
57 : : #define ISALLTRUE(x) ( ((GISTTYPE*)x)->flag & ALLISTRUE )
58 : :
59 : : #define GTHDRSIZE (VARHDRSZ + sizeof(int32))
60 : : #define CALCGTSIZE(flag, siglen) ( GTHDRSIZE+(((flag) & ALLISTRUE) ? 0 : (siglen)) )
61 : :
62 : : #define GETSIGN(x) ( (BITVECP)( (char*)x+GTHDRSIZE ) )
63 : :
64 : : #define SUMBIT(val) ( \
65 : : GETBITBYTE((val),0) + \
66 : : GETBITBYTE((val),1) + \
67 : : GETBITBYTE((val),2) + \
68 : : GETBITBYTE((val),3) + \
69 : : GETBITBYTE((val),4) + \
70 : : GETBITBYTE((val),5) + \
71 : : GETBITBYTE((val),6) + \
72 : : GETBITBYTE((val),7) \
73 : : )
74 : :
75 : : #define GETENTRY(vec,pos) ((GISTTYPE *) DatumGetPointer((vec)->vector[(pos)].key))
76 : :
77 : : #define WISH_F(a,b,c) (double)( -(double)(((a)-(b))*((a)-(b))*((a)-(b)))*(c) )
78 : :
79 : : /* shorthand for calculating CRC-32 of a single chunk of data. */
80 : : static pg_crc32
892 peter@eisentraut.org 81 :CBC 33069 : crc32_sz(const char *buf, int size)
82 : : {
83 : : pg_crc32 crc;
84 : :
4149 heikki.linnakangas@i 85 : 33069 : INIT_TRADITIONAL_CRC32(crc);
86 [ + + ]: 176340 : COMP_TRADITIONAL_CRC32(crc, buf, size);
87 : 33069 : FIN_TRADITIONAL_CRC32(crc);
88 : :
89 : 33069 : return crc;
90 : : }
91 : :
92 : :
7131 teodor@sigaev.ru 93 : 7 : PG_FUNCTION_INFO_V1(ghstore_in);
94 : 7 : PG_FUNCTION_INFO_V1(ghstore_out);
95 : :
96 : :
97 : : Datum
7102 bruce@momjian.us 98 :UBC 0 : ghstore_in(PG_FUNCTION_ARGS)
99 : : {
1191 tgl@sss.pgh.pa.us 100 [ # # ]: 0 : ereport(ERROR,
101 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
102 : : errmsg("cannot accept a value of type %s", "ghstore")));
103 : :
104 : : PG_RETURN_VOID(); /* keep compiler quiet */
105 : : }
106 : :
107 : : Datum
7102 bruce@momjian.us 108 : 0 : ghstore_out(PG_FUNCTION_ARGS)
109 : : {
1191 tgl@sss.pgh.pa.us 110 [ # # ]: 0 : ereport(ERROR,
111 : : (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
112 : : errmsg("cannot display a value of type %s", "ghstore")));
113 : :
114 : : PG_RETURN_VOID(); /* keep compiler quiet */
115 : : }
116 : :
117 : : static GISTTYPE *
2176 akorotkov@postgresql 118 :CBC 9772 : ghstore_alloc(bool allistrue, int siglen, BITVECP sign)
119 : : {
120 [ - + ]: 9772 : int flag = allistrue ? ALLISTRUE : 0;
121 [ + - ]: 9772 : int size = CALCGTSIZE(flag, siglen);
122 : 9772 : GISTTYPE *res = palloc(size);
123 : :
124 : 9772 : SET_VARSIZE(res, size);
125 : 9772 : res->flag = flag;
126 : :
127 [ + - ]: 9772 : if (!allistrue)
128 : : {
129 [ + + ]: 9772 : if (sign)
130 : 1440 : memcpy(GETSIGN(res), sign, siglen);
131 : : else
132 : 8332 : memset(GETSIGN(res), 0, siglen);
133 : : }
134 : :
135 : 9772 : return res;
136 : : }
137 : :
7131 teodor@sigaev.ru 138 : 8 : PG_FUNCTION_INFO_V1(ghstore_consistent);
139 : 8 : PG_FUNCTION_INFO_V1(ghstore_compress);
140 : 8 : PG_FUNCTION_INFO_V1(ghstore_decompress);
141 : 8 : PG_FUNCTION_INFO_V1(ghstore_penalty);
142 : 8 : PG_FUNCTION_INFO_V1(ghstore_picksplit);
143 : 8 : PG_FUNCTION_INFO_V1(ghstore_union);
144 : 8 : PG_FUNCTION_INFO_V1(ghstore_same);
2176 akorotkov@postgresql 145 : 8 : PG_FUNCTION_INFO_V1(ghstore_options);
146 : :
147 : : Datum
7102 bruce@momjian.us 148 : 8414 : ghstore_compress(PG_FUNCTION_ARGS)
149 : : {
7131 teodor@sigaev.ru 150 : 8414 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
2176 akorotkov@postgresql 151 [ + - ]: 8414 : int siglen = GET_SIGLEN();
7131 teodor@sigaev.ru 152 : 8414 : GISTENTRY *retval = entry;
153 : :
7102 bruce@momjian.us 154 [ + + ]: 8414 : if (entry->leafkey)
155 : : {
2176 akorotkov@postgresql 156 : 2002 : GISTTYPE *res = ghstore_alloc(false, siglen, NULL);
6010 tgl@sss.pgh.pa.us 157 : 2002 : HStore *val = DatumGetHStoreP(entry->key);
158 : 2002 : HEntry *hsent = ARRPTR(val);
159 : 2002 : char *ptr = STRPTR(val);
5861 bruce@momjian.us 160 : 2002 : int count = HS_COUNT(val);
161 : : int i;
162 : :
6010 tgl@sss.pgh.pa.us 163 [ + + ]: 11564 : for (i = 0; i < count; ++i)
164 : : {
165 : : int h;
166 : :
3769 167 [ + + ]: 9562 : h = crc32_sz((char *) HSTORE_KEY(hsent, ptr, i),
168 [ + + ]: 9562 : HSTORE_KEYLEN(hsent, i));
2176 akorotkov@postgresql 169 : 9562 : HASH(GETSIGN(res), h, siglen);
3769 tgl@sss.pgh.pa.us 170 [ + + ]: 9562 : if (!HSTORE_VALISNULL(hsent, i))
171 : : {
172 [ + - ]: 9560 : h = crc32_sz((char *) HSTORE_VAL(hsent, ptr, i),
173 [ - + ]: 9560 : HSTORE_VALLEN(hsent, i));
2176 akorotkov@postgresql 174 : 9560 : HASH(GETSIGN(res), h, siglen);
175 : : }
176 : : }
177 : :
100 michael@paquier.xyz 178 :GNC 2002 : retval = palloc_object(GISTENTRY);
7131 teodor@sigaev.ru 179 :CBC 2002 : gistentryinit(*retval, PointerGetDatum(res),
180 : : entry->rel, entry->page,
181 : : entry->offset,
182 : : false);
183 : : }
7102 bruce@momjian.us 184 [ + - ]: 6412 : else if (!ISALLTRUE(DatumGetPointer(entry->key)))
185 : : {
186 : : int32 i;
187 : : GISTTYPE *res;
188 : 6412 : BITVECP sign = GETSIGN(DatumGetPointer(entry->key));
189 : :
2176 akorotkov@postgresql 190 [ + - ]: 6412 : LOOPBYTE(siglen)
191 : : {
6694 bruce@momjian.us 192 [ + - ]: 6412 : if ((sign[i] & 0xff) != 0xff)
193 : 6412 : PG_RETURN_POINTER(retval);
194 : : }
195 : :
2176 akorotkov@postgresql 196 :UBC 0 : res = ghstore_alloc(true, siglen, NULL);
197 : :
100 michael@paquier.xyz 198 :UNC 0 : retval = palloc_object(GISTENTRY);
7131 teodor@sigaev.ru 199 :UBC 0 : gistentryinit(*retval, PointerGetDatum(res),
200 : : entry->rel, entry->page,
201 : : entry->offset,
202 : : false);
203 : : }
204 : :
7131 teodor@sigaev.ru 205 :CBC 2002 : PG_RETURN_POINTER(retval);
206 : : }
207 : :
208 : : /*
209 : : * Since type ghstore isn't toastable (and doesn't need to be),
210 : : * this function can be a no-op.
211 : : */
212 : : Datum
7102 bruce@momjian.us 213 : 49358 : ghstore_decompress(PG_FUNCTION_ARGS)
214 : : {
5659 tgl@sss.pgh.pa.us 215 : 49358 : PG_RETURN_POINTER(PG_GETARG_POINTER(0));
216 : : }
217 : :
218 : : Datum
7102 bruce@momjian.us 219 : 6330 : ghstore_same(PG_FUNCTION_ARGS)
220 : : {
7131 teodor@sigaev.ru 221 : 6330 : GISTTYPE *a = (GISTTYPE *) PG_GETARG_POINTER(0);
222 : 6330 : GISTTYPE *b = (GISTTYPE *) PG_GETARG_POINTER(1);
7102 bruce@momjian.us 223 : 6330 : bool *result = (bool *) PG_GETARG_POINTER(2);
2176 akorotkov@postgresql 224 [ + - ]: 6330 : int siglen = GET_SIGLEN();
225 : :
226 : :
7131 teodor@sigaev.ru 227 [ - + - - ]: 6330 : if (ISALLTRUE(a) && ISALLTRUE(b))
7131 teodor@sigaev.ru 228 :UBC 0 : *result = true;
7131 teodor@sigaev.ru 229 [ - + ]:CBC 6330 : else if (ISALLTRUE(a))
7131 teodor@sigaev.ru 230 :UBC 0 : *result = false;
7131 teodor@sigaev.ru 231 [ - + ]:CBC 6330 : else if (ISALLTRUE(b))
7131 teodor@sigaev.ru 232 :UBC 0 : *result = false;
233 : : else
234 : : {
235 : : int32 i;
7102 bruce@momjian.us 236 :CBC 6330 : BITVECP sa = GETSIGN(a),
237 : 6330 : sb = GETSIGN(b);
238 : :
7131 teodor@sigaev.ru 239 : 6330 : *result = true;
2176 akorotkov@postgresql 240 [ + + ]: 4732759 : LOOPBYTE(siglen)
241 : : {
6694 bruce@momjian.us 242 [ + + ]: 4731401 : if (sa[i] != sb[i])
243 : : {
244 : 4972 : *result = false;
245 : 4972 : break;
246 : : }
247 : : }
248 : : }
7131 teodor@sigaev.ru 249 : 6330 : PG_RETURN_POINTER(result);
250 : : }
251 : :
252 : : static int32
2176 akorotkov@postgresql 253 :UBC 0 : sizebitvec(BITVECP sign, int siglen)
254 : : {
5011 peter_e@gmx.net 255 : 0 : int32 size = 0,
256 : : i;
257 : :
2176 akorotkov@postgresql 258 [ # # ]: 0 : LOOPBYTE(siglen)
259 : : {
6694 bruce@momjian.us 260 : 0 : size += SUMBIT(sign);
261 : 0 : sign = (BITVECP) (((char *) sign) + 1);
262 : : }
7131 teodor@sigaev.ru 263 : 0 : return size;
264 : : }
265 : :
266 : : static int
2176 akorotkov@postgresql 267 :CBC 133266 : hemdistsign(BITVECP a, BITVECP b, int siglen)
268 : : {
269 : : int i,
7102 bruce@momjian.us 270 : 133266 : dist = 0;
271 : :
2176 akorotkov@postgresql 272 [ + + ]: 400189202 : LOOPBIT(siglen)
273 : : {
6694 bruce@momjian.us 274 [ + + ]: 400055936 : if (GETBIT(a, i) != GETBIT(b, i))
7131 teodor@sigaev.ru 275 : 3435598 : dist++;
276 : : }
277 : 133266 : return dist;
278 : : }
279 : :
280 : : static int
2176 akorotkov@postgresql 281 : 133266 : hemdist(GISTTYPE *a, GISTTYPE *b, int siglen)
282 : : {
7102 bruce@momjian.us 283 [ - + ]: 133266 : if (ISALLTRUE(a))
284 : : {
7131 teodor@sigaev.ru 285 [ # # ]:UBC 0 : if (ISALLTRUE(b))
286 : 0 : return 0;
287 : : else
2176 akorotkov@postgresql 288 : 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(b), siglen);
289 : : }
7102 bruce@momjian.us 290 [ - + ]:CBC 133266 : else if (ISALLTRUE(b))
2176 akorotkov@postgresql 291 :UBC 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(a), siglen);
292 : :
2176 akorotkov@postgresql 293 :CBC 133266 : return hemdistsign(GETSIGN(a), GETSIGN(b), siglen);
294 : : }
295 : :
296 : : static int32
297 : 12660 : unionkey(BITVECP sbase, GISTTYPE *add, int siglen)
298 : : {
299 : : int32 i;
7102 bruce@momjian.us 300 : 12660 : BITVECP sadd = GETSIGN(add);
301 : :
7131 teodor@sigaev.ru 302 [ - + ]: 12660 : if (ISALLTRUE(add))
7131 teodor@sigaev.ru 303 :UBC 0 : return 1;
2176 akorotkov@postgresql 304 [ + + ]:CBC 22435748 : LOOPBYTE(siglen)
6694 bruce@momjian.us 305 : 22423088 : sbase[i] |= sadd[i];
7131 teodor@sigaev.ru 306 : 12660 : return 0;
307 : : }
308 : :
309 : : Datum
7102 bruce@momjian.us 310 : 6330 : ghstore_union(PG_FUNCTION_ARGS)
311 : : {
312 : 6330 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
5011 peter_e@gmx.net 313 : 6330 : int32 len = entryvec->n;
314 : :
7102 bruce@momjian.us 315 : 6330 : int *size = (int *) PG_GETARG_POINTER(1);
2176 akorotkov@postgresql 316 [ + - ]: 6330 : int siglen = GET_SIGLEN();
317 : : int32 i;
318 : 6330 : GISTTYPE *result = ghstore_alloc(false, siglen, NULL);
319 : 6330 : BITVECP base = GETSIGN(result);
320 : :
7102 bruce@momjian.us 321 [ + + ]: 18990 : for (i = 0; i < len; i++)
322 : : {
2176 akorotkov@postgresql 323 [ - + ]: 12660 : if (unionkey(base, GETENTRY(entryvec, i), siglen))
324 : : {
2176 akorotkov@postgresql 325 :UBC 0 : result->flag |= ALLISTRUE;
326 : 0 : SET_VARSIZE(result, CALCGTSIZE(ALLISTRUE, siglen));
7131 teodor@sigaev.ru 327 : 0 : break;
328 : : }
329 : : }
330 : :
2176 akorotkov@postgresql 331 :CBC 6330 : *size = VARSIZE(result);
332 : :
7131 teodor@sigaev.ru 333 : 6330 : PG_RETURN_POINTER(result);
334 : : }
335 : :
336 : : Datum
7102 bruce@momjian.us 337 : 15966 : ghstore_penalty(PG_FUNCTION_ARGS)
338 : : {
7131 teodor@sigaev.ru 339 : 15966 : GISTENTRY *origentry = (GISTENTRY *) PG_GETARG_POINTER(0); /* always ISSIGNKEY */
340 : 15966 : GISTENTRY *newentry = (GISTENTRY *) PG_GETARG_POINTER(1);
7102 bruce@momjian.us 341 : 15966 : float *penalty = (float *) PG_GETARG_POINTER(2);
2176 akorotkov@postgresql 342 [ + - ]: 15966 : int siglen = GET_SIGLEN();
7131 teodor@sigaev.ru 343 : 15966 : GISTTYPE *origval = (GISTTYPE *) DatumGetPointer(origentry->key);
344 : 15966 : GISTTYPE *newval = (GISTTYPE *) DatumGetPointer(newentry->key);
345 : :
2176 akorotkov@postgresql 346 : 15966 : *penalty = hemdist(origval, newval, siglen);
7131 teodor@sigaev.ru 347 : 15966 : PG_RETURN_POINTER(penalty);
348 : : }
349 : :
350 : :
351 : : typedef struct
352 : : {
353 : : OffsetNumber pos;
354 : : int32 cost;
355 : : } SPLITCOST;
356 : :
357 : : static int
7102 bruce@momjian.us 358 : 6478 : comparecost(const void *a, const void *b)
359 : : {
758 nathan@postgresql.or 360 : 12956 : return pg_cmp_s32(((const SPLITCOST *) a)->cost,
361 : 6478 : ((const SPLITCOST *) b)->cost);
362 : : }
363 : :
364 : :
365 : : Datum
7102 bruce@momjian.us 366 : 720 : ghstore_picksplit(PG_FUNCTION_ARGS)
367 : : {
368 : 720 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
7131 teodor@sigaev.ru 369 : 720 : OffsetNumber maxoff = entryvec->n - 2;
370 : :
371 : 720 : GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
2176 akorotkov@postgresql 372 [ + - ]: 720 : int siglen = GET_SIGLEN();
373 : : OffsetNumber k,
374 : : j;
375 : : GISTTYPE *datum_l,
376 : : *datum_r;
377 : : BITVECP union_l,
378 : : union_r;
379 : : int32 size_alpha,
380 : : size_beta;
381 : : int32 size_waste,
7131 teodor@sigaev.ru 382 : 720 : waste = -1;
383 : : int32 nbytes;
384 : 720 : OffsetNumber seed_1 = 0,
385 : 720 : seed_2 = 0;
386 : : OffsetNumber *left,
387 : : *right;
388 : : BITVECP ptr;
389 : : int i;
390 : : SPLITCOST *costvector;
391 : : GISTTYPE *_k,
392 : : *_j;
393 : :
394 : 720 : nbytes = (maxoff + 2) * sizeof(OffsetNumber);
395 : 720 : v->spl_left = (OffsetNumber *) palloc(nbytes);
396 : 720 : v->spl_right = (OffsetNumber *) palloc(nbytes);
397 : :
7102 bruce@momjian.us 398 [ + + ]: 3160 : for (k = FirstOffsetNumber; k < maxoff; k = OffsetNumberNext(k))
399 : : {
7131 teodor@sigaev.ru 400 : 2440 : _k = GETENTRY(entryvec, k);
7102 bruce@momjian.us 401 [ + + ]: 107100 : for (j = OffsetNumberNext(k); j <= maxoff; j = OffsetNumberNext(j))
402 : : {
2176 akorotkov@postgresql 403 : 104660 : size_waste = hemdist(_k, GETENTRY(entryvec, j), siglen);
7102 bruce@momjian.us 404 [ + + ]: 104660 : if (size_waste > waste)
405 : : {
7131 teodor@sigaev.ru 406 : 1158 : waste = size_waste;
407 : 1158 : seed_1 = k;
408 : 1158 : seed_2 = j;
409 : : }
410 : : }
411 : : }
412 : :
413 : 720 : left = v->spl_left;
414 : 720 : v->spl_nleft = 0;
415 : 720 : right = v->spl_right;
416 : 720 : v->spl_nright = 0;
417 : :
418 [ + - - + ]: 720 : if (seed_1 == 0 || seed_2 == 0)
419 : : {
7131 teodor@sigaev.ru 420 :UBC 0 : seed_1 = 1;
421 : 0 : seed_2 = 2;
422 : : }
423 : :
424 : : /* form initial .. */
2176 akorotkov@postgresql 425 :CBC 720 : datum_l = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_1)), siglen,
426 : 720 : GETSIGN(GETENTRY(entryvec, seed_1)));
427 : 720 : datum_r = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_2)), siglen,
428 : 720 : GETSIGN(GETENTRY(entryvec, seed_2)));
429 : :
7131 teodor@sigaev.ru 430 : 720 : maxoff = OffsetNumberNext(maxoff);
431 : : /* sort before ... */
100 michael@paquier.xyz 432 :GNC 720 : costvector = palloc_array(SPLITCOST, maxoff);
7131 teodor@sigaev.ru 433 [ + + ]:CBC 4600 : for (j = FirstOffsetNumber; j <= maxoff; j = OffsetNumberNext(j))
434 : : {
435 : 3880 : costvector[j - 1].pos = j;
436 : 3880 : _j = GETENTRY(entryvec, j);
2176 akorotkov@postgresql 437 : 3880 : size_alpha = hemdist(datum_l, _j, siglen);
438 : 3880 : size_beta = hemdist(datum_r, _j, siglen);
7131 teodor@sigaev.ru 439 : 3880 : costvector[j - 1].cost = abs(size_alpha - size_beta);
440 : : }
1132 peter@eisentraut.org 441 : 720 : qsort(costvector, maxoff, sizeof(SPLITCOST), comparecost);
442 : :
7102 bruce@momjian.us 443 : 720 : union_l = GETSIGN(datum_l);
444 : 720 : union_r = GETSIGN(datum_r);
445 : :
446 [ + + ]: 4600 : for (k = 0; k < maxoff; k++)
447 : : {
7131 teodor@sigaev.ru 448 : 3880 : j = costvector[k].pos;
7102 bruce@momjian.us 449 [ + + ]: 3880 : if (j == seed_1)
450 : : {
7131 teodor@sigaev.ru 451 : 720 : *left++ = j;
452 : 720 : v->spl_nleft++;
453 : 720 : continue;
454 : : }
7102 bruce@momjian.us 455 [ + + ]: 3160 : else if (j == seed_2)
456 : : {
7131 teodor@sigaev.ru 457 : 720 : *right++ = j;
458 : 720 : v->spl_nright++;
459 : 720 : continue;
460 : : }
461 : 2440 : _j = GETENTRY(entryvec, j);
2176 akorotkov@postgresql 462 : 2440 : size_alpha = hemdist(datum_l, _j, siglen);
463 : 2440 : size_beta = hemdist(datum_r, _j, siglen);
464 : :
7102 bruce@momjian.us 465 [ + + ]: 2440 : if (size_alpha < size_beta + WISH_F(v->spl_nleft, v->spl_nright, 0.0001))
466 : : {
467 [ + - - + ]: 1125 : if (ISALLTRUE(datum_l) || ISALLTRUE(_j))
468 : : {
7131 teodor@sigaev.ru 469 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_l))
1132 peter@eisentraut.org 470 : 0 : memset(union_l, 0xff, siglen);
471 : : }
472 : : else
473 : : {
7102 bruce@momjian.us 474 :CBC 1125 : ptr = GETSIGN(_j);
2176 akorotkov@postgresql 475 [ + + ]: 1320309 : LOOPBYTE(siglen)
6694 bruce@momjian.us 476 : 1319184 : union_l[i] |= ptr[i];
477 : : }
7131 teodor@sigaev.ru 478 : 1125 : *left++ = j;
479 : 1125 : v->spl_nleft++;
480 : : }
481 : : else
482 : : {
7102 bruce@momjian.us 483 [ + - - + ]: 1315 : if (ISALLTRUE(datum_r) || ISALLTRUE(_j))
484 : : {
7131 teodor@sigaev.ru 485 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_r))
1132 peter@eisentraut.org 486 : 0 : memset(union_r, 0xff, siglen);
487 : : }
488 : : else
489 : : {
7102 bruce@momjian.us 490 :CBC 1315 : ptr = GETSIGN(_j);
2176 akorotkov@postgresql 491 [ + + ]: 1592611 : LOOPBYTE(siglen)
6694 bruce@momjian.us 492 : 1591296 : union_r[i] |= ptr[i];
493 : : }
7131 teodor@sigaev.ru 494 : 1315 : *right++ = j;
495 : 1315 : v->spl_nright++;
496 : : }
497 : : }
498 : :
499 : 720 : *right = *left = FirstOffsetNumber;
500 : :
501 : 720 : v->spl_ldatum = PointerGetDatum(datum_l);
502 : 720 : v->spl_rdatum = PointerGetDatum(datum_r);
503 : :
504 : 720 : PG_RETURN_POINTER(v);
505 : : }
506 : :
507 : :
508 : : Datum
7102 bruce@momjian.us 509 : 10522 : ghstore_consistent(PG_FUNCTION_ARGS)
510 : : {
511 : 10522 : GISTTYPE *entry = (GISTTYPE *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
6941 teodor@sigaev.ru 512 : 10522 : StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
513 : : #ifdef NOT_USED
514 : : Oid subtype = PG_GETARG_OID(3);
515 : : #endif
6544 tgl@sss.pgh.pa.us 516 : 10522 : bool *recheck = (bool *) PG_GETARG_POINTER(4);
2176 akorotkov@postgresql 517 [ + - ]: 10522 : int siglen = GET_SIGLEN();
7102 bruce@momjian.us 518 : 10522 : bool res = true;
519 : : BITVECP sign;
520 : :
521 : : /* All cases served by this function are inexact */
6544 tgl@sss.pgh.pa.us 522 : 10522 : *recheck = true;
523 : :
7102 bruce@momjian.us 524 [ - + ]: 10522 : if (ISALLTRUE(entry))
7131 teodor@sigaev.ru 525 :UBC 0 : PG_RETURN_BOOL(true);
526 : :
7102 bruce@momjian.us 527 :CBC 10522 : sign = GETSIGN(entry);
528 : :
6010 tgl@sss.pgh.pa.us 529 [ + + - + ]: 10522 : if (strategy == HStoreContainsStrategyNumber ||
530 : : strategy == HStoreOldContainsStrategyNumber)
7102 bruce@momjian.us 531 : 4723 : {
3100 tgl@sss.pgh.pa.us 532 : 4723 : HStore *query = PG_GETARG_HSTORE_P(1);
6941 teodor@sigaev.ru 533 : 4723 : HEntry *qe = ARRPTR(query);
534 : 4723 : char *qv = STRPTR(query);
5861 bruce@momjian.us 535 : 4723 : int count = HS_COUNT(query);
536 : : int i;
537 : :
6010 tgl@sss.pgh.pa.us 538 [ + + + + ]: 9657 : for (i = 0; res && i < count; ++i)
539 : : {
3769 540 [ + + ]: 4934 : int crc = crc32_sz((char *) HSTORE_KEY(qe, qv, i),
541 [ + + ]: 4934 : HSTORE_KEYLEN(qe, i));
542 : :
2176 akorotkov@postgresql 543 [ + + ]: 4934 : if (GETBIT(sign, HASHVAL(crc, siglen)))
544 : : {
3769 tgl@sss.pgh.pa.us 545 [ + + ]: 1969 : if (!HSTORE_VALISNULL(qe, i))
546 : : {
547 [ + - ]: 1154 : crc = crc32_sz((char *) HSTORE_VAL(qe, qv, i),
548 [ - + ]: 1154 : HSTORE_VALLEN(qe, i));
2176 akorotkov@postgresql 549 [ + + ]: 1154 : if (!GETBIT(sign, HASHVAL(crc, siglen)))
6941 teodor@sigaev.ru 550 : 557 : res = false;
551 : : }
552 : : }
553 : : else
554 : 2965 : res = false;
555 : : }
556 : : }
6694 bruce@momjian.us 557 [ + + ]: 5799 : else if (strategy == HStoreExistsStrategyNumber)
558 : : {
6010 tgl@sss.pgh.pa.us 559 : 1961 : text *query = PG_GETARG_TEXT_PP(1);
560 [ - + - - : 1961 : int crc = crc32_sz(VARDATA_ANY(query), VARSIZE_ANY_EXHDR(query));
- - - - -
+ - + ]
561 : :
2176 akorotkov@postgresql 562 : 1961 : res = (GETBIT(sign, HASHVAL(crc, siglen))) ? true : false;
563 : : }
6010 tgl@sss.pgh.pa.us 564 [ + + ]: 3838 : else if (strategy == HStoreExistsAllStrategyNumber)
565 : : {
5861 bruce@momjian.us 566 : 1580 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
567 : : Datum *key_datums;
568 : : bool *key_nulls;
569 : : int key_count;
570 : : int i;
571 : :
1353 peter@eisentraut.org 572 : 1580 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
573 : :
6010 tgl@sss.pgh.pa.us 574 [ + + + + ]: 3780 : for (i = 0; res && i < key_count; ++i)
575 : : {
576 : : int crc;
577 : :
578 [ - + ]: 2200 : if (key_nulls[i])
6010 tgl@sss.pgh.pa.us 579 :UBC 0 : continue;
222 peter@eisentraut.org 580 :GNC 2200 : crc = crc32_sz(VARDATA(DatumGetPointer(key_datums[i])), VARSIZE(DatumGetPointer(key_datums[i])) - VARHDRSZ);
2176 akorotkov@postgresql 581 [ + + ]:CBC 2200 : if (!(GETBIT(sign, HASHVAL(crc, siglen))))
3133 peter_e@gmx.net 582 : 1260 : res = false;
583 : : }
584 : : }
6010 tgl@sss.pgh.pa.us 585 [ + - ]: 2258 : else if (strategy == HStoreExistsAnyStrategyNumber)
586 : : {
5861 bruce@momjian.us 587 : 2258 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
588 : : Datum *key_datums;
589 : : bool *key_nulls;
590 : : int key_count;
591 : : int i;
592 : :
1353 peter@eisentraut.org 593 : 2258 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
594 : :
3133 peter_e@gmx.net 595 : 2258 : res = false;
596 : :
6010 tgl@sss.pgh.pa.us 597 [ + + + + ]: 5956 : for (i = 0; !res && i < key_count; ++i)
598 : : {
599 : : int crc;
600 : :
601 [ - + ]: 3698 : if (key_nulls[i])
6010 tgl@sss.pgh.pa.us 602 :UBC 0 : continue;
222 peter@eisentraut.org 603 :GNC 3698 : crc = crc32_sz(VARDATA(DatumGetPointer(key_datums[i])), VARSIZE(DatumGetPointer(key_datums[i])) - VARHDRSZ);
2176 akorotkov@postgresql 604 [ + + ]:CBC 3698 : if (GETBIT(sign, HASHVAL(crc, siglen)))
3133 peter_e@gmx.net 605 : 1232 : res = true;
606 : : }
607 : : }
608 : : else
6941 teodor@sigaev.ru 609 [ # # ]:UBC 0 : elog(ERROR, "Unsupported strategy number: %d", strategy);
610 : :
7131 teodor@sigaev.ru 611 :CBC 10522 : PG_RETURN_BOOL(res);
612 : : }
613 : :
614 : : Datum
2176 akorotkov@postgresql 615 : 10 : ghstore_options(PG_FUNCTION_ARGS)
616 : : {
617 : 10 : local_relopts *relopts = (local_relopts *) PG_GETARG_POINTER(0);
618 : :
619 : 10 : init_local_reloptions(relopts, sizeof(GistHstoreOptions));
620 : 10 : add_local_int_reloption(relopts, "siglen",
621 : : "signature length in bytes",
622 : : SIGLEN_DEFAULT, 1, SIGLEN_MAX,
623 : : offsetof(GistHstoreOptions, siglen));
624 : :
625 : 10 : PG_RETURN_VOID();
626 : : }
|
