Changeset View
Changeset View
Standalone View
Standalone View
swh/graph/backend.py
Show First 20 Lines • Show All 69 Lines • ▼ Show 20 Lines | def __enter__(self): | ||||
return self | return self | ||||
def __exit__(self, exc_type, exc_value, tb): | def __exit__(self, exc_type, exc_value, tb): | ||||
self.gateway.shutdown() | self.gateway.shutdown() | ||||
def stats(self): | def stats(self): | ||||
return self.entry.stats() | return self.entry.stats() | ||||
def count(self, ttype, direction, edges_fmt, src): | |||||
method = getattr(self.entry, 'count_' + ttype) | |||||
return method(direction, edges_fmt, src) | |||||
async def simple_traversal(self, ttype, direction, edges_fmt, src): | async def simple_traversal(self, ttype, direction, edges_fmt, src): | ||||
assert ttype in ('leaves', 'neighbors', 'visit_nodes') | assert ttype in ('leaves', 'neighbors', 'visit_nodes') | ||||
method = getattr(self.stream_proxy, ttype) | method = getattr(self.stream_proxy, ttype) | ||||
async for node_id in method(direction, edges_fmt, src): | async for node_id in method(direction, edges_fmt, src): | ||||
yield node_id | yield node_id | ||||
async def walk(self, direction, edges_fmt, algo, src, dst): | async def walk(self, direction, edges_fmt, algo, src, dst): | ||||
if dst in PID_TYPES: | if dst in PID_TYPES: | ||||
▲ Show 20 Lines • Show All 97 Lines • Show Last 20 Lines |