PolarDB-for-PostgreSQL

1
/*
2
 * contrib/btree_gist/btree_oid.c
3
 */
4
#include "postgres.h"
5

6
#include "btree_gist.h"
7
#include "btree_utils_num.h"
8

9
typedef struct
10
{
11
	Oid			lower;
12
	Oid			upper;
13
} oidKEY;
14

15
/*
16
** OID ops
17
*/
18
PG_FUNCTION_INFO_V1(gbt_oid_compress);
19
PG_FUNCTION_INFO_V1(gbt_oid_fetch);
20
PG_FUNCTION_INFO_V1(gbt_oid_union);
21
PG_FUNCTION_INFO_V1(gbt_oid_picksplit);
22
PG_FUNCTION_INFO_V1(gbt_oid_consistent);
23
PG_FUNCTION_INFO_V1(gbt_oid_distance);
24
PG_FUNCTION_INFO_V1(gbt_oid_penalty);
25
PG_FUNCTION_INFO_V1(gbt_oid_same);
26

27

28
static bool
29
gbt_oidgt(const void *a, const void *b, FmgrInfo *flinfo)
30
{
31
	return (*((const Oid *) a) > *((const Oid *) b));
32
}
33
static bool
34
gbt_oidge(const void *a, const void *b, FmgrInfo *flinfo)
35
{
36
	return (*((const Oid *) a) >= *((const Oid *) b));
37
}
38
static bool
39
gbt_oideq(const void *a, const void *b, FmgrInfo *flinfo)
40
{
41
	return (*((const Oid *) a) == *((const Oid *) b));
42
}
43
static bool
44
gbt_oidle(const void *a, const void *b, FmgrInfo *flinfo)
45
{
46
	return (*((const Oid *) a) <= *((const Oid *) b));
47
}
48
static bool
49
gbt_oidlt(const void *a, const void *b, FmgrInfo *flinfo)
50
{
51
	return (*((const Oid *) a) < *((const Oid *) b));
52
}
53

54
static int
55
gbt_oidkey_cmp(const void *a, const void *b, FmgrInfo *flinfo)
56
{
57
	oidKEY	   *ia = (oidKEY *) (((const Nsrt *) a)->t);
58
	oidKEY	   *ib = (oidKEY *) (((const Nsrt *) b)->t);
59

60
	if (ia->lower == ib->lower)
61
	{
62
		if (ia->upper == ib->upper)
63
			return 0;
64

65
		return (ia->upper > ib->upper) ? 1 : -1;
66
	}
67

68
	return (ia->lower > ib->lower) ? 1 : -1;
69
}
70

71
static float8
72
gbt_oid_dist(const void *a, const void *b, FmgrInfo *flinfo)
73
{
74
	Oid			aa = *(const Oid *) a;
75
	Oid			bb = *(const Oid *) b;
76

77
	if (aa < bb)
78
		return (float8) (bb - aa);
79
	else
80
		return (float8) (aa - bb);
81
}
82

83

84
static const gbtree_ninfo tinfo =
85
{
86
	gbt_t_oid,
87
	sizeof(Oid),
88
	8,							/* sizeof(gbtreekey8) */
89
	gbt_oidgt,
90
	gbt_oidge,
91
	gbt_oideq,
92
	gbt_oidle,
93
	gbt_oidlt,
94
	gbt_oidkey_cmp,
95
	gbt_oid_dist
96
};
97

98

99
PG_FUNCTION_INFO_V1(oid_dist);
100
Datum
101
oid_dist(PG_FUNCTION_ARGS)
102
{
103
	Oid			a = PG_GETARG_OID(0);
104
	Oid			b = PG_GETARG_OID(1);
105
	Oid			res;
106

107
	if (a < b)
108
		res = b - a;
109
	else
110
		res = a - b;
111
	PG_RETURN_OID(res);
112
}
113

114

115
/**************************************************
116
 * Oid ops
117
 **************************************************/
118

119

120
Datum
121
gbt_oid_compress(PG_FUNCTION_ARGS)
122
{
123
	GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
124

125
	PG_RETURN_POINTER(gbt_num_compress(entry, &tinfo));
126
}
127

