X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fnetconf%2Fnetconf-testtool%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fnetconf%2Ftest%2Ftool%2Frpc%2FSimulatedCreateSubscription.java;fp=opendaylight%2Fnetconf%2Fnetconf-testtool%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fnetconf%2Ftest%2Ftool%2Frpc%2FSimulatedCreateSubscription.java;h=abf51c476462996e9e4c709626893412c3a1995c;hb=a6eaa85b523c7d96fdbb3f695dc3511d848bfef7;hp=0000000000000000000000000000000000000000;hpb=64376d224aee3e6469300f8cc2815af6c1fa0f44;p=controller.git

diff --git a/opendaylight/netconf/netconf-testtool/src/main/java/org/opendaylight/controller/netconf/test/tool/rpc/SimulatedCreateSubscription.java b/opendaylight/netconf/netconf-testtool/src/main/java/org/opendaylight/controller/netconf/test/tool/rpc/SimulatedCreateSubscription.java
new file mode 100644
index 0000000000..abf51c4764
--- /dev/null
+++ b/opendaylight/netconf/netconf-testtool/src/main/java/org/opendaylight/controller/netconf/test/tool/rpc/SimulatedCreateSubscription.java
@@ -0,0 +1,173 @@
+/*
+ * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.controller.netconf.test.tool.rpc;
+
+import com.google.common.base.Optional;
+import com.google.common.base.Preconditions;
+import java.io.File;
+import java.io.IOException;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBException;
+import javax.xml.bind.Unmarshaller;
+import javax.xml.bind.annotation.XmlRootElement;
+import org.opendaylight.controller.netconf.api.NetconfDocumentedException;
+import org.opendaylight.controller.netconf.api.NetconfMessage;
+import org.opendaylight.controller.netconf.api.xml.XmlNetconfConstants;
+import org.opendaylight.controller.netconf.impl.NetconfServerSession;
+import org.opendaylight.controller.netconf.impl.mapping.operations.DefaultNetconfOperation;
+import org.opendaylight.controller.netconf.util.mapping.AbstractLastNetconfOperation;
+import org.opendaylight.controller.netconf.util.xml.XmlElement;
+import org.opendaylight.controller.netconf.util.xml.XmlUtil;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.xml.sax.SAXException;
+
+public class SimulatedCreateSubscription extends AbstractLastNetconfOperation implements DefaultNetconfOperation {
+
+    private NetconfServerSession session;
+    private final Optional<Notifications> notifications;
+    private ScheduledExecutorService scheduledExecutorService;
+
+    public SimulatedCreateSubscription(final String id, final Optional<File> notificationsFile) {
+        super(id);
+        if(notificationsFile.isPresent()) {
+            notifications = Optional.of(loadNotifications(notificationsFile.get()));
+            scheduledExecutorService = Executors.newScheduledThreadPool(1);
+        } else {
+            notifications = Optional.absent();
+        }
+    }
+
+    private Notifications loadNotifications(final File file) {
+        try {
+            final JAXBContext jaxbContext = JAXBContext.newInstance(Notifications.class);
+            final Unmarshaller jaxbUnmarshaller = jaxbContext.createUnmarshaller();
+            return (Notifications) jaxbUnmarshaller.unmarshal(file);
+        } catch (final JAXBException e) {
+            throw new IllegalArgumentException("Canot parse file " + file + " as a notifications file", e);
+        }
+    }
+
+    @Override
+    protected String getOperationName() {
+        return "create-subscription";
+    }
+
+    @Override
+    protected String getOperationNamespace() {
+        return "urn:ietf:params:xml:ns:netconf:notification:1.0";
+    }
+
+    @Override
+    protected Element handleWithNoSubsequentOperations(final Document document, final XmlElement operationElement) throws NetconfDocumentedException {
+
+
+        if(notifications.isPresent()) {
+            long delayAggregator = 0;
+            System.console().writer().println("Scheduling notifications " + notifications.get());
+
+            for (final Notification notification : notifications.get().getNotificationList()) {
+                for (int i = 0; i <= notification.getTimes(); i++) {
+
+                    delayAggregator += notification.getDelayInSeconds();
+
+                    System.console().writer().println("Times " + notification.getTimes());
+                    scheduledExecutorService.schedule(new Runnable() {
+                        @Override
+                        public void run() {
+                            try {
+                                System.console().writer().println("Sending actual notification " + notification);
+                                Preconditions.checkState(session != null, "Session is not set, cannot process notifications");
+                                session.sendMessage(parseNetconfNotification(notification.getContent()));
+                            } catch (IOException | SAXException e) {
+                                throw new IllegalStateException("Unable to process notification " + notification, e);
+                            }
+                        }
+                    }, delayAggregator, TimeUnit.SECONDS);
+                }
+            }
+        }
+        return XmlUtil.createElement(document, XmlNetconfConstants.OK, Optional.<String>absent());
+    }
+
+    private static NetconfMessage parseNetconfNotification(String content) throws IOException, SAXException {
+        final int startEventTime = content.indexOf("<eventTime>") + "<eventTime>".length();
+        final int endEventTime = content.indexOf("</eventTime>");
+        final String eventTime = content.substring(startEventTime, endEventTime);
+        if(eventTime.equals("XXXX")) {
+            content = content.replace(eventTime, new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ssXXX").format(new Date()));
+        }
+
+        return new NetconfMessage(XmlUtil.readXmlToDocument(content));
+    }
+
+    @Override
+    public void setNetconfSession(final NetconfServerSession s) {
+        this.session = s;
+    }
+
+    @XmlRootElement(name = "notifications")
+    public static final class Notifications {
+
+        @javax.xml.bind.annotation.XmlElement(nillable =  false, name = "notification", required = true)
+        private List<Notification> notificationList;
+
+        public List<Notification> getNotificationList() {
+            return notificationList;
+        }
+
+        @Override
+        public String toString() {
+            final StringBuffer sb = new StringBuffer("Notifications{");
+            sb.append("notificationList=").append(notificationList);
+            sb.append('}');
+            return sb.toString();
+        }
+    }
+
+    public static final class Notification {
+
+        @javax.xml.bind.annotation.XmlElement(nillable = false, name = "delay")
+        private long delayInSeconds;
+
+        @javax.xml.bind.annotation.XmlElement(nillable = false, name = "times")
+        private long times;
+
+        @javax.xml.bind.annotation.XmlElement(nillable = false, name = "content", required = true)
+        private String content;
+
+        public long getDelayInSeconds() {
+            return delayInSeconds;
+        }
+
+        public long getTimes() {
+            return times;
+        }
+
+        public String getContent() {
+            return content;
+        }
+
+        @Override
+        public String toString() {
+            final StringBuffer sb = new StringBuffer("Notification{");
+            sb.append("delayInSeconds=").append(delayInSeconds);
+            sb.append(", times=").append(times);
+            sb.append(", content='").append(content).append('\'');
+            sb.append('}');
+            return sb.toString();
+        }
+    }
+}