[フレーム]

jsonb_op.c File Reference

#include "postgres.h"
#include "catalog/pg_type.h"
#include "utils/fmgrprotos.h"
#include "utils/jsonb.h"

Include dependency graph for jsonb_op.c:

Go to the source code of this file.

Functions

Datum jsonb_exists (PG_FUNCTION_ARGS)

Datum jsonb_exists_any (PG_FUNCTION_ARGS)

Datum jsonb_exists_all (PG_FUNCTION_ARGS)

Datum jsonb_contains (PG_FUNCTION_ARGS)

Datum jsonb_contained (PG_FUNCTION_ARGS)

Datum jsonb_ne (PG_FUNCTION_ARGS)

Datum jsonb_lt (PG_FUNCTION_ARGS)

Datum jsonb_gt (PG_FUNCTION_ARGS)

Datum jsonb_le (PG_FUNCTION_ARGS)

Datum jsonb_ge (PG_FUNCTION_ARGS)

Datum jsonb_eq (PG_FUNCTION_ARGS)

Datum jsonb_cmp (PG_FUNCTION_ARGS)

Datum jsonb_hash (PG_FUNCTION_ARGS)

Datum jsonb_hash_extended (PG_FUNCTION_ARGS)

Function Documentation

◆ jsonb_cmp()

Datum jsonb_cmp ( PG_FUNCTION_ARGS )

Definition at line 236 of file jsonb_op.c.

237{

238 Jsonb *jba = PG_GETARG_JSONB_P(0);

239 Jsonb *jbb = PG_GETARG_JSONB_P(1);

240 int res;

241

242 res = compareJsonbContainers(&jba->root, &jbb->root);

243

244 PG_FREE_IF_COPY(jba, 0);

245 PG_FREE_IF_COPY(jbb, 1);

246 PG_RETURN_INT32(res);

247}

PG_FREE_IF_COPY

#define PG_FREE_IF_COPY(ptr, n)

Definition: fmgr.h:260

PG_RETURN_INT32

#define PG_RETURN_INT32(x)

Definition: fmgr.h:354

PG_GETARG_JSONB_P

#define PG_GETARG_JSONB_P(x)

Definition: jsonb.h:391

compareJsonbContainers

int compareJsonbContainers(JsonbContainer *a, JsonbContainer *b)

Definition: jsonb_util.c:191

Jsonb

Definition: jsonb.h:213

Jsonb::root

JsonbContainer root

Definition: jsonb.h:215

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_INT32, and Jsonb::root.

◆ jsonb_contained()

Datum jsonb_contained ( PG_FUNCTION_ARGS )

Definition at line 130 of file jsonb_op.c.

131{

132 /* Commutator of "contains" */

133 Jsonb *tmpl = PG_GETARG_JSONB_P(0);

134 Jsonb *val = PG_GETARG_JSONB_P(1);

135

136 JsonbIterator *it1,

137 *it2;

138

139 if (JB_ROOT_IS_OBJECT(val) != JB_ROOT_IS_OBJECT(tmpl))

140 PG_RETURN_BOOL(false);

141

142 it1 = JsonbIteratorInit(&val->root);

143 it2 = JsonbIteratorInit(&tmpl->root);

144

145 PG_RETURN_BOOL(JsonbDeepContains(&it1, &it2));

146}

PG_RETURN_BOOL

#define PG_RETURN_BOOL(x)

Definition: fmgr.h:359

val

long val

Definition: informix.c:689

JB_ROOT_IS_OBJECT

#define JB_ROOT_IS_OBJECT(jbp_)

Definition: jsonb.h:221

JsonbIteratorInit

JsonbIterator * JsonbIteratorInit(JsonbContainer *container)

Definition: jsonb_util.c:818

JsonbDeepContains

bool JsonbDeepContains(JsonbIterator **val, JsonbIterator **mContained)

Definition: jsonb_util.c:1072

JsonbIterator

Definition: jsonb.h:342

References JB_ROOT_IS_OBJECT, JsonbDeepContains(), JsonbIteratorInit(), PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, and val.

◆ jsonb_contains()

Datum jsonb_contains ( PG_FUNCTION_ARGS )

Definition at line 112 of file jsonb_op.c.

