netconf/netconf-client/src/main/java/org/opendaylight/netconf/client/NetconfClientDispatcherImpl.java

   1 /*
   2  * Copyright (c) 2013 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.netconf.client;
  10
  11 import io.netty.channel.EventLoopGroup;
  12 import io.netty.util.Timer;
  13 import io.netty.util.concurrent.Future;
  14 import java.io.Closeable;
  15 import org.opendaylight.netconf.client.conf.NetconfClientConfiguration;
  16 import org.opendaylight.netconf.client.conf.NetconfReconnectingClientConfiguration;
  17 import org.opendaylight.protocol.framework.AbstractDispatcher;
  18 import org.slf4j.Logger;
  19 import org.slf4j.LoggerFactory;
  20
  21 public class NetconfClientDispatcherImpl extends AbstractDispatcher<NetconfClientSession, NetconfClientSessionListener>
  22         implements NetconfClientDispatcher, Closeable {
  23
  24     private static final Logger LOG = LoggerFactory.getLogger(NetconfClientDispatcherImpl.class);
  25
  26     private final Timer timer;
  27
  28     public NetconfClientDispatcherImpl(final EventLoopGroup bossGroup, final EventLoopGroup workerGroup, final Timer timer) {
  29         super(bossGroup, workerGroup);
  30         this.timer = timer;
  31     }
  32
  33     protected Timer getTimer() {
  34         return timer;
  35     }
  36
  37     @Override
  38     public Future<NetconfClientSession> createClient(final NetconfClientConfiguration clientConfiguration) {
  39         switch (clientConfiguration.getProtocol()) {
  40         case TCP:
  41             return createTcpClient(clientConfiguration);
  42         case SSH:
  43             return createSshClient(clientConfiguration);
  44         }
  45         throw new IllegalArgumentException("Unknown client protocol " + clientConfiguration.getProtocol());
  46     }
  47
  48     @Override
  49     public Future<Void> createReconnectingClient(final NetconfReconnectingClientConfiguration clientConfiguration) {
  50         switch (clientConfiguration.getProtocol()) {
  51         case TCP:
  52             return createReconnectingTcpClient(clientConfiguration);
  53         case SSH:
  54             return createReconnectingSshClient(clientConfiguration);
  55         }
  56         throw new IllegalArgumentException("Unknown client protocol " + clientConfiguration.getProtocol());
  57     }
  58
  59     private Future<NetconfClientSession> createTcpClient(final NetconfClientConfiguration currentConfiguration) {
  60         LOG.debug("Creating TCP client with configuration: {}", currentConfiguration);
  61         return super.createClient(currentConfiguration.getAddress(), currentConfiguration.getReconnectStrategy(),
  62                 (ch, promise) -> new TcpClientChannelInitializer(getNegotiatorFactory(currentConfiguration), currentConfiguration
  63                                 .getSessionListener()).initialize(ch, promise));
  64     }
  65
  66     private Future<Void> createReconnectingTcpClient(final NetconfReconnectingClientConfiguration currentConfiguration) {
  67         LOG.debug("Creating reconnecting TCP client with configuration: {}", currentConfiguration);
  68         final TcpClientChannelInitializer init = new TcpClientChannelInitializer(getNegotiatorFactory(currentConfiguration),
  69                 currentConfiguration.getSessionListener());
  70
  71         return super.createReconnectingClient(currentConfiguration.getAddress(), currentConfiguration.getConnectStrategyFactory(),
  72                 currentConfiguration.getReconnectStrategy(), init::initialize);
  73     }
  74
  75     private Future<NetconfClientSession> createSshClient(final NetconfClientConfiguration currentConfiguration) {
  76         LOG.debug("Creating SSH client with configuration: {}", currentConfiguration);
  77         return super.createClient(currentConfiguration.getAddress(), currentConfiguration.getReconnectStrategy(),
  78                 (ch, sessionPromise) -> new SshClientChannelInitializer(currentConfiguration.getAuthHandler(),
  79                         getNegotiatorFactory(currentConfiguration), currentConfiguration.getSessionListener())
  80                         .initialize(ch, sessionPromise));
  81     }
  82
  83     private Future<Void> createReconnectingSshClient(final NetconfReconnectingClientConfiguration currentConfiguration) {
  84         LOG.debug("Creating reconnecting SSH client with configuration: {}", currentConfiguration);
  85         final SshClientChannelInitializer init = new SshClientChannelInitializer(currentConfiguration.getAuthHandler(),
  86                 getNegotiatorFactory(currentConfiguration), currentConfiguration.getSessionListener());
  87
  88         return super.createReconnectingClient(currentConfiguration.getAddress(), currentConfiguration.getConnectStrategyFactory(), currentConfiguration.getReconnectStrategy(),
  89                 init::initialize);
  90     }
  91
  92     protected NetconfClientSessionNegotiatorFactory getNegotiatorFactory(final NetconfClientConfiguration cfg) {
  93         return new NetconfClientSessionNegotiatorFactory(timer, cfg.getAdditionalHeader(),
  94                 cfg.getConnectionTimeoutMillis());
  95     }
  96 }