highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-10-11 05:13:47 +02:00
Code Issues Packages Projects Releases Wiki Activity

pool.SubMany(Eose)NonUnique()

Browse Source
This commit is contained in:
fiatjaf
2023-10-02 14:16:16 -03:00
parent c68e876b05
commit e45921c11a
1 changed files with 41 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
pool.go
View File

@@ -57,8 +57,17 @@ func (pool *SimplePool) EnsureRelay(url string) (*Relay, error) {
// SubMany opens a subscription with the given filters to multiple relays // SubMany opens a subscription with the given filters to multiple relays
// the subscriptions only end when the context is canceled // the subscriptions only end when the context is canceled
func (pool *SimplePool) SubMany(ctx context.Context, urls []string, filters Filters) chan IncomingEvent { func (pool *SimplePool) SubMany(ctx context.Context, urls []string, filters Filters, unique bool) chan IncomingEvent {
uniqueEvents := make(chan IncomingEvent) return pool.subMany(ctx, urls, filters, true)
}
// SubManyNonUnique is like SubMany, but returns duplicate events if they come from different relays
func (pool *SimplePool) SubManyNonUnique(ctx context.Context, urls []string, filters Filters, unique bool) chan IncomingEvent {
return pool.subMany(ctx, urls, filters, false)
}
func (pool *SimplePool) subMany(ctx context.Context, urls []string, filters Filters, unique bool) chan IncomingEvent {
events := make(chan IncomingEvent)
seenAlready := xsync.NewMapOf[bool]() seenAlready := xsync.NewMapOf[bool]()
pending := xsync.NewCounter() pending := xsync.NewCounter()
@@ -77,27 +86,43 @@ func (pool *SimplePool) SubMany(ctx context.Context, urls []string, filters Filt
} }
for evt := range sub.Events { for evt := range sub.Events {
// dispatch unique events to client stop := true
if _, ok := seenAlready.LoadOrStore(evt.ID, true); !ok { if unique {
uniqueEvents <- IncomingEvent{Event: evt, Relay: relay} _, stop = seenAlready.LoadOrStore(evt.ID, true)
}
if !stop {
select {
case events <- IncomingEvent{Event: evt, Relay: relay}:
case <-ctx.Done():
return
}
} }
} }
pending.Dec() pending.Dec()
if pending.Value() == 0 { if pending.Value() == 0 {
close(uniqueEvents) close(events)
} }
}(NormalizeURL(url)) }(NormalizeURL(url))
} }
return uniqueEvents return events
} }
// SubManyEose is like SubMany, but it stops subscriptions and closes the channel when gets a EOSE // SubManyEose is like SubMany, but it stops subscriptions and closes the channel when gets a EOSE
func (pool *SimplePool) SubManyEose(ctx context.Context, urls []string, filters Filters) chan IncomingEvent { func (pool *SimplePool) SubManyEose(ctx context.Context, urls []string, filters Filters) chan IncomingEvent {
return pool.subManyEose(ctx, urls, filters, true)
}
// SubManyEoseNonUnique is like SubManyEose, but returns duplicate events if they come from different relays
func (pool *SimplePool) SubManyEoseNonUnique(ctx context.Context, urls []string, filters Filters) chan IncomingEvent {
return pool.subManyEose(ctx, urls, filters, false)
}
func (pool *SimplePool) subManyEose(ctx context.Context, urls []string, filters Filters, unique bool) chan IncomingEvent {
ctx, cancel := context.WithCancel(ctx) ctx, cancel := context.WithCancel(ctx)
uniqueEvents := make(chan IncomingEvent) events := make(chan IncomingEvent)
seenAlready := xsync.NewMapOf[bool]() seenAlready := xsync.NewMapOf[bool]()
wg := sync.WaitGroup{} wg := sync.WaitGroup{}
wg.Add(len(urls)) wg.Add(len(urls))
@@ -106,7 +131,7 @@ func (pool *SimplePool) SubManyEose(ctx context.Context, urls []string, filters
// this will happen when all subscriptions get an eose (or when they die) // this will happen when all subscriptions get an eose (or when they die)
wg.Wait() wg.Wait()
cancel() cancel()
close(uniqueEvents) close(events)
}() }()
for _, url := range urls { for _, url := range urls {
@@ -135,10 +160,13 @@ func (pool *SimplePool) SubManyEose(ctx context.Context, urls []string, filters
return return
} }
// dispatch unique events to client stop := true
if _, ok := seenAlready.LoadOrStore(evt.ID, true); !ok { if unique {
_, stop = seenAlready.LoadOrStore(evt.ID, true)
}
if !stop {
select { select {
case uniqueEvents <- IncomingEvent{Event: evt, Relay: relay}: case events <- IncomingEvent{Event: evt, Relay: relay}:
case <-ctx.Done(): case <-ctx.Done():
return return
} }
@@ -148,7 +176,7 @@ func (pool *SimplePool) SubManyEose(ctx context.Context, urls []string, filters
}(NormalizeURL(url)) }(NormalizeURL(url))
} }
return uniqueEvents return events
} }
// QuerySingle returns the first event returned by the first relay, cancels everything else. // QuerySingle returns the first event returned by the first relay, cancels everything else.