feat(relay): add support for DMQ messages

Author: jpraynaud

Cargo.lock

@@ -11,7 +11,7 @@ repository = { workspace = true }
 
 [features]
 bundle_tls = ["reqwest/native-tls-vendored"]
-# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+future_dmq = ["dep:mithril-dmq"]
 
 [dependencies]
 anyhow = { workspace = true }
@@ -35,6 +35,7 @@ libp2p = { version = "0.55.0", features = [
     "yamux",
 ] }
 mithril-common = { path = "../mithril-common", features = ["full"] }
+mithril-dmq = { path = "../internal/mithril-dmq", optional = true }
 mithril-doc = { path = "../internal/mithril-doc" }
 mithril-test-http-server = { path = "../internal/tests/mithril-test-http-server" }
 reqwest = { workspace = true, features = [

mithril-relay/Cargo.toml

@@ -16,11 +16,14 @@ pub use relay::SignerRelayMode;
 
 /// The P2P topic names used by Mithril
 pub mod mithril_p2p_topic {
-    /// The topic name where signer registrations are published
-    pub const SIGNERS: &str = "mithril/signers";
+    /// The topic name where HTTP signer registrations are published
+    pub const SIGNERS_HTTP: &str = "mithril/signers/http";
 
-    /// The topic name where signatures are published
-    pub const SIGNATURES: &str = "mithril/signatures";
+    /// The topic name where HTTP signatures are published
+    pub const SIGNATURES_HTTP: &str = "mithril/signatures/http";
+
+    /// The topic name where DMQ signatures are published
+    pub const SIGNATURES_DMQ: &str = "mithril/signatures/dmq";
 }
 
 #[cfg(test)]

mithril-relay/src/lib.rs

@@ -14,6 +14,7 @@ use mithril_common::{
     logging::LoggerExtensions,
     messages::{RegisterSignatureMessageHttp, RegisterSignerMessage},
 };
+use mithril_dmq::DmqMessage;
 use serde::{Deserialize, Serialize};
 use slog::{Logger, debug, info};
 use std::{collections::HashMap, time::Duration};
@@ -63,11 +64,14 @@ pub type TopicName = String;
 /// The broadcast message received from a Gossip sub event
 #[derive(Serialize, Deserialize)]
 pub enum BroadcastMessage {
-    /// A signer registration message received from the Gossip sub
-    RegisterSigner(RegisterSignerMessage),
+    /// A HTTP signer registration message received from the Gossip sub
+    RegisterSignerHttp(RegisterSignerMessage),
 
-    /// A signature registration message received from the Gossip sub
-    RegisterSignature(RegisterSignatureMessageHttp),
+    /// A HTTP signature registration message received from the Gossip sub
+    RegisterSignatureHttp(RegisterSignatureMessageHttp),
+
+    /// A DMQ signature registration message received from the Gossip sub
+    RegisterSignatureDmq(DmqMessage),
 }
 
 /// A peer in the P2P network
@@ -95,12 +99,16 @@ impl Peer {
     fn build_topics() -> HashMap<TopicName, gossipsub::IdentTopic> {
         HashMap::from([
             (
-                mithril_p2p_topic::SIGNATURES.into(),
-                gossipsub::IdentTopic::new(mithril_p2p_topic::SIGNATURES),
+                mithril_p2p_topic::SIGNATURES_HTTP.into(),
+                gossipsub::IdentTopic::new(mithril_p2p_topic::SIGNATURES_HTTP),
+            ),
+            (
+                mithril_p2p_topic::SIGNATURES_DMQ.into(),
+                gossipsub::IdentTopic::new(mithril_p2p_topic::SIGNATURES_DMQ),
             ),
             (
-                mithril_p2p_topic::SIGNERS.into(),
-                gossipsub::IdentTopic::new(mithril_p2p_topic::SIGNERS),
+                mithril_p2p_topic::SIGNERS_HTTP.into(),
+                gossipsub::IdentTopic::new(mithril_p2p_topic::SIGNERS_HTTP),
             ),
         ])
     }
@@ -217,14 +225,25 @@ impl Peer {
         }
     }
 
-    /// Publish a signature on the P2P pubsub
-    pub fn publish_signature(
+    /// Publish a HTTP signature on the P2P pubsub
+    pub fn publish_signature_http(
         &mut self,
         message: &RegisterSignatureMessageHttp,
     ) -> StdResult<gossipsub::MessageId> {
         self.publish_broadcast_message(
-            &BroadcastMessage::RegisterSignature(message.to_owned()),
-            mithril_p2p_topic::SIGNATURES,
+            &BroadcastMessage::RegisterSignatureHttp(message.to_owned()),
+            mithril_p2p_topic::SIGNATURES_HTTP,
+        )
+    }
+
+    /// Publish a DMQ signature on the P2P pubsub
+    pub fn publish_signature_dmq(
+        &mut self,
+        message: &DmqMessage,
+    ) -> StdResult<gossipsub::MessageId> {
+        self.publish_broadcast_message(
+            &BroadcastMessage::RegisterSignatureDmq(message.to_owned()),
+            mithril_p2p_topic::SIGNATURES_DMQ,
         )
     }
 
@@ -270,8 +289,8 @@ impl Peer {
         message: &RegisterSignerMessage,
     ) -> StdResult<gossipsub::MessageId> {
         self.publish_broadcast_message(
-            &BroadcastMessage::RegisterSigner(message.to_owned()),
-            mithril_p2p_topic::SIGNERS,
+            &BroadcastMessage::RegisterSignerHttp(message.to_owned()),
+            mithril_p2p_topic::SIGNERS_HTTP,
         )
     }