PostgreSQL Source Code  git master
btree_uuid.c
Go to the documentation of this file.
1 /*
2  * contrib/btree_gist/btree_uuid.c
3  */
4 #include "postgres.h"
5 
6 #include "btree_gist.h"
7 #include "btree_utils_num.h"
8 #include "port/pg_bswap.h"
9 #include "utils/uuid.h"
10 
11 typedef struct
12 {
15 } uuidKEY;
16 
17 
18 /*
19  * UUID ops
20  */
28 
29 
30 static int
31 uuid_internal_cmp(const pg_uuid_t *arg1, const pg_uuid_t *arg2)
32 {
33  return memcmp(arg1->data, arg2->data, UUID_LEN);
34 }
35 
36 static bool
37 gbt_uuidgt(const void *a, const void *b, FmgrInfo *flinfo)
38 {
39  return uuid_internal_cmp((const pg_uuid_t *) a, (const pg_uuid_t *) b) > 0;
40 }
41 
42 static bool
43 gbt_uuidge(const void *a, const void *b, FmgrInfo *flinfo)
44 {
45  return uuid_internal_cmp((const pg_uuid_t *) a, (const pg_uuid_t *) b) >= 0;
46 }
47 
48 static bool
49 gbt_uuideq(const void *a, const void *b, FmgrInfo *flinfo)
50 {
51  return uuid_internal_cmp((const pg_uuid_t *) a, (const pg_uuid_t *) b) == 0;
52 }
53 
54 static bool
55 gbt_uuidle(const void *a, const void *b, FmgrInfo *flinfo)
56 {
57  return uuid_internal_cmp((const pg_uuid_t *) a, (const pg_uuid_t *) b) <= 0;
58 }
59 
60 static bool
61 gbt_uuidlt(const void *a, const void *b, FmgrInfo *flinfo)
62 {
63  return uuid_internal_cmp((const pg_uuid_t *) a, (const pg_uuid_t *) b) < 0;
64 }
65 
66 static int
67 gbt_uuidkey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
68 {
69  uuidKEY *ia = (uuidKEY *) (((const Nsrt *) a)->t);
70  uuidKEY *ib = (uuidKEY *) (((const Nsrt *) b)->t);
71  int res;
72 
73  res = uuid_internal_cmp(&ia->lower, &ib->lower);
74  if (res == 0)
75  res = uuid_internal_cmp(&ia->upper, &ib->upper);
76  return res;
77 }
78 
79 
80 static const gbtree_ninfo tinfo =
81 {
82  gbt_t_uuid,
83  UUID_LEN,
84  32, /* sizeof(gbtreekey32) */
85  gbt_uuidgt,
86  gbt_uuidge,
87  gbt_uuideq,
88  gbt_uuidle,
89  gbt_uuidlt,
91  NULL
92 };
93 
94 
95 /**************************************************
96  * uuid ops
97  **************************************************/
98 
99 
100 Datum
102 {
103  GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
104  GISTENTRY *retval;
105 
106  if (entry->leafkey)
107  {
108  char *r = (char *) palloc(2 * UUID_LEN);
109  pg_uuid_t *key = DatumGetUUIDP(entry->key);
110 
111  retval = palloc(sizeof(GISTENTRY));
112 
113  memcpy(r, key, UUID_LEN);
114  memcpy(r + UUID_LEN, key, UUID_LEN);
115  gistentryinit(*retval, PointerGetDatum(r),
116  entry->rel, entry->page,
117  entry->offset, false);
118  }
119  else
120  retval = entry;
121 
122  PG_RETURN_POINTER(retval);
123 }
124 
125 Datum
127 {
128  GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
129 
131 }
132 
133 Datum
135 {
136  GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
137  pg_uuid_t *query = PG_GETARG_UUID_P(1);
139 
140  /* Oid subtype = PG_GETARG_OID(3); */
141  bool *recheck = (bool *) PG_GETARG_POINTER(4);
142  uuidKEY *kkk = (uuidKEY *) DatumGetPointer(entry->key);
144 
145  /* All cases served by this function are exact */
146  *recheck = false;
147 
148  key.lower = (GBT_NUMKEY *) &kkk->lower;
149  key.upper = (GBT_NUMKEY *) &kkk->upper;
150 
151  PG_RETURN_BOOL(gbt_num_consistent(&key, (void *) query, &strategy,
152  GIST_LEAF(entry), &tinfo,
153  fcinfo->flinfo));
154 }
155 
156 Datum
158 {
160  void *out = palloc(sizeof(uuidKEY));
161 
162  *(int *) PG_GETARG_POINTER(1) = sizeof(uuidKEY);
163  PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo, fcinfo->flinfo));
164 }
165 
166 /*
167  * Convert a uuid to a "double" value for estimating sizes of ranges.
168  */
169 static double
171 {
172  uint64 uu[2];
173  const double two64 = 18446744073709551616.0; /* 2^64 */
174 
175  /* Source data may not be suitably aligned, so copy */
176  memcpy(uu, u->data, UUID_LEN);
177 
178  /*
179  * uuid values should be considered as big-endian numbers, since that
180  * corresponds to how memcmp will compare them. On a little-endian
181  * machine, byte-swap each half so we can use native uint64 arithmetic.
182  */
183 #ifndef WORDS_BIGENDIAN
184  uu[0] = pg_bswap64(uu[0]);
185  uu[1] = pg_bswap64(uu[1]);
186 #endif
187 
188  /*
189  * 2^128 is about 3.4e38, which in theory could exceed the range of
190  * "double" (POSIX only requires 1e37). To avoid any risk of overflow,
191  * put the decimal point between the two halves rather than treating the
192  * uuid value as a 128-bit integer.
193  */
194  return (double) uu[0] + (double) uu[1] / two64;
195 }
196 
197 Datum
199 {
200  uuidKEY *origentry = (uuidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
201  uuidKEY *newentry = (uuidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
202  float *result = (float *) PG_GETARG_POINTER(2);
203  double olower,
204  oupper,
205  nlower,
206  nupper;
207 
208  olower = uuid_2_double(&origentry->lower);
209  oupper = uuid_2_double(&origentry->upper);
210  nlower = uuid_2_double(&newentry->lower);
211  nupper = uuid_2_double(&newentry->upper);
212 
213  penalty_num(result, olower, oupper, nlower, nupper);
214 
215  PG_RETURN_POINTER(result);
216 }
217 
218 Datum
220 {
223  &tinfo, fcinfo->flinfo));
224 }
225 
226 Datum
228 {
229  uuidKEY *b1 = (uuidKEY *) PG_GETARG_POINTER(0);
230  uuidKEY *b2 = (uuidKEY *) PG_GETARG_POINTER(1);
231  bool *result = (bool *) PG_GETARG_POINTER(2);
232 
233  *result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
234  PG_RETURN_POINTER(result);
235 }
@ gbt_t_uuid
Definition: btree_gist.h:37
GISTENTRY * gbt_num_fetch(GISTENTRY *entry, const gbtree_ninfo *tinfo)
bool gbt_num_consistent(const GBT_NUMKEY_R *key, const void *query, const StrategyNumber *strategy, bool is_leaf, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
bool gbt_num_same(const GBT_NUMKEY *a, const GBT_NUMKEY *b, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
void * gbt_num_union(GBT_NUMKEY *out, const GistEntryVector *entryvec, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
GIST_SPLITVEC * gbt_num_picksplit(const GistEntryVector *entryvec, GIST_SPLITVEC *v, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
#define penalty_num(result, olower, oupper, nlower, nupper)
char GBT_NUMKEY
Datum gbt_uuid_consistent(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:134
Datum gbt_uuid_same(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:227
static bool gbt_uuidle(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:55
static bool gbt_uuideq(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:49
static bool gbt_uuidgt(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:37
Datum gbt_uuid_union(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:157
Datum gbt_uuid_penalty(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:198
static double uuid_2_double(const pg_uuid_t *u)
Definition: btree_uuid.c:170
PG_FUNCTION_INFO_V1(gbt_uuid_compress)
Datum gbt_uuid_picksplit(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:219
static const gbtree_ninfo tinfo
Definition: btree_uuid.c:80
Datum gbt_uuid_fetch(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:126
static int uuid_internal_cmp(const pg_uuid_t *arg1, const pg_uuid_t *arg2)
Definition: btree_uuid.c:31
static bool gbt_uuidge(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:43
static bool gbt_uuidlt(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:61
Datum gbt_uuid_compress(PG_FUNCTION_ARGS)
Definition: btree_uuid.c:101
static int gbt_uuidkey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_uuid.c:67
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_GETARG_UINT16(n)
Definition: fmgr.h:272
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:361
#define PG_FUNCTION_ARGS
Definition: fmgr.h:193
#define PG_RETURN_BOOL(x)
Definition: fmgr.h:359
#define GIST_LEAF(entry)
Definition: gist.h:170
#define gistentryinit(e, k, r, pg, o, l)
Definition: gist.h:244
int b
Definition: isn.c:69
int a
Definition: isn.c:68
void * palloc(Size size)
Definition: mcxt.c:1317
static uint64 pg_bswap64(uint64 x)
Definition: pg_bswap.h:89
static Datum PointerGetDatum(const void *X)
Definition: postgres.h:322
uintptr_t Datum
Definition: postgres.h:64
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:312
uint16 StrategyNumber
Definition: stratnum.h:22
Definition: fmgr.h:57
OffsetNumber offset
Definition: gist.h:163
Datum key
Definition: gist.h:160
Page page
Definition: gist.h:162
Relation rel
Definition: gist.h:161
bool leafkey
Definition: gist.h:164
Definition: uuid.h:21
unsigned char data[UUID_LEN]
Definition: uuid.h:22
pg_uuid_t upper
Definition: btree_uuid.c:14
pg_uuid_t lower
Definition: btree_uuid.c:13
#define UUID_LEN
Definition: uuid.h:18
static pg_uuid_t * DatumGetUUIDP(Datum X)
Definition: uuid.h:35
#define PG_GETARG_UUID_P(X)
Definition: uuid.h:40