highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-04-12 05:39:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

use xsync maps for pool relays to prevent concurrent access panics.

Browse Source
This commit is contained in:
fiatjaf 2023-11-16 14:51:33 -03:00
parent 9287b9fb5e
commit 307df51b9a
No known key found for this signature in database
GPG Key ID: BAD43C4BE5C1A3A1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pool.go
View File

@ -10,7 +10,7 @@ import (
)
type SimplePool struct {
Relays map[string]*Relay
Relays *xsync.MapOf[string, *Relay]
Context context.Context
cancel context.CancelFunc
@ -25,7 +25,7 @@ func NewSimplePool(ctx context.Context) *SimplePool {
ctx, cancel := context.WithCancel(ctx)
return &SimplePool{
Relays: make(map[string]*Relay),
Relays: xsync.NewMapOf[*Relay](),
Context: ctx,
cancel: cancel,
@ -37,7 +37,7 @@ func (pool *SimplePool) EnsureRelay(url string) (*Relay, error) {
defer namedLock(url)()
relay, ok := pool.Relays[nm]
relay, ok := pool.Relays.Load(nm)
if ok && relay.IsConnected() {
// already connected, unlock and return
return relay, nil
@ -50,7 +50,7 @@ func (pool *SimplePool) EnsureRelay(url string) (*Relay, error) {
return nil, fmt.Errorf("failed to connect: %w", err)
}
pool.Relays[nm] = relay
pool.Relays.Store(nm, relay)
return relay, nil
}
}