MapProWrapper.java

package railo.commons.collection;

import java.io.Serializable;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import railo.runtime.exp.ExpressionException;
import railo.runtime.exp.PageException;

public class MapProWrapper<K, V> implements MapPro<K, V>,Serializable {

	private final V NULL;
	private Map<K, V> map;

	public MapProWrapper(Map<K, V> map,V NULL){
		this.map=map;
		this.NULL=NULL;
	}
	
	@Override
	public void clear() {
		map.clear();
	}

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

	@Override
	public boolean containsValue(Object value) {
		if(value==null)value=NULL;
		return map.containsValue(value);
	}

	@Override
	public Set<java.util.Map.Entry<K, V>> entrySet() {
		Set<Entry<K, V>> src = map.entrySet();
		Iterator<Entry<K, V>> it = src.iterator();
		Entry<K, V> e;
		while(it.hasNext()){
			e = it.next();
			if(e.getValue()==NULL) e.setValue(null);
		}
		return src;
	}

	@Override
	public V get(Object key) {
		V v = map.get(key);
		if(v==NULL) return null;
		return v;
	}

	@Override
	public V g(K key, V defaultValue) {
		V v = map.get(key);
		if(v==NULL) return null;
		if(v==null) return defaultValue;
		return v;
	}

	@Override
	public V g(K key) throws PageException {
		V v = map.get(key);
		if(v==NULL) return null;
		if(v==null) throw invalidKey(this,key,false);
		return v;
	}

	@Override
	public V r(K key, V defaultValue) {
		V v = map.remove(key);
		if(v==NULL) return null;
		if(v==null) return defaultValue;
		return v;
	}

	@Override
	public V r(K key) throws PageException {
		V v = map.get(key);
		if(v==NULL) return null;
		if(v==null) throw invalidKey(this,key,true);
		return v;
	}
	
	private ExpressionException invalidKey(Map<K,V> map,K key, boolean remove) {

		StringBuilder sb=new StringBuilder();
		Iterator<K> it = map.keySet().iterator();
		K k;
		while(it.hasNext()){
			k = it.next();
			if(sb.length()>0)sb.append(',');
			sb.append(k.toString());
		}

		return new ExpressionException(
				(remove?
						"cannot remove key ["+key+"] from struct, key doesn't exist":
						"key [" + key + "] doesn't exist") +
				" (existing keys:" + sb.toString() + ")" );
	}
	
	@Override
	public boolean isEmpty() {
		return map.isEmpty();
	}

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

	@Override
	public V put(K key, V value) {
		V old;
		if(value==null) old=map.put(key, NULL);
		else old=map.put(key, value);
		
		if(old==NULL) return null;
		return old;
	}

	@Override
	public void putAll(Map<? extends K, ? extends V> m) {
		for (Map.Entry<? extends K, ? extends V> e : m.entrySet()) {
            put(e.getKey(), e.getValue());
		}
	}

	@Override
	public V remove(Object key) {
		return map.remove(key);
	}

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

	@Override
	public Collection<V> values() {
		Collection<V> src = map.values();
		Set<V> trg = new HashSet<V>();
		Iterator<V> it = src.iterator();
		V v;
		while(it.hasNext()){
			v = it.next();
			if(v==NULL) trg.add(null);
			else trg.add(v);
		}
		return trg;
	}

	@Override
	public boolean equals(Object arg0) {
		return map.equals(arg0);
	}

	@Override
	public int hashCode() {
		return map.hashCode();
	}

	@Override
	public String toString() {
		return map.toString();
	}
	
	public Map<K, V> getMap(){
		return map;
	}
}