Merge pull request #69 from m-chrzan/timer_scheduler

Timer scheduler

3 files changed, 97 insertions, 0 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
new file mode 100644
index 0000000..9744971
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
@@ -0,0 +1,31 @@
+package pl.edu.mimuw.cloudatlas.agent.modules;
+
+import pl.edu.mimuw.cloudatlas.agent.Executor;
+import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
+
+/*
+ * A Module is a (potentially stateful) event handler.
+ */
+public abstract class Module {
+    protected AgentMessage.AgentModule moduleType;
+    protected Executor executor;
+
+    public Module(AgentMessage.AgentModule moduleType) {
+        this.moduleType = moduleType;
+    }
+
+    public abstract void handle(AgentMessage event) throws InterruptedException ;
+
+    public void setExecutor(Executor executor) {
+        this.executor = executor;
+    }
+
+    public void sendMessage(AgentMessage event) throws InterruptedException {
+        this.executor.passMessage(event);
+    }
+
+    @Override
+    public String toString() {
+        return moduleType.toString();
+    }
+}
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
new file mode 100644
index 0000000..7a0da70
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java
@@ -0,0 +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);
+    }
+}
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduler.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduler.java
new file mode 100644
index 0000000..9390940
--- /dev/null
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduler.java
@@ -0,0 +1,47 @@
+package pl.edu.mimuw.cloudatlas.agent.modules;
+
+import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.TimerSchedulerMessage;
+
+import java.util.Timer;
+
+/**
+ * Initializes a timer within a constructor during its attachment to the executor
+ * Runs in a thread separate from executor - maybe refactor so that it's attached to executor's thread
+ *
+ * Handle used to attach tasks to schedule
+ * Tasks declared as inherited from TimerTask
+ *
+ * TODO: add request id and custom time
+ */
+public class TimerScheduler extends Module {
+    private Timer timer;
+
+    public TimerScheduler(AgentMessage.AgentModule moduleType) {
+        super(moduleType);
+        assert moduleType == AgentMessage.AgentModule.TIMER_SCHEDULER;
+        this.timer = new Timer();
+        System.out.println("TimerScheduler instance initialized");
+    }
+
+    @Override
+    public void handle(AgentMessage event) throws InterruptedException {
+        assert event.getDestinationModule() == event.getCorrectMessageType();
+        TimerSchedulerMessage timerEvent = (TimerSchedulerMessage) event;
+        addTask(timerEvent);
+    }
+
+    public void addTask(TimerSchedulerMessage msg) {
+        TimerScheduledTask task = msg.getTask();
+        task.setScheduler(this);
+        this.timer.schedule(task, msg.getDelay());
+        System.out.println("Task with delay " + msg.getDelay() + " scheduled");
+    }
+
+    // TODO
+    public void removeTask(String requestId) {}
+
+    public void passMessageFromTask(AgentMessage msg) throws InterruptedException {
+        this.executor.passMessage(msg);
+    }
+}