From e2cfe0d9643c3a02223a9e9c4f7d6ed29f0e363a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magdalena=20Grodzi=C5=84ska?= <mag.grodzinska@gmail.com>
Date: Fri, 10 Jan 2020 20:51:13 +0100
Subject: Add setting sender address in udp
---
.../mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java | 7 +++++++
.../java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java | 4 ++--
2 files changed, 9 insertions(+), 2 deletions(-)
(limited to 'src')
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java
index 4c223f5..1d7c0c6 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/RemoteGossipGirlMessage.java
@@ -2,9 +2,12 @@ package pl.edu.mimuw.cloudatlas.agent.messages;
import pl.edu.mimuw.cloudatlas.model.ValueTime;
+import java.net.InetAddress;
+
public class RemoteGossipGirlMessage extends GossipGirlMessage {
private ValueTime sentTimestamp;
private ValueTime receivedTimestamp;
+ private InetAddress senderAddress;
public RemoteGossipGirlMessage(String messageId, long timestamp, Type type) {
super(messageId, timestamp, type);
@@ -27,4 +30,8 @@ public class RemoteGossipGirlMessage extends GossipGirlMessage {
public ValueTime getReceivedTimestamp() {
return receivedTimestamp;
}
+
+ public InetAddress getSenderAddress() { return senderAddress; }
+
+ public void setSenderAddress(InetAddress senderAddress) { this.senderAddress = senderAddress; }
}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
index 94882e4..0c5bc86 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPServer.java
@@ -59,7 +59,6 @@ public class UDUPServer implements Runnable {
if (packetNo == 1 && packet.getLength() < this.bufSize) {
msg = this.serializer.deserialize(packetData);
- msg.getContent().setReceivedTimestamp(ValueUtils.currentTime());
System.out.println("UDP received message " + msg.getContent().getMessageId());
} else {
System.out.println("UDP received partial message with transmission id " + transmissionID + " packet no " + packetNo);
@@ -67,6 +66,8 @@ public class UDUPServer implements Runnable {
}
if (msg != null) {
+ msg.getContent().setReceivedTimestamp(ValueUtils.currentTime());
+ msg.getContent().setSenderAddress(packet.getAddress());
sendMessageFurther(msg);
}
}
@@ -137,7 +138,6 @@ public class UDUPServer implements Runnable {
try {
byte[] allPacketData = concatPacketData(transmissionID, newPacketNo, packetData);
msg = this.serializer.deserialize(allPacketData);
- msg.getContent().setReceivedTimestamp(ValueUtils.currentTime());
this.partialPackets.remove(transmissionID);
System.out.println("Kryo put together whole transmission for msg " + msg.getContent().getMessageId());
} catch (Error | Exception e) {
--
cgit v1.2.3