opendaylight/netconf/tools/netconf-testtool/src/main/java/org/opendaylight/controller/netconf/test/tool/client/stress/ConfigurableClientDispatcher.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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.controller.netconf.test.tool.client.stress;
  10
  11 import io.netty.channel.EventLoopGroup;
  12 import io.netty.util.Timer;
  13 import java.util.Set;
  14 import org.opendaylight.controller.netconf.client.NetconfClientDispatcherImpl;
  15 import org.opendaylight.controller.netconf.client.NetconfClientSessionNegotiatorFactory;
  16 import org.opendaylight.controller.netconf.client.conf.NetconfClientConfiguration;
  17
  18 public class ConfigurableClientDispatcher extends NetconfClientDispatcherImpl {
  19
  20     private final Set<String> capabilities;
  21
  22     private ConfigurableClientDispatcher(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer, final Set<String> capabilities) {
  23         super(bossGroup, workerGroup, timer);
  24         this.capabilities = capabilities;
  25     }
  26
  27     /**
  28      * EXI + chunked framing
  29      */
  30     public static ConfigurableClientDispatcher createChunkedExi(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer) {
  31         return new ConfigurableClientDispatcher(bossGroup, workerGroup, timer, NetconfClientSessionNegotiatorFactory.EXI_CLIENT_CAPABILITIES);
  32     }
  33
  34     /**
  35      * EXI + ]]>]]> framing
  36      */
  37     public static ConfigurableClientDispatcher createLegacyExi(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer) {
  38         return new ConfigurableClientDispatcher(bossGroup, workerGroup, timer, NetconfClientSessionNegotiatorFactory.LEGACY_EXI_CLIENT_CAPABILITIES);
  39     }
  40
  41     /**
  42      * Chunked framing
  43      */
  44     public static ConfigurableClientDispatcher createChunked(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer) {
  45         return new ConfigurableClientDispatcher(bossGroup, workerGroup, timer, NetconfClientSessionNegotiatorFactory.DEFAULT_CLIENT_CAPABILITIES);
  46     }
  47
  48     /**
  49      * ]]>]]> framing
  50      */
  51     public static ConfigurableClientDispatcher createLegacy(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer) {
  52         return new ConfigurableClientDispatcher(bossGroup, workerGroup, timer, NetconfClientSessionNegotiatorFactory.LEGACY_FRAMING_CLIENT_CAPABILITIES);
  53     }
  54
  55     @Override
  56     protected NetconfClientSessionNegotiatorFactory getNegotiatorFactory(final NetconfClientConfiguration cfg) {
  57         return new NetconfClientSessionNegotiatorFactory(getTimer(), cfg.getAdditionalHeader(), cfg.getConnectionTimeoutMillis(), capabilities);
  58     }
  59 }