Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-04-20 07:46:05 -04:00
Code Issues Packages Projects Releases Wiki Activity

Configure all peers once to receive direct messages in network stress test

Browse Source 
Instead on adding a listener once per selected random destination peer.
This commit is contained in:
Ivan Vilata-i-Balaguer 2016-05-03 11:06:38 +02:00
parent 3e34d23b9e
commit 53901ba798
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
View File

@ -186,6 +186,16 @@ public class NetworkStressTest {
final CountDownLatch receivedDirectLatch = new CountDownLatch(nPeers);
final Instant sendStart = Instant.now();
for (final P2PService srcPeer : peerNodes) {
// Make peer ready for receiving direct messages.
srcPeer.addDecryptedDirectMessageListener((decryptedMsgWithPubKey, peerNodeAddress) -> {
if (!(decryptedMsgWithPubKey.message instanceof StressTestDirectMessage))
return;
StressTestDirectMessage directMessage = (StressTestDirectMessage) (decryptedMsgWithPubKey.message);
if ((directMessage.getData().equals("test/" + srcPeer.getAddress())))
receivedDirectLatch.countDown();
});
// Select a random peer and send a direct message to it.
final int dstPeerIdx = (int) (Math.random() * nPeers);
final P2PService dstPeer = peerNodes.get(dstPeerIdx);
//print("sending direct message from peer %s to %s", srcPeer.getAddress(), dstPeer.getAddress());
@ -202,13 +212,6 @@ public class NetworkStressTest {
sentDirectLatch.countDown();
}
});
dstPeer.addDecryptedDirectMessageListener((decryptedMsgWithPubKey, peerNodeAddress) -> {
if (!(decryptedMsgWithPubKey.message instanceof StressTestDirectMessage))
return;
StressTestDirectMessage directMessage = (StressTestDirectMessage) (decryptedMsgWithPubKey.message);
if ((directMessage.getData().equals("test/" + dstPeer.getAddress())))
receivedDirectLatch.countDown();
});
}
// Since receiving is completed before sending is reported to be complete,
// all receiving checks should end before all sending checks to avoid deadlocking.