bgp/cli/src/main/java/org/opendaylight/protocol/bgp/cli/AbstractStatsCommandProvider.java

   1 /*
   2  * Copyright © 2016 Inocybe Technologies and others.  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
   9 package org.opendaylight.protocol.bgp.cli;
  10
  11 import java.io.IOException;
  12 import java.io.PrintStream;
  13 import java.lang.management.ManagementFactory;
  14 import java.util.Set;
  15 import javax.annotation.Nonnull;
  16 import javax.management.JMX;
  17 import javax.management.MBeanServer;
  18 import javax.management.ObjectName;
  19 import org.apache.karaf.shell.console.OsgiCommandSupport;
  20 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
  21 import org.opendaylight.controller.config.yang.bgp.rib.impl.BGPPeerModuleFactory;
  22 import org.opendaylight.controller.config.yang.bgp.rib.impl.BGPPeerRuntimeMXBean;
  23
  24 public abstract class AbstractStatsCommandProvider extends OsgiCommandSupport {
  25     private static final ObjectName BGP_PEER_MODULE_PATTERN = ObjectNameUtil.createRuntimeBeanPattern(BGPPeerModuleFactory.NAME, null);
  26
  27     private static final MBeanServer MBEAN_SERVER = ManagementFactory.getPlatformMBeanServer();
  28
  29     @Override
  30     protected final Object doExecute() throws Exception {
  31         final PrintStream out = this.session.getConsole();
  32         getObjectNames().forEach(objectName -> {
  33             onExecution(out, getRuntimeMXBean(objectName), objectName);
  34         });
  35         return null;
  36     }
  37
  38     protected abstract void onExecution(@Nonnull PrintStream out, @Nonnull BGPPeerRuntimeMXBean peerRuntimeMXBean, @Nonnull ObjectName objectName);
  39
  40     private static Set<ObjectName> getObjectNames() throws IOException {
  41         return MBEAN_SERVER.queryNames(BGP_PEER_MODULE_PATTERN, null);
  42     }
  43
  44     private static final BGPPeerRuntimeMXBean getRuntimeMXBean(final ObjectName objectName) {
  45         return JMX.newMXBeanProxy(MBEAN_SERVER, objectName, BGPPeerRuntimeMXBean.class);
  46     }
  47 }