-/**
+/*
* Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
-
package org.opendaylight.openflowplugin.impl.connection;
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.base.Preconditions;
+import com.google.common.util.concurrent.ThreadFactoryBuilder;
+import java.math.BigInteger;
import java.net.InetAddress;
+import java.time.LocalDateTime;
+import java.util.Collection;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ThreadFactory;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
+import org.opendaylight.mdsal.binding.api.DataBroker;
+import org.opendaylight.mdsal.binding.api.DataObjectModification;
+import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
+import org.opendaylight.mdsal.binding.api.DataTreeModification;
+import org.opendaylight.mdsal.binding.api.NotificationPublishService;
+import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
import org.opendaylight.openflowjava.protocol.api.connection.ConnectionReadyListener;
import org.opendaylight.openflowplugin.api.OFConstants;
import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionManager;
+import org.opendaylight.openflowplugin.api.openflow.connection.DeviceConnectionStatusProvider;
import org.opendaylight.openflowplugin.api.openflow.connection.HandshakeContext;
import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceConnectedHandler;
import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceDisconnectedHandler;
import org.opendaylight.openflowplugin.api.openflow.md.core.HandshakeListener;
import org.opendaylight.openflowplugin.api.openflow.md.core.HandshakeManager;
+import org.opendaylight.openflowplugin.impl.common.DeviceConnectionRateLimiter;
import org.opendaylight.openflowplugin.impl.connection.listener.ConnectionReadyListenerImpl;
import org.opendaylight.openflowplugin.impl.connection.listener.HandshakeListenerImpl;
import org.opendaylight.openflowplugin.impl.connection.listener.OpenflowProtocolListenerInitialImpl;
import org.opendaylight.openflowplugin.impl.connection.listener.SystemNotificationsListenerImpl;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OpenflowProtocolListener;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.SystemNotificationsListener;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.provider.config.rev160510.OpenflowProviderConfig;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
private static final Logger LOG = LoggerFactory.getLogger(ConnectionManagerImpl.class);
private static final boolean BITMAP_NEGOTIATION_ENABLED = true;
+ private final ThreadFactory threadFactory = new ThreadFactoryBuilder()
+ .setNameFormat("ConnectionHandler-%d")
+ .setDaemon(false)
+ .setUncaughtExceptionHandler((thread, ex) -> LOG.error("Uncaught exception {}", thread, ex))
+ .build();
+ private final ExecutorService executorsService = Executors.newCachedThreadPool(threadFactory);
+
private DeviceConnectedHandler deviceConnectedHandler;
private final OpenflowProviderConfig config;
private final ExecutorService executorService;
+ private final DeviceConnectionRateLimiter deviceConnectionRateLimiter;
+ private final DataBroker dataBroker;
+ private final int deviceConnectionHoldTime;
private DeviceDisconnectedHandler deviceDisconnectedHandler;
+ private DeviceConnectionStatusProvider deviceConnectionStatusProvider;
+ private final NotificationPublishService notificationPublishService;
- public ConnectionManagerImpl(final OpenflowProviderConfig config, final ExecutorService executorService) {
+ public ConnectionManagerImpl(final OpenflowProviderConfig config, final ExecutorService executorService,
+ final DataBroker dataBroker,
+ @NonNull final NotificationPublishService notificationPublishService) {
this.config = config;
this.executorService = executorService;
+ this.deviceConnectionRateLimiter = new DeviceConnectionRateLimiter(config);
+ this.dataBroker = dataBroker;
+ this.deviceConnectionHoldTime = config.getDeviceConnectionHoldTimeInSeconds().toJava();
+ deviceConnectionStatusProvider = new DeviceConnectionStatusProviderImpl();
+ deviceConnectionStatusProvider.init();
+ this.notificationPublishService = notificationPublishService;
}
@Override
public void onSwitchConnected(final ConnectionAdapter connectionAdapter) {
+ connectionAdapter.setExecutorService(executorsService);
LOG.trace("prepare connection context");
- final ConnectionContext connectionContext = new ConnectionContextImpl(connectionAdapter);
+ final ConnectionContext connectionContext = new ConnectionContextImpl(connectionAdapter,
+ deviceConnectionStatusProvider);
connectionContext.setDeviceDisconnectedHandler(this.deviceDisconnectedHandler);
HandshakeListener handshakeListener = new HandshakeListenerImpl(connectionContext, deviceConnectedHandler);
new OpenflowProtocolListenerInitialImpl(connectionContext, handshakeContext);
connectionAdapter.setMessageListener(ofMessageListener);
- final SystemNotificationsListener systemListener = new SystemNotificationsListenerImpl(
- connectionContext, config.getEchoReplyTimeout().getValue(), executorService);
+ final SystemNotificationsListener systemListener = new SystemNotificationsListenerImpl(connectionContext,
+ config.getEchoReplyTimeout().getValue().toJava(), executorService, notificationPublishService);
connectionAdapter.setSystemListener(systemListener);
LOG.trace("connection ballet finished");
final HandshakeListener handshakeListener) {
HandshakeManagerImpl handshakeManager = new HandshakeManagerImpl(connectionAdapter,
OFConstants.VERSION_ORDER.get(0),
- OFConstants.VERSION_ORDER);
- handshakeManager.setUseVersionBitmap(BITMAP_NEGOTIATION_ENABLED);
- handshakeManager.setHandshakeListener(handshakeListener);
- handshakeManager.setErrorHandler(new ErrorHandlerSimpleImpl());
+ OFConstants.VERSION_ORDER, new ErrorHandlerSimpleImpl(), handshakeListener, BITMAP_NEGOTIATION_ENABLED,
+ deviceConnectionRateLimiter, deviceConnectionHoldTime, deviceConnectionStatusProvider);
return handshakeManager;
}
public void setDeviceDisconnectedHandler(final DeviceDisconnectedHandler deviceDisconnectedHandler) {
this.deviceDisconnectedHandler = deviceDisconnectedHandler;
}
+
+ @VisibleForTesting
+ DeviceConnectionStatusProvider getDeviceConnectionStatusProvider() {
+ return deviceConnectionStatusProvider;
+ }
+
+ @Override
+ public void close() throws Exception {
+ if (deviceConnectionStatusProvider != null) {
+ deviceConnectionStatusProvider.close();
+ deviceConnectionStatusProvider = null;
+ }
+ if (executorsService != null) {
+ executorsService.shutdownNow();
+ }
+ }
+
+ class DeviceConnectionStatusProviderImpl implements DeviceConnectionStatusProvider,
+ ClusteredDataTreeChangeListener<Node> {
+ private final Map<BigInteger, LocalDateTime> deviceConnectionMap = new ConcurrentHashMap<>();
+
+ private ListenerRegistration<DeviceConnectionStatusProviderImpl> listenerRegistration;
+
+ @Override
+ @SuppressWarnings({"checkstyle:IllegalCatch"})
+ public void init() {
+ DataTreeIdentifier<Node> treeId = DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL,
+ getWildCardPath());
+ try {
+ listenerRegistration = dataBroker.registerDataTreeChangeListener(treeId, this);
+ } catch (Exception e) {
+ LOG.error("DeviceConnectionStatusProvider listener registration failed", e);
+ }
+ }
+
+ @Override
+ public LocalDateTime getDeviceLastConnectionTime(BigInteger nodeId) {
+ return deviceConnectionMap.get(nodeId);
+ }
+
+ @Override
+ public void addDeviceLastConnectionTime(BigInteger nodeId, LocalDateTime time) {
+ deviceConnectionMap.put(nodeId, time);
+ }
+
+ @Override
+ public void removeDeviceLastConnectionTime(BigInteger nodeId) {
+ deviceConnectionMap.remove(nodeId);
+ }
+
+ @Override
+ public void onDataTreeChanged(@NonNull Collection<DataTreeModification<Node>> changes) {
+ Preconditions.checkNotNull(changes, "Changes must not be null!");
+ for (DataTreeModification<Node> change : changes) {
+ final DataObjectModification<Node> mod = change.getRootNode();
+ switch (mod.getModificationType()) {
+ case DELETE:
+ break;
+ case SUBTREE_MODIFIED:
+ break;
+ case WRITE:
+ processNodeModification(change);
+ break;
+ default:
+ throw new IllegalArgumentException("Unhandled modification type " + mod.getModificationType());
+ }
+ }
+ }
+
+ private InstanceIdentifier<Node> getWildCardPath() {
+ return InstanceIdentifier.create(Nodes.class).child(Node.class);
+ }
+
+ private void processNodeModification(DataTreeModification<Node> change) {
+ final InstanceIdentifier<Node> key = change.getRootPath().getRootIdentifier();
+ final InstanceIdentifier<Node> nodeIdent = key.firstIdentifierOf(Node.class);
+ String[] nodeIdentity = nodeIdent.firstKeyOf(Node.class).getId().getValue().split(":");
+ String nodeId = nodeIdentity[1];
+ LOG.info("Clearing the device connection timer for the device {}", nodeId);
+ removeDeviceLastConnectionTime(new BigInteger(nodeId));
+ }
+
+ @Override
+ public void close() {
+ if (listenerRegistration != null) {
+ listenerRegistration.close();
+ listenerRegistration = null;
+ }
+ }
+ }
}
Code Review / openflowplugin.git / blobdiff