mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-08-20 10:52:25 +00:00
event: improve fix for the goroutine leak in resubscribe
This commit is contained in:
parent
da12747636
commit
5968e03e1e
1 changed files with 9 additions and 10 deletions
|
|
@ -143,9 +143,8 @@ func (s *resubscribeSub) loop() {
|
|||
}
|
||||
|
||||
func (s *resubscribeSub) subscribe() Subscription {
|
||||
subscribed := make(chan error, 1)
|
||||
subscribed := make(chan error)
|
||||
var sub Subscription
|
||||
retry:
|
||||
for {
|
||||
s.lastTry = mclock.Now()
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
|
|
@ -157,19 +156,19 @@ retry:
|
|||
select {
|
||||
case err := <-subscribed:
|
||||
cancel()
|
||||
if err != nil {
|
||||
// Subscribing failed, wait before launching the next try.
|
||||
if s.backoffWait() {
|
||||
return nil
|
||||
if err == nil {
|
||||
if sub == nil {
|
||||
panic("event: ResubscribeFunc returned nil subscription and no error")
|
||||
}
|
||||
continue retry
|
||||
return sub
|
||||
}
|
||||
if sub == nil {
|
||||
panic("event: ResubscribeFunc returned nil subscription and no error")
|
||||
// Subscribing failed, wait before launching the next try.
|
||||
if s.backoffWait() {
|
||||
return nil // unsubscribed during wait
|
||||
}
|
||||
return sub
|
||||
case <-s.unsub:
|
||||
cancel()
|
||||
<-subscribed // avoid leaking the s.fn goroutine.
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue