Move releaseConnection from ReusableBGPPeer to BGPSessionListener
[bgpcep.git] / bgp / testtool / src / main / java / org / opendaylight / protocol / bgp / testtool / Main.java
index dae066981f77f4a67081916104c1c6c287494cc7..e593770b0971661c953125b7f9a209e39e32d3f2 100644 (file)
@@ -8,24 +8,23 @@
 package org.opendaylight.protocol.bgp.testtool;
 
 import io.netty.channel.nio.NioEventLoopGroup;
-import io.netty.util.HashedWheelTimer;
 import io.netty.util.concurrent.GlobalEventExecutor;
-
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
+import java.net.UnknownHostException;
 import java.util.HashMap;
 import java.util.Map;
-
-import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
 import org.opendaylight.protocol.bgp.rib.impl.BGPDispatcherImpl;
 import org.opendaylight.protocol.bgp.rib.impl.BGPSessionProposalImpl;
+import org.opendaylight.protocol.bgp.rib.impl.StrictBGPPeerRegistry;
 import org.opendaylight.protocol.bgp.rib.impl.spi.BGPSessionPreferences;
+import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
 import org.opendaylight.protocol.framework.NeverReconnectStrategy;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev131125.LinkstateAddressFamily;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev131125.LinkstateSubsequentAddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateAddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateSubsequentAddressFamily;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
