BUG-5790: BGP Test tool

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

diff --git a/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMock.java b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMock.java
index fc0ec5d2271b4c112f782797ecd31f859be5179e..1efab2d56f6ed1e72d4e30c4eec404b4c44e43bc 100644 (file)
--- a/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMock.java
+++ b/bgp/bmp-mock/src/main/java/org/opendaylight/protocol/bmp/mock/BmpMock.java
@@ -8,9 +8,6 @@
 
 package org.opendaylight.protocol.bmp.mock;
 
-import ch.qos.logback.classic.LoggerContext;
-import com.google.common.base.Predicate;
-import com.google.common.collect.Iterables;
 import com.google.common.net.InetAddresses;
 import io.netty.channel.Channel;
 import java.net.InetAddress;
@@ -23,6 +20,7 @@ import org.opendaylight.protocol.bmp.impl.BmpActivator;
 import org.opendaylight.protocol.bmp.spi.registry.BmpExtensionProviderActivator;
 import org.opendaylight.protocol.bmp.spi.registry.BmpExtensionProviderContext;
 import org.opendaylight.protocol.bmp.spi.registry.SimpleBmpExtensionProviderContext;
+import org.opendaylight.protocol.util.LoggerUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -33,10 +31,12 @@ public final class BmpMock {
     private BmpMock() {
         throw new UnsupportedOperationException();
     }
+
     public static void main(final String[] args) {
         LOG.info("Starting BMP test tool.");
         final BmpMockArguments arguments = BmpMockArguments.parseArguments(args);
-        initiateLogger(arguments);
+        LoggerUtil.initiateLogger(arguments);
+
         final BmpMockDispatcher dispatcher = initiateMock(arguments);
         // now start the server / client
         if (arguments.isOnPassiveMode()) {
@@ -46,24 +46,21 @@ public final class BmpMock {
         }
     }
 
-    private static void initiateLogger(final BmpMockArguments arguments) {
-        getRootLogger((LoggerContext) LoggerFactory.getILoggerFactory()).setLevel(arguments.getLogLevel());
-    }
 
     private static BmpMockDispatcher initiateMock(final BmpMockArguments arguments) {
         final BmpExtensionProviderContext ctx = new SimpleBmpExtensionProviderContext();
         final BmpExtensionProviderActivator bmpActivator = new BmpActivator(
-                ServiceLoaderBGPExtensionProviderContext.getSingletonInstance());
+            ServiceLoaderBGPExtensionProviderContext.getSingletonInstance());
         bmpActivator.start(ctx);
 
         return new BmpMockDispatcher(ctx.getBmpMessageRegistry(),
-                new BmpSessionFactory() {
-                    @Override
-                    public BmpSession getSession(final Channel channel,
-                            final BmpSessionListenerFactory sessionListenerFactory) {
-                        return new BmpMockSession(arguments.getPeersCount(), arguments.getPrePolicyRoutesCount(), arguments.getPostPolicyRoutesCount());
-                    }
-                });
+            new BmpSessionFactory() {
+                @Override
+                public BmpSession getSession(final Channel channel,
+                    final BmpSessionListenerFactory sessionListenerFactory) {
+                    return new BmpMockSession(arguments.getPeersCount(), arguments.getPrePolicyRoutesCount(), arguments.getPostPolicyRoutesCount());
+                }
+            });
     }
 
     private static void deployClients(final BmpMockDispatcher dispatcher, final BmpMockArguments arguments) {
@@ -86,13 +83,4 @@ public final class BmpMock {
             currentLocal = InetAddresses.increment(currentLocal);
         }
     }
-
-    private static ch.qos.logback.classic.Logger getRootLogger(final LoggerContext lc) {
-        return Iterables.find(lc.getLoggerList(), new Predicate<Logger>() {
-            @Override
-            public boolean apply(final Logger input) {
-                return (input != null) ? input.getName().equals(Logger.ROOT_LOGGER_NAME) : false;
-            }
-        });
-    }
 }