From c7d6bd9a56f6958b2102ef93fcebc3187699a45f Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Thu, 7 Nov 2024 09:47:14 +0700
Subject: [PATCH] Count # of messages broadcasted by each node

---
 network-runner/src/node/mix/mod.rs   | 4 +++-
 network-runner/src/node/mix/state.rs | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/network-runner/src/node/mix/mod.rs b/network-runner/src/node/mix/mod.rs
index 9e70381..09078f9 100644
--- a/network-runner/src/node/mix/mod.rs
+++ b/network-runner/src/node/mix/mod.rs
@@ -133,6 +133,7 @@ impl MixNode {
                 node_id: id,
                 mock_counter: 0,
                 step_id: 0,
+                num_messages_broadcasted: 0,
             },
             persistent_sender,
             persistent_update_time_sender,
@@ -195,7 +196,8 @@ impl Node for MixNode {
                     persistent_sender.send(msg).unwrap();
                 }
                 MixOutgoingMessage::FullyUnwrapped(_) => {
-                    //TODO: increase counter and create a tracing event
+                    self.state.num_messages_broadcasted += 1;
+                    //TODO: create a tracing event
                 }
             }
         }
diff --git a/network-runner/src/node/mix/state.rs b/network-runner/src/node/mix/state.rs
index 10ce58f..688d39a 100644
--- a/network-runner/src/node/mix/state.rs
+++ b/network-runner/src/node/mix/state.rs
@@ -14,6 +14,7 @@ pub struct MixnodeState {
     pub node_id: NodeId,
     pub mock_counter: usize,
     pub step_id: usize,
+    pub num_messages_broadcasted: usize,
 }
 
 #[derive(Serialize)]