highperfocused/go-nostr
1
0
Fork 0
mirror of https://github.com/nbd-wtf/go-nostr.git synced 2025-03-17 21:32:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

prevent halting in some other places.

Browse Source
This commit is contained in:
fiatjaf 2023-07-11 11:04:48 -03:00
parent f6dddfd770
commit 6e79f72174
No known key found for this signature in database
GPG Key ID: BAD43C4BE5C1A3A1
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
relay.go
View File

@ -223,6 +223,9 @@ func (r *Relay) Connect(ctx context.Context) error {
writeRequest.answer <- err
}
close(writeRequest.answer)
case <-r.connectionContext.Done():
// stop here
return
}
}
}()
@ -286,7 +289,10 @@ func (r *Relay) Connect(ctx context.Context) error {
}
// dispatch this to the internal .events channel of the subscription
subscription.events <- &env.Event
select {
case subscription.events <- &env.Event:
case <-subscription.Context.Done():
}
}
case *EOSEEnvelope:
if subscription, ok := r.Subscriptions.Load(string(*env)); ok {
@ -313,7 +319,11 @@ func (r *Relay) Connect(ctx context.Context) error {
// Write queues a message to be sent to the relay.
func (r *Relay) Write(msg []byte) <-chan error {
ch := make(chan error)
r.writeQueue <- writeRequest{msg: msg, answer: ch}
select {
case r.writeQueue <- writeRequest{msg: msg, answer: ch}:
case <-r.connectionContext.Done():
go func() { ch <- fmt.Errorf("connection closed") }()
}
return ch
}