Thanks to visit codestin.com
Credit goes to doxygen.postgresql.org

PostgreSQL Source Code git master
btree_date.c
Go to the documentation of this file.
1/*
2 * contrib/btree_gist/btree_date.c
3 */
4#include "postgres.h"
5
6#include "btree_gist.h"
7#include "btree_utils_num.h"
8#include "utils/fmgrprotos.h"
9#include "utils/date.h"
10#include "utils/rel.h"
11#include "utils/sortsupport.h"
12
13typedef struct
14{
17} dateKEY;
18
19/* GiST support functions */
29
30static bool
31gbt_dategt(const void *a, const void *b, FmgrInfo *flinfo)
32{
34 DateADTGetDatum(*((const DateADT *) a)),
35 DateADTGetDatum(*((const DateADT *) b))));
36}
37
38static bool
39gbt_datege(const void *a, const void *b, FmgrInfo *flinfo)
40{
42 DateADTGetDatum(*((const DateADT *) a)),
43 DateADTGetDatum(*((const DateADT *) b))));
44}
45
46static bool
47gbt_dateeq(const void *a, const void *b, FmgrInfo *flinfo)
48{
50 DateADTGetDatum(*((const DateADT *) a)),
51 DateADTGetDatum(*((const DateADT *) b)))
52 );
53}
54
55static bool
56gbt_datele(const void *a, const void *b, FmgrInfo *flinfo)
57{
59 DateADTGetDatum(*((const DateADT *) a)),
60 DateADTGetDatum(*((const DateADT *) b))));
61}
62
63static bool
64gbt_datelt(const void *a, const void *b, FmgrInfo *flinfo)
65{
67 DateADTGetDatum(*((const DateADT *) a)),
68 DateADTGetDatum(*((const DateADT *) b))));
69}
70
71
72
73static int
74gbt_datekey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
75{
76 dateKEY *ia = (dateKEY *) (((const Nsrt *) a)->t);
77 dateKEY *ib = (dateKEY *) (((const Nsrt *) b)->t);
78 int res;
79
83 if (res == 0)
87
88 return res;
89}
90
91static float8
92gdb_date_dist(const void *a, const void *b, FmgrInfo *flinfo)
93{
94 /* we assume the difference can't overflow */
96 DateADTGetDatum(*((const DateADT *) a)),
97 DateADTGetDatum(*((const DateADT *) b)));
98
99 return (float8) abs(DatumGetInt32(diff));
100}
101
102
103static const gbtree_ninfo tinfo =
104{
106 sizeof(DateADT),
107 8, /* sizeof(gbtreekey8) */
115};
116
117
119Datum
121{
122 /* we assume the difference can't overflow */
125 PG_GETARG_DATUM(1));
126
127 PG_RETURN_INT32(abs(DatumGetInt32(diff)));
128}
129
130
131/**************************************************
132 * GiST support functions
133 **************************************************/
134
135Datum
137{
138 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
139
141}
142
143Datum
145{
146 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
147
149}
150
151Datum
153{
154 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
155 DateADT query = PG_GETARG_DATEADT(1);
157
158 /* Oid subtype = PG_GETARG_OID(3); */
159 bool *recheck = (bool *) PG_GETARG_POINTER(4);
160 dateKEY *kkk = (dateKEY *) DatumGetPointer(entry->key);
162
163 /* All cases served by this function are exact */
164 *recheck = false;
165
166 key.lower = (GBT_NUMKEY *) &kkk->lower;
167 key.upper = (GBT_NUMKEY *) &kkk->upper;
168
169 PG_RETURN_BOOL(gbt_num_consistent(&key, &query, &strategy,
170 GIST_LEAF(entry), &tinfo,
171 fcinfo->flinfo));
172}
173
174Datum
176{
177 GISTENTRY *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
178 DateADT query = PG_GETARG_DATEADT(1);
179
180 /* Oid subtype = PG_GETARG_OID(3); */
181 dateKEY *kkk = (dateKEY *) DatumGetPointer(entry->key);
183
184 key.lower = (GBT_NUMKEY *) &kkk->lower;
185 key.upper = (GBT_NUMKEY *) &kkk->upper;
186
188 &tinfo, fcinfo->flinfo));
189}
190
191Datum
193{
195 void *out = palloc(sizeof(dateKEY));
196
197 *(int *) PG_GETARG_POINTER(1) = sizeof(dateKEY);
198 PG_RETURN_POINTER(gbt_num_union(out, entryvec, &tinfo, fcinfo->flinfo));
199}
200
201Datum
203{
204 dateKEY *origentry = (dateKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
206 float *result = (float *) PG_GETARG_POINTER(2);
207 int32 diff,
208 res;
209
211 DateADTGetDatum(newentry->upper),
212 DateADTGetDatum(origentry->upper)));
213
214 res = Max(diff, 0);
215
217 DateADTGetDatum(origentry->lower),
218 DateADTGetDatum(newentry->lower)));
219
220 res += Max(diff, 0);
221
222 *result = 0.0;
223
224 if (res > 0)
225 {
227 DateADTGetDatum(origentry->upper),
228 DateADTGetDatum(origentry->lower)));
229 *result += FLT_MIN;
230 *result += (float) (res / ((double) (res + diff)));
231 *result *= (FLT_MAX / (((GISTENTRY *) PG_GETARG_POINTER(0))->rel->rd_att->natts + 1));
232 }
233
234 PG_RETURN_POINTER(result);
235}
236
237Datum
239{
242 &tinfo, fcinfo->flinfo));
243}
244
245Datum
247{
248 dateKEY *b1 = (dateKEY *) PG_GETARG_POINTER(0);
249 dateKEY *b2 = (dateKEY *) PG_GETARG_POINTER(1);
250 bool *result = (bool *) PG_GETARG_POINTER(2);
251
252 *result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
253 PG_RETURN_POINTER(result);
254}
255
256static int
258{
259 dateKEY *akey = (dateKEY *) DatumGetPointer(x);
260 dateKEY *bkey = (dateKEY *) DatumGetPointer(y);
261
262 /* for leaf items we expect lower == upper, so only compare lower */
264 DateADTGetDatum(akey->lower),
265 DateADTGetDatum(bkey->lower)));
266}
267
268Datum
270{
272
274 ssup->ssup_extra = NULL;
275
277}
static bool gbt_datelt(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:64
Datum gbt_date_fetch(PG_FUNCTION_ARGS)
Definition: btree_date.c:144
Datum gbt_date_consistent(PG_FUNCTION_ARGS)
Definition: btree_date.c:152
Datum gbt_date_penalty(PG_FUNCTION_ARGS)
Definition: btree_date.c:202
static int gbt_date_ssup_cmp(Datum x, Datum y, SortSupport ssup)
Definition: btree_date.c:257
Datum gbt_date_union(PG_FUNCTION_ARGS)
Definition: btree_date.c:192
Datum gbt_date_sortsupport(PG_FUNCTION_ARGS)
Definition: btree_date.c:269
static bool gbt_datege(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:39
PG_FUNCTION_INFO_V1(gbt_date_compress)
static int gbt_datekey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:74
static float8 gdb_date_dist(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:92
Datum gbt_date_distance(PG_FUNCTION_ARGS)
Definition: btree_date.c:175
static const gbtree_ninfo tinfo
Definition: btree_date.c:103
static bool gbt_dategt(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:31
Datum gbt_date_same(PG_FUNCTION_ARGS)
Definition: btree_date.c:246
Datum gbt_date_compress(PG_FUNCTION_ARGS)
Definition: btree_date.c:136
static bool gbt_datele(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:56
Datum date_dist(PG_FUNCTION_ARGS)
Definition: btree_date.c:120
static bool gbt_dateeq(const void *a, const void *b, FmgrInfo *flinfo)
Definition: btree_date.c:47
Datum gbt_date_picksplit(PG_FUNCTION_ARGS)
Definition: btree_date.c:238
@ gbt_t_date
Definition: btree_gist.h:27
GISTENTRY * gbt_num_compress(GISTENTRY *entry, const gbtree_ninfo *tinfo)
float8 gbt_num_distance(const GBT_NUMKEY_R *key, const void *query, bool is_leaf, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
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)
GISTENTRY * gbt_num_fetch(GISTENTRY *entry, const gbtree_ninfo *tinfo)
GIST_SPLITVEC * gbt_num_picksplit(const GistEntryVector *entryvec, GIST_SPLITVEC *v, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
void * gbt_num_union(GBT_NUMKEY *out, const GistEntryVector *entryvec, const gbtree_ninfo *tinfo, FmgrInfo *flinfo)
char GBT_NUMKEY
#define Max(x, y)
Definition: c.h:998
double float8
Definition: c.h:636
int32_t int32
Definition: c.h:535
Datum date_cmp(PG_FUNCTION_ARGS)
Definition: date.c:445
Datum date_le(PG_FUNCTION_ARGS)
Definition: date.c:418
Datum date_eq(PG_FUNCTION_ARGS)
Definition: date.c:391
Datum date_gt(PG_FUNCTION_ARGS)
Definition: date.c:427
Datum date_lt(PG_FUNCTION_ARGS)
Definition: date.c:409
Datum date_ge(PG_FUNCTION_ARGS)
Definition: date.c:436
Datum date_mi(PG_FUNCTION_ARGS)
Definition: date.c:552
static Datum DateADTGetDatum(DateADT X)
Definition: date.h:72
int32 DateADT
Definition: date.h:23
#define PG_GETARG_DATEADT(n)
Definition: date.h:89
#define PG_RETURN_VOID()
Definition: fmgr.h:349
#define DirectFunctionCall2(func, arg1, arg2)
Definition: fmgr.h:684
#define PG_RETURN_FLOAT8(x)
Definition: fmgr.h:367
#define PG_GETARG_POINTER(n)
Definition: fmgr.h:276
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:268
#define PG_GETARG_UINT16(n)
Definition: fmgr.h:272
#define PG_RETURN_INT32(x)
Definition: fmgr.h:354
#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:171
int y
Definition: isn.c:76
int b
Definition: isn.c:74
int x
Definition: isn.c:75
int a
Definition: isn.c:73
void * palloc(Size size)
Definition: mcxt.c:1365
static bool DatumGetBool(Datum X)
Definition: postgres.h:100
uint64_t Datum
Definition: postgres.h:70
static Pointer DatumGetPointer(Datum X)
Definition: postgres.h:322
static int32 DatumGetInt32(Datum X)
Definition: postgres.h:212
struct SortSupportData * SortSupport
Definition: sortsupport.h:58
uint16 StrategyNumber
Definition: stratnum.h:22
Definition: fmgr.h:57
Datum key
Definition: gist.h:161
int(* comparator)(Datum x, Datum y, SortSupport ssup)
Definition: sortsupport.h:106
void * ssup_extra
Definition: sortsupport.h:87
DateADT upper
Definition: btree_date.c:16
DateADT lower
Definition: btree_date.c:15