Add setting sender address in udp

1 files changed, 7 insertions, 0 deletions

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; }
 }