DataElement.java

/* AWE - Amanzi Wireless Explorer
 * http://awe.amanzi.org
 * (C) 2008-2009, AmanziTel AB
 *
 * This library is provided under the terms of the Eclipse Public License
 * as described at http://www.eclipse.org/legal/epl-v10.html. Any use,
 * reproduction or distribution of the library constitutes recipient's
 * acceptance of this agreement.
 *
 * This library is distributed WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 */

package org.amanzi.neo.impl.dto;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import org.amanzi.neo.dto.IDataElement;
import org.amanzi.neo.nodetypes.INodeType;
import org.neo4j.graphdb.Node;

/**
 * TODO Purpose of
 * <p>
 * </p>
 * 
 * @author Nikolay Lagutko (nikolay.lagutko@amanzitel.com)
 * @since 1.0.0
 */
public class DataElement implements IDataElement {

    private Node node;

    private final Map<String, Object> properties = new HashMap<String, Object>();

    private INodeType nodeType;

    private String name;

    public DataElement() {

    }

    public DataElement(final Node node) {
        this.node = node;
    }

    public Node getNode() {
        return node;
    }

    @Override
    public String toString() {
        return "Data element {NodeType: " + nodeType + "; Name: " + name + "}";
    }

    @Override
    public boolean equals(final Object anotherObject) {
        if (anotherObject instanceof DataElement) {
            final DataElement anotherElement = (DataElement)anotherObject;
            if (node == null) {
                return properties.equals(anotherElement.properties);
            } else {
                return node.equals(anotherElement.getNode());
            }
        }

        return false;
    }

    @Override
    public int hashCode() {
        return node == null ? super.hashCode() : node.hashCode();
    }

    @Override
    public Object get(final String header) {
        if ((node != null) && properties.isEmpty()) {
            loadProperties();
        }
        return properties.get(header);
    }

    protected void loadProperties() {
        for (final String propertyKey : node.getPropertyKeys()) {
            final Object value = node.getProperty(propertyKey);
            properties.put(propertyKey, value);
        }
    }

    @Override
    public Object put(final String key, final Object value) {
        return properties.put(key, value);
    }

    @Override
    public Set<String> keySet() {
        if (node != null && properties.isEmpty()) {
            loadProperties();
        }
        return properties.keySet();
    }

    public void setNodeType(final INodeType nodeType) {
        this.nodeType = nodeType;
    }

    @Override
    public INodeType getNodeType() {
        return nodeType;
    }

    @Override
    public long getId() {
        return node == null ? -1 : node.getId();
    }

    public void setName(final String name) {
        this.name = name;
    }

    @Override
    public String getName() {
        return name;
    }

    @Override
    public int compareTo(final IDataElement o) {
        if (name != null) {
            return name.compareTo(o.getName());
        } else if (node != null) {
            final DataElement element = (DataElement)o;

            return element.getNode() == null ? 1 : new Long(node.getId()).compareTo(element.getNode().getId());
        }
        return 0;
    }

    @Override
    public Map<String, Object> asMap() {
        if ((node != null) && properties.isEmpty()) {
            loadProperties();
        }
        return properties;
    }

    @Override
    public boolean contains(final String header) {
        if ((node != null) && properties.isEmpty()) {
            loadProperties();
        }
        return properties.containsKey(header);
    }
}