diff options
| author | Felix Hanley <felix@userspace.com.au> | 2018-03-20 11:37:09 +0000 |
|---|---|---|
| committer | Felix Hanley <felix@userspace.com.au> | 2018-03-20 11:37:09 +0000 |
| commit | e2e746843dca7874d287420336b3ec4830203ff5 (patch) | |
| tree | c806afa152d33f1124ea03e943db60ead0497b9e /dht/node.go | |
| parent | ef4aec50a8f6ace8076d4bfeef0ba64cf8598048 (diff) | |
| download | dhtsearch-e2e746843dca7874d287420336b3ec4830203ff5.tar.gz dhtsearch-e2e746843dca7874d287420336b3ec4830203ff5.tar.bz2 | |
Start working on SQL to relieve load
Diffstat (limited to 'dht/node.go')
| -rw-r--r-- | dht/node.go | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/dht/node.go b/dht/node.go index 829f244..b0dd9ec 100644 --- a/dht/node.go +++ b/dht/node.go @@ -40,7 +40,9 @@ type Node struct { blacklist *lru.ARCCache // OnAnnoucePeer is called for each peer that announces itself - OnAnnouncePeer func(p models.Peer) + OnAnnouncePeer func(models.Peer) + // OnBadPeer is called for each bad peer + OnBadPeer func(models.Peer) } // NewNode creates a new DHT node @@ -191,6 +193,10 @@ func (n *Node) packetWriter() { n.blacklist.Add(p.raddr.String(), true) // TODO reduce limit n.log.Warn("failed to write packet", "error", err) + if n.OnBadPeer != nil { + peer := models.Peer{Addr: p.raddr} + go n.OnBadPeer(peer) + } } } } @@ -245,7 +251,7 @@ func (n *Node) processPacket(p packet) error { } if _, black := n.blacklist.Get(p.raddr.String()); black { - return fmt.Errorf("blacklisted", "address", p.raddr.String()) + return fmt.Errorf("blacklisted: %s", p.raddr.String()) } switch y { |