Add bytes_transferred to each edge

stellar · Jun 15, 2020 · a382c69 · a382c69
1 parent b91e443
commit a382c69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/scripts/OverlaySurvey.py b/scripts/OverlaySurvey.py
@@ -15,10 +15,10 @@ def add_new_node(graph, label, version=""):
     graph.add_node(label, label=label, version=version)
 
 
-def add_new_edge(graph, u, v):
+def add_new_edge(graph, u, v, bytes_transferred):
     if graph.has_edge(u, v):
         return
-    graph.add_edge(u, v)
+    graph.add_edge(u, v, bytes_transferred=bytes_transferred)
 
 
 def next_peer(direction_tag, node_info):
@@ -52,7 +52,7 @@ def update_results(graph, parent_info, parent_key, results, is_inbound):
         results[direction_tag][other_key] = peer
         add_new_node(graph, parent_key)
         add_new_node(graph, other_key, peer["version"])
-        add_new_edge(graph, other_key, parent_key)
+        add_new_edge(graph, other_key, parent_key, peer["bytesWritten"] + peer["bytesRead"])
 
     if "numTotalInboundPeers" in parent_info:
         results["totalInbound"] = parent_info["numTotalInboundPeers"]