tbrugz.util.NonNullGetMap Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sqldump Show documentation

Utility to dump schema and data from a RDBMS

There is a newer version: 0.9.17

package tbrugz.util;

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

//XXXxx should this be a decorator? yes!
public class NonNullGetMap implements Map {
	
	final Map map;
	final Class initialValueClass;
	final GenericFactory factory;
	
	public NonNullGetMap(Map map, Class initialValueClass) {
		this.map = map;
		//this.initialValueClass = (Class) map.values().iterator().next().getClass();
		this.initialValueClass = initialValueClass;
		this.factory = null;
	}

	public NonNullGetMap(Map map, GenericFactory factory) {
		this.map = map;
		this.initialValueClass = null;
		this.factory = factory;
	}
	
	@SuppressWarnings("unchecked")
	@Override
	public V get(Object key) { //(K key)
		V v = map.get(key);
		if(v==null) {
			try {
				if(factory!=null) {
					v = factory.getInstance();
				}
				else {
					v = initialValueClass.newInstance();
				}
				put((K)key, v);
			} catch (InstantiationException e) {
				throw new RuntimeException(e);
			} catch (Exception e) {
				e.printStackTrace();
				throw new RuntimeException(e);
			}
		}
		return v;
	}

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

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

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

	@Override
	public boolean containsValue(Object value) {
		return map.containsValue(value);
	}

	@Override
	public V put(K key, V value) {
		return map.put(key, value);
	}

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

	@Override
	public void putAll(Map m) {
		map.putAll(m);
	}

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

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

	@Override
	public Collection values() {
		return map.values();
	}

	@Override
	public Set> entrySet() {
		return map.entrySet();
	}

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

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