logos-co
/
nomos-simulations
mirror of https://github.com/logos-co/nomos-simulations.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(blend): add logs for queue releases

This commit is contained in:
Youngjoon Lee 2024-12-20 11:52:55 +09:00
parent 9f9c6d0d34
commit 8529f425b4
No known key found for this signature in database
GPG Key ID: 303963A54A81DD4D
2 changed files with 31 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
simlib/blendnet-sims/Cargo.toml
View File

@ -17,9 +17,9 @@ serde_json = "1.0.132"
tracing = "0.1.40"
tracing-subscriber = { version = "0.3", features = ["json", "env-filter", "tracing-log"] }
netrunner = { path = "../netrunner" }
nomos-tracing = { git = "https://github.com/logos-co/nomos-node.git" }
nomos-blend = { git = "https://github.com/logos-co/nomos-node", package = "nomos-blend" }
nomos-blend-message = { git = "https://github.com/logos-co/nomos-node", package = "nomos-blend-message" }
nomos-tracing = { git = "https://github.com/logos-co/nomos-node.git", branch = "blend-mock-msg-payload" }
nomos-blend = { git = "https://github.com/logos-co/nomos-node", branch = "blend-mock-msg-payload", package = "nomos-blend" }
nomos-blend-message = { git = "https://github.com/logos-co/nomos-node", branch = "blend-mock-msg-payload", package = "nomos-blend-message" }
futures = "0.3.31"
rand_chacha = "0.3"
multiaddr = "0.18"

34
simlib/blendnet-sims/src/node/blend/mod.rs
View File

@ -254,6 +254,23 @@ impl BlendNode {
hasher.finalize().into()
}
fn schedule_persistent_transmission(&mut self, message: Vec<u8>) {
self.log_message(
"PersistentTransmissionScheduled",
&Self::parse_payload(&message),
);
self.persistent_sender.send(message).unwrap();
}
fn schedule_blend(&mut self, message: Vec<u8>) {
self.log_message("BlendScheduled", &Self::parse_payload(&message));
self.blend_sender.send(message).unwrap();
}
fn parse_payload(message: &[u8]) -> Payload {
Payload::load(MockBlendMessage::payload(message).unwrap())
}
fn update_time(&mut self, elapsed: Duration) {
self.data_msg_lottery_update_time_sender
.send(elapsed)
@ -268,6 +285,10 @@ impl BlendNode {
self.log_message(format!("{}MessageGenerated", msg_type).as_str(), payload);
}
fn log_message_released_from(&self, from: &str, payload: &Payload) {
self.log_message(format!("MessageReleasedFrom{}", from).as_str(), payload);
}
fn log_message_fully_unwrapped(&self, payload: &Payload) {
self.log_message("MessageFullyUnwrapped", payload);
}
@ -321,7 +342,7 @@ impl Node for BlendNode {
.crypto_processor
.wrap_message(payload.as_bytes())
.unwrap();
self.persistent_sender.send(message).unwrap();
self.schedule_persistent_transmission(message);
}
}
@ -332,19 +353,19 @@ impl Node for BlendNode {
Some(network_message.from),
None,
);
self.blend_sender
.send(network_message.into_payload().0)
.unwrap();
self.schedule_blend(network_message.into_payload().0);
}
// Proceed message blend
if let Poll::Ready(Some(msg)) = pin!(&mut self.blend_messages).poll_next(&mut cx) {
match msg {
BlendOutgoingMessage::Outbound(msg) => {
self.persistent_sender.send(msg).unwrap();
self.log_message_released_from("Blend", &Self::parse_payload(&msg));
self.schedule_persistent_transmission(msg);
}
BlendOutgoingMessage::FullyUnwrapped(payload) => {
let payload = Payload::load(payload);
self.log_message_released_from("Blend", &payload);
self.log_message_fully_unwrapped(&payload);
self.state.num_messages_fully_unwrapped += 1;
//TODO: create a tracing event
@ -360,13 +381,14 @@ impl Node for BlendNode {
.crypto_processor
.wrap_message(payload.as_bytes())
.unwrap();
self.persistent_sender.send(message).unwrap();
self.schedule_persistent_transmission(message);
}
// Proceed persistent transmission
if let Poll::Ready(Some(msg)) =
pin!(&mut self.persistent_transmission_messages).poll_next(&mut cx)
{
self.log_message_released_from("PersistentTransmission", &Self::parse_payload(&msg));
self.forward(
BlendMessage(msg),
None,