feat(dmq): add 'DmqMessage' type to wrap a 'DmqMsg'

Author: jpraynaud

Cargo.lock

@@ -16,10 +16,13 @@ crate-type = ["lib", "cdylib", "staticlib"]
 [dependencies]
 anyhow = { workspace = true }
 async-trait = { workspace = true }
+bincode = { version = "2.0.1" }
 blake2 = "0.10.6"
 mithril-cardano-node-chain = { path = "../cardano-node/mithril-cardano-node-chain" }
 mithril-common = { path = "../../mithril-common" }
 pallas-network = { git = "https://github.com/txpipe/pallas.git", branch = "main" }
+serde = { workspace = true }
+serde_bytes = "0.11.17"
 slog = { workspace = true }
 tokio = { workspace = true, features = ["sync"] }
 

internal/mithril-dmq/Cargo.toml

@@ -0,0 +1,68 @@
+use std::ops::{Deref, DerefMut};
+
+use pallas_network::miniprotocols::localmsgsubmission::DmqMsg;
+use serde::{Deserialize, Deserializer, Serialize, Serializer};
+
+/// Type alias for a DMQ message.
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub struct DmqMessage(DmqMsg);
+
+#[derive(Serialize, Deserialize)]
+struct RawBytes(#[serde(with = "serde_bytes")] Vec<u8>);
+
+impl Deref for DmqMessage {
+    type Target = DmqMsg;
+
+    fn deref(&self) -> &Self::Target {
+        &self.0
+    }
+}
+
+impl DerefMut for DmqMessage {
+    fn deref_mut(&mut self) -> &mut Self::Target {
+        &mut self.0
+    }
+}
+
+impl From<DmqMsg> for DmqMessage {
+    fn from(msg: DmqMsg) -> Self {
+        Self(msg)
+    }
+}
+
+impl From<DmqMessage> for DmqMsg {
+    fn from(msg: DmqMessage) -> Self {
+        msg.0
+    }
+}
+
+impl Serialize for DmqMessage {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: Serializer,
+    {
+        let raw_bytes = RawBytes(
+            bincode::serde::encode_to_vec(self, bincode::config::standard()).map_err(|e| {
+                serde::ser::Error::custom(format!("DMQ message serialization error: {e}"))
+            })?,
+        );
+
+        raw_bytes.serialize(serializer)
+    }
+}
+
+impl<'de> Deserialize<'de> for DmqMessage {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: Deserializer<'de>,
+    {
+        let raw_bytes = RawBytes::deserialize(deserializer)?;
+        let (res, _) =
+            bincode::serde::decode_from_slice::<Self, _>(&raw_bytes.0, bincode::config::standard())
+                .map_err(|e| {
+                    serde::de::Error::custom(format!("DMQ message deserialization error: {e}"))
+                })?;
+
+        Ok(res)
+    }
+}

internal/mithril-dmq/src/model/message.rs

@@ -1,3 +1,5 @@
 mod builder;
+mod message;
 
 pub use builder::*;
+pub use message::*;