Added notification benchmark (ntfbenchmark) and rpc benchmark models

[controller.git] / benchmark / ntfbenchmark / src / main / java / ntfbenchmark / impl / NtfbenchTestListener.java

diff --git a/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchTestListener.java b/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchTestListener.java
new file mode 100644 (file)
index 0000000..e2c876d
--- /dev/null
+++ b/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchTestListener.java
@@ -0,0 +1,42 @@
+/*
+ * Copyright (c) 2015 Cisco Systems 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 ntfbenchmark.impl;
+
+import java.util.concurrent.Future;
+
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ntfbench.payload.rev150709.Ntfbench;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ntfbench.payload.rev150709.NtfbenchPayloadListener;
+
+import com.google.common.util.concurrent.Futures;
+
+public class NtfbenchTestListener implements NtfbenchPayloadListener {
+
+    private final int expectedSize;
+    private int received = 0;
+
+    public NtfbenchTestListener(final int expectedSize) {
+        this.expectedSize = expectedSize;
+    }
+
+    @Override
+    public void onNtfbench(final Ntfbench notification) {
+        if (expectedSize == notification.getPayload().size()) {
+            received++;
+        }
+    };
+
+    public int getReceived() {
+        return received;
+    }
+
+    public Future<?> getAllDone() {
+        return Futures.immediateFuture(null);
+    }
+
+}