add dynamic_struct::{Builder, Reader}::downcast()

capnproto · Sep 23, 2024 · 0b7e946 · 0b7e946
1 parent ee107e9
commit 0b7e946
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 15 deletions.
diff --git a/capnp/src/dynamic_struct.rs b/capnp/src/dynamic_struct.rs
@@ -232,6 +232,18 @@ impl<'a> Reader<'a> {
         let field = self.schema.get_field_by_name(field_name)?;
         self.has(field)
     }
+
+    /// Downcasts the `Reader` into a specific struct type. Panics if the
+    /// expected type does not match the value.
+    pub fn downcast<T: crate::traits::OwnedStruct>(self) -> T::Reader<'a> {
+        assert!(
+            Into::<crate::introspect::Type>::into(crate::introspect::TypeVariant::Struct(
+                self.schema.raw
+            ))
+            .may_downcast_to(T::introspect())
+        );
+        self.reader.into()
+    }
 }
 
 /// A mutable dynamically-typed struct.
@@ -773,6 +785,18 @@ impl<'a> Builder<'a> {
         }
         Ok(())
     }
+
+    /// Downcasts the `Builder` into a specific struct type. Panics if the
+    /// expected type does not match the value.
+    pub fn downcast<T: crate::traits::OwnedStruct>(self) -> T::Builder<'a> {
+        assert!(
+            Into::<crate::introspect::Type>::into(crate::introspect::TypeVariant::Struct(
+                self.schema.raw
+            ))
+            .may_downcast_to(T::introspect())
+        );
+        self.builder.into()
+    }
 }
 
 impl<'a> crate::traits::SetterInput<crate::any_pointer::Owned> for Reader<'a> {

diff --git a/capnp/src/dynamic_value.rs b/capnp/src/dynamic_value.rs
@@ -77,14 +77,8 @@ impl<'a> Reader<'a> {
     /// the current way the `Introspect` and `OwnedStruct` traits are set up does not
     /// seem to allow this.
     pub fn downcast_struct<T: crate::traits::OwnedStruct>(self) -> T::Reader<'a> {
-        let sb: dynamic_struct::Reader = self.downcast();
-        assert!(
-            Into::<crate::introspect::Type>::into(crate::introspect::TypeVariant::Struct(
-                sb.schema.raw
-            ))
-            .may_downcast_to(T::introspect())
-        );
-        sb.reader.into()
+        let sr: dynamic_struct::Reader = self.downcast();
+        sr.downcast::<T>()
     }
 }
 
@@ -243,13 +237,7 @@ impl<'a> Builder<'a> {
     /// seem to allow this.
     pub fn downcast_struct<T: crate::traits::OwnedStruct>(self) -> T::Builder<'a> {
         let sb: dynamic_struct::Builder = self.downcast();
-        assert!(
-            Into::<crate::introspect::Type>::into(crate::introspect::TypeVariant::Struct(
-                sb.schema.raw
-            ))
-            .may_downcast_to(T::introspect())
-        );
-        sb.builder.into()
+        sb.downcast::<T>()
     }
 }