Added graph entities and graph entity property classes

Author: DvirDukhan

src/main/java/com/redislabs/redisgraph/impl/Edge.java

@@ -0,0 +1,92 @@
+package com.redislabs.redisgraph.impl;
+
+import java.util.Objects;
+
+/**
+ * A class represent an edge (graph entity). In addition to the base class id and properties, an edge shows its source,
+ * destination and relationship type
+ */
+public class Edge extends GraphEntity {
+
+    //memebers
+    String relationshipType;
+    int source;
+    int destination;
+
+
+    //getters & setters
+
+    /**
+     * @return the edge relationship type
+     */
+    public String getRelationshipType() {
+        return relationshipType;
+    }
+
+    /**
+     * @param relationshipType - the relationship type to be set.
+     */
+    public void setRelationshipType(String relationshipType) {
+        this.relationshipType = relationshipType;
+    }
+
+
+    /**
+     * @return The id of the source node
+     */
+    public int getSource() {
+        return source;
+    }
+
+    /**
+     * @param source - The id of the source node to be set
+     */
+    public void setSource(int source) {
+        this.source = source;
+    }
+
+    /**
+     *
+     * @return the id of the destination node
+     */
+    public int getDestination() {
+        return destination;
+    }
+
+    /**
+     *
+     * @param destination - The id of the destination node to be set
+     */
+    public void setDestination(int destination) {
+        this.destination = destination;
+    }
+
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Edge)) return false;
+        if (!super.equals(o)) return false;
+        Edge edge = (Edge) o;
+        return source == edge.source &&
+                destination == edge.destination &&
+                Objects.equals(relationshipType, edge.relationshipType);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(super.hashCode(), relationshipType, source, destination);
+    }
+
+    @Override
+    public String toString() {
+        final StringBuilder sb = new StringBuilder("Edge{");
+        sb.append("relationshipType='").append(relationshipType).append('\'');
+        sb.append(", source=").append(source);
+        sb.append(", destination=").append(destination);
+        sb.append(", id=").append(id);
+        sb.append(", propertyMap=").append(propertyMap);
+        sb.append('}');
+        return sb.toString();
+    }
+}

src/main/java/com/redislabs/redisgraph/impl/GraphEntity.java

@@ -0,0 +1,120 @@
+package com.redislabs.redisgraph.impl;
+
+
+import com.redislabs.redisgraph.ResultSet.ResultSetScalarTypes;
+
+import java.util.*;
+
+
+/**
+ * This is an abstract class for representing a graph entity.
+ * A graph entity has an id and a set of properties. The properties are mapped and accessed by their names.
+ */
+public abstract class GraphEntity {
+
+
+
+    //members
+
+    int id;
+    Map<String, Property> propertyMap = new HashMap<>();
+
+
+    //setters & getters
+
+    /**
+     *
+     * @return entity id
+     */
+    public int getId() {
+        return id;
+    }
+
+    /**
+     *
+     * @param id - entity id to be set
+     */
+    public void setId(int id) {
+        this.id = id;
+    }
+
+
+    /**
+     * Adds a property to the entity, by composing name, type and value to a property object
+     * @param name
+     * @param type
+     * @param value
+     */
+    public void addProperty(String name, ResultSetScalarTypes type, Object value){
+
+        addProperty(new Property(name, type, value));
+
+    }
+
+    /**
+     * Add a property to the entity
+     * @param property
+     */
+    public void addProperty (Property property){
+
+
+        propertyMap.put(property.name, property);
+    }
+
+    /**
+     *
+     * @return number of properties
+     */
+    public int getNumberOfProperties(){
+        return propertyMap.size();
+    }
+
+
+    /**
+     *
+     * @param propertyName - property name as lookup key (String)
+     * @return property object, or null if key is not found
+     */
+    public Property getProperty(String propertyName){
+        return propertyMap.get(propertyName);
+    }
+
+
+    /**
+     *
+     * @param name - the name of the property to be removed
+     */
+    public void removeProperty(String name){
+
+        propertyMap.remove(name);
+
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof GraphEntity)) return false;
+        GraphEntity that = (GraphEntity) o;
+        return id == that.id &&
+                Objects.equals(propertyMap, that.propertyMap);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(id, propertyMap);
+    }
+
+
+    /**
+     * Default toString implementation.
+     * @return
+     */
+    @Override
+    public String toString() {
+        final StringBuilder sb = new StringBuilder("GraphEntity{");
+        sb.append("id=").append(id);
+        sb.append(", propertyMap=").append(propertyMap);
+        sb.append('}');
+        return sb.toString();
+    }
+}

