implement envelope, event and filter parsing with simdjson-go.

filter_simdjson.go

@@ -0,0 +1,107 @@
+package nostr
+
+import (
+	"fmt"
+	"slices"
+
+	"github.com/minio/simdjson-go"
+)
+
+var (
+	attrIds     = []byte("ids")
+	attrAuthors = []byte("authors")
+	attrKinds   = []byte("kinds")
+	attrLimit   = []byte("limit")
+	attrSince   = []byte("since")
+	attrUntil   = []byte("until")
+	attrSearch  = []byte("search")
+)
+
+func (filter *Filter) UnmarshalSIMD(iter *simdjson.Iter) error {
+	obj, err := iter.Object(nil)
+	if err != nil {
+		return fmt.Errorf("unexpected at filter: %w", err)
+	}
+
+	var arr *simdjson.Array
+	for {
+		name, t, err := obj.NextElementBytes(iter)
+		if err != nil {
+			return err
+		} else if t == simdjson.TypeNone {
+			break
+		}
+
+		switch {
+		case slices.Equal(name, attrIds):
+			if arr, err = iter.Array(arr); err == nil {
+				filter.IDs, err = arr.AsString()
+			}
+		case slices.Equal(name, attrAuthors):
+			if arr, err = iter.Array(arr); err == nil {
+				filter.Authors, err = arr.AsString()
+			}
+		case slices.Equal(name, attrKinds):
+			if arr, err = iter.Array(arr); err == nil {
+				i := arr.Iter()
+				filter.Kinds = make([]int, 0, 6)
+				for {
+					t := i.Advance()
+					if t == simdjson.TypeNone {
+						break
+					}
+					if kind, err := i.Uint(); err != nil {
+						return err
+					} else {
+						filter.Kinds = append(filter.Kinds, int(kind))
+					}
+				}
+			}
+		case slices.Equal(name, attrSearch):
+			filter.Search, err = iter.String()
+		case slices.Equal(name, attrSince):
+			var tsu uint64
+			tsu, err = iter.Uint()
+			ts := Timestamp(tsu)
+			filter.Since = &ts
+		case slices.Equal(name, attrUntil):
+			var tsu uint64
+			tsu, err = iter.Uint()
+			ts := Timestamp(tsu)
+			filter.Until = &ts
+		case slices.Equal(name, attrLimit):
+			var limit uint64
+			limit, err = iter.Uint()
+			filter.Limit = int(limit)
+			if limit == 0 {
+				filter.LimitZero = true
+			}
+		default:
+			if len(name) > 1 && name[0] == '#' {
+				if filter.Tags == nil {
+					filter.Tags = make(TagMap, 1)
+				}
+
+				arr, err := iter.Array(arr)
+				if err != nil {
+					return err
+				}
+				vals, err := arr.AsString()
+				if err != nil {
+					return err
+				}
+
+				filter.Tags[string(name[1:])] = vals
+				continue
+			}
+
+			return fmt.Errorf("unexpected filter field '%s'", name)
+		}
+
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}