NeighborSet.java

package net.tomp2p.message;

import java.util.Collection;
import java.util.Iterator;

import net.tomp2p.peers.PeerAddress;
import net.tomp2p.utils.Utils;

public class NeighborSet {
    private final int neighborLimit;
    //this needs to be a collection as we want to process lists *and* sets
    private final Collection<PeerAddress> neighbors;

    public NeighborSet(final int neighborLimit, final Collection<PeerAddress> neighbors) {
        this.neighborLimit = neighborLimit;
        this.neighbors = neighbors;
        // remove neighbors that are over the limit
        int serializedSize = 1;
        // no need to cut if we don't provide a limit
        if (neighborLimit >= 0) {
			for (Iterator<PeerAddress> iterator = neighbors.iterator(); iterator.hasNext();) {
				PeerAddress neighbor = iterator.next();
				serializedSize += neighbor.size();
				if (serializedSize > neighborLimit) {
					iterator.remove();
				}
			}
        }
    }

    public Collection<PeerAddress> neighbors() {
        return neighbors;
    }

    public void addResult(Collection<PeerAddress> neighbors) {
        this.neighbors.addAll(neighbors);
    }

    public void add(PeerAddress neighbor) {
        this.neighbors.add(neighbor);
    }

    public int size() {
        return this.neighbors.size();
    }

    public int neighborLimit() {
        return neighborLimit;
    }

    @Override
    public boolean equals(Object obj) {
    	if (!(obj instanceof NeighborSet)) {
            return false;
        }
        if (obj == this) {
            return true;
        }
        final NeighborSet other = (NeighborSet) obj;

        boolean t1 = this.neighborLimit == other.neighborLimit;
        boolean t2 = Utils.isSameSets(this.neighbors, other.neighbors);

        return t1 && t2;
    }
    
    @Override
    public String toString() {
    	StringBuilder sb = new StringBuilder("NeighborSet:");
    	for(PeerAddress pa:neighbors) {
    		sb.append(pa).append(';');
    	}
    	return sb.toString();
    }
}