128
Datum
129
gbt_oid_fetch(PG_FUNCTION_ARGS)
130
{
131
	GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
132

133
	PG_RETURN_POINTER(gbt_num_fetch(entry, &tinfo));
134
}
135

136
Datum
137
gbt_oid_consistent(PG_FUNCTION_ARGS)
138
{
139
	GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
140
	Oid			query = PG_GETARG_OID(1);
141
	StrategyNumber strategy = (StrategyNumber) PG_GETARG_UINT16(2);
142

143
	/* Oid		subtype = PG_GETARG_OID(3); */
144
	bool	   *recheck = (bool *) PG_GETARG_POINTER(4);
145
	oidKEY	   *kkk = (oidKEY *) DatumGetPointer(entry->key);
146
	GBT_NUMKEY_R key;
147

148
	/* All cases served by this function are exact */
149
	*recheck = false;
150

151
	key.lower = (GBT_NUMKEY *) &kkk->lower;
152
	key.upper = (GBT_NUMKEY *) &kkk->upper;
153

154
	PG_RETURN_BOOL(
155
				   gbt_num_consistent(&key, (void *) &query, &strategy, GIST_LEAF(entry), &tinfo, fcinfo->flinfo)
156
		);
157
}
158

159

160
Datum
161
gbt_oid_distance(PG_FUNCTION_ARGS)
162
{
163
	GISTENTRY  *entry = (GISTENTRY *) PG_GETARG_POINTER(0);
164
	Oid			query = PG_GETARG_OID(1);
165

166
	/* Oid		subtype = PG_GETARG_OID(3); */
167
	oidKEY	   *kkk = (oidKEY *) DatumGetPointer(entry->key);
168
	GBT_NUMKEY_R key;
169

170
	key.lower = (GBT_NUMKEY *) &kkk->lower;
171
	key.upper = (GBT_NUMKEY *) &kkk->upper;
172

173
	PG_RETURN_FLOAT8(
174
					 gbt_num_distance(&key, (void *) &query, GIST_LEAF(entry), &tinfo, fcinfo->flinfo)
175
		);
176
}
177

178

179
Datum
180
gbt_oid_union(PG_FUNCTION_ARGS)
181
{
182
	GistEntryVector *entryvec = (GistEntryVector *) PG_GETARG_POINTER(0);
183
	void	   *out = palloc(sizeof(oidKEY));
184

185
	*(int *) PG_GETARG_POINTER(1) = sizeof(oidKEY);
186
	PG_RETURN_POINTER(gbt_num_union((void *) out, entryvec, &tinfo, fcinfo->flinfo));
187
}
188

189

190
Datum
191
gbt_oid_penalty(PG_FUNCTION_ARGS)
192
{
193
	oidKEY	   *origentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(0))->key);
194
	oidKEY	   *newentry = (oidKEY *) DatumGetPointer(((GISTENTRY *) PG_GETARG_POINTER(1))->key);
195
	float	   *result = (float *) PG_GETARG_POINTER(2);
196

197
	penalty_num(result, origentry->lower, origentry->upper, newentry->lower, newentry->upper);
198

199
	PG_RETURN_POINTER(result);
200
}
201

202
Datum
203
gbt_oid_picksplit(PG_FUNCTION_ARGS)
204
{
205
	PG_RETURN_POINTER(gbt_num_picksplit(
206
										(GistEntryVector *) PG_GETARG_POINTER(0),
207
										(GIST_SPLITVEC *) PG_GETARG_POINTER(1),
208
										&tinfo, fcinfo->flinfo
209
										));
210
}
211

212
Datum
213
gbt_oid_same(PG_FUNCTION_ARGS)
214
{
215
	oidKEY	   *b1 = (oidKEY *) PG_GETARG_POINTER(0);
216
	oidKEY	   *b2 = (oidKEY *) PG_GETARG_POINTER(1);
217
	bool	   *result = (bool *) PG_GETARG_POINTER(2);
218

219
	*result = gbt_num_same((void *) b1, (void *) b2, &tinfo, fcinfo->flinfo);
220
	PG_RETURN_POINTER(result);
221
}
222