opendaylight/netconf/netconf-testtool/src/main/java/org/opendaylight/controller/netconf/test/tool/TesttoolNegotiationFactory.java

   1 package org.opendaylight.controller.netconf.test.tool;
   2
   3 import io.netty.util.Timer;
   4 import java.net.SocketAddress;
   5 import java.util.HashMap;
   6 import java.util.Map;
   7 import java.util.Set;
   8 import org.opendaylight.controller.netconf.api.monitoring.NetconfMonitoringService;
   9 import org.opendaylight.controller.netconf.impl.CommitNotifier;
  10 import org.opendaylight.controller.netconf.impl.NetconfServerSessionNegotiatorFactory;
  11 import org.opendaylight.controller.netconf.impl.SessionIdProvider;
  12 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationService;
  13 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationServiceFactory;
  14
  15 public class TesttoolNegotiationFactory extends NetconfServerSessionNegotiatorFactory {
  16
  17     private final Map<SocketAddress, NetconfOperationService> cachedOperationServices = new HashMap<>();
  18
  19     public TesttoolNegotiationFactory(final Timer timer, final NetconfOperationServiceFactory netconfOperationProvider,
  20                                       final SessionIdProvider idProvider, final long connectionTimeoutMillis,
  21                                       final CommitNotifier commitNotifier, final NetconfMonitoringService monitoringService) {
  22         super(timer, netconfOperationProvider, idProvider, connectionTimeoutMillis, commitNotifier, monitoringService);
  23     }
  24
  25     public TesttoolNegotiationFactory(final Timer timer, final NetconfOperationServiceFactory netconfOperationProvider,
  26                                       final SessionIdProvider idProvider, final long connectionTimeoutMillis,
  27                                       final CommitNotifier commitNotifier, final NetconfMonitoringService monitoringService,
  28                                       final Set<String> baseCapabilities) {
  29         super(timer, netconfOperationProvider, idProvider, connectionTimeoutMillis, commitNotifier, monitoringService, baseCapabilities);
  30     }
  31
  32     @Override
  33     protected NetconfOperationService getOperationServiceForAddress(final String netconfSessionIdForReporting, final SocketAddress socketAddress) {
  34         if (cachedOperationServices.containsKey(socketAddress)) {
  35             return cachedOperationServices.get(socketAddress);
  36         } else {
  37             final NetconfOperationService service = getOperationServiceFactory().createService(netconfSessionIdForReporting);
  38             cachedOperationServices.put(socketAddress, service);
  39             return service;
  40         }
  41     }
  42 }