com.github.TKnudsen.ComplexDataObject.data.DataContainer Maven / Gradle / Ivy

Go to download
package com.github.TKnudsen.ComplexDataObject.data;

import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import com.github.TKnudsen.ComplexDataObject.data.interfaces.IKeyValueProvider;

/**
 * 
 * Title: DataContainer
 * 
 * 
 * 
 * Description: Stores and manages collections of IDObjects. A DataSchema
 * manages the keys/attributes of the collection iff the objects are instanceOf
 * IKeyValueProvider.
 * 
 * 
 * 
 * Copyright: Copyright (c) 2015-2016
 * 
 * 
 * @author Juergen Bernard
 * @version 1.0
 */
public class DataContainer> implements Iterable {

	private Map objectsMap = new HashMap();

	protected Map> attributeValues = new HashMap>();

	protected DataSchema dataSchema;

	public DataContainer(DataSchema dataSchema) {
		this.dataSchema = dataSchema;
	}

	public DataContainer(Map objectsMap) {
		this.objectsMap = objectsMap;
		dataSchema = new DataSchema();
		for (Long ID : objectsMap.keySet())
			extendDataSchema(objectsMap.get(ID));
	}

	public DataContainer(Iterable objects) {
		dataSchema = new DataSchema();
		for (T object : objects) {
			objectsMap.put(object.getID(), object);
			extendDataSchema(object);
		}
	}

	private void extendDataSchema(T object) {
		if (object instanceof IKeyValueProvider) {
			IKeyValueProvider keyValueProvider = (IKeyValueProvider) object;
			for (String string : keyValueProvider.keySet())
				if (string instanceof String)
					if (!dataSchema.contains(string) && keyValueProvider.getAttribute(string) != null)
						dataSchema.add(string, keyValueProvider.getAttribute(string).getClass());
		}
	}

	/**
	 * Introduces or updates a new attribute.
	 * 
	 *
	 * 
	 * @param attribute
	 *            the attribute name
	 * @param type
	 *            the expected data type.
	 * @param defaultValue
	 *            the default value in case the attribute is missing from a data
	 *            object.
	 * @return the data schema instance for call-chaining.
	 */
	public  DataSchema addAttribute(String attribute, Class type, A defaultValue) {
		// TODO t makes no sense. a new attribute is introduced here - not a new
		// object consisting of multiple attributes

		// either a new object is introduced or a new attribute.
		attributeValues = new HashMap>();

		dataSchema.add(attribute, type, defaultValue);

		Iterator objectIterator = iterator();
		while (objectIterator.hasNext()) {
			T next = objectIterator.next();
			if (next.getAttribute(attribute) == null)
				next.add(attribute, defaultValue);
		}

		return dataSchema;
	}

	/**
	 * Remove functionality. For test purposes. Maybe this functionality will be
	 * removed sometime.
	 * 
	 * @param object
	 * @return
	 */
	public boolean remove(T object) {
		if (object == null)
			return false;

		long id = object.getID();
		if (!objectsMap.containsKey(id))
			return false;

		for (String attribute : attributeValues.keySet()) {
			if (attributeValues.get(attribute) != null)
				attributeValues.get(attribute).remove(id);
		}

		objectsMap.remove(id);

		return true;
	}

	/**
	 * Removes an attribute from the container and the set of objects.
	 * 
	 * @param attribute
	 *            the attribute name.
	 * @return the data schema instance for call-chaining.
	 */
	public DataSchema remove(String attribute) {
		Iterator iterator = iterator();
		while (iterator.hasNext()) {
			T o = iterator.next();
			o.removeAttribute(attribute);
		}

		return dataSchema.remove(attribute);
	}

	@Override
	public Iterator iterator() {
		return objectsMap.values().iterator();
	}

	public Boolean isNumeric(String attribute) {
		if (!dataSchema.contains(attribute))
			return false;
		if (Number.class.isAssignableFrom(dataSchema.getType(attribute)))
			return true;
		return false;
	}

	public Boolean isBoolean(String attribute) {
		if (!dataSchema.contains(attribute))
			return false;
		if (Boolean.class.isAssignableFrom(dataSchema.getType(attribute)))
			return true;
		return false;
	}

	public Collection getAttributeNames() {
		return dataSchema.getAttributeNames();
	}

	public Map getAttributeValues(String attribute) {
		if (attributeValues.get(attribute) == null) {
			calculateEntities(attribute);
		}
		return attributeValues.get(attribute);
	}

	public boolean contains(T object) {
		if (objectsMap.containsKey(object.getID()))
			return true;
		return false;
	}

	private void calculateEntities(String attribute) {
		Map ent = new HashMap();

		Iterator iterator = iterator();
		while (iterator.hasNext()) {
			T o = iterator.next();
			if (o instanceof IKeyValueProvider)
				ent.put(o.getID(), o.getAttribute(attribute));
		}

		this.attributeValues.put(attribute, ent);
	}

	@Override
	public String toString() {
		String s = "Container with " + size() + " objects. Schema:\n";
		if (dataSchema == null)
			s += super.toString();
		else
			s += dataSchema.toString();
		return s;
	}

	public int size() {
		if (objectsMap == null)
			return 0;
		return objectsMap.size();
	}
}