提交 25a0d9c9 编写于 作者: E Eric Blake 提交者: Markus Armbruster

qapi: Use predicate callback to determine visit filtering

Previously, qapi-types and qapi-visit filtered out implicit
objects during visit_object_type() by using 'info' (works since
implicit objects do not [yet] have associated info); meanwhile
qapi-introspect filtered out all schema types on the first pass
by returning a python type from visit_begin(), which was then
used at a distance in QAPISchema.visit() to do the filtering.

Rather than keeping these ad hoc approaches, add a new visitor
callback visit_needed() which returns False to skip a given
entity, and which defaults to True unless overridden.  Use the
new mechanism to simplify all three filtering visitors.

No change to the generated code.
Suggested-by: NMarkus Armbruster <armbru@redhat.com>
Signed-off-by: NEric Blake <eblake@redhat.com>
Message-Id: <1444710158-8723-2-git-send-email-eblake@redhat.com>
Signed-off-by: NMarkus Armbruster <armbru@redhat.com>
上级 d08ac81a
...@@ -54,7 +54,6 @@ class QAPISchemaGenIntrospectVisitor(QAPISchemaVisitor): ...@@ -54,7 +54,6 @@ class QAPISchemaGenIntrospectVisitor(QAPISchemaVisitor):
self._jsons = [] self._jsons = []
self._used_types = [] self._used_types = []
self._name_map = {} self._name_map = {}
return QAPISchemaType # don't visit types for now
def visit_end(self): def visit_end(self):
# visit the types that are actually used # visit the types that are actually used
...@@ -82,6 +81,10 @@ const char %(c_name)s[] = %(c_string)s; ...@@ -82,6 +81,10 @@ const char %(c_name)s[] = %(c_string)s;
self._used_types = None self._used_types = None
self._name_map = None self._name_map = None
def visit_needed(self, entity):
# Ignore types on first pass; visit_end() will pick up used types
return not isinstance(entity, QAPISchemaType)
def _name(self, name): def _name(self, name):
if self._unmask: if self._unmask:
return name return name
......
...@@ -233,6 +233,10 @@ class QAPISchemaGenTypeVisitor(QAPISchemaVisitor): ...@@ -233,6 +233,10 @@ class QAPISchemaGenTypeVisitor(QAPISchemaVisitor):
self.decl = self._btin + self.decl self.decl = self._btin + self.decl
self._btin = None self._btin = None
def visit_needed(self, entity):
# Visit everything except implicit objects
return not isinstance(entity, QAPISchemaObjectType) or entity.info
def _gen_type_cleanup(self, name): def _gen_type_cleanup(self, name):
self.decl += gen_type_cleanup_decl(name) self.decl += gen_type_cleanup_decl(name)
self.defn += gen_type_cleanup(name) self.defn += gen_type_cleanup(name)
...@@ -254,14 +258,13 @@ class QAPISchemaGenTypeVisitor(QAPISchemaVisitor): ...@@ -254,14 +258,13 @@ class QAPISchemaGenTypeVisitor(QAPISchemaVisitor):
self._gen_type_cleanup(name) self._gen_type_cleanup(name)
def visit_object_type(self, name, info, base, members, variants): def visit_object_type(self, name, info, base, members, variants):
if info: self._fwdecl += gen_fwd_object_or_array(name)
self._fwdecl += gen_fwd_object_or_array(name) if variants:
if variants: assert not members # not implemented
assert not members # not implemented self.decl += gen_union(name, base, variants)
self.decl += gen_union(name, base, variants) else:
else: self.decl += gen_struct(name, base, members)
self.decl += gen_struct(name, base, members) self._gen_type_cleanup(name)
self._gen_type_cleanup(name)
def visit_alternate_type(self, name, info, variants): def visit_alternate_type(self, name, info, variants):
self._fwdecl += gen_fwd_object_or_array(name) self._fwdecl += gen_fwd_object_or_array(name)
......
...@@ -335,6 +335,10 @@ class QAPISchemaGenVisitVisitor(QAPISchemaVisitor): ...@@ -335,6 +335,10 @@ class QAPISchemaGenVisitVisitor(QAPISchemaVisitor):
self.decl = self._btin + self.decl self.decl = self._btin + self.decl
self._btin = None self._btin = None
def visit_needed(self, entity):
# Visit everything except implicit objects
return not isinstance(entity, QAPISchemaObjectType) or entity.info
def visit_enum_type(self, name, info, values, prefix): def visit_enum_type(self, name, info, values, prefix):
self.decl += gen_visit_decl(name, scalar=True) self.decl += gen_visit_decl(name, scalar=True)
self.defn += gen_visit_enum(name) self.defn += gen_visit_enum(name)
...@@ -351,13 +355,12 @@ class QAPISchemaGenVisitVisitor(QAPISchemaVisitor): ...@@ -351,13 +355,12 @@ class QAPISchemaGenVisitVisitor(QAPISchemaVisitor):
self.defn += defn self.defn += defn
def visit_object_type(self, name, info, base, members, variants): def visit_object_type(self, name, info, base, members, variants):
if info: self.decl += gen_visit_decl(name)
self.decl += gen_visit_decl(name) if variants:
if variants: assert not members # not implemented
assert not members # not implemented self.defn += gen_visit_union(name, base, variants)
self.defn += gen_visit_union(name, base, variants) else:
else: self.defn += gen_visit_struct(name, base, members)
self.defn += gen_visit_struct(name, base, members)
def visit_alternate_type(self, name, info, variants): def visit_alternate_type(self, name, info, variants):
self.decl += gen_visit_decl(name) self.decl += gen_visit_decl(name)
......
...@@ -811,6 +811,10 @@ class QAPISchemaVisitor(object): ...@@ -811,6 +811,10 @@ class QAPISchemaVisitor(object):
def visit_end(self): def visit_end(self):
pass pass
def visit_needed(self, entity):
# Default to visiting everything
return True
def visit_builtin_type(self, name, info, json_type): def visit_builtin_type(self, name, info, json_type):
pass pass
...@@ -1304,10 +1308,10 @@ class QAPISchema(object): ...@@ -1304,10 +1308,10 @@ class QAPISchema(object):
ent.check(self) ent.check(self)
def visit(self, visitor): def visit(self, visitor):
ignore = visitor.visit_begin(self) visitor.visit_begin(self)
for name in sorted(self._entity_dict.keys()): for (name, entity) in sorted(self._entity_dict.items()):
if not ignore or not isinstance(self._entity_dict[name], ignore): if visitor.visit_needed(entity):
self._entity_dict[name].visit(visitor) entity.visit(visitor)
visitor.visit_end() visitor.visit_end()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册