mirror of
https://github.com/vlang/v.git
synced 2025-09-13 22:42:26 +03:00
x.json2: optimise encoding to be faster than cJSON with -prod (#20052)
This commit is contained in:
parent
c4180d4b06
commit
7fc31591ad
9 changed files with 310 additions and 203 deletions
|
@ -113,7 +113,13 @@ fn (upd VlsUpdater) update_manifest(new_path string, from_source bool, timestamp
|
||||||
manifest['last_updated'] = json2.Any(timestamp.format_ss())
|
manifest['last_updated'] = json2.Any(timestamp.format_ss())
|
||||||
manifest['from_source'] = json2.Any(from_source)
|
manifest['from_source'] = json2.Any(from_source)
|
||||||
|
|
||||||
json_enc.encode_value(manifest, mut manifest_file)!
|
mut buffer := []u8{}
|
||||||
|
|
||||||
|
json_enc.encode_value(manifest, mut buffer)!
|
||||||
|
|
||||||
|
manifest_file.write(buffer)!
|
||||||
|
|
||||||
|
unsafe { buffer.free() }
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (upd VlsUpdater) init_download_prebuilt() ! {
|
fn (upd VlsUpdater) init_download_prebuilt() ! {
|
||||||
|
|
|
@ -5524,7 +5524,12 @@ fn (mut g Gen) const_decl_precomputed(mod string, name string, field_name string
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
escval := util.smart_quote(u8(rune_code).ascii_str(), false)
|
escval := util.smart_quote(u8(rune_code).ascii_str(), false)
|
||||||
g.const_decl_write_precomputed(mod, styp, cname, field_name, "'${escval}'")
|
|
||||||
|
g.global_const_defs[util.no_dots(field_name)] = GlobalConstDef{
|
||||||
|
mod: mod
|
||||||
|
def: "#define ${cname} '${escval}'"
|
||||||
|
order: -1
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
g.const_decl_write_precomputed(mod, styp, cname, field_name, u32(ct_value).str())
|
g.const_decl_write_precomputed(mod, styp, cname, field_name, u32(ct_value).str())
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,8 +5,8 @@ import time
|
||||||
import benchmark
|
import benchmark
|
||||||
|
|
||||||
// recommendations:
|
// recommendations:
|
||||||
// MAX_ITERATIONS=90_000 ./v run vlib/v/tests/bench/bench_json_vs_json2.v
|
// MAX_ITERATIONS=100_000 ./v run vlib/v/tests/bench/bench_json_vs_json2.v
|
||||||
// MAX_ITERATIONS=90_000 ./v -no-bounds-checking -prod -cc clang-15 crun vlib/v/tests/bench/bench_json_vs_json2.v
|
// MAX_ITERATIONS=100_000 ./v -no-bounds-checking -prod -cc clang-15 crun vlib/v/tests/bench/bench_json_vs_json2.v
|
||||||
|
|
||||||
const max_iterations = os.getenv_opt('MAX_ITERATIONS') or { '1000' }.int()
|
const max_iterations = os.getenv_opt('MAX_ITERATIONS') or { '1000' }.int()
|
||||||
|
|
||||||
|
@ -77,7 +77,7 @@ fn benchmark_measure_json_vs_json2_on_complex_struct() ! {
|
||||||
return error('json.decode ${p}')
|
return error('json.decode ${p}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure('json.decode')
|
b.measure('json.decode\n')
|
||||||
|
|
||||||
measure_json_encode_old_vs_new(json.decode(Person, s)!)!
|
measure_json_encode_old_vs_new(json.decode(Person, s)!)!
|
||||||
}
|
}
|
||||||
|
@ -124,7 +124,7 @@ fn benchmark_measure_decode_by_type() ! {
|
||||||
return error('json.decode ${d}')
|
return error('json.decode ${d}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure(' json.decode StructType[string]')
|
b.measure(' json.decode StructType[string]\n')
|
||||||
|
|
||||||
vb := '{"val": true}'
|
vb := '{"val": true}'
|
||||||
for _ in 0 .. max_iterations {
|
for _ in 0 .. max_iterations {
|
||||||
|
@ -140,7 +140,7 @@ fn benchmark_measure_decode_by_type() ! {
|
||||||
return error('json.decode ${d}')
|
return error('json.decode ${d}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure(' json.decode StructType[bool]')
|
b.measure(' json.decode StructType[bool]\n')
|
||||||
|
|
||||||
v0 := '{"val": 0}'
|
v0 := '{"val": 0}'
|
||||||
for _ in 0 .. max_iterations {
|
for _ in 0 .. max_iterations {
|
||||||
|
@ -156,7 +156,7 @@ fn benchmark_measure_decode_by_type() ! {
|
||||||
return error('json.decode ${d}')
|
return error('json.decode ${d}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure(' json.decode StructType[int]')
|
b.measure(' json.decode StructType[int]\n')
|
||||||
|
|
||||||
vt := '{"val": "2015-01-06 15:47:32"}'
|
vt := '{"val": "2015-01-06 15:47:32"}'
|
||||||
for _ in 0 .. max_iterations {
|
for _ in 0 .. max_iterations {
|
||||||
|
@ -172,7 +172,7 @@ fn benchmark_measure_decode_by_type() ! {
|
||||||
return error('json2.decode ${d}')
|
return error('json2.decode ${d}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure(' json.decode StructType[time.Time]')
|
b.measure(' json.decode StructType[time.Time]\n')
|
||||||
}
|
}
|
||||||
|
|
||||||
fn measure_json_encode_old_vs_new[T](val T) ! {
|
fn measure_json_encode_old_vs_new[T](val T) ! {
|
||||||
|
@ -191,5 +191,5 @@ fn measure_json_encode_old_vs_new[T](val T) ! {
|
||||||
return error('json.encode ${e}')
|
return error('json.encode ${e}')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
b.measure(' json.encode ${typename}')
|
b.measure(' json.encode ${typename}\n')
|
||||||
}
|
}
|
||||||
|
|
|
@ -150,3 +150,18 @@ The following list shows the possible outputs when casting a value to an incompa
|
||||||
3. Casting non-string values to string (`str()`) will return the
|
3. Casting non-string values to string (`str()`) will return the
|
||||||
JSON string representation of the value.
|
JSON string representation of the value.
|
||||||
4. Casting non-numeric values to int/float (`int()`/`i64()`/`f32()`/`f64()`) will return zero.
|
4. Casting non-numeric values to int/float (`int()`/`i64()`/`f32()`/`f64()`) will return zero.
|
||||||
|
|
||||||
|
## Encoding using string builder instead of []u8
|
||||||
|
To be more performant, `json2`, in PR 20052, decided to use buffers directly instead of Writers.
|
||||||
|
If you want to use Writers you can follow the steps below:
|
||||||
|
|
||||||
|
```v ignore
|
||||||
|
mut sb := strings.new_builder(64)
|
||||||
|
mut buffer := []u8{}
|
||||||
|
|
||||||
|
json2.encode_value(<some value to be encoded here>, mut buffer)!
|
||||||
|
|
||||||
|
sb.write(buffer)!
|
||||||
|
|
||||||
|
unsafe { buffer.free() }
|
||||||
|
```
|
||||||
|
|
|
@ -229,19 +229,19 @@ fn test_alias() {
|
||||||
|
|
||||||
fn test_pointer() {
|
fn test_pointer() {
|
||||||
mut string_initialized_with_reference := ''
|
mut string_initialized_with_reference := ''
|
||||||
assert json.encode(StructTypePointer[string]{ val: 0 }) == '{}'
|
assert json.encode(StructTypePointer[string]{ val: unsafe { nil } }) == '{}'
|
||||||
assert json.encode(StructTypePointer[string]{ val: &string_initialized_with_reference }) == '{"val":""}'
|
assert json.encode(StructTypePointer[string]{ val: &string_initialized_with_reference }) == '{"val":""}'
|
||||||
string_initialized_with_reference = 'a'
|
string_initialized_with_reference = 'a'
|
||||||
assert json.encode(StructTypePointer[string]{ val: &string_initialized_with_reference }) == '{"val":"a"}'
|
assert json.encode(StructTypePointer[string]{ val: &string_initialized_with_reference }) == '{"val":"a"}'
|
||||||
|
|
||||||
mut bool_initialized_with_reference := false
|
mut bool_initialized_with_reference := false
|
||||||
assert json.encode(StructTypePointer[bool]{ val: 0 }) == '{}'
|
assert json.encode(StructTypePointer[bool]{ val: unsafe { nil } }) == '{}'
|
||||||
assert json.encode(StructTypePointer[bool]{ val: &bool_initialized_with_reference }) == '{"val":false}'
|
assert json.encode(StructTypePointer[bool]{ val: &bool_initialized_with_reference }) == '{"val":false}'
|
||||||
bool_initialized_with_reference = true
|
bool_initialized_with_reference = true
|
||||||
assert json.encode(StructTypePointer[bool]{ val: &bool_initialized_with_reference }) == '{"val":true}'
|
assert json.encode(StructTypePointer[bool]{ val: &bool_initialized_with_reference }) == '{"val":true}'
|
||||||
|
|
||||||
mut int_initialized_with_reference := 0
|
mut int_initialized_with_reference := 0
|
||||||
assert json.encode(StructTypePointer[int]{ val: 0 }) == '{}'
|
assert json.encode(StructTypePointer[int]{ val: unsafe { nil } }) == '{}'
|
||||||
assert json.encode(StructTypePointer[int]{ val: &int_initialized_with_reference }) == '{"val":0}'
|
assert json.encode(StructTypePointer[int]{ val: &int_initialized_with_reference }) == '{"val":0}'
|
||||||
int_initialized_with_reference = 1
|
int_initialized_with_reference = 1
|
||||||
assert json.encode(StructTypePointer[int]{ val: &int_initialized_with_reference }) == '{"val":1}'
|
assert json.encode(StructTypePointer[int]{ val: &int_initialized_with_reference }) == '{"val":1}'
|
||||||
|
|
|
@ -3,8 +3,6 @@
|
||||||
// that can be found in the LICENSE file.
|
// that can be found in the LICENSE file.
|
||||||
module json2
|
module json2
|
||||||
|
|
||||||
import io
|
|
||||||
import strings
|
|
||||||
import time
|
import time
|
||||||
|
|
||||||
// Encoder encodes the an `Any` type into JSON representation.
|
// Encoder encodes the an `Any` type into JSON representation.
|
||||||
|
@ -15,167 +13,175 @@ pub struct Encoder {
|
||||||
escape_unicode bool = true
|
escape_unicode bool = true
|
||||||
}
|
}
|
||||||
|
|
||||||
pub const default_encoder = Encoder{}
|
|
||||||
|
|
||||||
// byte array versions of the most common tokens/chars to avoid reallocations
|
// byte array versions of the most common tokens/chars to avoid reallocations
|
||||||
const null_in_bytes = 'null'.bytes()
|
const null_in_bytes = 'null'
|
||||||
|
|
||||||
const true_in_bytes = 'true'.bytes()
|
const true_in_string = 'true'
|
||||||
|
|
||||||
const false_in_bytes = 'false'.bytes()
|
const false_in_string = 'false'
|
||||||
|
|
||||||
const zero_in_bytes = [u8(`0`)]
|
const zero_rune = `0`
|
||||||
|
|
||||||
const comma_bytes = [u8(`,`)]
|
const comma_rune = `,`
|
||||||
|
|
||||||
const colon_bytes = [u8(`:`)]
|
const colon_rune = `:`
|
||||||
|
|
||||||
const space_bytes = [u8(` `)]
|
const unicode_escape_chars = [`\\`, `u`]!
|
||||||
|
|
||||||
const unicode_escape_chars = [u8(`\\`), `u`]
|
const quote_rune = `"`
|
||||||
|
|
||||||
const quote_bytes = [u8(`"`)]
|
const escaped_chars = [(r'\b'), (r'\f'), (r'\n'), (r'\r'), (r'\t')]!
|
||||||
|
|
||||||
const escaped_chars = [(r'\b').bytes(), (r'\f').bytes(), (r'\n').bytes(),
|
const curly_open_rune = `{`
|
||||||
(r'\r').bytes(), (r'\t').bytes()]
|
|
||||||
|
|
||||||
const curly_open = [u8(`{`)]
|
const curly_close_rune = `}`
|
||||||
|
|
||||||
const curly_close = [u8(`}`)]
|
// encode_value encodes a value to the specific buffer.
|
||||||
|
pub fn (e &Encoder) encode_value[T](val T, mut buf []u8) ! {
|
||||||
// encode_value encodes a value to the specific writer.
|
e.encode_value_with_level[T](val, 1, mut buf)!
|
||||||
pub fn (e &Encoder) encode_value[T](val T, mut wr io.Writer) ! {
|
|
||||||
e.encode_value_with_level[T](val, 1, mut wr)!
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_newline(level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_newline(level int, mut buf []u8) ! {
|
||||||
if e.newline != 0 {
|
if e.newline != 0 {
|
||||||
wr.write([e.newline])!
|
buf << e.newline
|
||||||
for j := 0; j < level * e.newline_spaces_count; j++ {
|
for j := 0; j < level * e.newline_spaces_count; j++ {
|
||||||
wr.write(json2.space_bytes)!
|
buf << ` `
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_any(val Any, level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_any(val Any, level int, mut buf []u8) ! {
|
||||||
match val {
|
match val {
|
||||||
string {
|
string {
|
||||||
e.encode_string(val, mut wr)!
|
e.encode_string(val, mut buf)!
|
||||||
}
|
}
|
||||||
bool {
|
bool {
|
||||||
if val == true {
|
if val == true {
|
||||||
wr.write(json2.true_in_bytes)!
|
unsafe { buf.push_many(json2.true_in_string.str, json2.true_in_string.len) }
|
||||||
} else {
|
} else {
|
||||||
wr.write(json2.false_in_bytes)!
|
unsafe { buf.push_many(json2.false_in_string.str, json2.false_in_string.len) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
i8, i16, int, i64 {
|
i8, i16, int, i64 {
|
||||||
wr.write(val.str().bytes())!
|
str_int := val.str()
|
||||||
|
unsafe { buf.push_many(str_int.str, str_int.len) }
|
||||||
}
|
}
|
||||||
u8, u16, u32, u64 {
|
u8, u16, u32, u64 {
|
||||||
wr.write(val.str().bytes())!
|
str_int := val.str()
|
||||||
|
unsafe { buf.push_many(str_int.str, str_int.len) }
|
||||||
}
|
}
|
||||||
f32, f64 {
|
f32, f64 {
|
||||||
$if !nofloat ? {
|
$if !nofloat ? {
|
||||||
str_float := val.str().bytes()
|
str_float := val.str()
|
||||||
wr.write(str_float)!
|
unsafe { buf.push_many(str_float.str, str_float.len) }
|
||||||
if str_float[str_float.len - 1] == `.` {
|
if str_float[str_float.len - 1] == `.` {
|
||||||
wr.write(json2.zero_in_bytes)!
|
buf << json2.zero_rune
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
wr.write(json2.zero_in_bytes)!
|
buf << json2.zero_rune
|
||||||
}
|
}
|
||||||
map[string]Any {
|
map[string]Any {
|
||||||
wr.write(json2.curly_open)!
|
buf << json2.curly_open_rune
|
||||||
mut i := 0
|
mut i := 0
|
||||||
for k, v in val {
|
for k, v in val {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
e.encode_string(k, mut wr)!
|
e.encode_string(k, mut buf)!
|
||||||
wr.write(json2.colon_bytes)!
|
buf << json2.colon_rune
|
||||||
if e.newline != 0 {
|
if e.newline != 0 {
|
||||||
wr.write(json2.space_bytes)!
|
buf << ` `
|
||||||
}
|
}
|
||||||
e.encode_value_with_level(v, level + 1, mut wr)!
|
e.encode_value_with_level(v, level + 1, mut buf)!
|
||||||
if i < val.len - 1 {
|
if i < val.len - 1 {
|
||||||
wr.write(json2.comma_bytes)!
|
buf << json2.comma_rune
|
||||||
}
|
}
|
||||||
i++
|
i++
|
||||||
}
|
}
|
||||||
e.encode_newline(level - 1, mut wr)!
|
e.encode_newline(level - 1, mut buf)!
|
||||||
wr.write(json2.curly_close)!
|
buf << json2.curly_close_rune
|
||||||
}
|
}
|
||||||
[]Any {
|
[]Any {
|
||||||
wr.write([u8(`[`)])!
|
buf << `[`
|
||||||
for i in 0 .. val.len {
|
for i in 0 .. val.len {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
e.encode_value_with_level(val[i], level + 1, mut wr)!
|
e.encode_value_with_level(val[i], level + 1, mut buf)!
|
||||||
if i < val.len - 1 {
|
if i < val.len - 1 {
|
||||||
wr.write(json2.comma_bytes)!
|
buf << json2.comma_rune
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
e.encode_newline(level - 1, mut wr)!
|
e.encode_newline(level - 1, mut buf)!
|
||||||
wr.write([u8(`]`)])!
|
buf << `]`
|
||||||
}
|
}
|
||||||
time.Time {
|
time.Time {
|
||||||
wr.write(json2.quote_bytes)!
|
str_time := val.format_rfc3339()
|
||||||
wr.write(val.format_rfc3339().bytes())!
|
buf << `"`
|
||||||
wr.write(json2.quote_bytes)!
|
unsafe { buf.push_many(str_time.str, str_time.len) }
|
||||||
|
buf << `"`
|
||||||
}
|
}
|
||||||
Null {
|
Null {
|
||||||
wr.write(json2.null_in_bytes)!
|
buf << `n`
|
||||||
|
buf << `u`
|
||||||
|
buf << `l`
|
||||||
|
buf << `l`
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_map[T](value T, level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_map[T](value T, level int, mut buf []u8) ! {
|
||||||
wr.write(json2.curly_open)!
|
buf << json2.curly_open_rune
|
||||||
mut idx := 0
|
mut idx := 0
|
||||||
for k, v in value {
|
for k, v in value {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
e.encode_string(k.str(), mut wr)!
|
e.encode_string(k.str(), mut buf)!
|
||||||
wr.write(json2.colon_bytes)!
|
buf << json2.colon_rune
|
||||||
if e.newline != 0 {
|
if e.newline != 0 {
|
||||||
wr.write(json2.space_bytes)!
|
buf << ` `
|
||||||
}
|
}
|
||||||
e.encode_value_with_level(v, level + 1, mut wr)!
|
e.encode_value_with_level(v, level + 1, mut buf)!
|
||||||
if idx < value.len - 1 {
|
if idx < value.len - 1 {
|
||||||
wr.write(json2.comma_bytes)!
|
buf << json2.comma_rune
|
||||||
}
|
}
|
||||||
idx++
|
idx++
|
||||||
}
|
}
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
wr.write(json2.curly_close)!
|
buf << json2.curly_close_rune
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_value_with_level[T](val T, level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_value_with_level[T](val T, level int, mut buf []u8) ! {
|
||||||
$if T is string {
|
$if T is string {
|
||||||
e.encode_string(val, mut wr)!
|
e.encode_string(val, mut buf)!
|
||||||
} $else $if T is Any {
|
} $else $if T is Any {
|
||||||
e.encode_any(val, level, mut wr)!
|
e.encode_any(val, level, mut buf)!
|
||||||
} $else $if T is map[string]Any {
|
} $else $if T is map[string]Any {
|
||||||
// weird quirk but val is destructured immediately to Any
|
// weird quirk but val is destructured immediately to Any
|
||||||
e.encode_any(val, level, mut wr)!
|
e.encode_any(val, level, mut buf)!
|
||||||
} $else $if T is $map {
|
} $else $if T is $map {
|
||||||
e.encode_map(val, level, mut wr)!
|
e.encode_map(val, level, mut buf)!
|
||||||
} $else $if T is []Any {
|
} $else $if T is []Any {
|
||||||
e.encode_any(val, level, mut wr)!
|
e.encode_any(val, level, mut buf)!
|
||||||
} $else $if T is Encodable {
|
} $else $if T is Encodable {
|
||||||
wr.write(val.json_str().bytes())!
|
str_value := val.json_str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if T is $struct {
|
} $else $if T is $struct {
|
||||||
e.encode_struct(val, level, mut wr)!
|
e.encode_struct(val, level, mut buf)!
|
||||||
} $else $if T is $enum {
|
} $else $if T is $enum {
|
||||||
e.encode_any(Any(int(val)), level, mut wr)!
|
e.encode_any(Any(int(val)), level, mut buf)!
|
||||||
} $else $if T in [Null, bool, $float, $int] {
|
} $else $if T is bool {
|
||||||
e.encode_any(val, level, mut wr)!
|
e.encode_any(val, level, mut buf)!
|
||||||
|
} $else $if T is $float {
|
||||||
|
e.encode_any(val, level, mut buf)!
|
||||||
|
} $else $if T is $int {
|
||||||
|
e.encode_any(val, level, mut buf)!
|
||||||
|
} $else $if T is Null {
|
||||||
|
e.encode_any(val, level, mut buf)!
|
||||||
} $else {
|
} $else {
|
||||||
// dump(val.str())
|
// dump(val.str())
|
||||||
return error('cannot encode value with ${typeof(val).name} type')
|
return error('cannot encode value with ${typeof(val).name} type')
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_struct[U](val U, level int, mut buf []u8) ! {
|
||||||
wr.write(json2.curly_open)!
|
buf << json2.curly_open_rune
|
||||||
mut i := 0
|
mut i := 0
|
||||||
mut fields_len := 0
|
mut fields_len := 0
|
||||||
$for field in U.fields {
|
$for field in U.fields {
|
||||||
|
@ -201,33 +207,40 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
is_none := value.str() == 'Option(none)'
|
is_none := value.str() == 'Option(none)'
|
||||||
|
|
||||||
if !is_none {
|
if !is_none {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
if json_name != '' {
|
if json_name != '' {
|
||||||
e.encode_string(json_name, mut wr)!
|
e.encode_string(json_name, mut buf)!
|
||||||
} else {
|
} else {
|
||||||
e.encode_string(field.name, mut wr)!
|
e.encode_string(field.name, mut buf)!
|
||||||
}
|
}
|
||||||
wr.write(json2.colon_bytes)!
|
buf << json2.colon_rune
|
||||||
|
|
||||||
if e.newline != 0 {
|
if e.newline != 0 {
|
||||||
wr.write(json2.space_bytes)!
|
buf << ` `
|
||||||
}
|
}
|
||||||
|
|
||||||
$if field.typ is ?string {
|
$if field.typ is ?string {
|
||||||
e.encode_string(val.$(field.name) ?.str(), mut wr)!
|
e.encode_string(val.$(field.name) ?.str(), mut buf)!
|
||||||
} $else $if field.typ is ?bool || field.typ is ?f32 || field.typ is ?f64
|
} $else $if field.typ is ?bool {
|
||||||
|| field.typ is ?i8 || field.typ is ?i16 || field.typ is ?int
|
if value ?.str() == json2.true_in_string {
|
||||||
|| field.typ is ?i64 || field.typ is ?u8 || field.typ is ?u16
|
unsafe { buf.push_many(json2.true_in_string.str, json2.true_in_string.len) }
|
||||||
|| field.typ is ?u32 || field.typ is ?u64 {
|
} else {
|
||||||
wr.write(val.$(field.name) ?.str().bytes())!
|
unsafe { buf.push_many(json2.false_in_string.str, json2.false_in_string.len) }
|
||||||
|
}
|
||||||
|
} $else $if field.typ is ?f32 || field.typ is ?f64 || field.typ is ?i8
|
||||||
|
|| field.typ is ?i16 || field.typ is ?int || field.typ is ?i64
|
||||||
|
|| field.typ is ?u8 || field.typ is ?u16 || field.typ is ?u32
|
||||||
|
|| field.typ is ?u64 {
|
||||||
|
str_value := val.$(field.name) ?.str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if field.typ is ?time.Time {
|
} $else $if field.typ is ?time.Time {
|
||||||
option_value := val.$(field.name) as ?time.Time
|
option_value := val.$(field.name) as ?time.Time
|
||||||
parsed_time := option_value as time.Time
|
parsed_time := option_value as time.Time
|
||||||
e.encode_string(parsed_time.format_rfc3339(), mut wr)!
|
e.encode_string(parsed_time.format_rfc3339(), mut buf)!
|
||||||
} $else $if field.is_array {
|
} $else $if field.is_array {
|
||||||
e.encode_array(value, level + 1, mut wr)!
|
e.encode_array(value, level + 1, mut buf)!
|
||||||
} $else $if field.is_struct {
|
} $else $if field.is_struct {
|
||||||
e.encode_struct(value, level + 1, mut wr)!
|
e.encode_struct(value, level + 1, mut buf)!
|
||||||
} $else $if field.is_enum {
|
} $else $if field.is_enum {
|
||||||
// FIXME - checker and cast error
|
// FIXME - checker and cast error
|
||||||
// wr.write(int(val.$(field.name)?).str().bytes())!
|
// wr.write(int(val.$(field.name)?).str().bytes())!
|
||||||
|
@ -235,19 +248,21 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
} $else $if field.is_alias {
|
} $else $if field.is_alias {
|
||||||
match field.unaliased_typ {
|
match field.unaliased_typ {
|
||||||
typeof[string]().idx {
|
typeof[string]().idx {
|
||||||
e.encode_string(value.str(), mut wr)!
|
e.encode_string(value.str(), mut buf)!
|
||||||
}
|
}
|
||||||
typeof[bool]().idx, typeof[f32]().idx, typeof[f64]().idx, typeof[i8]().idx,
|
typeof[bool]().idx {}
|
||||||
typeof[i16]().idx, typeof[int]().idx, typeof[i64]().idx, typeof[u8]().idx,
|
typeof[f32]().idx, typeof[f64]().idx, typeof[i8]().idx, typeof[i16]().idx,
|
||||||
typeof[u16]().idx, typeof[u32]().idx, typeof[u64]().idx {
|
typeof[int]().idx, typeof[i64]().idx, typeof[u8]().idx, typeof[u16]().idx,
|
||||||
wr.write(value.str().bytes())!
|
typeof[u32]().idx, typeof[u64]().idx {
|
||||||
|
str_value := value.str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
}
|
}
|
||||||
typeof[[]int]().idx {
|
typeof[[]int]().idx {
|
||||||
// FIXME - error: could not infer generic type `U` in call to `encode_array`
|
// FIXME - error: could not infer generic type `U` in call to `encode_array`
|
||||||
// e.encode_array(value, level, mut wr)!
|
// e.encode_array(value, level, mut buf)!
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
// e.encode_value_with_level(value, level + 1, mut wr)!
|
// e.encode_value_with_level(value, level + 1, mut buf)!
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} $else {
|
} $else {
|
||||||
|
@ -259,53 +274,62 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
} $else {
|
} $else {
|
||||||
is_none := val.$(field.name).str() == 'unknown sum type value'
|
is_none := val.$(field.name).str() == 'unknown sum type value'
|
||||||
if !is_none && !is_nil {
|
if !is_none && !is_nil {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
if json_name != '' {
|
if json_name != '' {
|
||||||
e.encode_string(json_name, mut wr)!
|
e.encode_string(json_name, mut buf)!
|
||||||
} else {
|
} else {
|
||||||
e.encode_string(field.name, mut wr)!
|
e.encode_string(field.name, mut buf)!
|
||||||
}
|
}
|
||||||
wr.write(json2.colon_bytes)!
|
buf << json2.colon_rune
|
||||||
|
|
||||||
if e.newline != 0 {
|
if e.newline != 0 {
|
||||||
wr.write(json2.space_bytes)!
|
buf << ` `
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
$if field.indirections != 0 {
|
$if field.indirections != 0 {
|
||||||
if val.$(field.name) != unsafe { nil } {
|
if val.$(field.name) != unsafe { nil } {
|
||||||
$if field.indirections == 1 {
|
$if field.indirections == 1 {
|
||||||
e.encode_value_with_level(*val.$(field.name), level + 1, mut wr)!
|
e.encode_value_with_level(*val.$(field.name), level + 1, mut buf)!
|
||||||
}
|
}
|
||||||
$if field.indirections == 2 {
|
$if field.indirections == 2 {
|
||||||
e.encode_value_with_level(**val.$(field.name), level + 1, mut
|
e.encode_value_with_level(**val.$(field.name), level + 1, mut
|
||||||
wr)!
|
buf)!
|
||||||
}
|
}
|
||||||
$if field.indirections == 3 {
|
$if field.indirections == 3 {
|
||||||
e.encode_value_with_level(***val.$(field.name), level + 1, mut
|
e.encode_value_with_level(***val.$(field.name), level + 1, mut
|
||||||
wr)!
|
buf)!
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} $else $if field.typ is string {
|
} $else $if field.typ is string {
|
||||||
e.encode_string(val.$(field.name).str(), mut wr)!
|
e.encode_string(val.$(field.name).str(), mut buf)!
|
||||||
} $else $if field.typ is time.Time {
|
} $else $if field.typ is time.Time {
|
||||||
wr.write(json2.quote_bytes)!
|
str_value := val.$(field.name).format_rfc3339()
|
||||||
wr.write(val.$(field.name).format_rfc3339().bytes())!
|
buf << json2.quote_rune
|
||||||
wr.write(json2.quote_bytes)!
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if field.typ in [bool, $float, $int] {
|
buf << json2.quote_rune
|
||||||
wr.write(val.$(field.name).str().bytes())!
|
} $else $if field.typ is bool {
|
||||||
|
if value {
|
||||||
|
unsafe { buf.push_many(json2.true_in_string.str, json2.true_in_string.len) }
|
||||||
|
} else {
|
||||||
|
unsafe { buf.push_many(json2.false_in_string.str, json2.false_in_string.len) }
|
||||||
|
}
|
||||||
|
} $else $if field.typ in [$float, $int] {
|
||||||
|
str_value := val.$(field.name).str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if field.is_array {
|
} $else $if field.is_array {
|
||||||
// TODO - replace for `field.typ is $array`
|
// TODO - replace for `field.typ is $array`
|
||||||
e.encode_array(value, level + 1, mut wr)!
|
e.encode_array(value, level + 1, mut buf)!
|
||||||
} $else $if field.typ is $array {
|
} $else $if field.typ is $array {
|
||||||
// e.encode_array(value, level + 1, mut wr)! // FIXME - error: could not infer generic type `U` in call to `encode_array`
|
// e.encode_array(value, level + 1, mut buf)! // FIXME - error: could not infer generic type `U` in call to `encode_array`
|
||||||
} $else $if field.typ is $struct {
|
} $else $if field.typ is $struct {
|
||||||
e.encode_struct(value, level + 1, mut wr)!
|
e.encode_struct(value, level + 1, mut buf)!
|
||||||
} $else $if field.is_map {
|
} $else $if field.is_map {
|
||||||
e.encode_map(value, level + 1, mut wr)!
|
e.encode_map(value, level + 1, mut buf)!
|
||||||
} $else $if field.is_enum {
|
} $else $if field.is_enum {
|
||||||
// TODO - replace for `field.typ is $enum`
|
// TODO - replace for `field.typ is $enum`
|
||||||
wr.write(int(val.$(field.name)).str().bytes())!
|
str_value := int(val.$(field.name)).str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if field.typ is $enum {
|
} $else $if field.typ is $enum {
|
||||||
// wr.write(int(val.$(field.name)).str().bytes())! // FIXME - error: cannot cast string to `int`, use `val.$field.name.int()` instead.
|
// wr.write(int(val.$(field.name)).str().bytes())! // FIXME - error: cannot cast string to `int`, use `val.$field.name.int()` instead.
|
||||||
} $else $if field.typ is $sumtype {
|
} $else $if field.typ is $sumtype {
|
||||||
|
@ -324,9 +348,12 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
`0`...`9` {
|
`0`...`9` {
|
||||||
if sum_type_value.contains_any(' /:-') {
|
if sum_type_value.contains_any(' /:-') {
|
||||||
date_time_str := time.parse(sum_type_value)!
|
date_time_str := time.parse(sum_type_value)!
|
||||||
wr.write(date_time_str.format_rfc3339().bytes())!
|
unsafe {
|
||||||
|
str_value := date_time_str.format_rfc3339()
|
||||||
|
buf.push_many(str_value.str, str_value.len)
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
wr.write(sum_type_value.bytes())!
|
unsafe { buf.push_many(sum_type_value.str, sum_type_value.len) }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
`A`...`Z` {
|
`A`...`Z` {
|
||||||
|
@ -342,13 +369,13 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
if !sum_type_value.all_before('{').contains_any(' "\'') {
|
if !sum_type_value.all_before('{').contains_any(' "\'') {
|
||||||
// is_struct = true
|
// is_struct = true
|
||||||
// TODO
|
// TODO
|
||||||
// e.encode_struct_from_sumtype(value, level + 1, mut wr)!
|
// e.encode_struct_from_sumtype(value, level + 1, mut buf)!
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
`a`...`z` {
|
`a`...`z` {
|
||||||
if sum_type_value in ['true', 'false'] {
|
if sum_type_value in ['true', 'false'] {
|
||||||
wr.write(sum_type_value.bytes())!
|
unsafe { buf.push_many(sum_type_value.str, sum_type_value.len) }
|
||||||
} else {
|
} else {
|
||||||
// is_enum = true
|
// is_enum = true
|
||||||
}
|
}
|
||||||
|
@ -366,21 +393,28 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
// dump(is_enum)
|
// dump(is_enum)
|
||||||
// dump(is_array)
|
// dump(is_array)
|
||||||
if is_string {
|
if is_string {
|
||||||
e.encode_string(sum_type_value#[1..-1], mut wr)!
|
e.encode_string(sum_type_value#[1..-1], mut buf)!
|
||||||
}
|
}
|
||||||
} $else $if field.typ is $alias {
|
} $else $if field.typ is $alias {
|
||||||
$if field.unaliased_typ is string {
|
$if field.unaliased_typ is string {
|
||||||
e.encode_string(val.$(field.name).str(), mut wr)!
|
e.encode_string(val.$(field.name).str(), mut buf)!
|
||||||
} $else $if field.unaliased_typ is time.Time {
|
} $else $if field.unaliased_typ is time.Time {
|
||||||
parsed_time := time.parse(val.$(field.name).str()) or { time.Time{} }
|
parsed_time := time.parse(val.$(field.name).str()) or { time.Time{} }
|
||||||
e.encode_string(parsed_time.format_rfc3339(), mut wr)!
|
e.encode_string(parsed_time.format_rfc3339(), mut buf)!
|
||||||
} $else $if field.unaliased_typ in [bool, $float, $int] {
|
} $else $if field.unaliased_typ is bool {
|
||||||
wr.write(val.$(field.name).str().bytes())!
|
if val.$(field.name).str() == json2.true_in_string {
|
||||||
|
unsafe { buf.push_many(json2.true_in_string.str, json2.true_in_string.len) }
|
||||||
|
} else {
|
||||||
|
unsafe { buf.push_many(json2.false_in_string.str, json2.false_in_string.len) }
|
||||||
|
}
|
||||||
|
} $else $if field.unaliased_typ in [$float, $int] {
|
||||||
|
str_value := val.$(field.name).str()
|
||||||
|
unsafe { buf.push_many(str_value.str, str_value.len) }
|
||||||
} $else $if field.unaliased_typ is $array {
|
} $else $if field.unaliased_typ is $array {
|
||||||
// e.encode_array(val.$(field.name), level + 1, mut wr)! // FIXME - error: could not infer generic type `U` in call to `encode_array`
|
// e.encode_array(val.$(field.name), level + 1, mut buf)! // FIXME - error: could not infer generic type `U` in call to `encode_array`
|
||||||
} $else $if field.unaliased_typ is $struct {
|
} $else $if field.unaliased_typ is $struct {
|
||||||
// e.encode_struct(val.$(field.name), level + 1, mut wr)! // FIXME - error: cannot use `BoolAlias` as `StringAlias` in argument 1 to `x.json2.Encoder.encode_struct`
|
// e.encode_struct(val.$(field.name), level + 1, mut buf)! // FIXME - error: cannot use `BoolAlias` as `StringAlias` in argument 1 to `x.json2.Encoder.encode_struct`
|
||||||
e.encode_struct(value, level + 1, mut wr)!
|
e.encode_struct(value, level + 1, mut buf)!
|
||||||
} $else $if field.unaliased_typ is $enum {
|
} $else $if field.unaliased_typ is $enum {
|
||||||
// enum_value := val.$(field.name)
|
// enum_value := val.$(field.name)
|
||||||
// dump(int(val.$(field.name))) // FIXME
|
// dump(int(val.$(field.name))) // FIXME
|
||||||
|
@ -399,69 +433,70 @@ fn (e &Encoder) encode_struct[U](val U, level int, mut wr io.Writer) ! {
|
||||||
|
|
||||||
if i < fields_len - 1 && !ignore_field {
|
if i < fields_len - 1 && !ignore_field {
|
||||||
if !is_nil {
|
if !is_nil {
|
||||||
wr.write(json2.comma_bytes)!
|
buf << json2.comma_rune
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !ignore_field {
|
if !ignore_field {
|
||||||
i++
|
i++
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
e.encode_newline(level - 1, mut wr)!
|
e.encode_newline(level - 1, mut buf)!
|
||||||
wr.write(json2.curly_close)!
|
buf << json2.curly_close_rune
|
||||||
|
// b.measure('encode_struct')
|
||||||
}
|
}
|
||||||
|
|
||||||
fn (e &Encoder) encode_array[U](val []U, level int, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_array[U](val []U, level int, mut buf []u8) ! {
|
||||||
wr.write([u8(`[`)])!
|
buf << `[`
|
||||||
for i in 0 .. val.len {
|
for i in 0 .. val.len {
|
||||||
e.encode_newline(level, mut wr)!
|
e.encode_newline(level, mut buf)!
|
||||||
|
|
||||||
$if U is string {
|
$if U is string {
|
||||||
e.encode_any(val[i], level + 1, mut wr)!
|
e.encode_any(val[i], level + 1, mut buf)!
|
||||||
} $else $if U is bool {
|
} $else $if U is bool {
|
||||||
e.encode_any(bool(val[i]), level + 1, mut wr)!
|
e.encode_any(bool(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is f32 {
|
} $else $if U is f32 {
|
||||||
e.encode_any(f32(val[i]), level + 1, mut wr)!
|
e.encode_any(f32(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is f64 {
|
} $else $if U is f64 {
|
||||||
e.encode_any(f64(val[i]), level + 1, mut wr)!
|
e.encode_any(f64(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is i8 {
|
} $else $if U is i8 {
|
||||||
e.encode_any(i8(val[i]), level + 1, mut wr)!
|
e.encode_any(i8(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is i16 {
|
} $else $if U is i16 {
|
||||||
e.encode_any(i16(val[i]), level + 1, mut wr)!
|
e.encode_any(i16(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is int {
|
} $else $if U is int {
|
||||||
e.encode_any(int(val[i]), level + 1, mut wr)!
|
e.encode_any(int(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is i64 {
|
} $else $if U is i64 {
|
||||||
e.encode_any(i64(val[i]), level + 1, mut wr)!
|
e.encode_any(i64(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is u8 {
|
} $else $if U is u8 {
|
||||||
e.encode_any(u8(val[i]), level + 1, mut wr)!
|
e.encode_any(u8(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is u16 {
|
} $else $if U is u16 {
|
||||||
e.encode_any(u16(val[i]), level + 1, mut wr)!
|
e.encode_any(u16(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is u32 {
|
} $else $if U is u32 {
|
||||||
e.encode_any(u32(val[i]), level + 1, mut wr)!
|
e.encode_any(u32(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is u64 {
|
} $else $if U is u64 {
|
||||||
e.encode_any(u64(val[i]), level + 1, mut wr)!
|
e.encode_any(u64(val[i]), level + 1, mut buf)!
|
||||||
} $else $if U is $array {
|
} $else $if U is $array {
|
||||||
// FIXME - error: could not infer generic type `U` in call to `encode_array`
|
// FIXME - error: could not infer generic type `U` in call to `encode_array`
|
||||||
// e.encode_array(val[i], level + 1, mut wr)!
|
// e.encode_array(val[i], level + 1, mut buf)!
|
||||||
} $else $if U is $struct {
|
} $else $if U is $struct {
|
||||||
e.encode_struct(val[i], level + 1, mut wr)!
|
e.encode_struct(val[i], level + 1, mut buf)!
|
||||||
} $else $if U is $sumtype {
|
} $else $if U is $sumtype {
|
||||||
$if U is Any {
|
$if U is Any {
|
||||||
e.encode_any(val[i], level + 1, mut wr)!
|
e.encode_any(val[i], level + 1, mut buf)!
|
||||||
} $else {
|
} $else {
|
||||||
// TODO
|
// TODO
|
||||||
}
|
}
|
||||||
} $else $if U is $enum {
|
} $else $if U is $enum {
|
||||||
e.encode_any(i64(val[i]), level + 1, mut wr)!
|
e.encode_any(i64(val[i]), level + 1, mut buf)!
|
||||||
} $else {
|
} $else {
|
||||||
return error('type ${typeof(val).name} cannot be array encoded')
|
return error('type ${typeof(val).name} cannot be array encoded')
|
||||||
}
|
}
|
||||||
if i < val.len - 1 {
|
if i < val.len - 1 {
|
||||||
wr.write(json2.comma_bytes)!
|
buf << json2.comma_rune
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
e.encode_newline(level - 1, mut wr)!
|
e.encode_newline(level - 1, mut buf)!
|
||||||
wr.write([u8(`]`)])!
|
buf << `]`
|
||||||
}
|
}
|
||||||
|
|
||||||
// str returns the JSON string representation of the `map[string]Any` type.
|
// str returns the JSON string representation of the `map[string]Any` type.
|
||||||
|
@ -493,16 +528,16 @@ pub fn (f Any) json_str() string {
|
||||||
// prettify_json_str returns the pretty-formatted JSON string representation of the `Any` type.
|
// prettify_json_str returns the pretty-formatted JSON string representation of the `Any` type.
|
||||||
@[manualfree]
|
@[manualfree]
|
||||||
pub fn (f Any) prettify_json_str() string {
|
pub fn (f Any) prettify_json_str() string {
|
||||||
mut sb := strings.new_builder(4096)
|
mut buf := []u8{}
|
||||||
defer {
|
defer {
|
||||||
unsafe { sb.free() }
|
unsafe { buf.free() }
|
||||||
}
|
}
|
||||||
mut enc := Encoder{
|
mut enc := Encoder{
|
||||||
newline: `\n`
|
newline: `\n`
|
||||||
newline_spaces_count: 2
|
newline_spaces_count: 2
|
||||||
}
|
}
|
||||||
enc.encode_value(f, mut sb) or {}
|
enc.encode_value(f, mut buf) or {}
|
||||||
return sb.str()
|
return buf.bytestr()
|
||||||
}
|
}
|
||||||
|
|
||||||
// CharLengthIterator is an iterator that generates a char
|
// CharLengthIterator is an iterator that generates a char
|
||||||
|
@ -535,49 +570,58 @@ fn (mut iter CharLengthIterator) next() ?int {
|
||||||
// TODO - Need refactor. Is so slow. The longer the string, the lower the performance.
|
// TODO - Need refactor. Is so slow. The longer the string, the lower the performance.
|
||||||
// encode_string returns the JSON spec-compliant version of the string.
|
// encode_string returns the JSON spec-compliant version of the string.
|
||||||
@[manualfree]
|
@[manualfree]
|
||||||
fn (e &Encoder) encode_string(s string, mut wr io.Writer) ! {
|
fn (e &Encoder) encode_string(s string, mut buf []u8) ! {
|
||||||
mut char_lens := CharLengthIterator{
|
mut char_lens := CharLengthIterator{
|
||||||
text: s
|
text: s
|
||||||
}
|
}
|
||||||
mut i := 0
|
mut i := 0
|
||||||
wr.write(json2.quote_bytes)!
|
buf << json2.quote_rune
|
||||||
for char_len in char_lens {
|
for char_len in char_lens {
|
||||||
if char_len == 1 {
|
if char_len == 1 {
|
||||||
chr := s[i]
|
chr := s[i]
|
||||||
if chr in important_escapable_chars {
|
if chr in important_escapable_chars {
|
||||||
for j := 0; j < important_escapable_chars.len; j++ {
|
for j := 0; j < important_escapable_chars.len; j++ {
|
||||||
if chr == important_escapable_chars[j] {
|
if chr == important_escapable_chars[j] {
|
||||||
wr.write(json2.escaped_chars[j])!
|
unsafe { buf.push_many(json2.escaped_chars[j].str, json2.escaped_chars[j].len) }
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else if chr == `"` || chr == `/` || chr == `\\` {
|
} else if chr == `"` || chr == `/` || chr == `\\` {
|
||||||
wr.write([u8(`\\`), chr])!
|
buf << `\\`
|
||||||
|
buf << chr
|
||||||
} else if int(chr) < 0x20 {
|
} else if int(chr) < 0x20 {
|
||||||
hex_code := chr.hex().bytes()
|
// unsafe { buf.push_many(json2.unicode_escape_chars.str, json2.unicode_escape_chars.len) } // \u
|
||||||
wr.write(json2.unicode_escape_chars)! // \u
|
for r in json2.unicode_escape_chars {
|
||||||
wr.write(json2.zero_in_bytes)! // \u0
|
buf << r
|
||||||
wr.write(json2.zero_in_bytes)! // \u00
|
}
|
||||||
wr.write(hex_code)! // \u00xxxx
|
buf << json2.zero_rune // \u0
|
||||||
|
buf << json2.zero_rune // \u00
|
||||||
|
|
||||||
|
hex_code := chr.hex()
|
||||||
|
unsafe { buf.push_many(hex_code.str, hex_code.len) }
|
||||||
} else {
|
} else {
|
||||||
wr.write([u8(chr)])!
|
buf << chr
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
slice := s[i..i + char_len]
|
slice := s[i..i + char_len]
|
||||||
hex_code := slice.utf32_code().hex().bytes()
|
hex_code := slice.utf32_code().hex()
|
||||||
if !e.escape_unicode || hex_code.len < 4 {
|
if !e.escape_unicode || hex_code.len < 4 {
|
||||||
// unescaped non-ASCII char
|
// unescaped non-ASCII char
|
||||||
wr.write(slice.bytes())!
|
unsafe { buf.push_many(slice.str, slice.len) }
|
||||||
} else if hex_code.len == 4 {
|
} else if hex_code.len == 4 {
|
||||||
// a unicode endpoint
|
// a unicode endpoint
|
||||||
wr.write(json2.unicode_escape_chars)!
|
|
||||||
wr.write(hex_code)!
|
// unsafe { buf.push_many(json2.unicode_escape_chars.str, json2.unicode_escape_chars.len) }
|
||||||
|
for r in json2.unicode_escape_chars {
|
||||||
|
buf << r
|
||||||
|
}
|
||||||
|
unsafe { buf.push_many(hex_code.str, hex_code.len) }
|
||||||
} else {
|
} else {
|
||||||
// TODO: still figuring out what
|
// TODO: still figuring out what
|
||||||
// to do with more than 4 chars.
|
// to do with more than 4 chars
|
||||||
// According to https://www.json.org/json-en.html however, any codepoint is valid inside a string,
|
// According to https://www.json.org/json-en.html however, any codepoint is valid inside a string,
|
||||||
// so just passing it along should hopefully also work.
|
// so just passing it along should hopefully also work.
|
||||||
wr.write(slice.bytes())!
|
unsafe { buf.push_many(slice.str, slice.len) }
|
||||||
}
|
}
|
||||||
unsafe {
|
unsafe {
|
||||||
slice.free()
|
slice.free()
|
||||||
|
@ -587,5 +631,5 @@ fn (e &Encoder) encode_string(s string, mut wr io.Writer) ! {
|
||||||
i += char_len
|
i += char_len
|
||||||
}
|
}
|
||||||
|
|
||||||
wr.write(json2.quote_bytes)!
|
buf << json2.quote_rune
|
||||||
}
|
}
|
||||||
|
|
|
@ -139,6 +139,42 @@ fn test_encode_simple() {
|
||||||
assert json.encode(1) == '1'
|
assert json.encode(1) == '1'
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn test_encode_value() {
|
||||||
|
json_enc := json.Encoder{
|
||||||
|
newline: `\n`
|
||||||
|
newline_spaces_count: 2
|
||||||
|
escape_unicode: false
|
||||||
|
}
|
||||||
|
|
||||||
|
mut manifest := map[string]json.Any{}
|
||||||
|
|
||||||
|
manifest['server_path'] = json.Any('new_path')
|
||||||
|
manifest['last_updated'] = json.Any('timestamp.format_ss()')
|
||||||
|
manifest['from_source'] = json.Any('from_source')
|
||||||
|
|
||||||
|
mut sb := strings.new_builder(64)
|
||||||
|
mut buffer := []u8{}
|
||||||
|
json_enc.encode_value(manifest, mut buffer)!
|
||||||
|
|
||||||
|
assert buffer.len > 0
|
||||||
|
assert buffer == [u8(123), 10, 32, 32, 34, 115, 101, 114, 118, 101, 114, 95, 112, 97, 116,
|
||||||
|
104, 34, 58, 32, 34, 110, 101, 119, 95, 112, 97, 116, 104, 34, 44, 10, 32, 32, 34, 108,
|
||||||
|
97, 115, 116, 95, 117, 112, 100, 97, 116, 101, 100, 34, 58, 32, 34, 116, 105, 109, 101,
|
||||||
|
115, 116, 97, 109, 112, 46, 102, 111, 114, 109, 97, 116, 95, 115, 115, 40, 41, 34, 44,
|
||||||
|
10, 32, 32, 34, 102, 114, 111, 109, 95, 115, 111, 117, 114, 99, 101, 34, 58, 32, 34, 102,
|
||||||
|
114, 111, 109, 95, 115, 111, 117, 114, 99, 101, 34, 10, 125]
|
||||||
|
|
||||||
|
sb.write(buffer)!
|
||||||
|
|
||||||
|
unsafe { buffer.free() }
|
||||||
|
|
||||||
|
assert sb.str() == r'{
|
||||||
|
"server_path": "new_path",
|
||||||
|
"last_updated": "timestamp.format_ss()",
|
||||||
|
"from_source": "from_source"
|
||||||
|
}'
|
||||||
|
}
|
||||||
|
|
||||||
fn test_encode_time() {
|
fn test_encode_time() {
|
||||||
assert json.encode({
|
assert json.encode({
|
||||||
'bro': json.Any(time.Time{})
|
'bro': json.Any(time.Time{})
|
||||||
|
|
|
@ -3,7 +3,6 @@
|
||||||
// that can be found in the LICENSE file.
|
// that can be found in the LICENSE file.
|
||||||
module json2
|
module json2
|
||||||
|
|
||||||
import strings
|
|
||||||
import time
|
import time
|
||||||
|
|
||||||
// Decodes a JSON string into an `Any` type. Returns an option.
|
// Decodes a JSON string into an `Any` type. Returns an option.
|
||||||
|
@ -162,35 +161,37 @@ pub fn encode[T](val T) string {
|
||||||
$if T is $array {
|
$if T is $array {
|
||||||
return encode_array(val)
|
return encode_array(val)
|
||||||
} $else {
|
} $else {
|
||||||
mut sb := strings.new_builder(64)
|
mut buf := []u8{}
|
||||||
|
|
||||||
defer {
|
defer {
|
||||||
unsafe { sb.free() }
|
unsafe { buf.free() }
|
||||||
}
|
}
|
||||||
|
encoder := Encoder{}
|
||||||
|
|
||||||
default_encoder.encode_value(val, mut sb) or {
|
encoder.encode_value(val, mut buf) or {
|
||||||
println(err)
|
println(err)
|
||||||
default_encoder.encode_value[Null](null, mut sb) or {}
|
encoder.encode_value[string]('null', mut buf) or {}
|
||||||
}
|
}
|
||||||
|
|
||||||
return sb.str()
|
return buf.bytestr()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// encode_array is a generic function that encodes a array into a JSON string.
|
// encode_array is a generic function that encodes a array into a JSON string.
|
||||||
fn encode_array[T](val []T) string {
|
fn encode_array[T](val []T) string {
|
||||||
mut sb := strings.new_builder(64)
|
mut buf := []u8{}
|
||||||
|
|
||||||
defer {
|
defer {
|
||||||
unsafe { sb.free() }
|
unsafe { buf.free() }
|
||||||
}
|
}
|
||||||
|
|
||||||
default_encoder.encode_array(val, 1, mut sb) or {
|
encoder := Encoder{}
|
||||||
|
encoder.encode_array(val, 1, mut buf) or {
|
||||||
println(err)
|
println(err)
|
||||||
default_encoder.encode_value[Null](null, mut sb) or {}
|
encoder.encode_value[string]('null', mut buf) or {}
|
||||||
}
|
}
|
||||||
|
|
||||||
return sb.str()
|
return buf.bytestr()
|
||||||
}
|
}
|
||||||
|
|
||||||
// encode_pretty ...
|
// encode_pretty ...
|
||||||
|
|
|
@ -43,16 +43,16 @@ pub fn (t Token) full_col() int {
|
||||||
}
|
}
|
||||||
|
|
||||||
// list of characters commonly used in JSON.
|
// list of characters commonly used in JSON.
|
||||||
const char_list = [`{`, `}`, `[`, `]`, `,`, `:`]
|
const char_list = [`{`, `}`, `[`, `]`, `,`, `:`]!
|
||||||
// list of newlines to check when moving to a new position.
|
// list of newlines to check when moving to a new position.
|
||||||
const newlines = [`\r`, `\n`, `\t`]
|
const newlines = [`\r`, `\n`, `\t`]!
|
||||||
// list of escapable that needs to be escaped inside a JSON string.
|
// list of escapable that needs to be escaped inside a JSON string.
|
||||||
// double quotes and forward slashes are excluded intentionally since
|
// double quotes and forward slashes are excluded intentionally since
|
||||||
// they have their own separate checks for it in order to pass the
|
// they have their own separate checks for it in order to pass the
|
||||||
// JSON test suite (https://github.com/nst/JSONTestSuite/).
|
// JSON test suite (https://github.com/nst/JSONTestSuite/).
|
||||||
const important_escapable_chars = [`\b`, `\f`, `\n`, `\r`, `\t`]
|
const important_escapable_chars = [`\b`, `\f`, `\n`, `\r`, `\t`]!
|
||||||
// list of valid unicode escapes aside from \u{4-hex digits}
|
// list of valid unicode escapes aside from \u{4-hex digits}
|
||||||
const valid_unicode_escapes = [`b`, `f`, `n`, `r`, `t`, `\\`, `"`, `/`]
|
const valid_unicode_escapes = [`b`, `f`, `n`, `r`, `t`, `\\`, `"`, `/`]!
|
||||||
// used for transforming escapes into valid unicode (eg. n => \n)
|
// used for transforming escapes into valid unicode (eg. n => \n)
|
||||||
const unicode_transform_escapes = {
|
const unicode_transform_escapes = {
|
||||||
98: `\b`
|
98: `\b`
|
||||||
|
@ -64,7 +64,7 @@ const unicode_transform_escapes = {
|
||||||
34: `"`
|
34: `"`
|
||||||
47: `/`
|
47: `/`
|
||||||
}
|
}
|
||||||
const exp_signs = [u8(`-`), `+`]
|
const exp_signs = [u8(`-`), `+`]!
|
||||||
|
|
||||||
// move_pos proceeds to the next position.
|
// move_pos proceeds to the next position.
|
||||||
fn (mut s Scanner) move() {
|
fn (mut s Scanner) move() {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue