Add arena feature

protokit_binformat/Cargo.toml

@@ -8,6 +8,10 @@ repository = "https://github.com/semtexzv/protokit"
 authors = ["<semtexzv@gmail.com>"]
 description = "Usable protocol buffers"
 
+[features]
+default = ["arena"]
+arena = ["bumpalo", "bumpalo/boxed", "bumpalo/collections"]
+
 [dependencies.derive]
 package = "protokit_derive"
 path = "../protokit_derive"
@@ -16,3 +20,4 @@ version = "0.0.2"
 [dependencies]
 indexmap = "1"
 thiserror = "1"
+bumpalo = { version = "3.13.0", optional = true, default_features = false }

protokit_binformat/src/lib.rs

@@ -86,6 +86,23 @@ where
     }
 }
 
+#[cfg(feature = "arena")]
+impl<'buf, 'arena, T> BinProto<'buf> for bumpalo::boxed::Box<'arena, T>
+    where T: BinProto<'buf>
+{
+    fn merge_field(&mut self, tag_wire: u32, stream: &mut InputStream<'buf>) -> Result<()> {
+        self.deref_mut().merge_field(tag_wire, stream)
+    }
+
+    fn size(&self, stack: &mut SizeStack) -> usize {
+        self.deref().size(stack)
+    }
+
+    fn encode(&self, stream: &mut OutputStream) {
+        self.deref().encode(stream)
+    }
+}
+
 pub trait Varint: Default + Debug + Clone + Copy {
     fn from_u64(v: u64) -> Self;
     fn into_u64(self) -> u64;