Refactor handle (#75)

* Refactor passing messages to handler

* Add RMIMessage skeleton

* Handle RMI message in RMI module

1 files changed, 21 insertions, 3 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
index 9744971..a50a95f 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Module.java
@@ -2,19 +2,37 @@ package pl.edu.mimuw.cloudatlas.agent.modules;
 
 import pl.edu.mimuw.cloudatlas.agent.Executor;
 import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.TimerSchedulerMessage;
+import pl.edu.mimuw.cloudatlas.agent.messages.RMIMessage;
 
 /*
  * A Module is a (potentially stateful) event handler.
  */
 public abstract class Module {
-    protected AgentMessage.AgentModule moduleType;
+    public class InvalidMessageType extends Exception {
+        public InvalidMessageType(String message) {
+            super(message);
+        }
+    }
+
+    protected ModuleType moduleType;
     protected Executor executor;
 
-    public Module(AgentMessage.AgentModule moduleType) {
+    public Module(ModuleType moduleType) {
         this.moduleType = moduleType;
     }
 
-    public abstract void handle(AgentMessage event) throws InterruptedException ;
+    public void handle(AgentMessage event) throws InterruptedException, InvalidMessageType {
+        event.callMe(this);
+    }
+
+    public void handleTyped(TimerSchedulerMessage message) throws InterruptedException, InvalidMessageType {
+        throw new InvalidMessageType("Got a TimerSchedulerMessage in module " + moduleType.toString());
+    }
+
+    public void handleTyped(RMIMessage message) throws InterruptedException, InvalidMessageType {
+        throw new InvalidMessageType("Got an RMIMessage in module " + moduleType.toString());
+    }
 
     public void setExecutor(Executor executor) {
         this.executor = executor;