Fix wrong param in addToMap

Also track BundleOfEnvelopes
This commit is contained in:
chimp1984 2021-01-06 01:09:24 -05:00
parent d13b7e70fa
commit 082cc33ba4
No known key found for this signature in database
GPG key ID: 9801B4EC591F90E3

View file

@ -81,7 +81,9 @@ public class ConnectionStatistics implements MessageListener {
Connection connection) { Connection connection) {
lastMessageTimestamp = System.currentTimeMillis(); lastMessageTimestamp = System.currentTimeMillis();
if (networkEnvelope instanceof BundleOfEnvelopes) { if (networkEnvelope instanceof BundleOfEnvelopes) {
((BundleOfEnvelopes) networkEnvelope).getEnvelopes().forEach(e -> addToMap(networkEnvelope, receivedDataMap)); ((BundleOfEnvelopes) networkEnvelope).getEnvelopes().forEach(e -> addToMap(e, receivedDataMap));
// We want to track also number of BundleOfEnvelopes
addToMap(networkEnvelope, receivedDataMap);
} else { } else {
addToMap(networkEnvelope, receivedDataMap); addToMap(networkEnvelope, receivedDataMap);
} }
@ -91,7 +93,9 @@ public class ConnectionStatistics implements MessageListener {
public void onMessageSent(NetworkEnvelope networkEnvelope, Connection connection) { public void onMessageSent(NetworkEnvelope networkEnvelope, Connection connection) {
lastMessageTimestamp = System.currentTimeMillis(); lastMessageTimestamp = System.currentTimeMillis();
if (networkEnvelope instanceof BundleOfEnvelopes) { if (networkEnvelope instanceof BundleOfEnvelopes) {
((BundleOfEnvelopes) networkEnvelope).getEnvelopes().forEach(e -> addToMap(networkEnvelope, sentDataMap)); ((BundleOfEnvelopes) networkEnvelope).getEnvelopes().forEach(e -> addToMap(e, sentDataMap));
// We want to track also number of BundleOfEnvelopes
addToMap(networkEnvelope, sentDataMap);
} else { } else {
addToMap(networkEnvelope, sentDataMap); addToMap(networkEnvelope, sentDataMap);
} }