func (r *defaultRouter) remove(p peer.NodePeer) error {
r.mutex.Lock()
defer r.mutex.Unlock()
node := p.Node()
if _, ok := r.peers[node.String()]; !ok {
return fmt.Errorf("Peer Not found")
}
delete(r.peers, node.String())
delete(r.peerIDs, p.Id())
return nil
}
func (r *defaultRouter) accept(p peer.NodePeer) error {
r.mutex.Lock()
defer r.mutex.Unlock()
var node node.Node = p.Node()
if _, ok := r.peers[node.String()]; ok {
return fmt.Errorf("Peer: %s Already registered", node.String())
}
r.peers[node.String()] = p
r.peerIDs[p.Id()] = true
return nil
}