v2

vnt-core/src/protocol/mod.rs

@@ -0,0 +1,21 @@
+use bytes::BytesMut;
+use prost::Message;
+
+pub(crate) mod client_message;
+pub mod control_message;
+pub(crate) mod ip_packet_protocol;
+pub(crate) mod rpc_message;
+pub(crate) mod transmission;
+
+pub trait ProtoToBytesMut: Message {
+    fn encode_bytes_mut(&self) -> BytesMut
+    where
+        Self: Sized,
+    {
+        let mut bytes_mut = BytesMut::with_capacity(self.encoded_len());
+        self.encode_raw(&mut bytes_mut);
+        bytes_mut
+    }
+}
+
+impl<T: Message> ProtoToBytesMut for T {}