bmp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMockSessionListener.java

   1 /*
   2  * Copyright (c) 2017 AT&T Intellectual Property. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.protocol.bmp.mock;
   9
  10 import java.util.concurrent.atomic.AtomicBoolean;
  11 import java.util.concurrent.atomic.LongAdder;
  12 import org.opendaylight.protocol.bmp.api.BmpSession;
  13 import org.opendaylight.protocol.bmp.api.BmpSessionListener;
  14 import org.opendaylight.yangtools.yang.binding.Notification;
  15
  16 public final class BmpMockSessionListener implements BmpSessionListener {
  17     private final LongAdder counter = new LongAdder();
  18     private final AtomicBoolean up = new AtomicBoolean(false);
  19
  20     @Override
  21     public void onSessionUp(final BmpSession session) {
  22         up.set(true);
  23     }
  24
  25     @Override
  26     public void onSessionDown(final Exception exception) {
  27         up.set(false);
  28     }
  29
  30     @Override
  31     public void onMessage(final Notification<?> message) {
  32         counter.increment();
  33     }
  34
  35     public boolean getStatus() {
  36         return up.get();
  37     }
  38
  39     public long getNumberOfMessagesReceived() {
  40         return counter.longValue();
  41     }
  42 }