Add sending messages from modules

1 files changed, 21 insertions, 1 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/Module.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/Module.java
index 46df31b..167f3b7 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/Module.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/Module.java
@@ -6,5 +6,25 @@ import pl.edu.mimuw.cloudatlas.agent.message.AgentMessage;
  * A Module is a (potentially stateful) event handler.
  */
 public abstract class Module {
-    public abstract void handle(AgentMessage event);
+    private AgentMessage.AgentModule moduleType;
+    private Executor executor;
+
+    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();
+    }
 }