From 26c750550b9fa6bdbebc8acbb39a34281205c140 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magdalena=20Grodzi=C5=84ska?= <mag.grodzinska@gmail.com>
Date: Wed, 25 Dec 2019 12:53:39 +0100
Subject: Fix task type in timer message

---
 .../agent/messages/TimerSchedulerMessage.java          | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
index 9a558b7..333b381 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
@@ -1,14 +1,20 @@
 package pl.edu.mimuw.cloudatlas.agent.messages;
 
-import java.util.TimerTask;
+import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduledTask;
 
 public class TimerSchedulerMessage extends AgentMessage {
     private String requestId;
     private long delay;
     private long baseTime;
-    private TimerTask task;
-
-    public TimerSchedulerMessage(String messageId, AgentModule destinationModule, long timestamp, String requestId, long delay, long baseTime, TimerTask task) {
+    private TimerScheduledTask task;
+
+    public TimerSchedulerMessage(String messageId,
+                                 AgentModule destinationModule,
+                                 long timestamp,
+                                 String requestId,
+                                 long delay,
+                                 long baseTime,
+                                 TimerScheduledTask task) {
         super(messageId, destinationModule, timestamp);
         this.requestId = requestId;
         this.delay = delay;
@@ -32,11 +38,11 @@ public class TimerSchedulerMessage extends AgentMessage {
         this.baseTime = baseTime;
     }
 
-    public TimerTask getTask() {
+    public TimerScheduledTask getTask() {
         return task;
     }
 
-    public void setTask(TimerTask task) {
+    public void setTask(TimerScheduledTask task) {
         this.task = task;
     }
 
-- 
cgit v1.2.3