logos-blockchain/logos-blockchain-specs
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-specs.git synced 2026-01-07 07:33:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: cache msg_hash in sender as well

Browse Source
This commit is contained in:
Youngjoon Lee 2024-05-24 16:56:43 +09:00
parent b18191b11c
commit 06f8f0f7ca
No known key found for this signature in database
GPG Key ID: 09B750B5BD6F08A2
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mixnet/v2/sim/p2p.py
View File

@ -88,9 +88,12 @@ class GossipP2P(P2P):
if len(back) > 0:
neighbor = back[0]
back = back[1:]
else:
elif len(front) > 0:
neighbor = front[0]
front = front[1:]
else:
return
others = front + back
n = min(self.config.p2p.connection_density - 1, len(others))
conns = set(random.sample(others, n))
@ -100,6 +103,10 @@ class GossipP2P(P2P):
def broadcast(self, sender: "Node", msg: SphinxPacket | bytes):
yield from super().broadcast(sender, msg)
self.log(f"Node:{sender.id}: Gossiping a msg: {len(msg)} bytes")
msg_hash = hashlib.sha256(bytes(msg)).digest()
self.message_cache[sender].add(msg_hash)
for receiver in self.topology[sender]:
self.measurement.measure_egress(sender, msg)
self.env.process(self.send(msg, receiver))