2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.netconf.test.tool.client.stress;
10 import java.net.InetSocketAddress;
11 import java.util.List;
12 import java.util.concurrent.Callable;
13 import java.util.concurrent.ExecutionException;
14 import org.opendaylight.netconf.api.messages.NetconfHelloMessageAdditionalHeader;
15 import org.opendaylight.netconf.api.messages.NetconfMessage;
16 import org.opendaylight.netconf.client.NetconfClientDispatcherImpl;
17 import org.opendaylight.netconf.client.NetconfClientSession;
18 import org.opendaylight.netconf.client.conf.NetconfClientConfiguration;
19 import org.opendaylight.netconf.client.conf.NetconfClientConfigurationBuilder;
20 import org.opendaylight.netconf.client.mdsal.NetconfDeviceCommunicator;
21 import org.opendaylight.netconf.client.mdsal.api.RemoteDevice;
22 import org.opendaylight.netconf.client.mdsal.api.RemoteDeviceId;
23 import org.opendaylight.netconf.nettyutil.handler.ssh.authentication.LoginPasswordHandler;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
27 public class StressClientCallable implements Callable<Boolean> {
29 private static final Logger LOG = LoggerFactory.getLogger(StressClientCallable.class);
31 private final Parameters params;
32 private final NetconfDeviceCommunicator sessionListener;
33 private final NetconfClientDispatcherImpl netconfClientDispatcher;
34 private final NetconfClientConfiguration cfg;
35 private final NetconfClientSession netconfClientSession;
36 private final ExecutionStrategy executionStrategy;
38 public StressClientCallable(final Parameters params,
39 final NetconfClientDispatcherImpl netconfClientDispatcher,
40 final List<NetconfMessage> preparedMessages) {
42 sessionListener = getSessionListener(params.getInetAddress(), params.concurrentMessageLimit);
43 this.netconfClientDispatcher = netconfClientDispatcher;
44 cfg = getNetconfClientConfiguration(this.params, sessionListener);
46 LOG.info("Connecting to netconf server {}:{}", params.ip, params.port);
48 netconfClientSession = netconfClientDispatcher.createClient(cfg).get();
49 } catch (final InterruptedException e) {
50 throw new IllegalStateException(e);
51 } catch (final ExecutionException e) {
52 throw new IllegalStateException("Unable to connect", e);
54 executionStrategy = getExecutionStrategy(params, preparedMessages, sessionListener);
58 public Boolean call() {
59 executionStrategy.invoke();
60 netconfClientSession.close();
64 private static ExecutionStrategy getExecutionStrategy(final Parameters params,
65 final List<NetconfMessage> preparedMessages, final NetconfDeviceCommunicator sessionListener) {
67 return new AsyncExecutionStrategy(params, preparedMessages, sessionListener);
69 return new SyncExecutionStrategy(params, preparedMessages, sessionListener);
73 private static NetconfDeviceCommunicator getSessionListener(
74 final InetSocketAddress inetAddress, final int messageLimit) {
75 final RemoteDevice<NetconfDeviceCommunicator> loggingRemoteDevice = new StressClient.LoggingRemoteDevice();
76 return new NetconfDeviceCommunicator(
77 new RemoteDeviceId("secure-test", inetAddress), loggingRemoteDevice, messageLimit);
80 private static NetconfClientConfiguration getNetconfClientConfiguration(final Parameters params,
81 final NetconfDeviceCommunicator sessionListener) {
82 final var netconfClientConfigurationBuilder = NetconfClientConfigurationBuilder.create()
83 .withSessionListener(sessionListener)
84 .withAddress(params.getInetAddress())
85 .withProtocol(params.ssh ? NetconfClientConfiguration.NetconfClientProtocol.SSH
86 : NetconfClientConfiguration.NetconfClientProtocol.TCP)
87 .withAuthHandler(new LoginPasswordHandler(params.username, params.password))
88 .withConnectionTimeoutMillis(20000L);
90 if (params.tcpHeader != null) {
91 final String header = params.tcpHeader.replace("\"", "").trim() + "\n";
92 netconfClientConfigurationBuilder.withAdditionalHeader(
93 new NetconfHelloMessageAdditionalHeader(null, null, null, null, null) {
95 public String toFormattedString() {
96 LOG.debug("Sending TCP header {}", header);
101 return netconfClientConfigurationBuilder.build();