make simdjson great again.

now it is generally a little faster than the easyjson approach.

goos: linux
goarch: amd64
pkg: github.com/nbd-wtf/go-nostr
cpu: AMD Ryzen 3 3200G with Radeon Vega Graphics
BenchmarkParseMessage/stdlib-4         	      90	  15616341 ns/op
BenchmarkParseMessage/easyjson-4       	     110	  11306466 ns/op
BenchmarkParseMessage/simdjson-4       	     162	   7779856 ns/op
PASS
ok  	github.com/nbd-wtf/go-nostr	5.547s
This commit is contained in:
fiatjaf
2025-03-05 23:42:16 -03:00
parent de358e641c
commit 4fb6fcd9a2
6 changed files with 258 additions and 257 deletions
+16 -13
View File
@@ -17,16 +17,21 @@ var (
attrSig = []byte("sig")
)
func (event *Event) UnmarshalSIMD(iter *simdjson.Iter) error {
obj, err := iter.Object(nil)
func (event *Event) UnmarshalSIMD(
iter *simdjson.Iter,
obj *simdjson.Object,
arr *simdjson.Array,
subArr *simdjson.Array,
) (*simdjson.Object, *simdjson.Array, *simdjson.Array, error) {
obj, err := iter.Object(obj)
if err != nil {
return fmt.Errorf("unexpected at event: %w", err)
return obj, arr, subArr, fmt.Errorf("unexpected at event: %w", err)
}
for {
name, t, err := obj.NextElementBytes(iter)
if err != nil {
return err
return obj, arr, subArr, err
} else if t == simdjson.TypeNone {
break
}
@@ -49,36 +54,34 @@ func (event *Event) UnmarshalSIMD(iter *simdjson.Iter) error {
kind, err = iter.Uint()
event.Kind = int(kind)
case bytes.Equal(name, attrTags):
var arr *simdjson.Array
arr, err = iter.Array(nil)
arr, err = iter.Array(arr)
if err != nil {
return err
return obj, arr, subArr, err
}
event.Tags = make(Tags, 0, 10)
titer := arr.Iter()
var subArr *simdjson.Array
for {
if t := titer.Advance(); t == simdjson.TypeNone {
break
}
subArr, err = titer.Array(subArr)
if err != nil {
return err
return obj, arr, subArr, err
}
tag, err := subArr.AsString()
if err != nil {
return err
return obj, arr, subArr, err
}
event.Tags = append(event.Tags, tag)
}
default:
return fmt.Errorf("unexpected event field '%s'", name)
return obj, arr, subArr, fmt.Errorf("unexpected event field '%s'", name)
}
if err != nil {
return err
return obj, arr, subArr, err
}
}
return nil
return obj, arr, subArr, nil
}