Support enums in rust-transient-0

1 files changed, 18 insertions, 0 deletions

diff --git a/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs b/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
index c5bb458..fbfc82d 100644
--- a/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
+++ b/examples/10-transient-0/output/rust/src/gen/data/transient_0.rs
@@ -152,6 +152,24 @@ pub fn decode_struct_with_option(reader: &mut impl std::io::Read) -> ::std::io::
     std::io::Result::Ok(crate::data::StructWithOption { a: a, b: b })
 }
 
+pub fn encode_enum_with_variants(value: &crate::data::EnumWithVariants, writer: &mut impl std::io::Write) -> ::std::io::Result::<()> {
+    crate::transient_0::encode_int_32(&match value {
+        crate::data::EnumWithVariants::Zero => 0,
+        crate::data::EnumWithVariants::One => 1,
+        crate::data::EnumWithVariants::Two => 2,
+    }, writer)?;
+    ::std::io::Result::Ok(())
+}
+
+pub fn decode_enum_with_variants(reader: &mut impl std::io::Read) -> ::std::io::Result::<crate::data::EnumWithVariants> {
+    std::io::Result::Ok(match &crate::transient_0::decode_int_32(reader)? {
+        0 => crate::data::EnumWithVariants::Zero,
+        1 => crate::data::EnumWithVariants::One,
+        2 => crate::data::EnumWithVariants::Two,
+        _ => todo!(),
+    })
+}
+
 pub fn encode_variant_one(value: &crate::data::VariantOne, writer: &mut impl std::io::Write) -> ::std::io::Result::<()> {
     crate::transient_0::encode_int_32(&value.a, writer)?;
     ::std::io::Result::Ok(())