113{

114 Jsonb *val = PG_GETARG_JSONB_P(0);

115 Jsonb *tmpl = PG_GETARG_JSONB_P(1);

116

117 JsonbIterator *it1,

118 *it2;

119

120 if (JB_ROOT_IS_OBJECT(val) != JB_ROOT_IS_OBJECT(tmpl))

121 PG_RETURN_BOOL(false);

122

123 it1 = JsonbIteratorInit(&val->root);

124 it2 = JsonbIteratorInit(&tmpl->root);

125

126 PG_RETURN_BOOL(JsonbDeepContains(&it1, &it2));

127}

References JB_ROOT_IS_OBJECT, JsonbDeepContains(), JsonbIteratorInit(), PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, and val.

◆ jsonb_eq()

Datum jsonb_eq ( PG_FUNCTION_ARGS )

Definition at line 222 of file jsonb_op.c.

223{

224 Jsonb *jba = PG_GETARG_JSONB_P(0);

225 Jsonb *jbb = PG_GETARG_JSONB_P(1);

226 bool res;

227

228 res = (compareJsonbContainers(&jba->root, &jbb->root) == 0);

229

230 PG_FREE_IF_COPY(jba, 0);

231 PG_FREE_IF_COPY(jbb, 1);

232 PG_RETURN_BOOL(res);

233}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

◆ jsonb_exists()

Datum jsonb_exists ( PG_FUNCTION_ARGS )

Definition at line 21 of file jsonb_op.c.

22{

23 Jsonb *jb = PG_GETARG_JSONB_P(0);

24 text *key = PG_GETARG_TEXT_PP(1);

25 JsonbValue kval;

26 JsonbValue *v = NULL;

27

28 /*

29 * We only match Object keys (which are naturally always Strings), or

30 * string elements in arrays. In particular, we do not match non-string

31 * scalar elements. Existence of a key/element is only considered at the

32 * top level. No recursion occurs.

33 */

34 kval.type = jbvString;

35 kval.val.string.val = VARDATA_ANY(key);

36 kval.val.string.len = VARSIZE_ANY_EXHDR(key);

37

38 v = findJsonbValueFromContainer(&jb->root,

39 JB_FOBJECT | JB_FARRAY,

40 &kval);

41

42 PG_RETURN_BOOL(v != NULL);

43}

PG_GETARG_TEXT_PP

#define PG_GETARG_TEXT_PP(n)

Definition: fmgr.h:309

jbvString

@ jbvString

Definition: jsonb.h:229

JB_FARRAY

#define JB_FARRAY

Definition: jsonb.h:203

JB_FOBJECT

#define JB_FOBJECT

Definition: jsonb.h:202

findJsonbValueFromContainer

JsonbValue * findJsonbValueFromContainer(JsonbContainer *container, uint32 flags, JsonbValue *key)

Definition: jsonb_util.c:345

sort-test.key

key

Definition: sort-test.py:19

JsonbValue

Definition: jsonb.h:254

JsonbValue::type

enum jbvType type

Definition: jsonb.h:255

JsonbValue::val

char * val

Definition: jsonb.h:264

varlena

Definition: c.h:692

VARSIZE_ANY_EXHDR

static Size VARSIZE_ANY_EXHDR(const void *PTR)

Definition: varatt.h:472

VARDATA_ANY

static char * VARDATA_ANY(const void *PTR)

Definition: varatt.h:486

References findJsonbValueFromContainer(), JB_FARRAY, JB_FOBJECT, jbvString, sort-test::key, PG_GETARG_JSONB_P, PG_GETARG_TEXT_PP, PG_RETURN_BOOL, Jsonb::root, JsonbValue::type, JsonbValue::val, VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

◆ jsonb_exists_all()

Datum jsonb_exists_all ( PG_FUNCTION_ARGS )

Definition at line 79 of file jsonb_op.c.

