/*
* Copyright 2004-2010 Brian S O'Neill
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.cojen.util;
import java.lang.ref.Reference;
import java.lang.ref.ReferenceQueue;
import java.lang.ref.WeakReference;
import java.util.AbstractCollection;
import java.util.AbstractMap;
import java.util.AbstractSet;
import java.util.Collection;
import java.util.Collections;
import java.util.ConcurrentModificationException;
import java.util.Iterator;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Set;
/**
* WeakIdentityMap is like WeakHashMap, except it uses a key's identity
* hashcode and equals methods. WeakIdentityMap is not thread-safe and must be
* wrapped with Collections.synchronizedMap to be made thread-safe.
* <p>
* The documentation for WeakHashMap states that it is intended primarily for
* use with key objects whose equals methods test for object identity using the
* == operator. Because WeakIdentityMap strictly follows this behavior, it is
* better suited for this purpose.
* <p>
* Note: Weakly referenced entries may be automatically removed during
* either accessor or mutator operations, possibly causing a concurrent
* modification to be detected. Therefore, even if multiple threads are only
* accessing this map, be sure to synchronize this map first. Also, do not
* rely on the value returned by size() when using an iterator from this map.
* The iterators may return less entries than the amount reported by size().
*
* @author Brian S O'Neill
*/
public class WeakIdentityMap<K, V> extends AbstractMap<K, V> implements Map<K, V>, Cloneable {
// Types of Iterators
static final int KEYS = 0;
static final int VALUES = 1;
static final int ENTRIES = 2;
/**
* Converts a collection to string, supporting collections that contain
* self references
*/
static String toString(Collection c) {
if (c.size() == 0) {
return "[]";
}
StringBuffer buf = new StringBuffer(32 * c.size());
buf.append('[');
Iterator it = c.iterator();
boolean hasNext = it.hasNext();
while (hasNext) {
Object obj = it.next();
buf.append(obj == c ? "(this Collection)" : obj);
if (hasNext = it.hasNext()) {
buf.append(", ");
}
}
buf.append("]");
return buf.toString();
}
/**
* Converts a map to string, supporting maps that contain self references
*/
static String toString(Map m) {
if (m.size() == 0) {
return "{}";
}
StringBuffer buf = new StringBuffer(32 * m.size());
buf.append('{');
Iterator it = m.entrySet().iterator();
boolean hasNext = it.hasNext();
while (hasNext) {
Map.Entry entry = (Map.Entry)it.next();
Object key = entry.getKey();
Object value = entry.getValue();
buf.append(key == m ? "(this Map)" : key)
.append('=')
.append(value == m ? "(this Map)" : value);
hasNext = it.hasNext();
if (hasNext) {
buf.append(',').append(' ');
}
}
buf.append('}');
return buf.toString();
}
private transient Entry<K, V>[] table;
private transient int count;
private int threshold;
private final float loadFactor;
private final ReferenceQueue<K> queue;
private transient volatile int modCount;
// Views
private transient Set<K> keySet;
private transient Set<Map.Entry<K, V>> entrySet;
private transient Collection<V> values;
public WeakIdentityMap(int initialCapacity, float loadFactor) {
if (initialCapacity <= 0) {
throw new IllegalArgumentException("Initial capacity must be greater than 0");
}
if (loadFactor <= 0 || Float.isNaN(loadFactor)) {
throw new IllegalArgumentException("Load factor must be greater than 0");
}
this.loadFactor = loadFactor;
this.table = new Entry[initialCapacity];
this.threshold = (int)(initialCapacity * loadFactor);
this.queue = new ReferenceQueue();
}
public WeakIdentityMap(int initialCapacity) {
this(initialCapacity, 0.75f);
}
public WeakIdentityMap() {
this(11, 0.75f);
}
public WeakIdentityMap(Map<? extends K, ? extends V> t) {
this(Math.max(2 * t.size(), 11), 0.75f);
putAll(t);
}
public int size() {
// Cleanup right before, to report a more accurate size.
cleanup();
return this.count;
}
public boolean isEmpty() {
return this.count == 0;
}
public boolean containsValue(Object value) {
Entry[] tab = this.table;
if (value == null) {
for (int i = tab.length ; i-- > 0 ;) {
for (Entry e = tab[i], prev = null; e != null; e = e.next) {
if (e.get() == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[i] = e.next;
}
this.count--;
} else if (e.value == null) {
return true;
} else {
prev = e;
}
}
}
} else {
for (int i = tab.length ; i-- > 0 ;) {
for (Entry e = tab[i], prev = null; e != null; e = e.next) {
if (e.get() == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[i] = e.next;
}
this.count--;
} else if (value.equals(e.value)) {
return true;
} else {
prev = e;
}
}
}
}
return false;
}
public boolean containsKey(Object key) {
if (key == null) {
key = KeyFactory.NULL;
}
Entry[] tab = this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
Object entryKey = e.get();
if (entryKey == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
} else if (e.hash == hash && key == entryKey) {
return true;
} else {
prev = e;
}
}
return false;
}
public V get(Object key) {
if (key == null) {
key = KeyFactory.NULL;
}
Entry<K, V>[] tab = this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry<K, V> e = tab[index], prev = null; e != null; e = e.next) {
Object entryKey = e.get();
if (entryKey == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
} else if (e.hash == hash && key == entryKey) {
return e.value;
} else {
prev = e;
}
}
return null;
}
private void cleanup() {
// Cleanup after cleared References.
Entry[] tab = this.table;
ReferenceQueue queue = this.queue;
Reference ref;
while ((ref = queue.poll()) != null) {
// Since buckets are single-linked, traverse entire list and
// cleanup all cleared references in it.
int index = (((Entry) ref).hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
if (e.get() == null) {
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
} else {
prev = e;
}
}
}
}
private void rehash() {
int oldCapacity = this.table.length;
Entry[] oldMap = this.table;
int newCapacity = oldCapacity * 2 + 1;
if (newCapacity <= 0) {
// Overflow.
if ((newCapacity = Integer.MAX_VALUE) == oldCapacity) {
return;
}
}
Entry[] newMap = new Entry[newCapacity];
this.modCount++;
this.threshold = (int)(newCapacity * this.loadFactor);
this.table = newMap;
for (int i = oldCapacity ; i-- > 0 ;) {
for (Entry old = oldMap[i] ; old != null ; ) {
Entry e = old;
old = old.next;
// Only copy entry if its key hasn't been cleared.
if (e.get() == null) {
this.count--;
} else {
int index = (e.hash & 0x7fffffff) % newCapacity;
e.next = newMap[index];
newMap[index] = e;
}
}
}
}
public V put(K key, V value) {
if (key == null) {
key = (K) KeyFactory.NULL;
}
cleanup();
// Make sure the key is not already in the WeakIdentityMap.
Entry[] tab = this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
Object entryKey = e.get();
if (entryKey == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
} else if (e.hash == hash && key == entryKey) {
Object old = e.value;
e.value = value;
return (V) old;
} else {
prev = e;
}
}
this.modCount++;
if (this.count >= this.threshold) {
// Rehash the table if the threshold is still exceeded.
rehash();
tab = this.table;
index = (hash & 0x7fffffff) % tab.length;
}
// Creates the new entry.
Entry e = new Entry(hash, key, this.queue, value, tab[index]);
tab[index] = e;
this.count++;
return null;
}
public V remove(Object key) {
if (key == null) {
key = KeyFactory.NULL;
}
Entry<K, V>[] tab = this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry<K, V> e = tab[index], prev = null; e != null; e = e.next) {
Object entryKey = e.get();
if (entryKey == null) {
// Clean up after a cleared Reference.
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
} else if (e.hash == hash && key == entryKey) {
this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
this.count--;
V oldValue = e.value;
e.value = null;
return oldValue;
} else {
prev = e;
}
}
return null;
}
public void putAll(Map<? extends K, ? extends V> t) {
Iterator i = t.entrySet().iterator();
while (i.hasNext()) {
Map.Entry e = (Map.Entry) i.next();
put((K) e.getKey(), (V) e.getValue());
}
}
public void clear() {
Entry[] tab = this.table;
this.modCount++;
for (int index = tab.length; --index >= 0; ) {
tab[index] = null;
}
this.count = 0;
}
public Object clone() {
try {
WeakIdentityMap t = (WeakIdentityMap)super.clone();
t.table = new Entry[this.table.length];
for (int i = this.table.length ; i-- > 0 ; ) {
t.table[i] = (this.table[i] != null)
? (Entry)this.table[i].copy(this.queue) : null;
}
t.keySet = null;
t.entrySet = null;
t.values = null;
t.modCount = 0;
return t;
}
catch (CloneNotSupportedException e) {
// this shouldn't happen, since we are Cloneable
throw new InternalError();
}
}
public Set<K> keySet() {
if (this.keySet == null) {
this.keySet = new AbstractSet<K>() {
public Iterator iterator() {
return createHashIterator(KEYS);
}
public int size() {
return WeakIdentityMap.this.count;
}
public boolean contains(Object o) {
return containsKey(o);
}
public boolean remove(Object o) {
return o == null ? false : WeakIdentityMap.this.remove(o) == o;
}
public void clear() {
WeakIdentityMap.this.clear();
}
public String toString() {
return WeakIdentityMap.this.toString(this);
}
};
}
return this.keySet;
}
public Collection<V> values() {
if (this.values==null) {
this.values = new AbstractCollection<V>() {
public Iterator<V> iterator() {
return createHashIterator(VALUES);
}
public int size() {
return WeakIdentityMap.this.count;
}
public boolean contains(Object o) {
return containsValue(o);
}
public void clear() {
WeakIdentityMap.this.clear();
}
public String toString() {
return WeakIdentityMap.this.toString(this);
}
};
}
return this.values;
}
public Set<Map.Entry<K, V>> entrySet() {
if (this.entrySet==null) {
this.entrySet = new AbstractSet<Map.Entry<K, V>>() {
public Iterator<Map.Entry<K, V>> iterator() {
return createHashIterator(ENTRIES);
}
public boolean contains(Object o) {
if (!(o instanceof Map.Entry)) {
return false;
}
Map.Entry entry = (Map.Entry)o;
Object key = entry.getKey();
Entry[] tab = WeakIdentityMap.this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
Object entryKey = e.get();
if (entryKey == null) {
// Clean up after a cleared Reference.
WeakIdentityMap.this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
WeakIdentityMap.this.count--;
} else if (e.hash == hash && e.equals(entry)) {
return true;
} else {
prev = e;
}
}
return false;
}
public boolean remove(Object o) {
if (!(o instanceof Map.Entry)) {
return false;
}
Map.Entry entry = (Map.Entry)o;
Object key = entry.getKey();
Entry[] tab = WeakIdentityMap.this.table;
int hash = System.identityHashCode(key);
int index = (hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
if (e.get() == null) {
// Clean up after a cleared Reference.
WeakIdentityMap.this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
WeakIdentityMap.this.count--;
} else if (e.hash == hash && e.equals(entry)) {
WeakIdentityMap.this.modCount++;
if (prev != null) {
prev.next = e.next;
} else {
tab[index] = e.next;
}
WeakIdentityMap.this.count--;
e.value = null;
return true;
} else {
prev = e;
}
}
return false;
}
public int size() {
return WeakIdentityMap.this.count;
}
public void clear() {
WeakIdentityMap.this.clear();
}
public String toString() {
return WeakIdentityMap.toString(this);
}
};
}
return this.entrySet;
}
/**
* Gets the map as a String.
*
* @return a string version of the map
*/
public String toString() {
return toString(this);
}
private Iterator createHashIterator(int type) {
if (this.count == 0) {
return Collections.EMPTY_SET.iterator();
} else {
return new HashIterator(type);
}
}
/**
* WeakIdentityMap collision list entry.
*/
private static class Entry<K, V> extends WeakReference<K> implements Map.Entry<K, V> {
int hash;
V value;
Entry<K, V> next;
Entry(int hash, K key, ReferenceQueue<K> queue, V value, Entry<K, V> next) {
super(key, queue);
this.hash = hash;
this.value = value;
this.next = next;
}
public void clear() {
// Do nothing if reference is explicity cleared. This prevents
// backdoor modification of map entries.
}
public K getKey() {
K key = Entry.this.get();
return key == KeyFactory.NULL ? null : key;
}
public V getValue() {
return this.value;
}
public V setValue(V value) {
V oldValue = this.value;
this.value = value;
return oldValue;
}
public boolean equals(Object obj) {
if (!(obj instanceof Map.Entry)) {
return false;
}
return equals((Map.Entry)obj);
}
boolean equals(Map.Entry<K, V> e) {
Object thisKey = get();
if (thisKey == null) {
return false;
} else if (thisKey == KeyFactory.NULL) {
thisKey = null;
}
return (thisKey == e.getKey()) &&
(this.value == null ? e.getValue() == null : this.value.equals(e.getValue()));
}
public int hashCode() {
return this.hash ^ (this.value == null ? 0 : this.value.hashCode());
}
public String toString() {
return getKey() + "=" + this.value;
}
protected Object copy(ReferenceQueue queue) {
return new Entry(this.hash, get(), queue, this.value,
(this.next == null ? null : (Entry)this.next.copy(queue)));
}
}
private class HashIterator implements Iterator {
private final int type;
private final Entry[] table;
private int index;
// To ensure that the iterator doesn't return cleared entries, keep a
// hard reference to the key. Its existence will prevent the weak
// key from being cleared.
Object entryKey;
Entry entry;
Entry last;
/**
* The modCount value that the iterator believes that the backing
* List should have. If this expectation is violated, the iterator
* has detected concurrent modification.
*/
private int expectedModCount = WeakIdentityMap.this.modCount;
HashIterator(int type) {
this.table = WeakIdentityMap.this.table;
this.type = type;
this.index = table.length;
}
public boolean hasNext() {
while (this.entry == null || (this.entryKey = this.entry.get()) == null) {
if (this.entry != null) {
// Clean up after a cleared Reference.
remove(this.entry);
this.entry = this.entry.next;
}
else {
if (this.index <= 0) {
return false;
}
else {
this.entry = this.table[--this.index];
}
}
}
return true;
}
public Object next() {
if (WeakIdentityMap.this.modCount != this.expectedModCount) {
throw new ConcurrentModificationException();
}
if (!hasNext()) {
throw new NoSuchElementException();
}
this.last = this.entry;
this.entry = this.entry.next;
return this.type == KEYS ? this.last.getKey() :
(this.type == VALUES ? this.last.getValue() : this.last);
}
public void remove() {
if (this.last == null) {
throw new IllegalStateException();
}
if (WeakIdentityMap.this.modCount != this.expectedModCount) {
throw new ConcurrentModificationException();
}
remove(this.last);
this.last = null;
}
private void remove(Entry toRemove) {
Entry[] tab = this.table;
int index = (toRemove.hash & 0x7fffffff) % tab.length;
for (Entry e = tab[index], prev = null; e != null; e = e.next) {
if (e == toRemove) {
WeakIdentityMap.this.modCount++;
expectedModCount++;
if (prev == null) {
tab[index] = e.next;
} else {
prev.next = e.next;
}
WeakIdentityMap.this.count--;
return;
} else {
prev = e;
}
}
throw new ConcurrentModificationException();
}
public String toString() {
if (this.last != null) {
return "Iterator[" + this.last + ']';
} else {
return "Iterator[]";
}
}
}
}