Merge pull request #90 from m-chrzan/switch_udup_to_remotegosssipgirl

Make UDUP handle only gossip girl messages with its timestamps

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPClient.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPClient.java
index 2e4f0b4..089cad2 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPClient.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/UDUPClient.java
@@ -1,6 +1,8 @@
 package pl.edu.mimuw.cloudatlas.agent.modules;
 
 import pl.edu.mimuw.cloudatlas.agent.messages.UDUPMessage;
+import pl.edu.mimuw.cloudatlas.model.ValueTime;
+import pl.edu.mimuw.cloudatlas.model.ValueUtils;
 
 import javax.xml.crypto.Data;
 import java.io.IOException;
@@ -67,9 +69,12 @@ public class UDUPClient {
     public void sendMessage(UDUPMessage msg) throws IOException {
         int packetNo = 1;
         byte[] sendBuf;
-        byte[] dataBuf = this.serializer.serialize(msg);
+        byte[] dataBuf;
         this.lastTransmission++;
 
+        msg.getContent().setSentTimestamp(ValueUtils.currentTime());
+        dataBuf = this.serializer.serialize(msg);
+
         do {
             sendBuf = packSendBuffer(this.lastTransmission, packetNo, dataBuf);
             DatagramPacket packet = new DatagramPacket(sendBuf, 0, sendBuf.length, msg.getContact().getAddress(), this.serverPort);