From 6822da6e34cb88da2ede27eec1e45acd62357bbe Mon Sep 17 00:00:00 2001
From: Aryan Singh <asing271@illinois.edu>
Date: Sun, 27 Sep 2026 21:02:18 -0500
Subject: [PATCH] added total bytes r/s + updated teh view

---
 MP2/peer.py                      | 33 ++++++++++++++++++++++++++++----
 MP2/scripts/pretty_membership.py | 10 ++++++++++
 2 files changed, 39 insertions(+), 4 deletions(-)

diff --git a/MP2/peer.py b/MP2/peer.py
index 8da0826..bde0f9d 100644
--- a/MP2/peer.py
+++ b/MP2/peer.py
@@ -25,6 +25,10 @@ def log(message):
 def utc_time():
     return datetime.now(timezone.utc).timestamp()
 
+TOTAL_SENT_BYTES = 0
+TOTAL_RECEIVED_BYTES = 0
+traffic_lock = threading.Lock()
+
 #fixed
 INTRODUCER_IP = "172.22.156.3"
 GLOBAL_PORT = 8989
@@ -116,6 +120,7 @@ def global_switcher(cmd, value):
 
 #THREAD 1
 def worker_1():
+    global TOTAL_SENT_BYTES, TOTAL_RECEIVED_BYTES
     has_joined = (INTRODUCER_IP == MY_IP)
     last_join_attempt = 0.0
 
@@ -127,7 +132,10 @@ def worker_1():
             joining_message_json = json.dumps(joining_message_payload)
 
             try:
-                server_socket.sendto(joining_message_json.encode('utf-8'), (INTRODUCER_IP, GLOBAL_PORT))
+                joining_message_bytes = joining_message_json.encode('utf-8')
+                server_socket.sendto(joining_message_bytes, (INTRODUCER_IP, GLOBAL_PORT))
+                with traffic_lock:
+                    TOTAL_SENT_BYTES += len(joining_message_bytes)
             except Exception as e:
                 log(f"JOIN send failed: {e}")
             last_join_attempt = utc_time()
@@ -137,6 +145,9 @@ def worker_1():
         except socket.timeout:
             continue
 
+        with traffic_lock:
+            TOTAL_RECEIVED_BYTES += len(data)
+
         try:
             msg = json.loads(data.decode('utf-8'))
         except Exception:
@@ -195,6 +206,7 @@ def current_cluster_state():
         }
 
 def handle_join(msg):
+    global TOTAL_SENT_BYTES
     sender_id = tuple(msg["sender_id"])
     with members_list_lock:
         #check if the sender is in the dead members list
@@ -227,7 +239,10 @@ def handle_join(msg):
     }
 
     try:
-        server_socket.sendto(json.dumps(ack_payload).encode('utf-8'), (sender_id[0], sender_id[1]))
+        ack_payload_bytes = json.dumps(ack_payload).encode('utf-8')
+        server_socket.sendto(ack_payload_bytes, (sender_id[0], sender_id[1]))
+        with traffic_lock:
+            TOTAL_SENT_BYTES += len(ack_payload_bytes)
     except Exception as e:
         log(f"Failed to send GOSSIP response to {sender_id[0]}:{sender_id[1]}: {e}")
 
@@ -377,6 +392,7 @@ _peer_queue = []
 _peer_queue_lock = threading.Lock()
 
 def worker_3():
+    global TOTAL_SENT_BYTES
     global _peer_queue
     while not stop_threads.is_set():
         with members_list_lock:
@@ -407,7 +423,10 @@ def worker_3():
             }
 
             try:
-                server_socket.sendto(json.dumps(gossip_payload).encode('utf-8'), (peer[0], peer[1]))
+                gossip_payload_bytes = json.dumps(gossip_payload).encode('utf-8')
+                server_socket.sendto(gossip_payload_bytes, (peer[0], peer[1]))
+                with traffic_lock:
+                    TOTAL_SENT_BYTES += len(gossip_payload_bytes)
             except Exception as e:
                 log(f"Failed to send GOSSIP to {peer[0]}:{peer[1]}: {e}")
 
@@ -530,6 +549,10 @@ def worker_monitor():
             mode = cluster_state["mode"]
             drop_rate = cluster_state["drop_rate"]
 
+        with traffic_lock:
+            sent_bytes = TOTAL_SENT_BYTES
+            received_bytes = TOTAL_RECEIVED_BYTES
+
         try:
             with open("membership_list.txt", "w") as f:
                 for member_id, member_data in snapshot.items():
@@ -537,7 +560,9 @@ def worker_monitor():
 
                 f.write(
                     f"Mode: {mode}, "
-                    f"Drop Rate: {drop_rate}\n"
+                        f"Drop Rate: {drop_rate}, "
+                        f"Sent Bytes: {sent_bytes}, "
+                        f"Received Bytes: {received_bytes}\n"
                 )
 
         except Exception as e:
diff --git a/MP2/scripts/pretty_membership.py b/MP2/scripts/pretty_membership.py
index 5e4bc77..84d0426 100644
--- a/MP2/scripts/pretty_membership.py
+++ b/MP2/scripts/pretty_membership.py
@@ -73,6 +73,8 @@ def render(lines):
 
     mode = "?"
     drop_rate = "?"
+    sent_bytes = "?"
+    received_bytes = "?"
 
     now = time.time()
 
@@ -94,6 +96,10 @@ def render(lines):
 
                 mode = pieces[0].split(":", 1)[1].strip()
                 drop_rate = pieces[1].split(":", 1)[1].strip()
+                if len(pieces) > 2:
+                    sent_bytes = pieces[2].split(":", 1)[1].strip()
+                if len(pieces) > 3:
+                    received_bytes = pieces[3].split(":", 1)[1].strip()
 
             except Exception:
                 pass
@@ -166,6 +172,10 @@ def render(lines):
         f"{GRAY}|{RESET} "
         f"{BOLD}{CYAN}DROP:{RESET} "
         f"{WHITE}{drop_rate}%{RESET}"
+        f" {GRAY}|{RESET} "
+        f"{BOLD}{CYAN}TOTAL TX:{RESET} {WHITE}{sent_bytes} B{RESET}"
+        f" {GRAY}|{RESET} "
+        f"{BOLD}{CYAN}TOTAL RX:{RESET} {WHITE}{received_bytes} B{RESET}"
         f"{CLEAR_LINE}"
     )