Skip to content

Commit b3c70fb

Browse files
committed
refactor(dmq): promote 'message' module to directory
1 parent 8969368 commit b3c70fb

File tree

3 files changed

+10
-5
lines changed

3 files changed

+10
-5
lines changed

internal/mithril-dmq/src/lib.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,12 @@
22
//! This crate provides mechanisms to publish and consume messages of a Decentralized Message Queue network through a DMQ node.
33
44
mod consumer;
5-
mod message;
5+
mod model;
66
mod publisher;
77
pub mod test;
88

99
pub use consumer::{DmqConsumerClient, DmqConsumerClientPallas};
10-
pub use message::DmqMessageBuilder;
10+
pub use model::{DmqMessage, DmqMessageBuilder};
1111
pub use publisher::{DmqPublisherClient, DmqPublisherClientPallas};
1212

1313
#[cfg(test)]

internal/mithril-dmq/src/message.rs renamed to internal/mithril-dmq/src/model/builder.rs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@ use mithril_common::{
1010
crypto_helper::{KesSigner, TryToBytes},
1111
};
1212

13+
use crate::model::DmqMessage;
14+
1315
/// The TTL (Time To Live) for DMQ messages in blocks.
1416
const DMQ_MESSAGE_TTL_IN_BLOCKS: u16 = 100;
1517

@@ -38,7 +40,7 @@ impl DmqMessageBuilder {
3840
}
3941

4042
/// Builds a DMQ message from the provided message bytes.
41-
pub async fn build(&self, message_bytes: &[u8]) -> StdResult<DmqMsg> {
43+
pub async fn build(&self, message_bytes: &[u8]) -> StdResult<DmqMessage> {
4244
fn compute_msg_id(dmq_message: &DmqMsg) -> Vec<u8> {
4345
let mut hasher = Blake2b::<U64>::new();
4446
hasher.update(&dmq_message.msg_body);
@@ -76,7 +78,7 @@ impl DmqMessageBuilder {
7678
};
7779
dmq_message.msg_id = compute_msg_id(&dmq_message);
7880

79-
Ok(dmq_message)
81+
Ok(dmq_message.into())
8082
}
8183
}
8284

@@ -137,7 +139,7 @@ mod tests {
137139
},
138140
DmqMsg {
139141
msg_id: vec![],
140-
..dmq_message
142+
..dmq_message.into()
141143
}
142144
);
143145
}

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
mod builder;
2+
3+
pub use builder::*;

0 commit comments

Comments
 (0)