Added notification benchmark (ntfbenchmark) and rpc benchmark models

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

diff --git a/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchWTCListener.java b/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchWTCListener.java
new file mode 100644 (file)
index 0000000..a9d44bf
--- /dev/null
+++ b/benchmark/ntfbenchmark/src/main/java/ntfbenchmark/impl/NtfbenchWTCListener.java
@@ -0,0 +1,36 @@
+/*
+ * 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 org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ntfbench.payload.rev150709.Ntfbench;
+
+import com.google.common.util.concurrent.SettableFuture;
+
+public class NtfbenchWTCListener extends NtfbenchTestListener {
+    private final int expectedCount;
+    private final SettableFuture<?> allDone = SettableFuture.create();
+
+    public NtfbenchWTCListener(int expectedSize, int expectedCount) {
+        super(expectedSize);
+        this.expectedCount = expectedCount;
+    }
+
+    @Override
+    public void onNtfbench(Ntfbench notification) {
+        // TODO Auto-generated method stub
+        super.onNtfbench(notification);
+        if (expectedCount == getReceived()) {
+            allDone.set(null);
+        }
+    }
+
+    public SettableFuture<?> getAllDone() {
+        return allDone;
+    }
+}