forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 10:52:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make the Resub unsub channel buffered

Browse source
This commit is contained in:
inphi 2023-10-18 15:55:58 -04:00
parent 4964f1953e
commit ad2fa66e91
No known key found for this signature in database
GPG key ID: B61066A1A33F5D24
1 changed files with 6 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
event/subscription.go
View file

@ -120,7 +120,7 @@ func ResubscribeErr(backoffMax time.Duration, fn ResubscribeErrFunc) Subscriptio
backoffMax: backoffMax,
fn: fn,
err: make(chan error),
unsub: make(chan struct{}),
unsub: make(chan struct{}, 1),
}
go s.loop()
return s
@ -154,20 +154,13 @@ func (s *resubscribeSub) Err() <-chan error {
func (s *resubscribeSub) loop() {
var done bool
var unsubbed bool
defer func() {
close(s.err)
// Read unsub chan to avoid blocking Unsubscribe
if !unsubbed {
<-s.unsub
}
}()
defer close(s.err)
for !done {
sub := s.subscribe()
if sub == nil {
break
}
done, unsubbed = s.waitForError(sub)
done = s.waitForError(sub)
sub.Unsubscribe()
}
}
@ -204,14 +197,14 @@ func (s *resubscribeSub) subscribe() Subscription {
}
}
func (s *resubscribeSub) waitForError(sub Subscription) (bool, bool) {
func (s *resubscribeSub) waitForError(sub Subscription) bool {
defer sub.Unsubscribe()
select {
case err := <-sub.Err():
s.lastSubErr = err
return err == nil, false
return err == nil
case <-s.unsub:
return true, true
return true
}
}