openflowplugin-impl/src/main/java/org/opendaylight/openflowplugin/impl/rpc/RpcManagerImpl.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 package org.opendaylight.openflowplugin.impl.rpc;
   9
  10 import com.google.common.annotations.VisibleForTesting;
  11 import com.google.common.collect.Iterators;
  12 import java.util.Iterator;
  13 import java.util.concurrent.ConcurrentHashMap;
  14 import java.util.concurrent.ConcurrentMap;
  15 import org.eclipse.jdt.annotation.NonNull;
  16 import org.opendaylight.mdsal.binding.api.NotificationPublishService;
  17 import org.opendaylight.mdsal.binding.api.RpcProviderService;
  18 import org.opendaylight.openflowplugin.api.openflow.FlowGroupCacheManager;
  19 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  20 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  21 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
  22 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
  23 import org.opendaylight.openflowplugin.extension.api.core.extension.ExtensionConverterProvider;
  24 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 public class RpcManagerImpl implements RpcManager {
  30
  31     private static final Logger LOG = LoggerFactory.getLogger(RpcManagerImpl.class);
  32     private final OpenflowProviderConfig config;
  33     private final RpcProviderService rpcProviderRegistry;
  34     private final ConcurrentMap<DeviceInfo, RpcContext> contexts = new ConcurrentHashMap<>();
  35     private final ExtensionConverterProvider extensionConverterProvider;
  36     private final ConvertorExecutor convertorExecutor;
  37     private final NotificationPublishService notificationPublishService;
  38     private final FlowGroupCacheManager flowGroupCacheManager;
  39
  40     public RpcManagerImpl(final OpenflowProviderConfig config,
  41                           final RpcProviderService rpcProviderRegistry,
  42                           final ExtensionConverterProvider extensionConverterProvider,
  43                           final ConvertorExecutor convertorExecutor,
  44                           final NotificationPublishService notificationPublishService,
  45                           final FlowGroupCacheManager flowGroupCacheManager) {
  46         this.config = config;
  47         this.rpcProviderRegistry = rpcProviderRegistry;
  48         this.extensionConverterProvider = extensionConverterProvider;
  49         this.convertorExecutor = convertorExecutor;
  50         this.notificationPublishService = notificationPublishService;
  51         this.flowGroupCacheManager = flowGroupCacheManager;
  52     }
  53
  54     @Override
  55     public void close() {
  56         for (final Iterator<RpcContext> iterator = Iterators.consumingIterator(contexts.values().iterator());
  57                 iterator.hasNext();) {
  58             iterator.next().close();
  59         }
  60     }
  61
  62     /**
  63      * This method is only for testing.
  64      */
  65     @VisibleForTesting
  66     void addRecordToContexts(final DeviceInfo deviceInfo, final RpcContext rpcContexts) {
  67         if (!contexts.containsKey(deviceInfo)) {
  68             this.contexts.put(deviceInfo, rpcContexts);
  69         }
  70     }
  71
  72     @Override
  73     public RpcContext createContext(final @NonNull DeviceContext deviceContext) {
  74         final RpcContextImpl rpcContext = new RpcContextImpl(
  75                 rpcProviderRegistry,
  76                 config.getRpcRequestsQuota().getValue().toJava(),
  77                 deviceContext,
  78                 extensionConverterProvider,
  79                 convertorExecutor,
  80                 notificationPublishService,
  81                 config.isIsStatisticsRpcEnabled(),
  82                 flowGroupCacheManager);
  83
  84         contexts.put(deviceContext.getDeviceInfo(), rpcContext);
  85         return rpcContext;
  86     }
  87
  88     @Override
  89     public void onDeviceRemoved(final DeviceInfo deviceInfo) {
  90         contexts.remove(deviceInfo);
  91         if (LOG.isDebugEnabled()) {
  92             LOG.debug("Rpc context removed for node {}", deviceInfo);
  93         }
  94     }
  95 }