changeset 1498:1b809d2fdf03

add CacheMap
author Franklin Schmidt <fschmidt@gmail.com>
date Fri, 08 May 2020 18:07:14 -0600
parents f04bfbb08721
children 22e15cf73040
files src/goodjava/util/CacheMap.java src/goodjava/util/SoftCacheMap.java src/goodjava/util/WeakCacheMap.java src/goodjava/webserver/handlers/DomainHandler.java
diffstat 4 files changed, 231 insertions(+), 10 deletions(-) [+]
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());
+		}
+	}
+
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/goodjava/util/SoftCacheMap.java	Fri May 08 18:07:14 2020 -0600
@@ -0,0 +1,35 @@
+package goodjava.util;
+
+import java.lang.ref.ReferenceQueue;
+import java.lang.ref.SoftReference;
+
+
+public final class SoftCacheMap<K,V> extends CacheMap<K,V> {
+
+	static final class MySoftReference<K,V> extends SoftReference<V> implements MyReference<K,V> {
+		private final K key;
+
+		MySoftReference(K key,V value,ReferenceQueue<V> q) {
+			super(value,q);
+			this.key = key;
+		}
+
+		public K key() {
+			return key;
+		}
+
+		public boolean equals(Object obj) {
+			Object o = this.get();
+			if( o==null )
+				return false;
+			SoftReference ref = (SoftReference)obj;
+			return o.equals(ref.get());
+		}
+
+	}
+
+	protected MyReference<K,V> newReference(K key,V value,ReferenceQueue<V> q) {
+		return new MySoftReference<K,V>(key,value,q);
+	}
+
+}
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/goodjava/util/WeakCacheMap.java	Fri May 08 18:07:14 2020 -0600
@@ -0,0 +1,35 @@
+package goodjava.util;
+
+import java.lang.ref.ReferenceQueue;
+import java.lang.ref.WeakReference;
+
+
+public class WeakCacheMap<K,V> extends CacheMap<K,V> {
+
+	static final class MyWeakReference<K,V> extends WeakReference<V> implements MyReference<K,V> {
+		private final K key;
+
+		MyWeakReference(K key,V value,ReferenceQueue<V> q) {
+			super(value,q);
+			this.key = key;
+		}
+
+		public K key() {
+			return key;
+		}
+
+		public boolean equals(Object obj) {
+			Object o = this.get();
+			if( o==null )
+				return false;
+			WeakReference ref = (WeakReference)obj;
+			return o.equals(ref.get());
+		}
+
+	}
+
+	protected MyReference<K,V> newReference(K key,V value,ReferenceQueue<V> q) {
+		return new MyWeakReference<K,V>(key,value,q);
+	}
+
+}
--- a/src/goodjava/webserver/handlers/DomainHandler.java	Fri May 08 16:54:30 2020 -0600
+++ b/src/goodjava/webserver/handlers/DomainHandler.java	Fri May 08 18:07:14 2020 -0600
@@ -2,16 +2,13 @@
 
 import java.io.Closeable;
 import java.io.IOException;
-import java.lang.ref.Reference;
-//import java.lang.ref.WeakReference;
-import java.lang.ref.SoftReference;
-import java.lang.ref.ReferenceQueue;
 import java.util.Map;
 import java.util.HashMap;
 import java.util.Set;
 import java.util.Timer;
 import java.util.TimerTask;
 import java.util.concurrent.ConcurrentHashMap;
+import goodjava.util.SoftCacheMap;
 import goodjava.logging.Logger;
 import goodjava.logging.LoggerFactory;
 import goodjava.webserver.Handler;
@@ -40,7 +37,7 @@
 	}
 
 	private static final long HOUR = 1000L*60*60;
-	private final Map<String,Reference<Handler>> map = new HashMap<String,Reference<Handler>>();
+	private final Map<String,Handler> map = new SoftCacheMap<String,Handler>();
 	private final Set<Handler> dontGc = ConcurrentHashMap.newKeySet();
 	private final Timer timer = new Timer();
 
@@ -71,14 +68,13 @@
 	public Handler getHandler(String domain) {
 		domain = domain.toLowerCase();
 		synchronized(map) {
-			Reference<Handler> ref = map.get(domain);
-			Handler handler = ref==null ? null : ref.get();
+			Handler handler = map.get(domain);
 			if( handler == null ) {
 				//if(ref!=null) logger.info("gc "+domain);
 				handler = factory.newHandler(domain);
 				if( handler == null )
 					return null;
-				map.put(domain,new SoftReference<Handler>(handler));
+				map.put(domain,handler);
 			}
 			return handler;
 		}
@@ -88,8 +84,7 @@
 		//logger.info("removeHandler "+domain);
 		domain = domain.toLowerCase();
 		synchronized(map) {
-			Reference<Handler> ref = map.remove(domain);
-			Handler handler = ref==null ? null : ref.get();
+			Handler handler = map.remove(domain);
 			if( handler != null ) {
 				close(handler);
 			}