package org.opendaylight.openflowplugin.impl.rpc;
import com.google.common.annotations.VisibleForTesting;
-import com.google.common.base.Preconditions;
-import com.google.common.base.Verify;
import com.google.common.collect.Iterators;
import java.util.Iterator;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
+import javax.annotation.Nonnull;
import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
-import org.opendaylight.openflowplugin.api.openflow.OFPContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
-import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceInitializationPhaseHandler;
-import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceTerminationPhaseHandler;
-import org.opendaylight.openflowplugin.api.openflow.lifecycle.LifecycleService;
import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
import org.opendaylight.openflowplugin.extension.api.core.extension.ExtensionConverterProvider;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class RpcManagerImpl implements RpcManager {
private static final Logger LOG = LoggerFactory.getLogger(RpcManagerImpl.class);
+ private final OpenflowProviderConfig config;
private final RpcProviderRegistry rpcProviderRegistry;
- private DeviceInitializationPhaseHandler deviceInitPhaseHandler;
- private DeviceTerminationPhaseHandler deviceTerminPhaseHandler;
- private final int maxRequestsQuota;
private final ConcurrentMap<DeviceInfo, RpcContext> contexts = new ConcurrentHashMap<>();
- private boolean isStatisticsRpcEnabled;
private final ExtensionConverterProvider extensionConverterProvider;
private final ConvertorExecutor convertorExecutor;
private final NotificationPublishService notificationPublishService;
-
- public RpcManagerImpl(
- final RpcProviderRegistry rpcProviderRegistry,
- final int quotaValue,
- final ExtensionConverterProvider extensionConverterProvider,
- final ConvertorExecutor convertorExecutor,
- final NotificationPublishService notificationPublishService) {
+ public RpcManagerImpl(final OpenflowProviderConfig config,
+ final RpcProviderRegistry rpcProviderRegistry,
+ final ExtensionConverterProvider extensionConverterProvider,
+ final ConvertorExecutor convertorExecutor,
+ final NotificationPublishService notificationPublishService) {
+ this.config = config;
this.rpcProviderRegistry = rpcProviderRegistry;
- maxRequestsQuota = quotaValue;
this.extensionConverterProvider = extensionConverterProvider;
this.convertorExecutor = convertorExecutor;
this.notificationPublishService = notificationPublishService;
}
- @Override
- public void setDeviceInitializationPhaseHandler(final DeviceInitializationPhaseHandler handler) {
- deviceInitPhaseHandler = handler;
- }
-
- @Override
- public void onDeviceContextLevelUp(final DeviceInfo deviceInfo, final LifecycleService lifecycleService) throws Exception {
-
- final DeviceContext deviceContext = Preconditions.checkNotNull(lifecycleService.getDeviceContext());
-
- final RpcContext rpcContext = new RpcContextImpl(
- deviceInfo,
- rpcProviderRegistry,
- deviceContext.getMessageSpy(),
- maxRequestsQuota,
- deviceInfo.getNodeInstanceIdentifier(),
- deviceContext,
- extensionConverterProvider,
- convertorExecutor,
- notificationPublishService);
-
- Verify.verify(contexts.putIfAbsent(deviceInfo, rpcContext) == null, "RpcCtx still not closed for node {}", deviceInfo.getNodeId());
- lifecycleService.setRpcContext(rpcContext);
- rpcContext.setStatisticsRpcEnabled(isStatisticsRpcEnabled);
-
- // finish device initialization cycle back to DeviceManager
- deviceInitPhaseHandler.onDeviceContextLevelUp(deviceInfo, lifecycleService);
- }
-
@Override
public void close() {
for (final Iterator<RpcContext> iterator = Iterators.consumingIterator(contexts.values().iterator());
}
}
- @Override
- public void onDeviceContextLevelDown(final DeviceInfo deviceInfo) {
- final RpcContext removedContext = contexts.remove(deviceInfo);
- if (removedContext != null) {
- LOG.debug("Unregister RPCs services for node {}", deviceInfo.getLOGValue());
- removedContext.close();
- }
- deviceTerminPhaseHandler.onDeviceContextLevelDown(deviceInfo);
- }
-
- @Override
- public void setDeviceTerminationPhaseHandler(final DeviceTerminationPhaseHandler handler) {
- this.deviceTerminPhaseHandler = handler;
- }
-
/**
- * This method is only for testing
+ * This method is only for testing.
*/
@VisibleForTesting
void addRecordToContexts(DeviceInfo deviceInfo, RpcContext rpcContexts) {
- if(!contexts.containsKey(deviceInfo)) {
- this.contexts.put(deviceInfo,rpcContexts);
+ if (!contexts.containsKey(deviceInfo)) {
+ this.contexts.put(deviceInfo, rpcContexts);
}
}
@Override
- public <T extends OFPContext> T gainContext(DeviceInfo deviceInfo) {
- return (T) contexts.get(deviceInfo);
- }
+ public RpcContext createContext(final @Nonnull DeviceContext deviceContext) {
+ final RpcContextImpl rpcContext = new RpcContextImpl(
+ rpcProviderRegistry,
+ config.getRpcRequestsQuota().getValue(),
+ deviceContext,
+ extensionConverterProvider,
+ convertorExecutor,
+ notificationPublishService,
+ config.isIsStatisticsRpcEnabled());
+ contexts.put(deviceContext.getDeviceInfo(), rpcContext);
+ return rpcContext;
+ }
@Override
- public void setStatisticsRpcEnabled(boolean statisticsRpcEnabled) {
- isStatisticsRpcEnabled = statisticsRpcEnabled;
+ public void onDeviceRemoved(final DeviceInfo deviceInfo) {
+ contexts.remove(deviceInfo);
+ if (LOG.isDebugEnabled()) {
+ LOG.debug("Rpc context removed for node {}", deviceInfo);
+ }
}
}