80{

81 Jsonb *jb = PG_GETARG_JSONB_P(0);

82 ArrayType *keys = PG_GETARG_ARRAYTYPE_P(1);

83 int i;

84 Datum *key_datums;

85 bool *key_nulls;

86 int elem_count;

87

88 deconstruct_array_builtin(keys, TEXTOID, &key_datums, &key_nulls, &elem_count);

89

90 for (i = 0; i < elem_count; i++)

91 {

92 JsonbValue strVal;

93

94 if (key_nulls[i])

95 continue;

96

97 strVal.type = jbvString;

98 /* We rely on the array elements not being toasted */

99 strVal.val.string.val = VARDATA_ANY(DatumGetPointer(key_datums[i]));

100 strVal.val.string.len = VARSIZE_ANY_EXHDR(DatumGetPointer(key_datums[i]));

101

102 if (findJsonbValueFromContainer(&jb->root,

103 JB_FOBJECT | JB_FARRAY,

104 &strVal) == NULL)

105 PG_RETURN_BOOL(false);

106 }

107

108 PG_RETURN_BOOL(true);

109}

PG_GETARG_ARRAYTYPE_P

#define PG_GETARG_ARRAYTYPE_P(n)

Definition: array.h:263

deconstruct_array_builtin

void deconstruct_array_builtin(ArrayType *array, Oid elmtype, Datum **elemsp, bool **nullsp, int *nelemsp)

Definition: arrayfuncs.c:3698

i

int i

Definition: isn.c:77

Datum

uint64_t Datum

Definition: postgres.h:70

DatumGetPointer

static Pointer DatumGetPointer(Datum X)

Definition: postgres.h:322

ArrayType

Definition: array.h:93

strVal

#define strVal(v)

Definition: value.h:82

References DatumGetPointer(), deconstruct_array_builtin(), findJsonbValueFromContainer(), i, JB_FARRAY, JB_FOBJECT, jbvString, PG_GETARG_ARRAYTYPE_P, PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, strVal, VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

◆ jsonb_exists_any()

Datum jsonb_exists_any ( PG_FUNCTION_ARGS )

Definition at line 46 of file jsonb_op.c.

47{

48 Jsonb *jb = PG_GETARG_JSONB_P(0);

49 ArrayType *keys = PG_GETARG_ARRAYTYPE_P(1);

50 int i;

51 Datum *key_datums;

52 bool *key_nulls;

53 int elem_count;

54

55 deconstruct_array_builtin(keys, TEXTOID, &key_datums, &key_nulls, &elem_count);

56

57 for (i = 0; i < elem_count; i++)

58 {

59 JsonbValue strVal;

60

61 if (key_nulls[i])

62 continue;

63

64 strVal.type = jbvString;

65 /* We rely on the array elements not being toasted */

66 strVal.val.string.val = VARDATA_ANY(DatumGetPointer(key_datums[i]));

67 strVal.val.string.len = VARSIZE_ANY_EXHDR(DatumGetPointer(key_datums[i]));

68

69 if (findJsonbValueFromContainer(&jb->root,

70 JB_FOBJECT | JB_FARRAY,

71 &strVal) != NULL)

72 PG_RETURN_BOOL(true);

73 }

74

75 PG_RETURN_BOOL(false);

76}

References DatumGetPointer(), deconstruct_array_builtin(), findJsonbValueFromContainer(), i, JB_FARRAY, JB_FOBJECT, jbvString, PG_GETARG_ARRAYTYPE_P, PG_GETARG_JSONB_P, PG_RETURN_BOOL, Jsonb::root, strVal, VARDATA_ANY(), and VARSIZE_ANY_EXHDR().

◆ jsonb_ge()

Datum jsonb_ge ( PG_FUNCTION_ARGS )

Definition at line 208 of file jsonb_op.c.

209{

210 Jsonb *jba = PG_GETARG_JSONB_P(0);

211 Jsonb *jbb = PG_GETARG_JSONB_P(1);

212 bool res;

213

214 res = (compareJsonbContainers(&jba->root, &jbb->root) >= 0);

215

216 PG_FREE_IF_COPY(jba, 0);

217 PG_FREE_IF_COPY(jbb, 1);

218 PG_RETURN_BOOL(res);

219}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

◆ jsonb_gt()

Datum jsonb_gt ( PG_FUNCTION_ARGS )

Definition at line 180 of file jsonb_op.c.

