Add to aai-common query builder get parent step

[aai/aai-common.git] / aai-core / src / main / java / org / onap / aai / query / builder / GremlinQueryBuilder.java

diff --git a/aai-core/src/main/java/org/onap/aai/query/builder/GremlinQueryBuilder.java b/aai-core/src/main/java/org/onap/aai/query/builder/GremlinQueryBuilder.java
index 55faa4d..010dfa5 100644 (file)
--- a/aai-core/src/main/java/org/onap/aai/query/builder/GremlinQueryBuilder.java
+++ b/aai-core/src/main/java/org/onap/aai/query/builder/GremlinQueryBuilder.java
@@ -25,6 +25,8 @@ import java.util.List;
 import java.util.Map;
 import java.util.Set;
 
+import com.google.common.base.Joiner;
+import org.apache.tinkerpop.gremlin.process.traversal.Path;
 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal;
 import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource;
 import org.apache.tinkerpop.gremlin.structure.Direction;
@@ -525,6 +527,16 @@ public abstract class GremlinQueryBuilder<E> extends QueryBuilder<E> {
                stepIndex++;
                return this;
        }
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public QueryBuilder<Path> path(){
+               this.list.add(".path()");
+               stepIndex++;
+               return (QueryBuilder<Path>)this;
+       }
        
        @Override
        public QueryBuilder<Edge> outE() {
@@ -678,5 +690,11 @@ public abstract class GremlinQueryBuilder<E> extends QueryBuilder<E> {
                
                return (List<E>)this.completeTraversal.toList();
        }
+
+       protected QueryBuilder<Edge> has(String key, String value) {
+               this.list.add(".has('" + key + "','" + value + "')");
+
+               return (QueryBuilder<Edge>)this;
+       }
        
 }