From c44f0e747f00ec71bd9c773bfea7d287e50ed77b Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 19 Jul 2026 21:41:55 +0100 Subject: Support sums in rust-transient-0 --- .../output/rust/src/gen/data/transient_0.rs | 30 ++++++++++++++++++++++ examples/10-transient-0/output/rust/src/lib.rs | 17 +++++++++++- 2 files changed, 46 insertions(+), 1 deletion(-) (limited to 'examples') 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 fbfc82d..0ea3f6f 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 @@ -170,6 +170,36 @@ pub fn decode_enum_with_variants(reader: &mut impl std::io::Read) -> ::std::io:: }) } +pub fn encode_sum_with_variants(value: &crate::data::SumWithVariants, writer: &mut impl std::io::Write) -> ::std::io::Result::<()> { + match value { + crate::data::SumWithVariants::VariantOne(value) => { + { + crate::transient_0::encode_int_32(&0, writer)?; + }; + { + crate::data::transient_0::encode_variant_one(value, writer)?; + }; + }, + crate::data::SumWithVariants::VariantTwo(value) => { + { + crate::transient_0::encode_int_32(&1, writer)?; + }; + { + crate::data::transient_0::encode_variant_two(value, writer)?; + }; + }, + }; + ::std::io::Result::Ok(()) +} + +pub fn decode_sum_with_variants(reader: &mut impl std::io::Read) -> ::std::io::Result:: { + std::io::Result::Ok(match &crate::transient_0::decode_int_32(reader)? { + 0 => crate::data::SumWithVariants::VariantOne(crate::data::transient_0::decode_variant_one(reader)?), + 1 => crate::data::SumWithVariants::VariantTwo(crate::data::transient_0::decode_variant_two(reader)?), + _ => 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(()) diff --git a/examples/10-transient-0/output/rust/src/lib.rs b/examples/10-transient-0/output/rust/src/lib.rs index c3da14c..3aa9751 100644 --- a/examples/10-transient-0/output/rust/src/lib.rs +++ b/examples/10-transient-0/output/rust/src/lib.rs @@ -4,7 +4,7 @@ pub mod transient_0; #[cfg(test)] mod test { use std::io::Cursor; - use super::data::{EnumWithVariants, InnerStruct, OuterStruct, StructWithBool, StructWithInt32, StructWithInt64, StructWithList, StructWithOption, StructWithString}; + use super::data::{EnumWithVariants, InnerStruct, OuterStruct, StructWithBool, StructWithInt32, StructWithInt64, StructWithList, StructWithOption, StructWithString, SumWithVariants, VariantOne, VariantTwo }; #[test] fn struct_with_bool() { @@ -128,6 +128,21 @@ mod test { ); } + #[test] + fn sum_with_variants() { + assert_round_trip_encoding( + SumWithVariants::VariantOne(VariantOne { a: 10 }), + super::data::transient_0::encode_sum_with_variants, + super::data::transient_0::decode_sum_with_variants, + ); + + assert_round_trip_encoding( + SumWithVariants::VariantTwo(VariantTwo { a: 25 }), + super::data::transient_0::encode_sum_with_variants, + super::data::transient_0::decode_sum_with_variants, + ); + } + fn assert_round_trip_encoding( value: T, encode: impl Fn(&T, &mut Cursor>) -> std::io::Result<()>, -- cgit v1.2.3