181{

182 Jsonb *jba = PG_GETARG_JSONB_P(0);

183 Jsonb *jbb = PG_GETARG_JSONB_P(1);

184 bool res;

185

186 res = (compareJsonbContainers(&jba->root, &jbb->root) > 0);

187

188 PG_FREE_IF_COPY(jba, 0);

189 PG_FREE_IF_COPY(jbb, 1);

190 PG_RETURN_BOOL(res);

191}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

◆ jsonb_hash()

Datum jsonb_hash ( PG_FUNCTION_ARGS )

Definition at line 253 of file jsonb_op.c.

254{

255 Jsonb *jb = PG_GETARG_JSONB_P(0);

256 JsonbIterator *it;

257 JsonbValue v;

258 JsonbIteratorToken r;

259 uint32 hash = 0;

260

261 if (JB_ROOT_COUNT(jb) == 0)

262 PG_RETURN_INT32(0);

263

264 it = JsonbIteratorInit(&jb->root);

265

266 while ((r = JsonbIteratorNext(&it, &v, false)) != WJB_DONE)

267 {

268 switch (r)

269 {

270 /* Rotation is left to JsonbHashScalarValue() */

271 case WJB_BEGIN_ARRAY:

272 hash ^= JB_FARRAY;

273 break;

274 case WJB_BEGIN_OBJECT:

275 hash ^= JB_FOBJECT;

276 break;

277 case WJB_KEY:

278 case WJB_VALUE:

279 case WJB_ELEM:

280 JsonbHashScalarValue(&v, &hash);

281 break;

282 case WJB_END_ARRAY:

283 case WJB_END_OBJECT:

284 break;

285 default:

286 elog(ERROR, "invalid JsonbIteratorNext rc: %d", (int) r);

287 }

288 }

289

290 PG_FREE_IF_COPY(jb, 0);

291 PG_RETURN_INT32(hash);

292}

uint32

uint32_t uint32

Definition: c.h:538

ERROR

#define ERROR

Definition: elog.h:39

elog

#define elog(elevel,...)

Definition: elog.h:226

JsonbIteratorToken

Definition: jsonb.h:21

WJB_KEY

@ WJB_KEY

Definition: jsonb.h:23

WJB_DONE

@ WJB_DONE

Definition: jsonb.h:22

WJB_END_ARRAY

@ WJB_END_ARRAY

Definition: jsonb.h:27

WJB_VALUE

@ WJB_VALUE

Definition: jsonb.h:24

WJB_END_OBJECT

@ WJB_END_OBJECT

Definition: jsonb.h:29

WJB_ELEM

@ WJB_ELEM

Definition: jsonb.h:25

WJB_BEGIN_OBJECT

@ WJB_BEGIN_OBJECT

Definition: jsonb.h:28

WJB_BEGIN_ARRAY

@ WJB_BEGIN_ARRAY

Definition: jsonb.h:26

JB_ROOT_COUNT

#define JB_ROOT_COUNT(jbp_)

Definition: jsonb.h:219

JsonbHashScalarValue

void JsonbHashScalarValue(const JsonbValue *scalarVal, uint32 *hash)

Definition: jsonb_util.c:1326

JsonbIteratorNext

JsonbIteratorToken JsonbIteratorNext(JsonbIterator **it, JsonbValue *val, bool skipNested)

Definition: jsonb_util.c:856

hash

static unsigned hash(unsigned *uv, int n)

Definition: rege_dfa.c:715

