diff src/goodjava/util/CacheMap.java @ 1498:1b809d2fdf03

add CacheMap
author Franklin Schmidt <fschmidt@gmail.com>
date Fri, 08 May 2020 18:07:14 -0600
parents
children fa066aaa068c
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/goodjava/util/CacheMap.java	Fri May 08 18:07:14 2020 -0600
@@ -0,0 +1,156 @@
+package goodjava.util;
+
+import java.lang.ref.ReferenceQueue;
+import java.util.Map;
+import java.util.AbstractMap;
+import java.util.HashMap;
+import java.util.Set;
+import java.util.AbstractSet;
+import java.util.Iterator;
+
+
+public abstract class CacheMap<K,V> extends AbstractMap<K,V> {
+
+	protected static interface MyReference<K,V> {
+		public K key();
+		public V get();
+		public void clear();
+	}
+
+	protected abstract MyReference<K,V> newReference(K key,V value,ReferenceQueue<V> q);
+ 
+	private final Map<K,MyReference<K,V>> cache = new HashMap<K,MyReference<K,V>>();
+	private final ReferenceQueue<V> queue = new ReferenceQueue<V>();
+
+	private void sweep() {
+		while(true) {
+			@SuppressWarnings("unchecked")
+			MyReference<K,V> ref = (MyReference<K,V>)queue.poll();
+			if( ref == null )
+				return;
+			MyReference<K,V> mappedRef = cache.remove(ref.key());
+			if( mappedRef != ref && mappedRef != null )
+				cache.put( mappedRef.key(), mappedRef );  // put it back
+		}
+	}
+
+	public int size() {
+		return cache.size();
+	}
+
+	public boolean isEmpty() {
+		return cache.isEmpty();
+	}
+
+	public boolean containsKey(Object key) {
+		return cache.containsKey(key);
+	}
+
+	public V get(Object key) {
+		MyReference<K,V> ref = cache.get(key);
+		return ref==null ? null : ref.get();
+	}
+
+	public V put(K key,V value) {
+		sweep();
+		MyReference<K,V> ref = cache.put( key, newReference(key,value,queue) );
+		return ref==null ? null : ref.get();
+	}
+
+	public V remove(Object key) {
+		sweep();
+		MyReference<K,V> ref = cache.remove(key);
+		return ref==null ? null : ref.get();
+	}
+
+	public void clear() {
+		sweep();
+		cache.clear();
+	}
+
+/*
+	public Object clone() {
+		GCCacheMap map = new GCCacheMap();
+		map.cache = (HashMap)cache.clone();
+		return map;
+	}
+*/
+	public Set<K> keySet() {
+		return cache.keySet();
+	}
+
+	public Set<Map.Entry<K,V>> entrySet() {
+		return new MySet();
+	}
+
+
+	private class MySet extends AbstractSet<Map.Entry<K,V>> {
+
+		public int size() {
+			return CacheMap.this.size();
+		}
+
+		public Iterator<Map.Entry<K,V>> iterator() {
+			return new MyIterator(cache.entrySet().iterator());
+		}
+
+	}
+
+	private class MyIterator implements Iterator<Map.Entry<K,V>> {
+		Iterator<Map.Entry<K,MyReference<K,V>>> iter;
+
+		MyIterator(Iterator<Map.Entry<K,MyReference<K,V>>> iter) {
+			this.iter = iter;
+		}
+
+		public boolean hasNext() {
+			return iter.hasNext();
+		}
+
+		public void remove() {
+			iter.remove();
+		}
+
+		public Map.Entry<K,V> next() {
+			return new MyEntry( iter.next() );
+		}
+	}
+
+	private class MyEntry implements Map.Entry<K,V> {
+		Map.Entry<K,MyReference<K,V>> entry;
+
+		MyEntry(Map.Entry<K,MyReference<K,V>> entry) {
+			this.entry = entry;
+		}
+
+		public K getKey() {
+			return entry.getKey();
+		}
+
+		public V getValue() {
+			MyReference<K,V> ref = entry.getValue();
+			return ref.get();
+		}
+
+		public V setValue(V value) {
+			MyReference<K,V> ref = entry.setValue( newReference(getKey(),value,queue) );
+			return ref.get();
+		}
+
+		public boolean equals(Object o) {
+			if( o==null || !(o instanceof CacheMap.MyEntry) )
+				return false;
+			@SuppressWarnings("unchecked")
+			MyEntry m = (MyEntry)o;
+			return entry.equals(m.entry);
+		}
+
+		public int hashCode() {
+			K key = getKey();
+			V value = getValue();
+			return (key==null ? 0 : key.hashCode()) ^
+					(value==null ? 0 : value.hashCode());
+		}
+	}
+
+}