Sha256: 025f56c09d0404de4749072be916ee5a4e2291370d66ee50c14210ad1d62a114
Contents?: true
Size: 1.97 KB
Versions: 7
Compression:
Stored size: 1.97 KB
Contents
module Pacer
module Neo4j2
module Algo
class PathPipe < Pacer::Pipes::RubyPipe
import org.neo4j.graphdb::Node
import org.neo4j.graphdb::Relationship
import com.tinkerpop.blueprints.impls.neo4j2.Neo4j2Vertex
import com.tinkerpop.blueprints.impls.neo4j2.Neo4j2Edge
attr_reader :algo, :target, :graph, :max_hits
attr_accessor :current_paths, :hits
def initialize(algo, graph, target, max_hits)
super()
@algo = algo
@max_hits = max_hits || -1
@graph = graph.blueprints_graph
@target = unwrap target if target
end
def processNextStart
next_raw_path.map do |e|
if e.is_a? Node
Neo4j2Vertex.new e, graph
elsif e.is_a? Relationship
Neo4j2Edge.new e, graph
else
e
end
end
end
def next_raw_path
loop do
if current_paths
if hits == 0
self.current_paths = nil
elsif current_paths.hasNext
self.hits -= 1
return current_paths.next
else
self.current_paths = nil
end
else
self.hits = max_hits
self.current_paths = @algo.findAllPaths(next_raw, target).iterator
end
end
end
def next_raw
unwrap starts.next
end
def unwrap(vertex)
if vertex.respond_to? :element
vertex.element.raw_element
else
vertex.raw_element
end
end
end
class PathFromPathPipe < PathPipe
attr_writer :target
def initialize(algo, graph, max_hits)
super(algo, graph, nil, max_hits)
end
def next_raw
path = starts.next
self.target = unwrap path.to_a.last
unwrap path[0]
end
end
end
end
end
Version data entries
7 entries across 7 versions & 1 rubygems