traininfoservice/banedk/TimeoutMap.java

package dk.thoerup.traininfoservice.banedk;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;


public class TimeoutMap<K,V> implements Map<K,V>{

        class CacheItem<T> {
                
                public CacheItem(T v) {
                        value = v;
                        lastupdate = System.currentTimeMillis();
                }
                
                public long lastupdate;
                public T value;
        }
        
        private ConcurrentHashMap<K,CacheItem<V>> cache = new ConcurrentHashMap<K,CacheItem<V>>();
        private long timeout;
        
        public TimeoutMap(int timeout) {
                this.timeout = timeout; 
        }


        @Override
        public void clear() {
                cache.clear();
                
        }

        @Override
        public boolean containsKey(Object key) {
                return cache.containsKey(key);
        }

        @Override
        public boolean containsValue(Object arg0) {
                //return values().contains(arg0);
                throw new UnsupportedOperationException();
        }

        @Override
        public Set<java.util.Map.Entry<K, V>> entrySet() {
                //TODO someday implement this
                throw new UnsupportedOperationException();
        }

        @Override
        public boolean isEmpty() {
                return cache.isEmpty();
        }

        @Override
        public Set<K> keySet() {
                return cache.keySet();
        }

        @Override
        public void putAll(Map<? extends K, ? extends V> arg0) {
                for(K key : arg0.keySet()) {
                        this.put(key, arg0.get(key) );
                }
        }

        @Override
        public V remove(Object arg0) {
                return cache.remove(arg0).value;
        }

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

        @Override
        public Collection<V> values() {
                ArrayList<V> values = new ArrayList<V>();
                for (CacheItem<V> item : cache.values()) {
                        values.add( item.value );
                }
                
                return values;

        }

        @Override
        public V get(Object key) {
                long now = System.currentTimeMillis();
                
                CacheItem<V> item = cache.get(key);
                                
                
                if (item != null) {
                        if ( (item.lastupdate+timeout) < now) { //item too old
                                return null;
                        } else {
                                return item.value; //item still good
                        }
                } else {
                        return null; // no item found
                }
        }

        @Override
        public V put(K key, V value) {          
                CacheItem<V> item= new CacheItem<V>(value);
                
                item = cache.put(key, item);
                
                if (item != null)
                        return item.value;
                else
                        return null; 
        }
        
}
1	torben	387	package dk.thoerup.traininfoservice.banedk;
2
3	torben	430	import java.util.ArrayList;
4	torben	428	import java.util.Collection;
5			import java.util.Map;
6			import java.util.Set;
7	torben	430	import java.util.concurrent.ConcurrentHashMap;
8	torben	387
9
10	torben	429
11	torben	428	public class TimeoutMap<K,V> implements Map<K,V>{
12	torben	387
13			class CacheItem<T> {
14
15			public CacheItem(T v) {
16			value = v;
17			lastupdate = System.currentTimeMillis();
18			}
19
20			public long lastupdate;
21			public T value;
22			}
23
24			private ConcurrentHashMap<K,CacheItem<V>> cache = new ConcurrentHashMap<K,CacheItem<V>>();
25			private long timeout;
26
27	torben	428	public TimeoutMap(int timeout) {
28	torben	387	this.timeout = timeout;
29			}
30	torben	428
31
32			@Override
33			public void clear() {
34			cache.clear();
35
36	torben	387	}
37	torben	428
38			@Override
39	torben	804	public boolean containsKey(Object key) {
40			return cache.containsKey(key);
41	torben	428	}
42
43			@Override
44			public boolean containsValue(Object arg0) {
45	torben	966	//return values().contains(arg0);
46			throw new UnsupportedOperationException();
47	torben	428	}
48
49			@Override
50			public Set<java.util.Map.Entry<K, V>> entrySet() {
51			//TODO someday implement this
52			throw new UnsupportedOperationException();
53			}
54
55			@Override
56			public boolean isEmpty() {
57			return cache.isEmpty();
58			}
59
60			@Override
61			public Set<K> keySet() {
62			return cache.keySet();
63			}
64
65			@Override
66			public void putAll(Map<? extends K, ? extends V> arg0) {
67	torben	430	for(K key : arg0.keySet()) {
68			this.put(key, arg0.get(key) );
69			}
70	torben	428	}
71
72			@Override
73			public V remove(Object arg0) {
74			return cache.remove(arg0).value;
75			}
76
77			@Override
78			public int size() {
79			return cache.size();
80			}
81
82			@Override
83			public Collection<V> values() {
84	torben	430	ArrayList<V> values = new ArrayList<V>();
85			for (CacheItem<V> item : cache.values()) {
86			values.add( item.value );
87			}
88
89			return values;
90
91	torben	428	}
92
93			@Override
94			public V get(Object key) {
95	torben	387	long now = System.currentTimeMillis();
96
97	torben	428	CacheItem<V> item = cache.get(key);
98	torben	387
99
100			if (item != null) {
101			if ( (item.lastupdate+timeout) < now) { //item too old
102			return null;
103			} else {
104			return item.value; //item still good
105			}
106			} else {
107			return null; // no item found
108			}
109	torben	428	}
110
111			@Override
112			public V put(K key, V value) {
113			CacheItem<V> item= new CacheItem<V>(value);
114	torben	387
115	torben	432	item = cache.put(key, item);
116
117			if (item != null)
118			return item.value;
119			else
120			return null;
121	torben	387	}
122
123			}