References elog, ERROR, hash(), JB_FARRAY, JB_FOBJECT, JB_ROOT_COUNT, JsonbHashScalarValue(), JsonbIteratorInit(), JsonbIteratorNext(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_INT32, Jsonb::root, WJB_BEGIN_ARRAY, WJB_BEGIN_OBJECT, WJB_DONE, WJB_ELEM, WJB_END_ARRAY, WJB_END_OBJECT, WJB_KEY, and WJB_VALUE.

◆ jsonb_hash_extended()

Datum jsonb_hash_extended ( PG_FUNCTION_ARGS )

Definition at line 295 of file jsonb_op.c.

296{

297 Jsonb *jb = PG_GETARG_JSONB_P(0);

298 uint64 seed = PG_GETARG_INT64(1);

299 JsonbIterator *it;

300 JsonbValue v;

301 JsonbIteratorToken r;

302 uint64 hash = 0;

303

304 if (JB_ROOT_COUNT(jb) == 0)

305 PG_RETURN_UINT64(seed);

306

307 it = JsonbIteratorInit(&jb->root);

308

309 while ((r = JsonbIteratorNext(&it, &v, false)) != WJB_DONE)

310 {

311 switch (r)

312 {

313 /* Rotation is left to JsonbHashScalarValueExtended() */

314 case WJB_BEGIN_ARRAY:

315 hash ^= ((uint64) JB_FARRAY) << 32 | JB_FARRAY;

316 break;

317 case WJB_BEGIN_OBJECT:

318 hash ^= ((uint64) JB_FOBJECT) << 32 | JB_FOBJECT;

319 break;

320 case WJB_KEY:

321 case WJB_VALUE:

322 case WJB_ELEM:

323 JsonbHashScalarValueExtended(&v, &hash, seed);

324 break;

325 case WJB_END_ARRAY:

326 case WJB_END_OBJECT:

327 break;

328 default:

329 elog(ERROR, "invalid JsonbIteratorNext rc: %d", (int) r);

330 }

331 }

332

333 PG_FREE_IF_COPY(jb, 0);

334 PG_RETURN_UINT64(hash);

335}

uint64

uint64_t uint64

Definition: c.h:539

PG_GETARG_INT64

#define PG_GETARG_INT64(n)

Definition: fmgr.h:283

PG_RETURN_UINT64

#define PG_RETURN_UINT64(x)

Definition: fmgr.h:369

JsonbHashScalarValueExtended

void JsonbHashScalarValueExtended(const JsonbValue *scalarVal, uint64 *hash, uint64 seed)

Definition: jsonb_util.c:1369

References elog, ERROR, hash(), JB_FARRAY, JB_FOBJECT, JB_ROOT_COUNT, JsonbHashScalarValueExtended(), JsonbIteratorInit(), JsonbIteratorNext(), PG_FREE_IF_COPY, PG_GETARG_INT64, PG_GETARG_JSONB_P, PG_RETURN_UINT64, Jsonb::root, WJB_BEGIN_ARRAY, WJB_BEGIN_OBJECT, WJB_DONE, WJB_ELEM, WJB_END_ARRAY, WJB_END_OBJECT, WJB_KEY, and WJB_VALUE.

◆ jsonb_le()

Datum jsonb_le ( PG_FUNCTION_ARGS )

Definition at line 194 of file jsonb_op.c.

195{

196 Jsonb *jba = PG_GETARG_JSONB_P(0);

197 Jsonb *jbb = PG_GETARG_JSONB_P(1);

198 bool res;

199

200 res = (compareJsonbContainers(&jba->root, &jbb->root) <= 0);

201

202 PG_FREE_IF_COPY(jba, 0);

203 PG_FREE_IF_COPY(jbb, 1);

204 PG_RETURN_BOOL(res);

205}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

◆ jsonb_lt()

Datum jsonb_lt ( PG_FUNCTION_ARGS )

Definition at line 166 of file jsonb_op.c.

167{

168 Jsonb *jba = PG_GETARG_JSONB_P(0);

169 Jsonb *jbb = PG_GETARG_JSONB_P(1);

170 bool res;

171

172 res = (compareJsonbContainers(&jba->root, &jbb->root) < 0);

173

174 PG_FREE_IF_COPY(jba, 0);

175 PG_FREE_IF_COPY(jbb, 1);

176 PG_RETURN_BOOL(res);

177}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.

◆ jsonb_ne()

Datum jsonb_ne ( PG_FUNCTION_ARGS )

Definition at line 149 of file jsonb_op.c.

150{

151 Jsonb *jba = PG_GETARG_JSONB_P(0);

152 Jsonb *jbb = PG_GETARG_JSONB_P(1);

153 bool res;

154

155 res = (compareJsonbContainers(&jba->root, &jbb->root) != 0);

156

157 PG_FREE_IF_COPY(jba, 0);

158 PG_FREE_IF_COPY(jbb, 1);

159 PG_RETURN_BOOL(res);

160}

References compareJsonbContainers(), PG_FREE_IF_COPY, PG_GETARG_JSONB_P, PG_RETURN_BOOL, and Jsonb::root.