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
192 peter@eisentraut.org 81 :GNC 33034 : crc32_sz(const char *buf, int size)
82 : : {
83 : : pg_crc32 crc;
84 : :
3449 heikki.linnakangas@i 85 :CBC 33034 : INIT_TRADITIONAL_CRC32(crc);
86 [ + + ]: 175983 : COMP_TRADITIONAL_CRC32(crc, buf, size);
87 : 33034 : FIN_TRADITIONAL_CRC32(crc);
88 : :
89 : 33034 : return crc;
90 : : }
91 : :
92 : :
6431 teodor@sigaev.ru 93 : 7 : PG_FUNCTION_INFO_V1(ghstore_in);
94 : 7 : PG_FUNCTION_INFO_V1(ghstore_out);
95 : :
96 : :
97 : : Datum
6402 bruce@momjian.us 98 :UBC 0 : ghstore_in(PG_FUNCTION_ARGS)
99 : : {
491 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
6402 bruce@momjian.us 108 : 0 : ghstore_out(PG_FUNCTION_ARGS)
109 : : {
491 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 *
1476 akorotkov@postgresql 118 :CBC 9511 : ghstore_alloc(bool allistrue, int siglen, BITVECP sign)
119 : : {
120 [ - + ]: 9511 : int flag = allistrue ? ALLISTRUE : 0;
121 [ + - ]: 9511 : int size = CALCGTSIZE(flag, siglen);
122 : 9511 : GISTTYPE *res = palloc(size);
123 : :
124 : 9511 : SET_VARSIZE(res, size);
125 : 9511 : res->flag = flag;
126 : :
127 [ + - ]: 9511 : if (!allistrue)
128 : : {
129 [ + + ]: 9511 : if (sign)
130 : 1336 : memcpy(GETSIGN(res), sign, siglen);
131 : : else
132 : 8175 : memset(GETSIGN(res), 0, siglen);
133 : : }
134 : :
135 : 9511 : return res;
136 : : }
137 : :
6431 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);
1476 akorotkov@postgresql 145 : 8 : PG_FUNCTION_INFO_V1(ghstore_options);
146 : :
147 : : Datum
6402 bruce@momjian.us 148 : 8181 : ghstore_compress(PG_FUNCTION_ARGS)
149 : : {
6431 teodor@sigaev.ru 150 : 8181 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
1476 akorotkov@postgresql 151 [ + - ]: 8181 : int siglen = GET_SIGLEN();
6431 teodor@sigaev.ru 152 : 8181 : GISTENTRY *retval = entry;
153 : :
6402 bruce@momjian.us 154 [ + + ]: 8181 : if (entry->leafkey)
155 : : {
1476 akorotkov@postgresql 156 : 2002 : GISTTYPE *res = ghstore_alloc(false, siglen, NULL);
5310 tgl@sss.pgh.pa.us 157 : 2002 : HStore *val = DatumGetHStoreP(entry->key);
158 : 2002 : HEntry *hsent = ARRPTR(val);
159 : 2002 : char *ptr = STRPTR(val);
5161 bruce@momjian.us 160 : 2002 : int count = HS_COUNT(val);
161 : : int i;
162 : :
5310 tgl@sss.pgh.pa.us 163 [ + + ]: 11564 : for (i = 0; i < count; ++i)
164 : : {
165 : : int h;
166 : :
3069 167 [ + + ]: 9562 : h = crc32_sz((char *) HSTORE_KEY(hsent, ptr, i),
168 [ + + ]: 9562 : HSTORE_KEYLEN(hsent, i));
1476 akorotkov@postgresql 169 : 9562 : HASH(GETSIGN(res), h, siglen);
3069 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));
1476 akorotkov@postgresql 174 : 9560 : HASH(GETSIGN(res), h, siglen);
175 : : }
176 : : }
177 : :
6431 teodor@sigaev.ru 178 : 2002 : retval = (GISTENTRY *) palloc(sizeof(GISTENTRY));
179 : 2002 : gistentryinit(*retval, PointerGetDatum(res),
180 : : entry->rel, entry->page,
181 : : entry->offset,
182 : : false);
183 : : }
6402 bruce@momjian.us 184 [ + - ]: 6179 : else if (!ISALLTRUE(DatumGetPointer(entry->key)))
185 : : {
186 : : int32 i;
187 : : GISTTYPE *res;
188 : 6179 : BITVECP sign = GETSIGN(DatumGetPointer(entry->key));
189 : :
1476 akorotkov@postgresql 190 [ + - ]: 6183 : LOOPBYTE(siglen)
191 : : {
5994 bruce@momjian.us 192 [ + + ]: 6183 : if ((sign[i] & 0xff) != 0xff)
193 : 6179 : PG_RETURN_POINTER(retval);
194 : : }
195 : :
1476 akorotkov@postgresql 196 :UBC 0 : res = ghstore_alloc(true, siglen, NULL);
197 : :
6431 teodor@sigaev.ru 198 : 0 : retval = (GISTENTRY *) palloc(sizeof(GISTENTRY));
199 : 0 : gistentryinit(*retval, PointerGetDatum(res),
200 : : entry->rel, entry->page,
201 : : entry->offset,
202 : : false);
203 : : }
204 : :
6431 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
6402 bruce@momjian.us 213 : 48593 : ghstore_decompress(PG_FUNCTION_ARGS)
214 : : {
4959 tgl@sss.pgh.pa.us 215 : 48593 : PG_RETURN_POINTER(PG_GETARG_POINTER(0));
216 : : }
217 : :
218 : : Datum
6402 bruce@momjian.us 219 : 6173 : ghstore_same(PG_FUNCTION_ARGS)
220 : : {
6431 teodor@sigaev.ru 221 : 6173 : GISTTYPE *a = (GISTTYPE *) PG_GETARG_POINTER(0);
222 : 6173 : GISTTYPE *b = (GISTTYPE *) PG_GETARG_POINTER(1);
6402 bruce@momjian.us 223 : 6173 : bool *result = (bool *) PG_GETARG_POINTER(2);
1476 akorotkov@postgresql 224 [ + - ]: 6173 : int siglen = GET_SIGLEN();
225 : :
226 : :
6431 teodor@sigaev.ru 227 [ - + - - ]: 6173 : if (ISALLTRUE(a) && ISALLTRUE(b))
6431 teodor@sigaev.ru 228 :UBC 0 : *result = true;
6431 teodor@sigaev.ru 229 [ - + ]:CBC 6173 : else if (ISALLTRUE(a))
6431 teodor@sigaev.ru 230 :UBC 0 : *result = false;
6431 teodor@sigaev.ru 231 [ - + ]:CBC 6173 : else if (ISALLTRUE(b))
6431 teodor@sigaev.ru 232 :UBC 0 : *result = false;
233 : : else
234 : : {
235 : : int32 i;
6402 bruce@momjian.us 236 :CBC 6173 : BITVECP sa = GETSIGN(a),
237 : 6173 : sb = GETSIGN(b);
238 : :
6431 teodor@sigaev.ru 239 : 6173 : *result = true;
1476 akorotkov@postgresql 240 [ + + ]: 4526351 : LOOPBYTE(siglen)
241 : : {
5994 bruce@momjian.us 242 [ + + ]: 4525021 : if (sa[i] != sb[i])
243 : : {
244 : 4843 : *result = false;
245 : 4843 : break;
246 : : }
247 : : }
248 : : }
6431 teodor@sigaev.ru 249 : 6173 : PG_RETURN_POINTER(result);
250 : : }
251 : :
252 : : static int32
1476 akorotkov@postgresql 253 :UBC 0 : sizebitvec(BITVECP sign, int siglen)
254 : : {
4311 peter_e@gmx.net 255 : 0 : int32 size = 0,
256 : : i;
257 : :
1476 akorotkov@postgresql 258 [ # # ]: 0 : LOOPBYTE(siglen)
259 : : {
5994 bruce@momjian.us 260 : 0 : size += SUMBIT(sign);
261 : 0 : sign = (BITVECP) (((char *) sign) + 1);
262 : : }
6431 teodor@sigaev.ru 263 : 0 : return size;
264 : : }
265 : :
266 : : static int
1476 akorotkov@postgresql 267 :CBC 132486 : hemdistsign(BITVECP a, BITVECP b, int siglen)
268 : : {
269 : : int i,
6402 bruce@momjian.us 270 : 132486 : dist = 0;
271 : :
1476 akorotkov@postgresql 272 [ + + ]: 386498438 : LOOPBIT(siglen)
273 : : {
5994 bruce@momjian.us 274 [ + + ]: 386365952 : if (GETBIT(a, i) != GETBIT(b, i))
6431 teodor@sigaev.ru 275 : 3304686 : dist++;
276 : : }
277 : 132486 : return dist;
278 : : }
279 : :
280 : : static int
1476 akorotkov@postgresql 281 : 132486 : hemdist(GISTTYPE *a, GISTTYPE *b, int siglen)
282 : : {
6402 bruce@momjian.us 283 [ - + ]: 132486 : if (ISALLTRUE(a))
284 : : {
6431 teodor@sigaev.ru 285 [ # # ]:UBC 0 : if (ISALLTRUE(b))
286 : 0 : return 0;
287 : : else
1476 akorotkov@postgresql 288 : 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(b), siglen);
289 : : }
6402 bruce@momjian.us 290 [ - + ]:CBC 132486 : else if (ISALLTRUE(b))
1476 akorotkov@postgresql 291 :UBC 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(a), siglen);
292 : :
1476 akorotkov@postgresql 293 :CBC 132486 : return hemdistsign(GETSIGN(a), GETSIGN(b), siglen);
294 : : }
295 : :
296 : : static int32
297 : 12346 : unionkey(BITVECP sbase, GISTTYPE *add, int siglen)
298 : : {
299 : : int32 i;
6402 bruce@momjian.us 300 : 12346 : BITVECP sadd = GETSIGN(add);
301 : :
6431 teodor@sigaev.ru 302 [ - + ]: 12346 : if (ISALLTRUE(add))
6431 teodor@sigaev.ru 303 :UBC 0 : return 1;
1476 akorotkov@postgresql 304 [ + + ]:CBC 21799898 : LOOPBYTE(siglen)
5994 bruce@momjian.us 305 : 21787552 : sbase[i] |= sadd[i];
6431 teodor@sigaev.ru 306 : 12346 : return 0;
307 : : }
308 : :
309 : : Datum
6402 bruce@momjian.us 310 : 6173 : ghstore_union(PG_FUNCTION_ARGS)
311 : : {
312 : 6173 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
4311 peter_e@gmx.net 313 : 6173 : int32 len = entryvec->n;
314 : :
6402 bruce@momjian.us 315 : 6173 : int *size = (int *) PG_GETARG_POINTER(1);
1476 akorotkov@postgresql 316 [ + - ]: 6173 : int siglen = GET_SIGLEN();
317 : : int32 i;
318 : 6173 : GISTTYPE *result = ghstore_alloc(false, siglen, NULL);
319 : 6173 : BITVECP base = GETSIGN(result);
320 : :
6402 bruce@momjian.us 321 [ + + ]: 18519 : for (i = 0; i < len; i++)
322 : : {
1476 akorotkov@postgresql 323 [ - + ]: 12346 : if (unionkey(base, GETENTRY(entryvec, i), siglen))
324 : : {
1476 akorotkov@postgresql 325 :UBC 0 : result->flag |= ALLISTRUE;
326 : 0 : SET_VARSIZE(result, CALCGTSIZE(ALLISTRUE, siglen));
6431 teodor@sigaev.ru 327 : 0 : break;
328 : : }
329 : : }
330 : :
1476 akorotkov@postgresql 331 :CBC 6173 : *size = VARSIZE(result);
332 : :
6431 teodor@sigaev.ru 333 : 6173 : PG_RETURN_POINTER(result);
334 : : }
335 : :
336 : : Datum
6402 bruce@momjian.us 337 : 15966 : ghstore_penalty(PG_FUNCTION_ARGS)
338 : : {
6431 teodor@sigaev.ru 339 : 15966 : GISTENTRY *origentry = (GISTENTRY *) PG_GETARG_POINTER(0); /* always ISSIGNKEY */
340 : 15966 : GISTENTRY *newentry = (GISTENTRY *) PG_GETARG_POINTER(1);
6402 bruce@momjian.us 341 : 15966 : float *penalty = (float *) PG_GETARG_POINTER(2);
1476 akorotkov@postgresql 342 [ + - ]: 15966 : int siglen = GET_SIGLEN();
6431 teodor@sigaev.ru 343 : 15966 : GISTTYPE *origval = (GISTTYPE *) DatumGetPointer(origentry->key);
344 : 15966 : GISTTYPE *newval = (GISTTYPE *) DatumGetPointer(newentry->key);
345 : :
1476 akorotkov@postgresql 346 : 15966 : *penalty = hemdist(origval, newval, siglen);
6431 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
6402 bruce@momjian.us 358 : 5962 : comparecost(const void *a, const void *b)
359 : : {
58 nathan@postgresql.or 360 :GNC 11924 : return pg_cmp_s32(((const SPLITCOST *) a)->cost,
361 : 5962 : ((const SPLITCOST *) b)->cost);
362 : : }
363 : :
364 : :
365 : : Datum
6402 bruce@momjian.us 366 :CBC 668 : ghstore_picksplit(PG_FUNCTION_ARGS)
367 : : {
368 : 668 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
6431 teodor@sigaev.ru 369 : 668 : OffsetNumber maxoff = entryvec->n - 2;
370 : :
371 : 668 : GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
1476 akorotkov@postgresql 372 [ + - ]: 668 : 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,
6431 teodor@sigaev.ru 382 : 668 : waste = -1;
383 : : int32 nbytes;
384 : 668 : OffsetNumber seed_1 = 0,
385 : 668 : seed_2 = 0;
386 : : OffsetNumber *left,
387 : : *right;
388 : : BITVECP ptr;
389 : : int i;
390 : : SPLITCOST *costvector;
391 : : GISTTYPE *_k,
392 : : *_j;
393 : :
394 : 668 : nbytes = (maxoff + 2) * sizeof(OffsetNumber);
395 : 668 : v->spl_left = (OffsetNumber *) palloc(nbytes);
396 : 668 : v->spl_right = (OffsetNumber *) palloc(nbytes);
397 : :
6402 bruce@momjian.us 398 [ + + ]: 3004 : for (k = FirstOffsetNumber; k < maxoff; k = OffsetNumberNext(k))
399 : : {
6431 teodor@sigaev.ru 400 : 2336 : _k = GETENTRY(entryvec, k);
6402 bruce@momjian.us 401 [ + + ]: 106840 : for (j = OffsetNumberNext(k); j <= maxoff; j = OffsetNumberNext(j))
402 : : {
1476 akorotkov@postgresql 403 : 104504 : size_waste = hemdist(_k, GETENTRY(entryvec, j), siglen);
6402 bruce@momjian.us 404 [ + + ]: 104504 : if (size_waste > waste)
405 : : {
6431 teodor@sigaev.ru 406 : 1084 : waste = size_waste;
407 : 1084 : seed_1 = k;
408 : 1084 : seed_2 = j;
409 : : }
410 : : }
411 : : }
412 : :
413 : 668 : left = v->spl_left;
414 : 668 : v->spl_nleft = 0;
415 : 668 : right = v->spl_right;
416 : 668 : v->spl_nright = 0;
417 : :
418 [ + - - + ]: 668 : if (seed_1 == 0 || seed_2 == 0)
419 : : {
6431 teodor@sigaev.ru 420 :UBC 0 : seed_1 = 1;
421 : 0 : seed_2 = 2;
422 : : }
423 : :
424 : : /* form initial .. */
1476 akorotkov@postgresql 425 :CBC 668 : datum_l = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_1)), siglen,
426 : 668 : GETSIGN(GETENTRY(entryvec, seed_1)));
427 : 668 : datum_r = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_2)), siglen,
428 : 668 : GETSIGN(GETENTRY(entryvec, seed_2)));
429 : :
6431 teodor@sigaev.ru 430 : 668 : maxoff = OffsetNumberNext(maxoff);
431 : : /* sort before ... */
432 : 668 : costvector = (SPLITCOST *) palloc(sizeof(SPLITCOST) * maxoff);
433 [ + + ]: 4340 : for (j = FirstOffsetNumber; j <= maxoff; j = OffsetNumberNext(j))
434 : : {
435 : 3672 : costvector[j - 1].pos = j;
436 : 3672 : _j = GETENTRY(entryvec, j);
1476 akorotkov@postgresql 437 : 3672 : size_alpha = hemdist(datum_l, _j, siglen);
438 : 3672 : size_beta = hemdist(datum_r, _j, siglen);
6431 teodor@sigaev.ru 439 : 3672 : costvector[j - 1].cost = abs(size_alpha - size_beta);
440 : : }
432 peter@eisentraut.org 441 : 668 : qsort(costvector, maxoff, sizeof(SPLITCOST), comparecost);
442 : :
6402 bruce@momjian.us 443 : 668 : union_l = GETSIGN(datum_l);
444 : 668 : union_r = GETSIGN(datum_r);
445 : :
446 [ + + ]: 4340 : for (k = 0; k < maxoff; k++)
447 : : {
6431 teodor@sigaev.ru 448 : 3672 : j = costvector[k].pos;
6402 bruce@momjian.us 449 [ + + ]: 3672 : if (j == seed_1)
450 : : {
6431 teodor@sigaev.ru 451 : 668 : *left++ = j;
452 : 668 : v->spl_nleft++;
453 : 668 : continue;
454 : : }
6402 bruce@momjian.us 455 [ + + ]: 3004 : else if (j == seed_2)
456 : : {
6431 teodor@sigaev.ru 457 : 668 : *right++ = j;
458 : 668 : v->spl_nright++;
459 : 668 : continue;
460 : : }
461 : 2336 : _j = GETENTRY(entryvec, j);
1476 akorotkov@postgresql 462 : 2336 : size_alpha = hemdist(datum_l, _j, siglen);
463 : 2336 : size_beta = hemdist(datum_r, _j, siglen);
464 : :
6402 bruce@momjian.us 465 [ + + ]: 2336 : if (size_alpha < size_beta + WISH_F(v->spl_nleft, v->spl_nright, 0.0001))
466 : : {
467 [ + - - + ]: 1157 : if (ISALLTRUE(datum_l) || ISALLTRUE(_j))
468 : : {
6431 teodor@sigaev.ru 469 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_l))
432 peter@eisentraut.org 470 : 0 : memset(union_l, 0xff, siglen);
471 : : }
472 : : else
473 : : {
6402 bruce@momjian.us 474 :CBC 1157 : ptr = GETSIGN(_j);
1476 akorotkov@postgresql 475 [ + + ]: 1290733 : LOOPBYTE(siglen)
5994 bruce@momjian.us 476 : 1289576 : union_l[i] |= ptr[i];
477 : : }
6431 teodor@sigaev.ru 478 : 1157 : *left++ = j;
479 : 1157 : v->spl_nleft++;
480 : : }
481 : : else
482 : : {
6402 bruce@momjian.us 483 [ + - - + ]: 1179 : if (ISALLTRUE(datum_r) || ISALLTRUE(_j))
484 : : {
6431 teodor@sigaev.ru 485 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_r))
432 peter@eisentraut.org 486 : 0 : memset(union_r, 0xff, siglen);
487 : : }
488 : : else
489 : : {
6402 bruce@momjian.us 490 :CBC 1179 : ptr = GETSIGN(_j);
1476 akorotkov@postgresql 491 [ + + ]: 1411587 : LOOPBYTE(siglen)
5994 bruce@momjian.us 492 : 1410408 : union_r[i] |= ptr[i];
493 : : }
6431 teodor@sigaev.ru 494 : 1179 : *right++ = j;
495 : 1179 : v->spl_nright++;
496 : : }
497 : : }
498 : :
499 : 668 : *right = *left = FirstOffsetNumber;
500 : :
501 : 668 : v->spl_ldatum = PointerGetDatum(datum_l);
502 : 668 : v->spl_rdatum = PointerGetDatum(datum_r);
503 : :
504 : 668 : PG_RETURN_POINTER(v);
505 : : }
506 : :
507 : :
508 : : Datum
6402 bruce@momjian.us 509 : 10436 : ghstore_consistent(PG_FUNCTION_ARGS)
510 : : {
511 : 10436 : GISTTYPE *entry = (GISTTYPE *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
6241 teodor@sigaev.ru 512 : 10436 : StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
513 : :
514 : : /* Oid subtype = PG_GETARG_OID(3); */
5844 tgl@sss.pgh.pa.us 515 : 10436 : bool *recheck = (bool *) PG_GETARG_POINTER(4);
1476 akorotkov@postgresql 516 [ + - ]: 10436 : int siglen = GET_SIGLEN();
6402 bruce@momjian.us 517 : 10436 : bool res = true;
518 : : BITVECP sign;
519 : :
520 : : /* All cases served by this function are inexact */
5844 tgl@sss.pgh.pa.us 521 : 10436 : *recheck = true;
522 : :
6402 bruce@momjian.us 523 [ - + ]: 10436 : if (ISALLTRUE(entry))
6431 teodor@sigaev.ru 524 :UBC 0 : PG_RETURN_BOOL(true);
525 : :
6402 bruce@momjian.us 526 :CBC 10436 : sign = GETSIGN(entry);
527 : :
5310 tgl@sss.pgh.pa.us 528 [ + + - + ]: 10436 : if (strategy == HStoreContainsStrategyNumber ||
529 : : strategy == HStoreOldContainsStrategyNumber)
6402 bruce@momjian.us 530 : 4683 : {
2400 tgl@sss.pgh.pa.us 531 : 4683 : HStore *query = PG_GETARG_HSTORE_P(1);
6241 teodor@sigaev.ru 532 : 4683 : HEntry *qe = ARRPTR(query);
533 : 4683 : char *qv = STRPTR(query);
5161 bruce@momjian.us 534 : 4683 : int count = HS_COUNT(query);
535 : : int i;
536 : :
5310 tgl@sss.pgh.pa.us 537 [ + + + + ]: 9579 : for (i = 0; res && i < count; ++i)
538 : : {
3069 539 [ + + ]: 4896 : int crc = crc32_sz((char *) HSTORE_KEY(qe, qv, i),
540 [ + + ]: 4896 : HSTORE_KEYLEN(qe, i));
541 : :
1476 akorotkov@postgresql 542 [ + + ]: 4896 : if (GETBIT(sign, HASHVAL(crc, siglen)))
543 : : {
3069 tgl@sss.pgh.pa.us 544 [ + + ]: 1974 : if (!HSTORE_VALISNULL(qe, i))
545 : : {
546 [ + - ]: 1202 : crc = crc32_sz((char *) HSTORE_VAL(qe, qv, i),
547 [ - + ]: 1202 : HSTORE_VALLEN(qe, i));
1476 akorotkov@postgresql 548 [ + + ]: 1202 : if (!GETBIT(sign, HASHVAL(crc, siglen)))
6241 teodor@sigaev.ru 549 : 595 : res = false;
550 : : }
551 : : }
552 : : else
553 : 2922 : res = false;
554 : : }
555 : : }
5994 bruce@momjian.us 556 [ + + ]: 5753 : else if (strategy == HStoreExistsStrategyNumber)
557 : : {
5310 tgl@sss.pgh.pa.us 558 : 1934 : text *query = PG_GETARG_TEXT_PP(1);
559 [ - + - - : 1934 : int crc = crc32_sz(VARDATA_ANY(query), VARSIZE_ANY_EXHDR(query));
- - - - -
+ - + ]
560 : :
1476 akorotkov@postgresql 561 : 1934 : res = (GETBIT(sign, HASHVAL(crc, siglen))) ? true : false;
562 : : }
5310 tgl@sss.pgh.pa.us 563 [ + + ]: 3819 : else if (strategy == HStoreExistsAllStrategyNumber)
564 : : {
5161 bruce@momjian.us 565 : 1572 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
566 : : Datum *key_datums;
567 : : bool *key_nulls;
568 : : int key_count;
569 : : int i;
570 : :
653 peter@eisentraut.org 571 : 1572 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
572 : :
5310 tgl@sss.pgh.pa.us 573 [ + + + + ]: 3748 : for (i = 0; res && i < key_count; ++i)
574 : : {
575 : : int crc;
576 : :
577 [ - + ]: 2176 : if (key_nulls[i])
5310 tgl@sss.pgh.pa.us 578 :UBC 0 : continue;
5310 tgl@sss.pgh.pa.us 579 :CBC 2176 : crc = crc32_sz(VARDATA(key_datums[i]), VARSIZE(key_datums[i]) - VARHDRSZ);
1476 akorotkov@postgresql 580 [ + + ]: 2176 : if (!(GETBIT(sign, HASHVAL(crc, siglen))))
2433 peter_e@gmx.net 581 : 1263 : res = false;
582 : : }
583 : : }
5310 tgl@sss.pgh.pa.us 584 [ + - ]: 2247 : else if (strategy == HStoreExistsAnyStrategyNumber)
585 : : {
5161 bruce@momjian.us 586 : 2247 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
587 : : Datum *key_datums;
588 : : bool *key_nulls;
589 : : int key_count;
590 : : int i;
591 : :
653 peter@eisentraut.org 592 : 2247 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
593 : :
2433 peter_e@gmx.net 594 : 2247 : res = false;
595 : :
5310 tgl@sss.pgh.pa.us 596 [ + + + + ]: 5951 : for (i = 0; !res && i < key_count; ++i)
597 : : {
598 : : int crc;
599 : :
600 [ - + ]: 3704 : if (key_nulls[i])
5310 tgl@sss.pgh.pa.us 601 :UBC 0 : continue;
5310 tgl@sss.pgh.pa.us 602 :CBC 3704 : crc = crc32_sz(VARDATA(key_datums[i]), VARSIZE(key_datums[i]) - VARHDRSZ);
1476 akorotkov@postgresql 603 [ + + ]: 3704 : if (GETBIT(sign, HASHVAL(crc, siglen)))
2433 peter_e@gmx.net 604 : 1209 : res = true;
605 : : }
606 : : }
607 : : else
6241 teodor@sigaev.ru 608 [ # # ]:UBC 0 : elog(ERROR, "Unsupported strategy number: %d", strategy);
609 : :
6431 teodor@sigaev.ru 610 :CBC 10436 : PG_RETURN_BOOL(res);
611 : : }
612 : :
613 : : Datum
1476 akorotkov@postgresql 614 : 10 : ghstore_options(PG_FUNCTION_ARGS)
615 : : {
616 : 10 : local_relopts *relopts = (local_relopts *) PG_GETARG_POINTER(0);
617 : :
618 : 10 : init_local_reloptions(relopts, sizeof(GistHstoreOptions));
619 : 10 : add_local_int_reloption(relopts, "siglen",
620 : : "signature length in bytes",
621 : : SIGLEN_DEFAULT, 1, SIGLEN_MAX,
622 : : offsetof(GistHstoreOptions, siglen));
623 : :
624 : 10 : PG_RETURN_VOID();
625 : : }
|
