def execute_entities(self, graph, query, ctx):
entities_link = query.fields_map['_entities']
query = entities_link.node
representations = entities_link.options['representations']
queue = Queue(self.executor)
task_set = queue.fork(None)
query_workflow = Query(queue, task_set, graph, query, Context(ctx))
type_ids_map = defaultdict(list)
for rep in representations:
typename = rep['__typename']
for key in get_keys(graph, typename):
if key not in rep:
continue
ident = rep[key]
type_ids_map[typename].append(ident)
for typename in type_ids_map:
ids = type_ids_map[typename]
node = graph.nodes_map[typename]
query_workflow.process_node(node, query, ids)
return self.executor.process(queue, query_workflow)
def execute_query(self, graph, query, ctx):
query = InitOptions(graph).visit(query)
queue = Queue(self.executor)
task_set = queue.fork(None)
query_workflow = Query(queue, task_set, graph, query, Context(ctx))
query_workflow.start()
return self.executor.process(queue, query_workflow)