TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-12-13 15:45:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

network: an address should either be good or bad, but not both

Browse source
This commit is contained in:
Roman Khimov 2020-10-13 14:16:06 +03:00
parent 4ff5bb361a
commit b36371ed94
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkg/network/discovery.go
View file

@ -107,6 +107,7 @@ func (d *DefaultDiscovery) RegisterBadAddr(addr string) {
} else { } else {
d.badAddrs[addr] = true d.badAddrs[addr] = true
delete(d.unconnectedAddrs, addr) delete(d.unconnectedAddrs, addr)
delete(d.goodAddrs, addr)
} }
d.lock.Unlock() d.lock.Unlock()
} }
@ -150,6 +151,7 @@ func (d *DefaultDiscovery) GoodPeers() []string {
func (d *DefaultDiscovery) RegisterGoodAddr(s string) { func (d *DefaultDiscovery) RegisterGoodAddr(s string) {
d.lock.Lock() d.lock.Lock()
d.goodAddrs[s] = true d.goodAddrs[s] = true
delete(d.badAddrs, s)
d.lock.Unlock() d.lock.Unlock()
} }

4
pkg/network/discovery_test.go
View file

@ -134,14 +134,14 @@ func TestDefaultDiscoverer(t *testing.T) {
time.Sleep(time.Second) time.Sleep(time.Second)
} }
assert.Equal(t, len(set1), len(d.BadPeers())) assert.Equal(t, len(set1), len(d.BadPeers()))
assert.Equal(t, len(set1), len(d.GoodPeers())) assert.Equal(t, 0, len(d.GoodPeers()))
assert.Equal(t, 0, len(d.UnconnectedPeers())) assert.Equal(t, 0, len(d.UnconnectedPeers()))
// Re-adding bad addresses is a no-op. // Re-adding bad addresses is a no-op.
d.BackFill(set1...) d.BackFill(set1...)
assert.Equal(t, 0, len(d.UnconnectedPeers())) assert.Equal(t, 0, len(d.UnconnectedPeers()))
assert.Equal(t, len(set1), len(d.BadPeers())) assert.Equal(t, len(set1), len(d.BadPeers()))
assert.Equal(t, len(set1), len(d.GoodPeers())) assert.Equal(t, 0, len(d.GoodPeers()))
require.Equal(t, 0, d.PoolCount()) require.Equal(t, 0, d.PoolCount())
// Close should work and subsequent RequestRemote is a no-op. // Close should work and subsequent RequestRemote is a no-op.