coracle/nostrlib
2
1
Fork 2
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

khatru: byAuthor and byKind as xsync maps.

Browse Source
This commit is contained in:
fiatjaf
2026-04-16 07:20:48 -03:00
parent d56bdba3ff
commit 31473172a9
1 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
khatru/listener.go
+14 -14
View File
@@ -33,16 +33,16 @@ type subscription struct {
type dispatcher struct { type dispatcher struct {
serial int serial int
subscriptions *xsync.MapOf[int, subscription] subscriptions *xsync.MapOf[int, subscription]
byAuthor map[nostr.PubKey]set.Set[int] byAuthor *xsync.MapOf[nostr.PubKey, set.Set[int]]
byKind map[nostr.Kind]set.Set[int] byKind *xsync.MapOf[nostr.Kind, set.Set[int]]
fallback set.Set[int] fallback set.Set[int]
} }
func newDispatcher() dispatcher { func newDispatcher() dispatcher {
return dispatcher{ return dispatcher{
subscriptions: xsync.NewMapOf[int, subscription](), subscriptions: xsync.NewMapOf[int, subscription](),
byAuthor: make(map[nostr.PubKey]set.Set[int]), byAuthor: xsync.NewMapOf[nostr.PubKey, set.Set[int]](),
byKind: make(map[nostr.Kind]set.Set[int]), byKind: xsync.NewMapOf[nostr.Kind, set.Set[int]](),
fallback: set.NewSliceSet[int](), fallback: set.NewSliceSet[int](),
} }
} }
@@ -57,10 +57,10 @@ func (d *dispatcher) addSubscription(sub subscription) int {
if sub.filter.Authors != nil { if sub.filter.Authors != nil {
indexed = true indexed = true
for _, author := range sub.filter.Authors { for _, author := range sub.filter.Authors {
s, ok := d.byAuthor[author] s, ok := d.byAuthor.Load(author)
if !ok { if !ok {
s = set.NewSliceSet[int]() s = set.NewSliceSet[int]()
d.byAuthor[author] = s d.byAuthor.Store(author, s)
} }
s.Add(ssid) s.Add(ssid)
} }
@@ -69,10 +69,10 @@ func (d *dispatcher) addSubscription(sub subscription) int {
if sub.filter.Kinds != nil { if sub.filter.Kinds != nil {
indexed = true indexed = true
for _, kind := range sub.filter.Kinds { for _, kind := range sub.filter.Kinds {
s, ok := d.byKind[kind] s, ok := d.byKind.Load(kind)
if !ok { if !ok {
s = set.NewSliceSet[int]() s = set.NewSliceSet[int]()
d.byKind[kind] = s d.byKind.Store(kind, s)
} }
s.Add(ssid) s.Add(ssid)
} }
@@ -95,13 +95,13 @@ func (d *dispatcher) removeSubscription(ssid int) {
if sub.filter.Authors != nil { if sub.filter.Authors != nil {
indexed = true indexed = true
for _, author := range sub.filter.Authors { for _, author := range sub.filter.Authors {
s, ok := d.byAuthor[author] s, ok := d.byAuthor.Load(author)
if !ok { if !ok {
return return
} }
s.Remove(ssid) s.Remove(ssid)
if s.Len() == 0 { if s.Len() == 0 {
delete(d.byAuthor, author) d.byAuthor.Delete(author)
} }
} }
} }
@@ -109,13 +109,13 @@ func (d *dispatcher) removeSubscription(ssid int) {
if sub.filter.Kinds != nil { if sub.filter.Kinds != nil {
indexed = true indexed = true
for _, kind := range sub.filter.Kinds { for _, kind := range sub.filter.Kinds {
s, ok := d.byKind[kind] s, ok := d.byKind.Load(kind)
if !ok { if !ok {
return return
} }
s.Remove(ssid) s.Remove(ssid)
if s.Len() == 0 { if s.Len() == 0 {
delete(d.byKind, kind) d.byKind.Delete(kind)
} }
} }
} }
@@ -127,8 +127,8 @@ func (d *dispatcher) removeSubscription(ssid int) {
func (d *dispatcher) candidates(event nostr.Event) iter.Seq[subscription] { func (d *dispatcher) candidates(event nostr.Event) iter.Seq[subscription] {
return func(yield func(subscription) bool) { return func(yield func(subscription) bool) {
authorSubs, hasAuthorSubs := d.byAuthor[event.PubKey] authorSubs, hasAuthorSubs := d.byAuthor.Load(event.PubKey)
kindSubs, hasKindSubs := d.byKind[event.Kind] kindSubs, hasKindSubs := d.byKind.Load(event.Kind)
if hasAuthorSubs && hasKindSubs { if hasAuthorSubs && hasKindSubs {
for _, ssid := range authorSubs.Slice() { for _, ssid := range authorSubs.Slice() {