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 javax.annotation.Nonnull;
  16 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
  17 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
  18 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
  19 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
  20 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
  21 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
  22 import org.opendaylight.openflowplugin.extension.api.core.extension.ExtensionConverterProvider;
  23 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
  24 import org.slf4j.Logger;
  25 import org.slf4j.LoggerFactory;
  26
  27 public class RpcManagerImpl implements RpcManager {
  28
  29     private static final Logger LOG = LoggerFactory.getLogger(RpcManagerImpl.class);
  30     private final RpcProviderRegistry rpcProviderRegistry;
  31     private int rpcRequestQuota;
  32     private final ConcurrentMap<DeviceInfo, RpcContext> contexts = new ConcurrentHashMap<>();
  33     private boolean isStatisticsRpcEnabled;
  34     private final ExtensionConverterProvider extensionConverterProvider;
  35     private final ConvertorExecutor convertorExecutor;
  36     private final NotificationPublishService notificationPublishService;
  37
  38
  39     public RpcManagerImpl(final RpcProviderRegistry rpcProviderRegistry,
  40                           final ExtensionConverterProvider extensionConverterProvider,
  41                           final ConvertorExecutor convertorExecutor,
  42                           final NotificationPublishService notificationPublishService) {
  43         this.rpcProviderRegistry = rpcProviderRegistry;
  44         this.extensionConverterProvider = extensionConverterProvider;
  45         this.convertorExecutor = convertorExecutor;
  46         this.notificationPublishService = notificationPublishService;
  47     }
  48
  49     @Override
  50     public void close() {
  51         for (final Iterator<RpcContext> iterator = Iterators.consumingIterator(contexts.values().iterator());
  52                 iterator.hasNext();) {
  53             iterator.next().close();
  54         }
  55     }
  56
  57     /**
  58      * This method is only for testing
  59      */
  60     @VisibleForTesting
  61     void addRecordToContexts(DeviceInfo deviceInfo, RpcContext rpcContexts) {
  62         if(!contexts.containsKey(deviceInfo)) {
  63             this.contexts.put(deviceInfo,rpcContexts);
  64         }
  65     }
  66
  67
  68     @Override
  69     public void setStatisticsRpcEnabled(boolean statisticsRpcEnabled) {
  70         isStatisticsRpcEnabled = statisticsRpcEnabled;
  71     }
  72
  73     @Override
  74     public void setRpcRequestQuota(final int rpcRequestQuota) {
  75         this.rpcRequestQuota = rpcRequestQuota;
  76     }
  77
  78     public RpcContext createContext(final @Nonnull DeviceInfo deviceInfo, final @Nonnull DeviceContext deviceContext) {
  79         return new RpcContextImpl(
  80                 rpcProviderRegistry,
  81                 rpcRequestQuota,
  82                 deviceContext,
  83                 extensionConverterProvider,
  84                 convertorExecutor,
  85                 notificationPublishService,
  86                 this.isStatisticsRpcEnabled);
  87     }
  88
  89     @Override
  90     public void onDeviceRemoved(final DeviceInfo deviceInfo) {
  91         contexts.remove(deviceInfo);
  92         if (LOG.isDebugEnabled()) {
  93             LOG.debug("Rpc context removed for node {}", deviceInfo.getLOGValue());
  94         }
  95     }
  96 }