netconf/tools/netconf-testtool/src/main/java/org/opendaylight/netconf/test/tool/TesttoolNegotiationFactory.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.netconf.test.tool;
  10
  11 import io.netty.util.Timer;
  12 import java.net.SocketAddress;
  13 import java.util.HashMap;
  14 import java.util.Map;
  15 import java.util.Set;
  16 import org.opendaylight.netconf.api.monitoring.NetconfMonitoringService;
  17 import org.opendaylight.netconf.impl.NetconfServerSessionNegotiatorFactory;
  18 import org.opendaylight.netconf.impl.NetconfServerSessionNegotiatorFactoryBuilder;
  19 import org.opendaylight.netconf.impl.SessionIdProvider;
  20 import org.opendaylight.netconf.mapping.api.NetconfOperationService;
  21 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
  22
  23 public class TesttoolNegotiationFactory extends NetconfServerSessionNegotiatorFactory {
  24
  25     private final Map<SocketAddress, NetconfOperationService> cachedOperationServices = new HashMap<>();
  26
  27     public TesttoolNegotiationFactory(final Timer timer, final NetconfOperationServiceFactory netconfOperationProvider,
  28                                       final SessionIdProvider idProvider, final long connectionTimeoutMillis,
  29                                       final NetconfMonitoringService monitoringService) {
  30         super(timer, netconfOperationProvider, idProvider, connectionTimeoutMillis, monitoringService, NetconfServerSessionNegotiatorFactory.DEFAULT_BASE_CAPABILITIES);
  31     }
  32
  33     public TesttoolNegotiationFactory(final Timer timer, final NetconfOperationServiceFactory netconfOperationProvider,
  34                                       final SessionIdProvider idProvider, final long connectionTimeoutMillis,
  35                                       final NetconfMonitoringService monitoringService, final Set<String> baseCapabilities) {
  36         super(timer, netconfOperationProvider, idProvider, connectionTimeoutMillis, monitoringService, baseCapabilities);
  37     }
  38
  39     @Override
  40     protected NetconfOperationService getOperationServiceForAddress(final String netconfSessionIdForReporting, final SocketAddress socketAddress) {
  41         if (cachedOperationServices.containsKey(socketAddress)) {
  42             return cachedOperationServices.get(socketAddress);
  43         } else {
  44             final NetconfOperationService service = getOperationServiceFactory().createService(netconfSessionIdForReporting);
  45             cachedOperationServices.put(socketAddress, service);
  46             return service;
  47         }
  48     }
  49 }