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.base.Preconditions;
  11 import com.google.common.base.Verify;
  12 import com.google.common.collect.Iterators;
  13 import java.util.Iterator;
  14 import java.util.concurrent.ConcurrentHashMap;
  15 import java.util.concurrent.ConcurrentMap;
  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.handlers.DeviceInitializationPhaseHandler;
  20 import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceTerminationPhaseHandler;
  21 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
  22 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
  23 import org.opendaylight.openflowplugin.impl.LifecycleConductor;
  24 import org.opendaylight.openflowplugin.impl.util.MdSalRegistrationUtils;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.role.service.rev150727.OfpRole;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29
  30 public class RpcManagerImpl implements RpcManager {
  31
  32     private static final Logger LOG = LoggerFactory.getLogger(RpcManagerImpl.class);
  33     private final RpcProviderRegistry rpcProviderRegistry;
  34     private DeviceInitializationPhaseHandler deviceInitPhaseHandler;
  35     private DeviceTerminationPhaseHandler deviceTerminPhaseHandler;
  36     private final int maxRequestsQuota;
  37     private final ConcurrentMap<NodeId, RpcContext> contexts = new ConcurrentHashMap<>();
  38     private boolean isStatisticsRpcEnabled;
  39     private NotificationPublishService notificationPublishService;
  40
  41     public RpcManagerImpl(final RpcProviderRegistry rpcProviderRegistry,
  42                           final int quotaValue) {
  43         this.rpcProviderRegistry = rpcProviderRegistry;
  44         maxRequestsQuota = quotaValue;
  45     }
  46
  47     @Override
  48     public void setDeviceInitializationPhaseHandler(final DeviceInitializationPhaseHandler handler) {
  49         deviceInitPhaseHandler = handler;
  50     }
  51
  52     @Override
  53     public void onDeviceContextLevelUp(final NodeId nodeId) throws Exception {
  54
  55         DeviceContext deviceContext = Preconditions.checkNotNull(LifecycleConductor.getInstance().getDeviceContext(nodeId));
  56
  57         final RpcContext rpcContext = new RpcContextImpl(
  58                 rpcProviderRegistry,
  59                 deviceContext,
  60                 maxRequestsQuota,
  61                 isStatisticsRpcEnabled,
  62                 notificationPublishService);
  63
  64         Verify.verify(contexts.putIfAbsent(nodeId, rpcContext) == null, "RpcCtx still not closed for node {}", nodeId);
  65
  66         // finish device initialization cycle back to DeviceManager
  67         deviceInitPhaseHandler.onDeviceContextLevelUp(nodeId);
  68     }
  69
  70     @Override
  71     public void close() {
  72         for (final Iterator<RpcContext> iterator = Iterators.consumingIterator(contexts.values().iterator());
  73                 iterator.hasNext();) {
  74             iterator.next().close();
  75         }
  76     }
  77
  78
  79     @Override
  80     public void onDeviceContextLevelDown(final DeviceContext deviceContext) {
  81         final RpcContext removedContext = contexts.remove(deviceContext.getDeviceState().getNodeId());
  82         if (removedContext != null) {
  83             LOG.info("Unregister RPCs services for device context closure");
  84             removedContext.close();
  85         }
  86         deviceTerminPhaseHandler.onDeviceContextLevelDown(deviceContext);
  87     }
  88     @Override
  89     public void setStatisticsRpcEnabled(final boolean isStatisticsRpcEnabled) {
  90         this.isStatisticsRpcEnabled = isStatisticsRpcEnabled;
  91     }
  92
  93     @Override
  94     public void setNotificationPublishService(final NotificationPublishService notificationPublishService) {
  95         this.notificationPublishService = notificationPublishService;
  96     }
  97
  98     @Override
  99     public void setDeviceTerminationPhaseHandler(final DeviceTerminationPhaseHandler handler) {
 100         this.deviceTerminPhaseHandler = handler;
 101     }
 102 }