com.tinkerpop.gremlin.tinkergraph.process.graph.TinkerGraphTraversal Maven / Gradle / Ivy
package com.tinkerpop.gremlin.tinkergraph.process.graph;
import com.tinkerpop.gremlin.process.graph.util.DefaultGraphTraversal;
import com.tinkerpop.gremlin.structure.Element;
import com.tinkerpop.gremlin.tinkergraph.process.graph.step.sideEffect.TinkerGraphStep;
import com.tinkerpop.gremlin.tinkergraph.process.graph.strategy.TinkerGraphStepStrategy;
import com.tinkerpop.gremlin.tinkergraph.structure.TinkerGraph;
/**
* @author Marko A. Rodriguez (http://markorodriguez.com)
*/
public class TinkerGraphTraversal extends DefaultGraphTraversal {
public TinkerGraphTraversal(final TinkerGraph graph, final Class elementClass) {
super(graph);
this.strategies().register(TinkerGraphStepStrategy.instance());
this.addStep(new TinkerGraphStep<>(this, elementClass));
}
@Override
public void prepareForGraphComputer() {
super.prepareForGraphComputer();
this.strategies().unregister(TinkerGraphStepStrategy.class);
}
}