Merge pull request #82 from m-chrzan/agent-modules

Agent modules

1 files changed, 6 insertions, 9 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java
index 1b078cf..392e0a1 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java
@@ -2,12 +2,13 @@ package pl.edu.mimuw.cloudatlas.agent;
 
 import java.util.ArrayList;
 import java.util.HashMap;
-import java.util.Iterator;
 import java.util.Map;
 
 import pl.edu.mimuw.cloudatlas.agent.modules.Module;
 import pl.edu.mimuw.cloudatlas.agent.modules.ModuleType;
+import pl.edu.mimuw.cloudatlas.agent.modules.Qurnik;
 import pl.edu.mimuw.cloudatlas.agent.modules.RMI;
+import pl.edu.mimuw.cloudatlas.agent.modules.Stanik;
 import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduler;
 
 public class Agent {
@@ -16,6 +17,8 @@ public class Agent {
         HashMap<ModuleType, Module> modules = new HashMap<ModuleType, Module>();
         modules.put(ModuleType.TIMER_SCHEDULER, new TimerScheduler(ModuleType.TIMER_SCHEDULER));
         modules.put(ModuleType.RMI, new RMI(ModuleType.RMI));
+        modules.put(ModuleType.STATE, new Stanik());
+        modules.put(ModuleType.QUERY, new Qurnik());
         // TODO add modules as we implement them
         return modules;
     }
@@ -23,11 +26,8 @@ public class Agent {
     public static HashMap<ModuleType, Executor> initializeExecutors(
             HashMap<ModuleType, Module> modules) {
         HashMap<ModuleType, Executor> executors = new HashMap<ModuleType, Executor>();
-        Iterator it = modules.entrySet().iterator();
 
-        while (it.hasNext()) {
-            Map.Entry<ModuleType, Module> moduleEntry =
-                    (Map.Entry<ModuleType, Module>) it.next();
+        for (Map.Entry<ModuleType, Module> moduleEntry : modules.entrySet()) {
             Module module = moduleEntry.getValue();
             Executor executor = new Executor(module);
             executors.put(moduleEntry.getKey(), executor);
@@ -38,11 +38,8 @@ public class Agent {
 
     public static ArrayList<Thread>  initializeExecutorThreads(HashMap<ModuleType, Executor> executors) {
         ArrayList<Thread> executorThreads = new ArrayList<Thread>();
-        Iterator it = executors.entrySet().iterator();
 
-        while (it.hasNext()) {
-            Map.Entry<ModuleType, Executor> executorEntry =
-                    (Map.Entry<ModuleType, Executor>) it.next();
+        for (Map.Entry<ModuleType, Executor> executorEntry : executors.entrySet()) {
             Thread thread = new Thread(executorEntry.getValue());
             thread.setDaemon(true);
             System.out.println("Initializing executor " + executorEntry.getKey());