IdentitySet.java

package eu.stratosphere.util;

import java.util.AbstractSet;
import java.util.IdentityHashMap;
import java.util.Iterator;

/**
 * Provides a {@link Set} that uses reference-equality and thus intentionally violates the general Set contract.
 * 
 * @author Arvid Heise
 * @param <E>
 *        the type of the elements
 * @see Set
 * @see IdentityHashMap
 */
public class IdentitySet<E> extends AbstractSet<E> {
	private final IdentityHashMap<E, Object> backing = new IdentityHashMap<E, Object>();

	@Override
	public boolean add(final E e) {
		return this.backing.put(e, null) == null;
	}

	@Override
	public void clear() {
		this.backing.clear();
	}

	@Override
	public boolean contains(final Object o) {
		return this.backing.containsKey(o);
	};

	@Override
	public Iterator<E> iterator() {
		return this.backing.keySet().iterator();
	}

	@Override
	public boolean remove(final Object o) {
		return this.backing.keySet().remove(o);
	}

	@Override
	public int size() {
		return this.backing.size();
	}
}