BGPCEP-737: Implement BMP client reconnection

[bgpcep.git] / bgp / bmp-mock / src / main / java / org / opendaylight / protocol / bmp / mock / BmpMockSessionListener.java

diff --git a/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockSessionListener.java b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockSessionListener.java
new file mode 100644 (file)
index 0000000..8e7fe80
--- /dev/null
+++ b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockSessionListener.java
@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2017 AT&T Intellectual Property. 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.protocol.bmp.mock;
+
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.LongAdder;
+import javax.annotation.concurrent.GuardedBy;
+import org.opendaylight.protocol.bmp.api.BmpSession;
+import org.opendaylight.protocol.bmp.api.BmpSessionListener;
+import org.opendaylight.yangtools.yang.binding.Notification;
+
+public final class BmpMockSessionListener implements BmpSessionListener {
+    private final LongAdder counter = new LongAdder();
+    @GuardedBy("this")
+    private AtomicBoolean up = new AtomicBoolean(false);
+
+    @Override
+    public void onSessionUp(final BmpSession session) {
+        this.up.set(true);
+    }
+
+    @Override
+    public void onSessionDown(final Exception exception) {
+        this.up.set(false);
+    }
+
+    @Override
+    public void onMessage(final Notification message) {
+        this.counter.increment();
+    }
+
+    public boolean getStatus() {
+        return this.up.get();
+    }
+
+    public long getNumberOfMessagesReceived() {
+        return this.counter.longValue();
+    }
+}