From cb173ac60b8307247efd83aaed5a0bb44a107766 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Magdalena=20Grodzi=C5=84ska?= <mag.grodzinska@gmail.com>
Date: Mon, 23 Dec 2019 19:06:41 +0100
Subject: Add sending messages from modules

---
 .../java/pl/edu/mimuw/cloudatlas/agent/Module.java | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'src/main/java/pl/edu/mimuw/cloudatlas/agent/Module.java')

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();
+    }
 }
-- 
cgit v1.2.3