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
702 peter@eisentraut.org 81 :CBC 33097 : crc32_sz(const char *buf, int size)
82 : : {
83 : : pg_crc32 crc;
84 : :
3959 heikki.linnakangas@i 85 : 33097 : INIT_TRADITIONAL_CRC32(crc);
86 [ + + ]: 176455 : COMP_TRADITIONAL_CRC32(crc, buf, size);
87 : 33097 : FIN_TRADITIONAL_CRC32(crc);
88 : :
89 : 33097 : return crc;
90 : : }
91 : :
92 : :
6941 teodor@sigaev.ru 93 : 7 : PG_FUNCTION_INFO_V1(ghstore_in);
94 : 7 : PG_FUNCTION_INFO_V1(ghstore_out);
95 : :
96 : :
97 : : Datum
6912 bruce@momjian.us 98 :UBC 0 : ghstore_in(PG_FUNCTION_ARGS)
99 : : {
1001 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
6912 bruce@momjian.us 108 : 0 : ghstore_out(PG_FUNCTION_ARGS)
109 : : {
1001 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 *
1986 akorotkov@postgresql 118 :CBC 9498 : ghstore_alloc(bool allistrue, int siglen, BITVECP sign)
119 : : {
120 [ - + ]: 9498 : int flag = allistrue ? ALLISTRUE : 0;
121 [ + - ]: 9498 : int size = CALCGTSIZE(flag, siglen);
122 : 9498 : GISTTYPE *res = palloc(size);
123 : :
124 : 9498 : SET_VARSIZE(res, size);
125 : 9498 : res->flag = flag;
126 : :
127 [ + - ]: 9498 : if (!allistrue)
128 : : {
129 [ + + ]: 9498 : if (sign)
130 : 1338 : memcpy(GETSIGN(res), sign, siglen);
131 : : else
132 : 8160 : memset(GETSIGN(res), 0, siglen);
133 : : }
134 : :
135 : 9498 : return res;
136 : : }
137 : :
6941 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);
1986 akorotkov@postgresql 145 : 8 : PG_FUNCTION_INFO_V1(ghstore_options);
146 : :
147 : : Datum
6912 bruce@momjian.us 148 : 8161 : ghstore_compress(PG_FUNCTION_ARGS)
149 : : {
6941 teodor@sigaev.ru 150 : 8161 : GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
1986 akorotkov@postgresql 151 [ + - ]: 8161 : int siglen = GET_SIGLEN();
6941 teodor@sigaev.ru 152 : 8161 : GISTENTRY *retval = entry;
153 : :
6912 bruce@momjian.us 154 [ + + ]: 8161 : if (entry->leafkey)
155 : : {
1986 akorotkov@postgresql 156 : 2002 : GISTTYPE *res = ghstore_alloc(false, siglen, NULL);
5820 tgl@sss.pgh.pa.us 157 : 2002 : HStore *val = DatumGetHStoreP(entry->key);
158 : 2002 : HEntry *hsent = ARRPTR(val);
159 : 2002 : char *ptr = STRPTR(val);
5671 bruce@momjian.us 160 : 2002 : int count = HS_COUNT(val);
161 : : int i;
162 : :
5820 tgl@sss.pgh.pa.us 163 [ + + ]: 11564 : for (i = 0; i < count; ++i)
164 : : {
165 : : int h;
166 : :
3579 167 [ + + ]: 9562 : h = crc32_sz((char *) HSTORE_KEY(hsent, ptr, i),
168 [ + + ]: 9562 : HSTORE_KEYLEN(hsent, i));
1986 akorotkov@postgresql 169 : 9562 : HASH(GETSIGN(res), h, siglen);
3579 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));
1986 akorotkov@postgresql 174 : 9560 : HASH(GETSIGN(res), h, siglen);
175 : : }
176 : : }
177 : :
6941 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 : : }
6912 bruce@momjian.us 184 [ + - ]: 6159 : else if (!ISALLTRUE(DatumGetPointer(entry->key)))
185 : : {
186 : : int32 i;
187 : : GISTTYPE *res;
188 : 6159 : BITVECP sign = GETSIGN(DatumGetPointer(entry->key));
189 : :
1986 akorotkov@postgresql 190 [ + - ]: 6159 : LOOPBYTE(siglen)
191 : : {
6504 bruce@momjian.us 192 [ + - ]: 6159 : if ((sign[i] & 0xff) != 0xff)
193 : 6159 : PG_RETURN_POINTER(retval);
194 : : }
195 : :
1986 akorotkov@postgresql 196 :UBC 0 : res = ghstore_alloc(true, siglen, NULL);
197 : :
6941 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 : :
6941 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
6912 bruce@momjian.us 213 : 48762 : ghstore_decompress(PG_FUNCTION_ARGS)
214 : : {
5469 tgl@sss.pgh.pa.us 215 : 48762 : PG_RETURN_POINTER(PG_GETARG_POINTER(0));
216 : : }
217 : :
218 : : Datum
6912 bruce@momjian.us 219 : 6158 : ghstore_same(PG_FUNCTION_ARGS)
220 : : {
6941 teodor@sigaev.ru 221 : 6158 : GISTTYPE *a = (GISTTYPE *) PG_GETARG_POINTER(0);
222 : 6158 : GISTTYPE *b = (GISTTYPE *) PG_GETARG_POINTER(1);
6912 bruce@momjian.us 223 : 6158 : bool *result = (bool *) PG_GETARG_POINTER(2);
1986 akorotkov@postgresql 224 [ + - ]: 6158 : int siglen = GET_SIGLEN();
225 : :
226 : :
6941 teodor@sigaev.ru 227 [ - + - - ]: 6158 : if (ISALLTRUE(a) && ISALLTRUE(b))
6941 teodor@sigaev.ru 228 :UBC 0 : *result = true;
6941 teodor@sigaev.ru 229 [ - + ]:CBC 6158 : else if (ISALLTRUE(a))
6941 teodor@sigaev.ru 230 :UBC 0 : *result = false;
6941 teodor@sigaev.ru 231 [ - + ]:CBC 6158 : else if (ISALLTRUE(b))
6941 teodor@sigaev.ru 232 :UBC 0 : *result = false;
233 : : else
234 : : {
235 : : int32 i;
6912 bruce@momjian.us 236 :CBC 6158 : BITVECP sa = GETSIGN(a),
237 : 6158 : sb = GETSIGN(b);
238 : :
6941 teodor@sigaev.ru 239 : 6158 : *result = true;
1986 akorotkov@postgresql 240 [ + + ]: 4580555 : LOOPBYTE(siglen)
241 : : {
6504 bruce@momjian.us 242 [ + + ]: 4579218 : if (sa[i] != sb[i])
243 : : {
244 : 4821 : *result = false;
245 : 4821 : break;
246 : : }
247 : : }
248 : : }
6941 teodor@sigaev.ru 249 : 6158 : PG_RETURN_POINTER(result);
250 : : }
251 : :
252 : : static int32
1986 akorotkov@postgresql 253 :UBC 0 : sizebitvec(BITVECP sign, int siglen)
254 : : {
4821 peter_e@gmx.net 255 : 0 : int32 size = 0,
256 : : i;
257 : :
1986 akorotkov@postgresql 258 [ # # ]: 0 : LOOPBYTE(siglen)
259 : : {
6504 bruce@momjian.us 260 : 0 : size += SUMBIT(sign);
261 : 0 : sign = (BITVECP) (((char *) sign) + 1);
262 : : }
6941 teodor@sigaev.ru 263 : 0 : return size;
264 : : }
265 : :
266 : : static int
1986 akorotkov@postgresql 267 :CBC 132657 : hemdistsign(BITVECP a, BITVECP b, int siglen)
268 : : {
269 : : int i,
6912 bruce@momjian.us 270 : 132657 : dist = 0;
271 : :
1986 akorotkov@postgresql 272 [ + + ]: 385283569 : LOOPBIT(siglen)
273 : : {
6504 bruce@momjian.us 274 [ + + ]: 385150912 : if (GETBIT(a, i) != GETBIT(b, i))
6941 teodor@sigaev.ru 275 : 3412085 : dist++;
276 : : }
277 : 132657 : return dist;
278 : : }
279 : :
280 : : static int
1986 akorotkov@postgresql 281 : 132657 : hemdist(GISTTYPE *a, GISTTYPE *b, int siglen)
282 : : {
6912 bruce@momjian.us 283 [ - + ]: 132657 : if (ISALLTRUE(a))
284 : : {
6941 teodor@sigaev.ru 285 [ # # ]:UBC 0 : if (ISALLTRUE(b))
286 : 0 : return 0;
287 : : else
1986 akorotkov@postgresql 288 : 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(b), siglen);
289 : : }
6912 bruce@momjian.us 290 [ - + ]:CBC 132657 : else if (ISALLTRUE(b))
1986 akorotkov@postgresql 291 :UBC 0 : return SIGLENBIT(siglen) - sizebitvec(GETSIGN(a), siglen);
292 : :
1986 akorotkov@postgresql 293 :CBC 132657 : return hemdistsign(GETSIGN(a), GETSIGN(b), siglen);
294 : : }
295 : :
296 : : static int32
297 : 12316 : unionkey(BITVECP sbase, GISTTYPE *add, int siglen)
298 : : {
299 : : int32 i;
6912 bruce@momjian.us 300 : 12316 : BITVECP sadd = GETSIGN(add);
301 : :
6941 teodor@sigaev.ru 302 [ - + ]: 12316 : if (ISALLTRUE(add))
6941 teodor@sigaev.ru 303 :UBC 0 : return 1;
1986 akorotkov@postgresql 304 [ + + ]:CBC 21739148 : LOOPBYTE(siglen)
6504 bruce@momjian.us 305 : 21726832 : sbase[i] |= sadd[i];
6941 teodor@sigaev.ru 306 : 12316 : return 0;
307 : : }
308 : :
309 : : Datum
6912 bruce@momjian.us 310 : 6158 : ghstore_union(PG_FUNCTION_ARGS)
311 : : {
312 : 6158 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
4821 peter_e@gmx.net 313 : 6158 : int32 len = entryvec->n;
314 : :
6912 bruce@momjian.us 315 : 6158 : int *size = (int *) PG_GETARG_POINTER(1);
1986 akorotkov@postgresql 316 [ + - ]: 6158 : int siglen = GET_SIGLEN();
317 : : int32 i;
318 : 6158 : GISTTYPE *result = ghstore_alloc(false, siglen, NULL);
319 : 6158 : BITVECP base = GETSIGN(result);
320 : :
6912 bruce@momjian.us 321 [ + + ]: 18474 : for (i = 0; i < len; i++)
322 : : {
1986 akorotkov@postgresql 323 [ - + ]: 12316 : if (unionkey(base, GETENTRY(entryvec, i), siglen))
324 : : {
1986 akorotkov@postgresql 325 :UBC 0 : result->flag |= ALLISTRUE;
326 : 0 : SET_VARSIZE(result, CALCGTSIZE(ALLISTRUE, siglen));
6941 teodor@sigaev.ru 327 : 0 : break;
328 : : }
329 : : }
330 : :
1986 akorotkov@postgresql 331 :CBC 6158 : *size = VARSIZE(result);
332 : :
6941 teodor@sigaev.ru 333 : 6158 : PG_RETURN_POINTER(result);
334 : : }
335 : :
336 : : Datum
6912 bruce@momjian.us 337 : 16122 : ghstore_penalty(PG_FUNCTION_ARGS)
338 : : {
6941 teodor@sigaev.ru 339 : 16122 : GISTENTRY *origentry = (GISTENTRY *) PG_GETARG_POINTER(0); /* always ISSIGNKEY */
340 : 16122 : GISTENTRY *newentry = (GISTENTRY *) PG_GETARG_POINTER(1);
6912 bruce@momjian.us 341 : 16122 : float *penalty = (float *) PG_GETARG_POINTER(2);
1986 akorotkov@postgresql 342 [ + - ]: 16122 : int siglen = GET_SIGLEN();
6941 teodor@sigaev.ru 343 : 16122 : GISTTYPE *origval = (GISTTYPE *) DatumGetPointer(origentry->key);
344 : 16122 : GISTTYPE *newval = (GISTTYPE *) DatumGetPointer(newentry->key);
345 : :
1986 akorotkov@postgresql 346 : 16122 : *penalty = hemdist(origval, newval, siglen);
6941 teodor@sigaev.ru 347 : 16122 : PG_RETURN_POINTER(penalty);
348 : : }
349 : :
350 : :
351 : : typedef struct
352 : : {
353 : : OffsetNumber pos;
354 : : int32 cost;
355 : : } SPLITCOST;
356 : :
357 : : static int
6912 bruce@momjian.us 358 : 6020 : comparecost(const void *a, const void *b)
359 : : {
568 nathan@postgresql.or 360 : 12040 : return pg_cmp_s32(((const SPLITCOST *) a)->cost,
361 : 6020 : ((const SPLITCOST *) b)->cost);
362 : : }
363 : :
364 : :
365 : : Datum
6912 bruce@momjian.us 366 : 669 : ghstore_picksplit(PG_FUNCTION_ARGS)
367 : : {
368 : 669 : GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
6941 teodor@sigaev.ru 369 : 669 : OffsetNumber maxoff = entryvec->n - 2;
370 : :
371 : 669 : GIST_SPLITVEC *v = (GIST_SPLITVEC *) PG_GETARG_POINTER(1);
1986 akorotkov@postgresql 372 [ + - ]: 669 : 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,
6941 teodor@sigaev.ru 382 : 669 : waste = -1;
383 : : int32 nbytes;
384 : 669 : OffsetNumber seed_1 = 0,
385 : 669 : seed_2 = 0;
386 : : OffsetNumber *left,
387 : : *right;
388 : : BITVECP ptr;
389 : : int i;
390 : : SPLITCOST *costvector;
391 : : GISTTYPE *_k,
392 : : *_j;
393 : :
394 : 669 : nbytes = (maxoff + 2) * sizeof(OffsetNumber);
395 : 669 : v->spl_left = (OffsetNumber *) palloc(nbytes);
396 : 669 : v->spl_right = (OffsetNumber *) palloc(nbytes);
397 : :
6912 bruce@momjian.us 398 [ + + ]: 3007 : for (k = FirstOffsetNumber; k < maxoff; k = OffsetNumberNext(k))
399 : : {
6941 teodor@sigaev.ru 400 : 2338 : _k = GETENTRY(entryvec, k);
6912 bruce@momjian.us 401 [ + + ]: 106845 : for (j = OffsetNumberNext(k); j <= maxoff; j = OffsetNumberNext(j))
402 : : {
1986 akorotkov@postgresql 403 : 104507 : size_waste = hemdist(_k, GETENTRY(entryvec, j), siglen);
6912 bruce@momjian.us 404 [ + + ]: 104507 : if (size_waste > waste)
405 : : {
6941 teodor@sigaev.ru 406 : 1127 : waste = size_waste;
407 : 1127 : seed_1 = k;
408 : 1127 : seed_2 = j;
409 : : }
410 : : }
411 : : }
412 : :
413 : 669 : left = v->spl_left;
414 : 669 : v->spl_nleft = 0;
415 : 669 : right = v->spl_right;
416 : 669 : v->spl_nright = 0;
417 : :
418 [ + - - + ]: 669 : if (seed_1 == 0 || seed_2 == 0)
419 : : {
6941 teodor@sigaev.ru 420 :UBC 0 : seed_1 = 1;
421 : 0 : seed_2 = 2;
422 : : }
423 : :
424 : : /* form initial .. */
1986 akorotkov@postgresql 425 :CBC 669 : datum_l = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_1)), siglen,
426 : 669 : GETSIGN(GETENTRY(entryvec, seed_1)));
427 : 669 : datum_r = ghstore_alloc(ISALLTRUE(GETENTRY(entryvec, seed_2)), siglen,
428 : 669 : GETSIGN(GETENTRY(entryvec, seed_2)));
429 : :
6941 teodor@sigaev.ru 430 : 669 : maxoff = OffsetNumberNext(maxoff);
431 : : /* sort before ... */
432 : 669 : costvector = (SPLITCOST *) palloc(sizeof(SPLITCOST) * maxoff);
433 [ + + ]: 4345 : for (j = FirstOffsetNumber; j <= maxoff; j = OffsetNumberNext(j))
434 : : {
435 : 3676 : costvector[j - 1].pos = j;
436 : 3676 : _j = GETENTRY(entryvec, j);
1986 akorotkov@postgresql 437 : 3676 : size_alpha = hemdist(datum_l, _j, siglen);
438 : 3676 : size_beta = hemdist(datum_r, _j, siglen);
6941 teodor@sigaev.ru 439 : 3676 : costvector[j - 1].cost = abs(size_alpha - size_beta);
440 : : }
942 peter@eisentraut.org 441 : 669 : qsort(costvector, maxoff, sizeof(SPLITCOST), comparecost);
442 : :
6912 bruce@momjian.us 443 : 669 : union_l = GETSIGN(datum_l);
444 : 669 : union_r = GETSIGN(datum_r);
445 : :
446 [ + + ]: 4345 : for (k = 0; k < maxoff; k++)
447 : : {
6941 teodor@sigaev.ru 448 : 3676 : j = costvector[k].pos;
6912 bruce@momjian.us 449 [ + + ]: 3676 : if (j == seed_1)
450 : : {
6941 teodor@sigaev.ru 451 : 669 : *left++ = j;
452 : 669 : v->spl_nleft++;
453 : 669 : continue;
454 : : }
6912 bruce@momjian.us 455 [ + + ]: 3007 : else if (j == seed_2)
456 : : {
6941 teodor@sigaev.ru 457 : 669 : *right++ = j;
458 : 669 : v->spl_nright++;
459 : 669 : continue;
460 : : }
461 : 2338 : _j = GETENTRY(entryvec, j);
1986 akorotkov@postgresql 462 : 2338 : size_alpha = hemdist(datum_l, _j, siglen);
463 : 2338 : size_beta = hemdist(datum_r, _j, siglen);
464 : :
6912 bruce@momjian.us 465 [ + + ]: 2338 : if (size_alpha < size_beta + WISH_F(v->spl_nleft, v->spl_nright, 0.0001))
466 : : {
467 [ + - - + ]: 1106 : if (ISALLTRUE(datum_l) || ISALLTRUE(_j))
468 : : {
6941 teodor@sigaev.ru 469 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_l))
942 peter@eisentraut.org 470 : 0 : memset(union_l, 0xff, siglen);
471 : : }
472 : : else
473 : : {
6912 bruce@momjian.us 474 :CBC 1106 : ptr = GETSIGN(_j);
1986 akorotkov@postgresql 475 [ + + ]: 1263762 : LOOPBYTE(siglen)
6504 bruce@momjian.us 476 : 1262656 : union_l[i] |= ptr[i];
477 : : }
6941 teodor@sigaev.ru 478 : 1106 : *left++ = j;
479 : 1106 : v->spl_nleft++;
480 : : }
481 : : else
482 : : {
6912 bruce@momjian.us 483 [ + - - + ]: 1232 : if (ISALLTRUE(datum_r) || ISALLTRUE(_j))
484 : : {
6941 teodor@sigaev.ru 485 [ # # ]:UBC 0 : if (!ISALLTRUE(datum_r))
942 peter@eisentraut.org 486 : 0 : memset(union_r, 0xff, siglen);
487 : : }
488 : : else
489 : : {
6912 bruce@momjian.us 490 :CBC 1232 : ptr = GETSIGN(_j);
1986 akorotkov@postgresql 491 [ + + ]: 1442608 : LOOPBYTE(siglen)
6504 bruce@momjian.us 492 : 1441376 : union_r[i] |= ptr[i];
493 : : }
6941 teodor@sigaev.ru 494 : 1232 : *right++ = j;
495 : 1232 : v->spl_nright++;
496 : : }
497 : : }
498 : :
499 : 669 : *right = *left = FirstOffsetNumber;
500 : :
501 : 669 : v->spl_ldatum = PointerGetDatum(datum_l);
502 : 669 : v->spl_rdatum = PointerGetDatum(datum_r);
503 : :
504 : 669 : PG_RETURN_POINTER(v);
505 : : }
506 : :
507 : :
508 : : Datum
6912 bruce@momjian.us 509 : 10490 : ghstore_consistent(PG_FUNCTION_ARGS)
510 : : {
511 : 10490 : GISTTYPE *entry = (GISTTYPE *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
6751 teodor@sigaev.ru 512 : 10490 : StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
513 : :
514 : : /* Oid subtype = PG_GETARG_OID(3); */
6354 tgl@sss.pgh.pa.us 515 : 10490 : bool *recheck = (bool *) PG_GETARG_POINTER(4);
1986 akorotkov@postgresql 516 [ + - ]: 10490 : int siglen = GET_SIGLEN();
6912 bruce@momjian.us 517 : 10490 : bool res = true;
518 : : BITVECP sign;
519 : :
520 : : /* All cases served by this function are inexact */
6354 tgl@sss.pgh.pa.us 521 : 10490 : *recheck = true;
522 : :
6912 bruce@momjian.us 523 [ - + ]: 10490 : if (ISALLTRUE(entry))
6941 teodor@sigaev.ru 524 :UBC 0 : PG_RETURN_BOOL(true);
525 : :
6912 bruce@momjian.us 526 :CBC 10490 : sign = GETSIGN(entry);
527 : :
5820 tgl@sss.pgh.pa.us 528 [ + + - + ]: 10490 : if (strategy == HStoreContainsStrategyNumber ||
529 : : strategy == HStoreOldContainsStrategyNumber)
6912 bruce@momjian.us 530 : 4715 : {
2910 tgl@sss.pgh.pa.us 531 : 4715 : HStore *query = PG_GETARG_HSTORE_P(1);
6751 teodor@sigaev.ru 532 : 4715 : HEntry *qe = ARRPTR(query);
533 : 4715 : char *qv = STRPTR(query);
5671 bruce@momjian.us 534 : 4715 : int count = HS_COUNT(query);
535 : : int i;
536 : :
5820 tgl@sss.pgh.pa.us 537 [ + + + + ]: 9645 : for (i = 0; res && i < count; ++i)
538 : : {
3579 539 [ + + ]: 4930 : int crc = crc32_sz((char *) HSTORE_KEY(qe, qv, i),
540 [ + + ]: 4930 : HSTORE_KEYLEN(qe, i));
541 : :
1986 akorotkov@postgresql 542 [ + + ]: 4930 : if (GETBIT(sign, HASHVAL(crc, siglen)))
543 : : {
3579 tgl@sss.pgh.pa.us 544 [ + + ]: 1977 : if (!HSTORE_VALISNULL(qe, i))
545 : : {
546 [ + - ]: 1188 : crc = crc32_sz((char *) HSTORE_VAL(qe, qv, i),
547 [ - + ]: 1188 : HSTORE_VALLEN(qe, i));
1986 akorotkov@postgresql 548 [ + + ]: 1188 : if (!GETBIT(sign, HASHVAL(crc, siglen)))
6751 teodor@sigaev.ru 549 : 576 : res = false;
550 : : }
551 : : }
552 : : else
553 : 2953 : res = false;
554 : : }
555 : : }
6504 bruce@momjian.us 556 [ + + ]: 5775 : else if (strategy == HStoreExistsStrategyNumber)
557 : : {
5820 tgl@sss.pgh.pa.us 558 : 1924 : text *query = PG_GETARG_TEXT_PP(1);
559 [ - + - - : 1924 : int crc = crc32_sz(VARDATA_ANY(query), VARSIZE_ANY_EXHDR(query));
- - - - -
+ - + ]
560 : :
1986 akorotkov@postgresql 561 : 1924 : res = (GETBIT(sign, HASHVAL(crc, siglen))) ? true : false;
562 : : }
5820 tgl@sss.pgh.pa.us 563 [ + + ]: 3851 : else if (strategy == HStoreExistsAllStrategyNumber)
564 : : {
5671 bruce@momjian.us 565 : 1608 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
566 : : Datum *key_datums;
567 : : bool *key_nulls;
568 : : int key_count;
569 : : int i;
570 : :
1163 peter@eisentraut.org 571 : 1608 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
572 : :
5820 tgl@sss.pgh.pa.us 573 [ + + + + ]: 3846 : for (i = 0; res && i < key_count; ++i)
574 : : {
575 : : int crc;
576 : :
577 [ - + ]: 2238 : if (key_nulls[i])
5820 tgl@sss.pgh.pa.us 578 :UBC 0 : continue;
32 peter@eisentraut.org 579 :GNC 2238 : crc = crc32_sz(VARDATA(DatumGetPointer(key_datums[i])), VARSIZE(DatumGetPointer(key_datums[i])) - VARHDRSZ);
1986 akorotkov@postgresql 580 [ + + ]:CBC 2238 : if (!(GETBIT(sign, HASHVAL(crc, siglen))))
2943 peter_e@gmx.net 581 : 1284 : res = false;
582 : : }
583 : : }
5820 tgl@sss.pgh.pa.us 584 [ + - ]: 2243 : else if (strategy == HStoreExistsAnyStrategyNumber)
585 : : {
5671 bruce@momjian.us 586 : 2243 : ArrayType *query = PG_GETARG_ARRAYTYPE_P(1);
587 : : Datum *key_datums;
588 : : bool *key_nulls;
589 : : int key_count;
590 : : int i;
591 : :
1163 peter@eisentraut.org 592 : 2243 : deconstruct_array_builtin(query, TEXTOID, &key_datums, &key_nulls, &key_count);
593 : :
2943 peter_e@gmx.net 594 : 2243 : res = false;
595 : :
5820 tgl@sss.pgh.pa.us 596 [ + + + + ]: 5938 : for (i = 0; !res && i < key_count; ++i)
597 : : {
598 : : int crc;
599 : :
600 [ - + ]: 3695 : if (key_nulls[i])
5820 tgl@sss.pgh.pa.us 601 :UBC 0 : continue;
32 peter@eisentraut.org 602 :GNC 3695 : crc = crc32_sz(VARDATA(DatumGetPointer(key_datums[i])), VARSIZE(DatumGetPointer(key_datums[i])) - VARHDRSZ);
1986 akorotkov@postgresql 603 [ + + ]:CBC 3695 : if (GETBIT(sign, HASHVAL(crc, siglen)))
2943 peter_e@gmx.net 604 : 1208 : res = true;
605 : : }
606 : : }
607 : : else
6751 teodor@sigaev.ru 608 [ # # ]:UBC 0 : elog(ERROR, "Unsupported strategy number: %d", strategy);
609 : :
6941 teodor@sigaev.ru 610 :CBC 10490 : PG_RETURN_BOOL(res);
611 : : }
612 : :
613 : : Datum
1986 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 : : }
|
