Merge changes I34a6851c,I046e6272,I169f6b40,Iec24bf37,I89933b0c

[openflowjava.git] / simple-client / src / main / java / org / opendaylight / openflowjava / protocol / impl / clients / ScenarioServiceImpl.java

diff --git a/simple-client/src/main/java/org/opendaylight/openflowjava/protocol/impl/clients/ScenarioServiceImpl.java b/simple-client/src/main/java/org/opendaylight/openflowjava/protocol/impl/clients/ScenarioServiceImpl.java
new file mode 100644 (file)
index 0000000..9e1df90
--- /dev/null
+++ b/simple-client/src/main/java/org/opendaylight/openflowjava/protocol/impl/clients/ScenarioServiceImpl.java
@@ -0,0 +1,81 @@
+package org.opendaylight.openflowjava.protocol.impl.clients;
+
+import com.google.common.base.Preconditions;
+import java.io.File;
+import java.io.IOException;
+import java.util.SortedMap;
+import java.util.TreeMap;
+import javax.xml.XMLConstants;
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+import javax.xml.bind.Unmarshaller;
+import javax.xml.validation.Schema;
+import javax.xml.validation.SchemaFactory;
+import org.opendaylight.openflowjava.util.ByteBufUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.xml.sax.SAXException;
+
+/**
+ * @author Jozef Bacigal
+ *         Date: 9.3.2016
+ */
+public class ScenarioServiceImpl implements ScenarioService {
+
+    private static final Logger LOG = LoggerFactory.getLogger(ScenarioServiceImpl.class);
+
+    private String XML_FILE_PATH_WITH_FILE_NAME = SIMPLE_CLIENT_SRC_MAIN_RESOURCES + SCENARIO_XML;
+
+    public ScenarioServiceImpl(String scenarioFile){
+        if (null != scenarioFile && !scenarioFile.isEmpty()) {
+            this.XML_FILE_PATH_WITH_FILE_NAME = scenarioFile;
+        }
+    }
+
+    @Override
+    public Scenario unMarshallData(String scenarioName) throws SAXException, JAXBException {
+        SchemaFactory sf = SchemaFactory.newInstance(XMLConstants.W3C_XML_SCHEMA_NS_URI);
+        Schema schema = sf.newSchema(new File(XSD_SCHEMA_PATH_WITH_FILE_NAME));
+        LOG.debug("Loading schema from: {}", XSD_SCHEMA_PATH_WITH_FILE_NAME);
+
+        JAXBContext jc = JAXBContext.newInstance(Scenarios.class);
+
+        Unmarshaller unmarshaller = jc.createUnmarshaller();
+        unmarshaller.setSchema(schema);
+
+        Scenarios scenarios = (Scenarios) unmarshaller.unmarshal(new File(XML_FILE_PATH_WITH_FILE_NAME));
+        LOG.debug("Scenarios ({}) are un-marshaled from {}", scenarios.getScenario().size(), XML_FILE_PATH_WITH_FILE_NAME);
+
+        boolean foundConfiguration = false;
+        Scenario scenarioType = null;
+        for (Scenario scenario : scenarios.getScenario()) {
+            if (scenario.getName().equals(scenarioName)) {
+                scenarioType = scenario;
+                foundConfiguration = true;
+            }
+        }
+        if (!foundConfiguration) {
+            LOG.warn("Scenario {} not found.", scenarioName);
+        } else {
+            LOG.info("Scenario {} found with {} steps.", scenarioName, scenarioType.getStep().size());
+        }
+        return scenarioType;
+    }
+
+    @Override
+    public SortedMap<Integer, ClientEvent> getEventsFromScenario(Scenario scenario) throws IOException {
+        Preconditions.checkNotNull(scenario, "Scenario name not found. Check XML file, scenario name or directories.");
+        SortedMap<Integer, ClientEvent> events = new TreeMap<>();
+        Integer counter = 0;
+        for (Step step : scenario.getStep()) {
+            LOG.debug("Step {}: {}, type {}, bytes {}", step.getOrder(), step.getName(), step.getEvent().value(), step.getBytes().toArray());
+            switch (step.getEvent()) {
+                case SLEEP_EVENT: events.put(counter++, new SleepEvent(1000)); break;
+                case SEND_EVENT: events.put(counter++, new SendEvent(ByteBufUtils.serializeList(step.getBytes()))); break;
+                case WAIT_FOR_MESSAGE_EVENT: events.put(counter++, new WaitForMessageEvent(ByteBufUtils.serializeList(step.getBytes()))); break;
+            }
+        }
+        return events;
+    }
+
+}