def visit_FuncCall(self, node):
self.write(common.qname(*node.name))
self.write('(')
if node.agg_distinct:
self.write('DISTINCT ')
self.visit_list(node.args, newlines=False)
if node.agg_order:
self.write(' ORDER BY ')
self.visit_list(node.agg_order, newlines=False)
self.write(')')
if node.agg_filter:
self.write(' FILTER (WHERE ')
self.visit(node.agg_filter)
self.write(')')
if node.over:
self.write(' OVER (')
if node.over.partition_clause:
self.write('PARTITION BY ')
self.visit_list(node.over.partition_clause, newlines=False)
if node.over.order_clause:
self.write(' ORDER BY ')
self.visit_list(node.over.order_clause, newlines=False)
# XXX: add support for frame definition
self.write(')')
if node.with_ordinality:
self.write(' WITH ORDINALITY')
def visit_FuncCall(self, node):
self.write(common.qname(*node.name))
self.write('(')
if node.agg_distinct:
self.write('DISTINCT ')
self.visit_list(node.args, newlines=False)
if node.agg_order:
self.write(' ORDER BY ')
self.visit_list(node.agg_order, newlines=False)
self.write(')')
if node.agg_filter:
self.write(' FILTER (WHERE ')
self.visit(node.agg_filter)
self.write(')')
if node.over:
self.write(' OVER (')
if node.over.partition_clause:
self.write('PARTITION BY ')
self.visit_list(node.over.partition_clause, newlines=False)
if node.over.order_clause:
self.write(' ORDER BY ')
self.visit_list(node.over.order_clause, newlines=False)
# XXX: add support for frame definition
self.write(')')
if node.with_ordinality:
self.write(' WITH ORDINALITY')
def visit_TypeName(self, node):
self.write(common.qname(*node.name))
if node.array_bounds:
for array_bound in node.array_bounds:
self.write('[')
if array_bound >= 0:
self.write(array_bound)
self.write(']')
def visit_ColumnRef(self, node):
names = node.name
if isinstance(names[-1], pgast.Star):
self.write(common.qname(*names[:-1]))
if len(names) > 1:
self.write('.')
self.write('*')
else:
if names == ['VALUE']:
self.write(names[0])
elif names[0] in {'OLD', 'NEW'}:
self.write(names[0])
if len(names) > 1:
self.write('.')
self.write(common.qname(*names[1:]))
else:
self.write(common.qname(*names))
def visit_TypeName(self, node):
self.write(common.qname(*node.name))
if node.array_bounds:
for array_bound in node.array_bounds:
self.write('[')
if array_bound >= 0:
self.write(array_bound)
self.write(']')
def visit_ColumnRef(self, node):
names = node.name
if isinstance(names[-1], pgast.Star):
self.write(common.qname(*names[:-1]))
if len(names) > 1:
self.write('.')
self.write('*')
else:
if names == ['VALUE']:
self.write(names[0])
elif names[0] in {'OLD', 'NEW'}:
self.write(names[0])
if len(names) > 1:
self.write('.')
self.write(common.qname(*names[1:]))
else:
self.write(common.qname(*names))
async def creation_code(self, context):
link_map = await context.get_class_map()
ids = tuple(sorted(list(link_map[n] for n in self.link_names)))
id_str = '_'.join(str(i) for i in ids)
name = '%s_%s_%s_cardinality_idx' % (self.name_prefix, id_str,
self.cardinality)
name = common.edgedb_name_to_pg_name(name)
predicate = 'ptr_item_id IN (%s)' % ', '.join(str(id) for id in ids)
code = '''
CREATE {unique} INDEX {name} ON {table}s ({cols}) {predicate}
'''.format(unique='UNIQUE',
name=common.qname(name),
table=common.qname(*self.table_name),
cols=', '.join(common.quote_ident(c) for c in self.columns),
predicate=('WHERE {}'.format(predicate)))
return code
async def creation_code(self, context):
link_map = await context.get_class_map()
ids = tuple(sorted(list(link_map[n] for n in self.link_names)))
id_str = '_'.join(str(i) for i in ids)
name = '%s_%s_%s_cardinality_idx' % (
self.name_prefix, id_str, self.cardinality)
name = common.edgedb_name_to_pg_name(name)
predicate = 'ptr_item_id IN (%s)' % ', '.join(str(id) for id in ids)
code = '''
CREATE {unique} INDEX {name} ON {table}s ({cols}) {predicate}
'''.format(unique='UNIQUE',
name=common.qname(name),
table=common.qname(*self.table_name),
cols=', '.join(common.quote_ident(c) for c in self.columns),
predicate=('WHERE {}'.format(predicate)))
return code
def visit_Relation(self, node):
if node.schemaname is None:
self.write(common.qname(node.name))
else:
self.write(common.qname(node.schemaname, node.name))
def visit_Relation(self, node):
if node.schemaname is None:
self.write(common.qname(node.name))
else:
self.write(common.qname(node.schemaname, node.name))