src/main/java/com/redislabs/redisgraph/impl/Node.java

@@ -0,0 +1,70 @@
+package com.redislabs.redisgraph.impl;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+
+/**
+ * * A class represent an node (graph entity). In addition to the base class id and properties, a node has labels.
+ */
+public class Node extends GraphEntity {
+
+    //members
+    List<String> labels = new ArrayList<>();
+
+    /**
+     * @param label - a label to be add
+     */
+    public void addLabel(String label) {
+        labels.add(label);
+    }
+
+    /**
+     * @param label - a label to be removed
+     */
+    public void removeLabel(String label) {
+        labels.remove(label);
+    }
+
+    /**
+     * @param index - label index
+     * @return the proprty label
+     * @throws IndexOutOfBoundsException if the index is out of range
+     *                                   ({@code index < 0 || index >= getNumberOfLabels()})
+     */
+    public String getLabel(int index) throws IndexOutOfBoundsException{
+        return labels.get(index);
+    }
+
+    /**
+     *
+     * @return the number of labels
+     */
+    public int getNumberOfLabels() {
+        return labels.size();
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Node)) return false;
+        if (!super.equals(o)) return false;
+        Node node = (Node) o;
+        return Objects.equals(labels, node.labels);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(super.hashCode(), labels);
+    }
+
+    @Override
+    public String toString() {
+        final StringBuilder sb = new StringBuilder("Node{");
+        sb.append("labels=").append(labels);
+        sb.append(", id=").append(id);
+        sb.append(", propertyMap=").append(propertyMap);
+        sb.append('}');
+        return sb.toString();
+    }
+}

src/main/java/com/redislabs/redisgraph/impl/Property.java

@@ -0,0 +1,113 @@
+package com.redislabs.redisgraph.impl;
+
+import com.redislabs.redisgraph.ResultSet;
+
+import java.util.Objects;
+
+/**
+ * A Graph entity property. Has a name, type, and value
+ */
+public class Property {
+
+    //members
+    String name;
+    ResultSet.ResultSetScalarTypes type;
+    Object value;
+
+
+    /**
+     * Default constructor
+     */
+    public Property() {
+
+    }
+
+    /**
+     * Parameterized constructor
+     *
+     * @param name
+     * @param type
+     * @param value
+     */
+    public Property(String name, ResultSet.ResultSetScalarTypes type, Object value) {
+        this.name = name;
+        this.type = type;
+        this.value = value;
+    }
+
+    //getters & setters
+
+    /**
+     * @return property name
+     */
+    public String getName() {
+        return name;
+    }
+
+    /**
+     * @param name - property name to be set
+     */
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    /**
+     * @return property type
+     */
+    public ResultSet.ResultSetScalarTypes getType() {
+        return type;
+    }
+
+    /**
+     * @param type property type to be set
+     */
+    public void setType(ResultSet.ResultSetScalarTypes type) {
+        this.type = type;
+    }
+
+
+    /**
+     * @return property value
+     */
+    public Object getValue() {
+        return value;
+    }
+
+
+    /**
+     * @param value property value to be set
+     */
+    public void setValue(Object value) {
+        this.value = value;
+    }
+
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Property)) return false;
+        Property property = (Property) o;
+        return Objects.equals(name, property.name) &&
+                type == property.type &&
+                Objects.equals(value, property.value);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(name, type, value);
+    }
+
+    /**
+     * Default toString implementation
+     * @return
+     */
+    @Override
+    public String toString() {
+        final StringBuilder sb = new StringBuilder("Property{");
+        sb.append("name='").append(name).append('\'');
+        sb.append(", type=").append(type);
+        sb.append(", value=").append(value);
+        sb.append('}');
+        return sb.toString();
+    }
+}