1 files changed, 12 insertions, 0 deletions
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java
index 4b933b5..7a0da70 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java
@@ -1,7 +1,19 @@
 package pl.edu.mimuw.cloudatlas.agent.modules;
 
+import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
+
 import java.util.TimerTask;
 
 public abstract class TimerScheduledTask extends TimerTask {
+    private TimerScheduler scheduler;
+
     public abstract void run();
+
+    public void setScheduler(TimerScheduler scheduler) {
+        this.scheduler = scheduler;
+    }
+
+    protected void sendMessage(AgentMessage msg) throws InterruptedException {
+        this.scheduler.passMessageFromTask(msg);
+    }
 }