package dk.thoerup.traininfoservice.banedk;

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

import com.sun.xml.rpc.processor.schema.UnimplementedFeatureException;

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 arg0) {
		CacheItem<V> item = cache.get(arg0);
		
		return (item != null);
	}

	@Override
	public boolean containsValue(Object arg0) {
		//TODO someday implement this
		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) {
		//TODO someday implement this
		throw new UnsupportedOperationException();
	}

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

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

	@Override
	public Collection<V> values() {
		//TODO someday implement this
		throw new UnsupportedOperationException();
	}

	@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);
		
		return cache.put(key, item).value;
	}
	
}