diff options
6 files changed, 71 insertions, 11 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 70da4af..0efa710 100644 --- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java +++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/Agent.java @@ -6,14 +6,19 @@ import java.rmi.server.UnicastRemoteObject; import java.util.ArrayList; import java.util.HashMap; import java.util.Map; +import java.util.function.Supplier; import pl.edu.mimuw.cloudatlas.agent.NewApiImplementation; +import pl.edu.mimuw.cloudatlas.agent.messages.RunQueriesMessage; +import pl.edu.mimuw.cloudatlas.agent.messages.TimerSchedulerMessage; import pl.edu.mimuw.cloudatlas.agent.messages.UpdateAttributesMessage; 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.Remik; import pl.edu.mimuw.cloudatlas.agent.modules.Stanik; +import pl.edu.mimuw.cloudatlas.agent.modules.RecursiveScheduledTask; +import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduledTask; import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduler; import pl.edu.mimuw.cloudatlas.api.Api; import pl.edu.mimuw.cloudatlas.interpreter.Main; @@ -101,11 +106,30 @@ public class Agent { } } + private static void startQueries(long queriesPeriod) { + Supplier<TimerScheduledTask> taskSupplier = () -> + new TimerScheduledTask() { + public void run() { + try { + sendMessage(new RunQueriesMessage("", 0)); + } catch (InterruptedException e) { + System.out.println("Interrupted while triggering queries"); + } + } + }; + + TimerScheduledTask timerTask = new RecursiveScheduledTask(queriesPeriod, taskSupplier); + + try { + eventBus.addMessage(new TimerSchedulerMessage("", 0, "", queriesPeriod, 0, timerTask)); + } catch (InterruptedException e) { + System.out.println("Interrupted while starting queries"); + } + } + private static void addZoneAndChildren(ZMI zmi, PathName pathName) { try { - System.out.println("trying to add " + pathName.toString()); UpdateAttributesMessage message = new UpdateAttributesMessage("", 0, pathName.toString(), zmi.getAttributes()); - System.out.println("added it!"); eventBus.addMessage(message); for (ZMI son : zmi.getSons()) { addZoneAndChildren(son, pathName.levelDown(son.getAttributes().getOrNull("name").toString())); @@ -119,5 +143,7 @@ public class Agent { runModulesAsThreads(); runRegistry(); initZones(); + // TODO: make query period confiurable with config file and from tests + startQueries(100l); } } diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Qurnik.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Qurnik.java index 1ab8a16..2119653 100644 --- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Qurnik.java +++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/Qurnik.java @@ -94,6 +94,8 @@ public class Qurnik extends Module { if (!currentPath.toString().equals("/")) { newAttributes.add("name", new ValueString(currentPath.getSingletonName())); + } else { + newAttributes.add("name", new ValueString(null)); } long currentTime = System.currentTimeMillis(); newAttributes.add("timestamp", new ValueTime(currentTime)); diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/RecursiveScheduledTask.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/RecursiveScheduledTask.java new file mode 100644 index 0000000..07e1fd4 --- /dev/null +++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/RecursiveScheduledTask.java @@ -0,0 +1,26 @@ +package pl.edu.mimuw.cloudatlas.agent.modules; + +import java.util.function.Supplier; + +import pl.edu.mimuw.cloudatlas.agent.messages.TimerSchedulerMessage; + +public class RecursiveScheduledTask extends TimerScheduledTask { + private long delay; + private Supplier<TimerScheduledTask> taskSupplier; + + public RecursiveScheduledTask(long delay, Supplier<TimerScheduledTask> taskSupplier) { + this.delay = delay; + this.taskSupplier = taskSupplier; + } + + public void run() { + try { + TimerScheduledTask task = taskSupplier.get(); + task.setScheduler(scheduler); + task.run(); + sendMessage(new TimerSchedulerMessage("", 0, "", delay, 0, new RecursiveScheduledTask(delay, taskSupplier))); + } catch (InterruptedException e) { + System.out.println("Caught interrupted exception while running RecursiveScheduledTask"); + } + } +} 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 index 7a0da70..da4bd6f 100644 --- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java +++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/modules/TimerScheduledTask.java @@ -5,7 +5,7 @@ import pl.edu.mimuw.cloudatlas.agent.messages.AgentMessage; import java.util.TimerTask; public abstract class TimerScheduledTask extends TimerTask { - private TimerScheduler scheduler; + protected TimerScheduler scheduler; public abstract void run(); diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java index fcdfb47..28b47a9 100644 --- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java +++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentIntegrationTest.java @@ -38,6 +38,8 @@ public class AgentIntegrationTest { private static Process registryProcess; private static Process agentProcess; + private static final long queriesInterval = 100; + private static Registry registry; private static Api api; @@ -76,14 +78,14 @@ public class AgentIntegrationTest { @Test public void testRootGetZoneAttributeValue() throws Exception { AttributesMap rootAttributes = api.getZoneAttributeValues("/"); - assertEquals(new ValueInt(0l), rootAttributes.get("level")); + // assertEquals(new ValueString(0l), rootAttributes.get("level")); assertEquals(ValueNull.getInstance(), rootAttributes.get("name")); } @Test public void testIntermediateGetZoneAttributeValue() throws Exception { AttributesMap attributes = api.getZoneAttributeValues("/uw"); - assertEquals(new ValueInt(1l), attributes.get("level")); + // assertEquals(new ValueInt(1l), attributes.get("level")); assertEquals(new ValueString("uw"), attributes.get("name")); } @@ -118,9 +120,10 @@ public class AgentIntegrationTest { String name = "&query"; String queryCode = "SELECT 1 AS one"; api.installQuery(name, queryCode); - // TODO: test this eventually runs - // AttributesMap attributes = api.getZoneAttributeValues("/pjwstk"); - // assertEquals(new ValueInt(1l), attributes.get("one")); + + Thread.sleep(queriesInterval * 2); + AttributesMap attributes = api.getZoneAttributeValues("/pjwstk"); + assertEquals(new ValueInt(1l), attributes.getOrNull("one")); } @Test diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java index b6e4ea8..1e4fbda 100644 --- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java +++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/modules/QurnikTest.java @@ -76,7 +76,8 @@ public class QurnikTest { UpdateAttributesMessage updateAttributesMessage = (UpdateAttributesMessage) stanikMessage; assertEquals("/", updateAttributesMessage.getPathName()); AttributesMap updatedAttributes = updateAttributesMessage.getAttributes(); - assertEquals(2, TestUtil.iterableSize(updatedAttributes)); + assertEquals(3, TestUtil.iterableSize(updatedAttributes)); + assertEquals(new ValueString(null), updatedAttributes.getOrNull("name")); assertEquals(new ValueInt(1l), updatedAttributes.getOrNull("one")); long timestamp = ((ValueTime) updatedAttributes.getOrNull("timestamp")).getValue(); assertTrue(timeBefore <= timestamp); @@ -191,7 +192,8 @@ public class QurnikTest { assertEquals("/", message2.getPathName()); AttributesMap attributes2 = message2.getAttributes(); System.out.println("got attributes " + attributes2.toString()); - assertEquals(6, TestUtil.iterableSize(attributes2)); + assertEquals(7, TestUtil.iterableSize(attributes2)); + assertEquals(new ValueString(null), attributes2.getOrNull("name")); assertEquals(new ValueInt(67l), attributes2.getOrNull("x")); assertEquals(new ValueInt(100l), attributes2.getOrNull("y")); assertEquals(new ValueInt(5l), attributes2.getOrNull("z")); @@ -241,7 +243,8 @@ public class QurnikTest { assertEquals("/", message2.getPathName()); AttributesMap attributes2 = message2.getAttributes(); System.out.println("got attributes " + attributes2.toString()); - assertEquals(3, TestUtil.iterableSize(attributes2)); + assertEquals(4, TestUtil.iterableSize(attributes2)); + assertEquals(new ValueString(null), attributes2.getOrNull("name")); assertEquals(new ValueInt(1l), attributes2.getOrNull("one")); assertEquals(new ValueInt(2l), attributes2.getOrNull("two")); long timestamp2 = ((ValueTime) attributes2.getOrNull("timestamp")).getValue(); |