@@ -38,79 +37,81 @@ import org.slf4j.LoggerFactory;
  */
 public final class Main {
 
-       private static final Logger LOG = LoggerFactory.getLogger(Main.class);
+    private static final Logger LOG = LoggerFactory.getLogger(Main.class);
+
+    private static final String USAGE = "DESCRIPTION:\n"
+        + "\tCreates a server with given parameters. As long as it runs, it accepts connections " + "from PCCs.\n" + "USAGE:\n"
+        + "\t-a, --address\n" + "\t\tthe ip address to which is this server bound.\n"
+        + "\t\tFormat: x.x.x.x:y where y is port number.\n\n"
+        + "\t\tThis IP address will appear in BGP Open message as BGP Identifier of the server.\n" +
 
-       private static final String USAGE = "DESCRIPTION:\n"
-                       + "\tCreates a server with given parameters. As long as it runs, it accepts connections " + "from PCCs.\n" + "USAGE:\n"
-                       + "\t-a, --address\n" + "\t\tthe ip address to which is this server bound.\n"
-                       + "\t\tFormat: x.x.x.x:y where y is port number.\n\n"
-                       + "\t\tThis IP address will appear in BGP Open message as BGP Identifier of the server.\n" +
+            "\t-as\n" + "\t\t value of AS in the initial open message\n\n" +
 
-                       "\t-as\n" + "\t\t value of AS in the initial open message\n\n" +
+            "\t-h, --holdtimer\n" + "\t\tin seconds, value of the desired holdtimer\n"
+            + "\t\tAccording to RFC4271, recommended value for deadtimer is 90 seconds.\n"
+            + "\t\tIf not set, this will be the default value.\n\n" +
 
-                       "\t-h, --holdtimer\n" + "\t\tin seconds, value of the desired holdtimer\n"
-                       + "\t\tAccording to RFC4271, recommended value for deadtimer is 90 seconds.\n"
-                       + "\t\tIf not set, this will be the default value.\n\n" +
+            "\t--help\n" + "\t\tdisplay this help and exits\n\n" +
 
-                       "\t--help\n" + "\t\tdisplay this help and exits\n\n" +
+            "With no parameters, this help is printed.";
 
-                       "With no parameters, this help is printed.";
+    private final BGPDispatcherImpl dispatcher;
 
-       private final BGPDispatcherImpl dispatcher;
+    private static final int INITIAL_HOLD_TIME = 90;
 
-       private static final int INITIAL_HOLD_TIME = 90;
+    private static final int RECONNECT_MILLIS = 5000;
 
-       private static final int RECONNECT_MILLIS = 5000;
+    private Main() {
+        this.dispatcher = new BGPDispatcherImpl(ServiceLoaderBGPExtensionProviderContext.getSingletonInstance().getMessageRegistry(), new NioEventLoopGroup(), new NioEventLoopGroup());
+    }
 
-       private Main() throws Exception {
-               this.dispatcher = new BGPDispatcherImpl(ServiceLoaderBGPExtensionProviderContext.createConsumerContext().getMessageRegistry(),
-                               new HashedWheelTimer(), new NioEventLoopGroup(), new NioEventLoopGroup());
-       }
+    public static void main(final String[] args) throws UnknownHostException {
+        if (args.length == 0 || (args.length == 1 && args[0].equalsIgnoreCase("--help"))) {
+            LOG.info(Main.USAGE);
+            return;
+        }
 
-       public static void main(final String[] args) throws Exception {
-               if (args.length == 0 || (args.length == 1 && args[0].equalsIgnoreCase("--help"))) {
-                       System.err.println(Main.USAGE);
-                       return;
-               }
+        InetSocketAddress address = null;
+        short holdTimerValue = INITIAL_HOLD_TIME;
+        AsNumber as = null;
 
-               InetSocketAddress address = null;
-               short holdTimerValue = INITIAL_HOLD_TIME;
-               AsNumber as = null;
+        int i = 0;
+        while (i < args.length) {
+            if (args[i].equalsIgnoreCase("-a") || args[i].equalsIgnoreCase("--address")) {
+                final String[] ip = args[i + 1].split(":");
+                address = new InetSocketAddress(InetAddress.getByName(ip[0]), Integer.parseInt(ip[1]));
+                i++;
+            } else if (args[i].equalsIgnoreCase("-h") || args[i].equalsIgnoreCase("--holdtimer")) {
+                holdTimerValue = Short.valueOf(args[i + 1]);
+                i++;
+            } else if (args[i].equalsIgnoreCase("-as")) {
+                as = new AsNumber(Long.valueOf(args[i + 1]));
+                i++;
+            } else {
+                LOG.error("WARNING: Unrecognized argument: {}", args[i]);
+            }
+            i++;
+        }
 
-               int i = 0;
-               while (i < args.length) {
-                       if (args[i].equalsIgnoreCase("-a") || args[i].equalsIgnoreCase("--address")) {
-                               final String[] ip = args[i + 1].split(":");
-                               address = new InetSocketAddress(InetAddress.getByName(ip[0]), Integer.valueOf(ip[1]));
-                               i++;
-                       } else if (args[i].equalsIgnoreCase("-h") || args[i].equalsIgnoreCase("--holdtimer")) {
-                               holdTimerValue = Short.valueOf(args[i + 1]);
-                               i++;
-                       } else if (args[i].equalsIgnoreCase("-as")) {
-                               as = new AsNumber(Long.valueOf(args[i + 1]));
-                               i++;
-                       } else {
-                               LOG.error("WARNING: Unrecognized argument: " + args[i]);
-                       }
-                       i++;
-               }
+        final Main m = new Main();
 
-               final Main m = new Main();
+        final BGPSessionListener sessionListener = new TestingListener();
 
-               final BGPSessionListener sessionListener = new TestingListener();
+        final Map<Class<? extends AddressFamily>, Class<? extends SubsequentAddressFamily>> tables = new HashMap<>();
+        tables.put(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
+        tables.put(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class);
 
-               final Map<Class<? extends AddressFamily>, Class<? extends SubsequentAddressFamily>> tables = new HashMap<>();
-               tables.put(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
-               tables.put(LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class);
+        final BGPSessionProposalImpl prop = new BGPSessionProposalImpl(holdTimerValue, as, new Ipv4Address("25.25.25.2"), tables, as);
 
-               final BGPSessionProposalImpl prop = new BGPSessionProposalImpl(holdTimerValue, as, new Ipv4Address("25.25.25.2"), tables);
+        final BGPSessionPreferences proposal = prop.getProposal();
 
-               final BGPSessionPreferences proposal = prop.getProposal();
+        LOG.debug("{} {} {}", address, sessionListener, proposal);
 
-               LOG.debug("{} {} {}", address, sessionListener, proposal);
+        final InetSocketAddress addr = address;
+        final StrictBGPPeerRegistry strictBGPPeerRegistry = new StrictBGPPeerRegistry();
+        strictBGPPeerRegistry.addPeer(StrictBGPPeerRegistry.getIpAddress(address), sessionListener, proposal);
 
-               final InetSocketAddress addr = address;
-               m.dispatcher.createClient(addr, proposal, as, sessionListener,
-                               new NeverReconnectStrategy(GlobalEventExecutor.INSTANCE, RECONNECT_MILLIS));
-       }
+        m.dispatcher.createClient(addr, strictBGPPeerRegistry,
+            new NeverReconnectStrategy(GlobalEventExecutor.INSTANCE, RECONNECT_MILLIS));
+    }
 }