TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

network: implement connection retries

Browse source 
It's worth to try a bit more than once.
This commit is contained in:
Roman Khimov 2019-09-12 16:29:50 +03:00
parent be6c905e5d
commit 85f19936dd
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pkg/network/discovery.go
View file

@ -6,6 +6,7 @@ import (
const ( const (
maxPoolSize = 200 maxPoolSize = 200
connRetries = 3
) )
// Discoverer is an interface that is responsible for maintaining // Discoverer is an interface that is responsible for maintaining
@ -25,7 +26,7 @@ type DefaultDiscovery struct {
dialTimeout time.Duration dialTimeout time.Duration
addrs map[string]bool addrs map[string]bool
badAddrs map[string]bool badAddrs map[string]bool
unconnectedAddrs map[string]bool unconnectedAddrs map[string]int
requestCh chan int requestCh chan int
connectedCh chan string connectedCh chan string
backFill chan string backFill chan string
@ -40,7 +41,7 @@ func NewDefaultDiscovery(dt time.Duration, ts Transporter) *DefaultDiscovery {
dialTimeout: dt, dialTimeout: dt,
addrs: make(map[string]bool), addrs: make(map[string]bool),
badAddrs: make(map[string]bool), badAddrs: make(map[string]bool),
unconnectedAddrs: make(map[string]bool), unconnectedAddrs: make(map[string]int),
requestCh: make(chan int), requestCh: make(chan int),
connectedCh: make(chan string), connectedCh: make(chan string),
backFill: make(chan string), backFill: make(chan string),
@ -143,12 +144,17 @@ func (d *DefaultDiscovery) run() {
} }
if _, ok := d.addrs[addr]; !ok { if _, ok := d.addrs[addr]; !ok {
d.addrs[addr] = true d.addrs[addr] = true
d.unconnectedAddrs[addr] = true d.unconnectedAddrs[addr] = connRetries
d.pool <- addr d.pool <- addr
} }
case addr := <-d.badAddrCh: case addr := <-d.badAddrCh:
d.badAddrs[addr] = true d.unconnectedAddrs[addr]--
delete(d.unconnectedAddrs, addr) if d.unconnectedAddrs[addr] > 0 {
d.pool <- addr
} else {
d.badAddrs[addr] = true
delete(d.unconnectedAddrs, addr)
}
d.RequestRemote(1) d.RequestRemote(1)
case addr := <-d.connectedCh: case addr := <-d.connectedCh: