Add FlatbufferSchema.has_type()

Reviewed By: ginfung Differential Revision: D33409610 fbshipit-source-id: 960032bb5114e51f07115aef8fd25118db48a68e
facebook · Jan 5, 2022 · 2ea681c · 2ea681c
1 parent ebfa5b1
commit 2ea681c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/compiler/crates/schema/src/flatbuffer/wrapper.rs b/compiler/crates/schema/src/flatbuffer/wrapper.rs
@@ -161,6 +161,10 @@ impl SchemaWrapper {
         result
     }
 
+    pub fn has_type(&self, type_name: StringKey) -> bool {
+        self.flatbuffer_schema().has_type(type_name)
+    }
+
     pub fn has_directive(&self, name: StringKey) -> bool {
         self.get_directive(name).is_some()
     }

diff --git a/compiler/crates/schema/src/schema.rs b/compiler/crates/schema/src/schema.rs
@@ -345,7 +345,7 @@ impl SDLSchema {
 
     pub fn has_type(&self, type_name: StringKey) -> bool {
         match self {
-            SDLSchema::FlatBuffer(_schema) => todo!(),
+            SDLSchema::FlatBuffer(schema) => schema.has_type(type_name),
             SDLSchema::InMemory(schema) => schema.has_type(type_name),
         }
     }