1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
# Copyright (C) 2016-present the asyncpg authors and contributors
# <see AUTHORS file>
#
# This module is part of asyncpg and is released under
# the Apache 2.0 License: http://www.apache.org/licenses/LICENSE-2.0
cdef jsonb_encode(CodecContext settings, WriteBuffer buf, obj):
cdef:
char *str
ssize_t size
if settings.is_encoding_json():
obj = settings.get_json_encoder().encode(obj)
as_pg_string_and_size(settings, obj, &str, &size)
if size > 0x7fffffff - 1:
raise ValueError('string too long')
buf.write_int32(<int32_t>size + 1)
buf.write_byte(1) # JSONB format version
buf.write_cstr(str, size)
cdef jsonb_decode(CodecContext settings, FRBuffer *buf):
cdef uint8_t format = <uint8_t>(frb_read(buf, 1)[0])
if format != 1:
raise ValueError('unexpected JSONB format: {}'.format(format))
rv = text_decode(settings, buf)
if settings.is_decoding_json():
rv = settings.get_json_decoder().decode(rv)
return rv
cdef json_encode(CodecContext settings, WriteBuffer buf, obj):
cdef:
char *str
ssize_t size
if settings.is_encoding_json():
obj = settings.get_json_encoder().encode(obj)
text_encode(settings, buf, obj)
cdef json_decode(CodecContext settings, FRBuffer *buf):
rv = text_decode(settings, buf)
if settings.is_decoding_json():
rv = settings.get_json_decoder().decode(rv)
return rv
|
