Bump odlparent to 5.0.0
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / lifecycle / ContextChainHolderImpl.java
index 44c9565b8fce9f819bb0ca229d99a6f49a7b3786..56045af70992f6ea48fdde65d4b9bb576efe3f8c 100644 (file)
@@ -8,28 +8,25 @@
 package org.opendaylight.openflowplugin.impl.lifecycle;
 
 import com.google.common.annotations.VisibleForTesting;
-import com.google.common.base.Function;
-import com.google.common.base.Verify;
+import com.google.common.util.concurrent.FutureCallback;
 import com.google.common.util.concurrent.Futures;
-import com.google.common.util.concurrent.ListenableFuture;
-import io.netty.util.HashedWheelTimer;
-import io.netty.util.Timeout;
-import io.netty.util.TimerTask;
-import io.netty.util.internal.ConcurrentSet;
-import java.util.ArrayList;
+import com.google.common.util.concurrent.MoreExecutors;
+import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
-import java.util.Set;
+import java.util.Optional;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.ExecutorService;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
+import java.util.stream.Collectors;
 import javax.annotation.Nonnull;
-import javax.annotation.Nullable;
-import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipChange;
-import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListenerRegistration;
-import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipService;
-import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
+import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipChange;
+import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipListenerRegistration;
+import org.opendaylight.mdsal.eos.binding.api.EntityOwnershipService;
 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
 import org.opendaylight.openflowplugin.api.openflow.OFPManager;
 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
@@ -40,333 +37,327 @@ import org.opendaylight.openflowplugin.api.openflow.device.DeviceManager;
 import org.opendaylight.openflowplugin.api.openflow.lifecycle.ContextChain;
 import org.opendaylight.openflowplugin.api.openflow.lifecycle.ContextChainHolder;
 import org.opendaylight.openflowplugin.api.openflow.lifecycle.ContextChainMastershipState;
-import org.opendaylight.openflowplugin.api.openflow.lifecycle.LifecycleService;
+import org.opendaylight.openflowplugin.api.openflow.lifecycle.MasterChecker;
+import org.opendaylight.openflowplugin.api.openflow.lifecycle.OwnershipChangeListener;
+import org.opendaylight.openflowplugin.api.openflow.role.RoleContext;
+import org.opendaylight.openflowplugin.api.openflow.role.RoleManager;
 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcManager;
 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsManager;
 import org.opendaylight.openflowplugin.impl.util.DeviceStateUtil;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FeaturesReply;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.mdsal.core.general.entity.rev150930.Entity;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflowplugin.rf.state.rev170713.ResultState;
+import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class ContextChainHolderImpl implements ContextChainHolder {
-
+public class ContextChainHolderImpl implements ContextChainHolder, MasterChecker {
     private static final Logger LOG = LoggerFactory.getLogger(ContextChainHolderImpl.class);
 
     private static final String CONTEXT_CREATED_FOR_CONNECTION = " context created for connection: {}";
-    private static final long DEFAULT_CHECK_ROLE_MASTER = 10000L;
-    private static final String SERVICE_ENTITY_TYPE = "org.opendaylight.mdsal.ServiceEntityType";
+    private static final long REMOVE_DEVICE_FROM_DS_TIMEOUT = 5000L;
     private static final String ASYNC_SERVICE_ENTITY_TYPE = "org.opendaylight.mdsal.AsyncServiceCloseEntityType";
 
-    private final ConcurrentHashMap<DeviceInfo, ContextChain> contextChainMap = new ConcurrentHashMap<>();
-    private final ConcurrentHashMap<DeviceInfo, ContextChain> withoutRoleChains = new ConcurrentHashMap<>();
-    private final List<DeviceInfo> markToBeRemoved = new ArrayList<>();
-    private final HashedWheelTimer timer;
-    private final Long checkRoleMaster;
-
+    private final Map<DeviceInfo, ContextChain> contextChainMap = new ConcurrentHashMap<>();
+    private final Map<DeviceInfo, ? super ConnectionContext> connectingDevices = new ConcurrentHashMap<>();
+    private final EntityOwnershipListenerRegistration eosListenerRegistration;
+    private final ClusterSingletonServiceProvider singletonServiceProvider;
+    private final ExecutorService executorService;
+    private final OwnershipChangeListener ownershipChangeListener;
     private DeviceManager deviceManager;
     private RpcManager rpcManager;
     private StatisticsManager statisticsManager;
-    private EntityOwnershipListenerRegistration eosListenerRegistration;
-    private ClusterSingletonServiceProvider singletonServicesProvider;
-    private boolean timerIsRunningRole;
-
-    public ContextChainHolderImpl(final HashedWheelTimer timer) {
-        this.timerIsRunningRole = false;
-        this.timer = timer;
-        this.checkRoleMaster = DEFAULT_CHECK_ROLE_MASTER;
+    private RoleManager roleManager;
+
+    public ContextChainHolderImpl(final ExecutorService executorService,
+                                  final ClusterSingletonServiceProvider singletonServiceProvider,
+                                  final EntityOwnershipService entityOwnershipService,
+                                  final OwnershipChangeListener ownershipChangeListener) {
+        this.singletonServiceProvider = singletonServiceProvider;
+        this.executorService = executorService;
+        this.ownershipChangeListener = ownershipChangeListener;
+        this.ownershipChangeListener.setMasterChecker(this);
+        this.eosListenerRegistration = Objects
+                .requireNonNull(entityOwnershipService.registerListener(ASYNC_SERVICE_ENTITY_TYPE, this));
     }
 
     @Override
     public <T extends OFPManager> void addManager(final T manager) {
-        if (Objects.isNull(deviceManager) && manager instanceof DeviceManager) {
+        if (deviceManager == null && manager instanceof DeviceManager) {
             LOG.trace("Context chain holder: Device manager OK.");
             deviceManager = (DeviceManager) manager;
-        } else if (Objects.isNull(rpcManager) && manager instanceof RpcManager) {
+        } else if (rpcManager == null && manager instanceof RpcManager) {
             LOG.trace("Context chain holder: RPC manager OK.");
             rpcManager = (RpcManager) manager;
-        } else if (Objects.isNull(statisticsManager) && manager instanceof StatisticsManager) {
+        } else if (statisticsManager == null && manager instanceof StatisticsManager) {
             LOG.trace("Context chain holder: Statistics manager OK.");
             statisticsManager = (StatisticsManager) manager;
+        } else if (roleManager == null && manager instanceof RoleManager) {
+            LOG.trace("Context chain holder: Role manager OK.");
+            roleManager = (RoleManager) manager;
         }
     }
 
-    @Override
-    public ContextChain createContextChain(final ConnectionContext connectionContext) {
-
+    @VisibleForTesting
+    void createContextChain(final ConnectionContext connectionContext) {
         final DeviceInfo deviceInfo = connectionContext.getDeviceInfo();
-        final String deviceInfoLOGValue = deviceInfo.getLOGValue();
-
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("Creating a new chain" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfoLOGValue);
-        }
-
-        final ContextChain contextChain = new ContextChainImpl(connectionContext);
-        final LifecycleService lifecycleService = new LifecycleServiceImpl(this);
-        lifecycleService.registerDeviceRemovedHandler(deviceManager);
-        lifecycleService.registerDeviceRemovedHandler(rpcManager);
-        lifecycleService.registerDeviceRemovedHandler(statisticsManager);
-
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("Lifecycle services" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfoLOGValue);
-        }
 
         final DeviceContext deviceContext = deviceManager.createContext(connectionContext);
-
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("Device" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfoLOGValue);
-        }
-
-        final RpcContext rpcContext = rpcManager.createContext(connectionContext.getDeviceInfo(), deviceContext);
-
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("RPC" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfoLOGValue);
-        }
-
-        final StatisticsContext statisticsContext
-                = statisticsManager.createContext(deviceContext);
-
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("Statistics" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfoLOGValue);
-        }
-
-        deviceContext.setLifecycleInitializationPhaseHandler(statisticsContext);
-        statisticsContext.setLifecycleInitializationPhaseHandler(rpcContext);
-        statisticsContext.setInitialSubmitHandler(deviceContext);
-
-        contextChain.addLifecycleService(lifecycleService);
+        deviceContext.registerMastershipWatcher(this);
+        LOG.debug("Device" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfo);
+
+        final RpcContext rpcContext = rpcManager.createContext(deviceContext);
+        rpcContext.registerMastershipWatcher(this);
+        LOG.debug("RPC" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfo);
+
+        final StatisticsContext statisticsContext = statisticsManager
+                .createContext(deviceContext, ownershipChangeListener.isReconciliationFrameworkRegistered());
+        statisticsContext.registerMastershipWatcher(this);
+        LOG.debug("Statistics" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfo);
+
+        final RoleContext roleContext = roleManager.createContext(deviceContext);
+        roleContext.registerMastershipWatcher(this);
+        LOG.debug("Role" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfo);
+
+        final ContextChain contextChain = new ContextChainImpl(this, connectionContext, executorService);
+        contextChain.registerDeviceRemovedHandler(deviceManager);
+        contextChain.registerDeviceRemovedHandler(rpcManager);
+        contextChain.registerDeviceRemovedHandler(statisticsManager);
+        contextChain.registerDeviceRemovedHandler(roleManager);
+        contextChain.registerDeviceRemovedHandler(this);
         contextChain.addContext(deviceContext);
         contextChain.addContext(rpcContext);
         contextChain.addContext(statisticsContext);
-        this.withoutRoleChains.put(deviceInfo, contextChain);
-        if (!this.timerIsRunningRole) {
-            this.startTimerRole();
-        }
-        deviceContext.onPublished();
-        contextChain.registerServices(this.singletonServicesProvider);
+        contextChain.addContext(roleContext);
+        contextChainMap.put(deviceInfo, contextChain);
+        connectingDevices.remove(deviceInfo);
+        LOG.debug("Context chain" + CONTEXT_CREATED_FOR_CONNECTION, deviceInfo);
 
-        return contextChain;
+        deviceContext.onPublished();
+        contextChain.registerServices(singletonServiceProvider);
     }
 
     @Override
-    public ListenableFuture<Void> destroyContextChain(final DeviceInfo deviceInfo) {
-        ContextChain chain = contextChainMap.remove(deviceInfo);
-        if (chain != null) {
-            chain.close();
-        }
-        if (markToBeRemoved.contains(deviceInfo)) {
-            markToBeRemoved.remove(deviceInfo);
-            LOG.info("Removing device: {} from DS", deviceInfo.getLOGValue());
-            return deviceManager.removeDeviceFromOperationalDS(deviceInfo);
+    public ConnectionStatus deviceConnected(final ConnectionContext connectionContext) {
+        final DeviceInfo deviceInfo = connectionContext.getDeviceInfo();
+        final ContextChain contextChain = contextChainMap.get(deviceInfo);
+        final FeaturesReply featuresReply = connectionContext.getFeatures();
+        final Short auxiliaryId = featuresReply != null ? featuresReply.getAuxiliaryId() : null;
+
+        if (auxiliaryId != null && auxiliaryId != 0) {
+            if (contextChain == null) {
+                LOG.warn("An auxiliary connection for device {}, but no primary connection. Refusing connection.",
+                         deviceInfo);
+                return ConnectionStatus.REFUSING_AUXILIARY_CONNECTION;
+            } else {
+                if (contextChain.addAuxiliaryConnection(connectionContext)) {
+                    LOG.info("An auxiliary connection was added to device: {}", deviceInfo);
+                    return ConnectionStatus.MAY_CONTINUE;
+                } else {
+                    LOG.warn("Not able to add auxiliary connection to the device {}", deviceInfo);
+                    return ConnectionStatus.REFUSING_AUXILIARY_CONNECTION;
+                }
+            }
         } else {
-            return Futures.immediateFuture(null);
-        }
-    }
+            LOG.info("Device {} connected.", deviceInfo);
+            final boolean contextExists = contextChain != null;
+            final boolean isClosing = contextExists && contextChain.isClosing();
 
-    @Override
-    public ConnectionStatus deviceConnected(final ConnectionContext connectionContext) throws Exception {
-
-        DeviceInfo deviceInfo = connectionContext.getDeviceInfo();
-        LOG.info("Device {} connected.", deviceInfo.getLOGValue());
-        ContextChain contextChain = contextChainMap.get(deviceInfo);
-        if (contextChain != null) {
-            if (contextChain.addAuxiliaryConnection(connectionContext)) {
-                LOG.info("An auxiliary connection was added to device: {}", deviceInfo.getLOGValue());
-                return ConnectionStatus.MAY_CONTINUE;
-            } else {
-                LOG.warn("Device {} already connected. Closing all connection to the device.", deviceInfo.getLOGValue());
-                destroyContextChain(deviceInfo);
+            if (!isClosing && connectingDevices.putIfAbsent(deviceInfo, connectionContext) != null) {
+                LOG.warn("Device {} is already trying to connect, wait until succeeded or disconnected.", deviceInfo);
                 return ConnectionStatus.ALREADY_CONNECTED;
             }
-        } else {
-            if (LOG.isDebugEnabled()) {
-                LOG.debug("No context chain found for device: {}, creating new.", deviceInfo.getLOGValue());
+
+            if (contextExists) {
+                if (isClosing) {
+                    LOG.warn("Device {} is already in termination state, closing all incoming connections.",
+                             deviceInfo);
+                    return ConnectionStatus.CLOSING;
+                }
+
+                LOG.warn("Device {} already connected. Closing previous connection", deviceInfo);
+                destroyContextChain(deviceInfo);
+                LOG.info("Old connection dropped, creating new context chain for device {}", deviceInfo);
+                createContextChain(connectionContext);
+            } else {
+                LOG.info("No context chain found for device: {}, creating new.", deviceInfo);
+                createContextChain(connectionContext);
             }
-            contextChainMap.put(deviceInfo, createContextChain(connectionContext));
+
+            return ConnectionStatus.MAY_CONTINUE;
         }
 
-        return ConnectionStatus.MAY_CONTINUE;
     }
 
     @Override
-    public void addSingletonServicesProvider(final ClusterSingletonServiceProvider singletonServicesProvider) {
-        this.singletonServicesProvider = singletonServicesProvider;
-    }
+    public void onNotAbleToStartMastership(@Nonnull final DeviceInfo deviceInfo, @Nonnull final String reason,
+                                           final boolean mandatory) {
+        LOG.warn("Not able to set MASTER role on device {}, reason: {}", deviceInfo, reason);
 
-    @Override
-    public void onNotAbleToStartMastership(final DeviceInfo deviceInfo, @Nonnull final String reason, final boolean mandatory) {
-        this.withoutRoleChains.remove(deviceInfo);
-        LOG.warn("Not able to set MASTER role on device {}, reason: {}", deviceInfo.getLOGValue(), reason);
-        if (mandatory && contextChainMap.containsKey(deviceInfo)) {
-            LOG.warn("This mastering is mandatory, destroying context chain and closing connection.");
-            Futures.transform(contextChainMap.get(deviceInfo).stopChain(), new Function<Void, Object>() {
-                        @Nullable
-                        @Override
-                        public Object apply(@Nullable Void aVoid) {
-                            destroyContextChain(deviceInfo);
-                            return null;
-                        }
-                    });
+        if (!mandatory) {
+            return;
         }
+
+        Optional.ofNullable(contextChainMap.get(deviceInfo)).ifPresent(contextChain -> {
+            LOG.warn("This mastering is mandatory, destroying context chain and closing connection for device {}.",
+                     deviceInfo);
+            destroyContextChain(deviceInfo);
+        });
     }
 
     @Override
-    public void onMasterRoleAcquired(final DeviceInfo deviceInfo,
+    public void onMasterRoleAcquired(@Nonnull final DeviceInfo deviceInfo,
                                      @Nonnull final ContextChainMastershipState mastershipState) {
-        this.withoutRoleChains.remove(deviceInfo);
-        ContextChain contextChain = contextChainMap.get(deviceInfo);
-        if (contextChain != null) {
-            if (contextChain.isMastered(mastershipState)) {
-                LOG.info("Role MASTER was granted to device {}", deviceInfo.getLOGValue());
-                this.sendNotificationNodeAdded(deviceInfo);
+        Optional.ofNullable(contextChainMap.get(deviceInfo)).ifPresent(contextChain -> {
+            if (ownershipChangeListener.isReconciliationFrameworkRegistered()
+                    && !ContextChainMastershipState.INITIAL_SUBMIT.equals(mastershipState)) {
+                if (contextChain.isMastered(mastershipState, true)) {
+                    Futures.addCallback(ownershipChangeListener.becomeMasterBeforeSubmittedDS(deviceInfo),
+                                        reconciliationFrameworkCallback(deviceInfo, contextChain),
+                                        MoreExecutors.directExecutor());
+                }
+            } else if (contextChain.isMastered(mastershipState, false)) {
+                LOG.info("Role MASTER was granted to device {}", deviceInfo);
+                ownershipChangeListener.becomeMaster(deviceInfo);
+                deviceManager.sendNodeAddedNotification(deviceInfo.getNodeInstanceIdentifier());
             }
-        }
+        });
     }
 
     @Override
     public void onSlaveRoleAcquired(final DeviceInfo deviceInfo) {
-        this.withoutRoleChains.remove(deviceInfo);
-        ContextChain contextChain = contextChainMap.get(deviceInfo);
-        if (contextChain != null) {
-            contextChain.makeContextChainStateSlave();
-        }
+        ownershipChangeListener.becomeSlaveOrDisconnect(deviceInfo);
+        LOG.info("Role SLAVE was granted to device {}", deviceInfo);
+        Optional.ofNullable(contextChainMap.get(deviceInfo)).ifPresent(ContextChain::makeContextChainStateSlave);
     }
 
     @Override
-    public void onSlaveRoleNotAcquired(final DeviceInfo deviceInfo) {
-        this.withoutRoleChains.remove(deviceInfo);
-        ContextChain contextChain = contextChainMap.get(deviceInfo);
-        if (contextChain != null) {
-            destroyContextChain(deviceInfo);
-        }
+    public void onSlaveRoleNotAcquired(final DeviceInfo deviceInfo, final String reason) {
+        LOG.warn("Not able to set SLAVE role on device {}, reason: {}", deviceInfo, reason);
+        Optional.ofNullable(contextChainMap.get(deviceInfo)).ifPresent(contextChain -> destroyContextChain(deviceInfo));
     }
 
     @Override
     public void onDeviceDisconnected(final ConnectionContext connectionContext) {
-
         final DeviceInfo deviceInfo = connectionContext.getDeviceInfo();
-        if (deviceInfo != null) {
-            ContextChain chain = contextChainMap.get(deviceInfo);
-            if (chain != null) {
-                if (chain.auxiliaryConnectionDropped(connectionContext)) {
-                    LOG.info("Auxiliary connection from device {} disconnected.", deviceInfo.getLOGValue());
-                } else {
-                    LOG.info("Device {} disconnected.", deviceInfo.getLOGValue());
-                    Futures.transform(chain.connectionDropped(), new Function<Void, Object>() {
-                        @Nullable
-                        @Override
-                        public Object apply(@Nullable Void aVoid) {
-                            destroyContextChain(deviceInfo);
-                            return null;
-                        }
-                    });
-                }
+
+        Optional.ofNullable(connectionContext.getDeviceInfo()).map(contextChainMap::get).ifPresent(contextChain -> {
+            if (contextChain.auxiliaryConnectionDropped(connectionContext)) {
+                LOG.info("Auxiliary connection from device {} disconnected.", deviceInfo);
+            } else {
+                LOG.info("Device {} disconnected.", deviceInfo);
+                destroyContextChain(deviceInfo);
             }
-        }
+        });
     }
 
-    @Override
-    public void changeEntityOwnershipService(final EntityOwnershipService entityOwnershipService) {
-        if (Objects.nonNull(this.eosListenerRegistration)) {
-            LOG.warn("EOS Listener already registered.");
-        } else {
-            this.eosListenerRegistration = Verify.verifyNotNull(entityOwnershipService.registerListener
-                    (ASYNC_SERVICE_ENTITY_TYPE, this));
-        }
+    @VisibleForTesting
+    boolean checkAllManagers() {
+        return deviceManager != null && rpcManager != null && statisticsManager != null && roleManager != null;
     }
 
-    private void startTimerRole() {
-        this.timerIsRunningRole = true;
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("There is a context chain without role, starting timer.");
-        }
-        timer.newTimeout(new RoleTimerTask(), this.checkRoleMaster, TimeUnit.MILLISECONDS);
+    @Override
+    public ContextChain getContextChain(final DeviceInfo deviceInfo) {
+        return contextChainMap.get(deviceInfo);
     }
 
-    private void stopTimerRole() {
-        this.timerIsRunningRole = false;
-        if (LOG.isDebugEnabled()) {
-            LOG.debug("There are no context chains, stopping timer.");
-        }
+    @Override
+    public void close() {
+        Map<DeviceInfo, ContextChain> copyOfChains = new HashMap<>(contextChainMap);
+        copyOfChains.keySet().forEach(this::destroyContextChain);
+        copyOfChains.clear();
+        contextChainMap.clear();
+        eosListenerRegistration.close();
     }
 
-    private void timerTickRole() {
-        if (!withoutRoleChains.isEmpty()) {
-            this.withoutRoleChains.forEach((deviceInfo, contextChain) -> contextChain.makeDeviceSlave());
-            timer.newTimeout(new RoleTimerTask(), this.checkRoleMaster, TimeUnit.MILLISECONDS);
-        } else {
-            this.stopTimerRole();
+    @Override
+    @SuppressFBWarnings("BC_UNCONFIRMED_CAST_OF_RETURN_VALUE")
+    public void ownershipChanged(EntityOwnershipChange entityOwnershipChange) {
+        if (entityOwnershipChange.getState().hasOwner()) {
+            return;
         }
-    }
 
-    @VisibleForTesting
-    boolean checkAllManagers() {
-        return Objects.nonNull(deviceManager) && Objects.nonNull(rpcManager) && Objects.nonNull(statisticsManager);
+        // Findbugs flags a false violation for "Unchecked/unconfirmed cast" from GenericEntity to Entity hence the
+        // suppression above. The suppression is temporary until EntityOwnershipChange is modified to eliminate the
+        // violation.
+        final String entityName = entityOwnershipChange
+                .getEntity()
+                .getIdentifier()
+                .firstKeyOf(Entity.class)
+                .getName();
+
+        if (entityName != null) {
+            LOG.debug("Entity {} has no owner", entityName);
+            try {
+                //TODO:Remove notifications
+                final KeyedInstanceIdentifier<Node, NodeKey> nodeInstanceIdentifier =
+                        DeviceStateUtil.createNodeInstanceIdentifier(new NodeId(entityName));
+                deviceManager.sendNodeRemovedNotification(nodeInstanceIdentifier);
+
+                LOG.info("Try to remove device {} from operational DS", entityName);
+                deviceManager.removeDeviceFromOperationalDS(nodeInstanceIdentifier)
+                        .get(REMOVE_DEVICE_FROM_DS_TIMEOUT, TimeUnit.MILLISECONDS);
+                LOG.info("Removing device from operational DS {} was successful", entityName);
+            } catch (TimeoutException | ExecutionException | NullPointerException | InterruptedException e) {
+                LOG.warn("Not able to remove device {} from operational DS. ", entityName, e);
+            }
+        }
     }
 
-    @Override
-    public void close() throws Exception {
-        this.contextChainMap.forEach((deviceInfo, contextChain) -> {
-            if (contextChain.isMastered(ContextChainMastershipState.CHECK)) {
-                contextChain.stopChain();
-            }
+    private void destroyContextChain(final DeviceInfo deviceInfo) {
+        ownershipChangeListener.becomeSlaveOrDisconnect(deviceInfo);
+        Optional.ofNullable(contextChainMap.get(deviceInfo)).ifPresent(contextChain -> {
+            deviceManager.sendNodeRemovedNotification(deviceInfo.getNodeInstanceIdentifier());
             contextChain.close();
+            connectingDevices.remove(deviceInfo);
         });
-        if (Objects.nonNull(eosListenerRegistration)) {
-            eosListenerRegistration.close();
-        }
     }
 
     @Override
-    public void ownershipChanged(EntityOwnershipChange entityOwnershipChange) {
-        if (!entityOwnershipChange.hasOwner()) {
-            final YangInstanceIdentifier yii = entityOwnershipChange.getEntity().getId();
-            final YangInstanceIdentifier.NodeIdentifierWithPredicates niiwp =
-                    (YangInstanceIdentifier.NodeIdentifierWithPredicates) yii.getLastPathArgument();
-            String entityName =  niiwp.getKeyValues().values().iterator().next().toString();
-            if (LOG.isDebugEnabled()) {
-                LOG.debug("Entity {} has no owner", entityName);
-            }
-
-            if (entityName != null ){
-                final NodeId nodeId = new NodeId(entityName);
-                DeviceInfo inMap = null;
-                for (Map.Entry<DeviceInfo, ContextChain> entry : contextChainMap.entrySet()) {
-                    if (entry.getKey().getNodeId().equals(nodeId)) {
-                        inMap = entry.getKey();
-                        break;
-                    }
-                }
-                if (Objects.nonNull(inMap)) {
-                    markToBeRemoved.add(inMap);
-                } else {
-                    try {
-                        LOG.info("Removing device: {} from DS", nodeId);
-                        deviceManager
-                                .removeDeviceFromOperationalDS(DeviceStateUtil.createNodeInstanceIdentifier(nodeId))
-                                .checkedGet(5L, TimeUnit.SECONDS);
-                    } catch (TimeoutException | TransactionCommitFailedException e) {
-                        LOG.info("Not able to remove device {} from DS. Probably removed by another cluster node.",
-                                nodeId);
-                    }
-                }
-            }
-        }
+    public List<DeviceInfo> listOfMasteredDevices() {
+        return contextChainMap.entrySet().stream()
+                .filter(deviceInfoContextChainEntry -> deviceInfoContextChainEntry.getValue()
+                        .isMastered(ContextChainMastershipState.CHECK, false)).map(Map.Entry::getKey)
+                .collect(Collectors.toList());
     }
 
-    private void sendNotificationNodeAdded(final DeviceInfo deviceInfo) {
-        this.deviceManager.sendNodeAddedNotification(deviceInfo);
+    @Override
+    public boolean isAnyDeviceMastered() {
+        return contextChainMap.entrySet().stream().findAny()
+                .filter(deviceInfoContextChainEntry -> deviceInfoContextChainEntry.getValue()
+                        .isMastered(ContextChainMastershipState.CHECK, false)).isPresent();
     }
 
-    private class RoleTimerTask implements TimerTask {
+    @Override
+    public void onDeviceRemoved(final DeviceInfo deviceInfo) {
+        contextChainMap.remove(deviceInfo);
+        LOG.debug("Context chain removed for node {}", deviceInfo);
+    }
 
-        @Override
-        public void run(Timeout timeout) throws Exception {
-            timerTickRole();
-        }
+    private FutureCallback<ResultState> reconciliationFrameworkCallback(@Nonnull DeviceInfo deviceInfo,
+                                                                        ContextChain contextChain) {
+        return new FutureCallback<ResultState>() {
+            @Override
+            public void onSuccess(ResultState result) {
+                if (ResultState.DONOTHING == result) {
+                    LOG.info("Device {} connection is enabled by reconciliation framework.", deviceInfo);
+                    contextChain.continueInitializationAfterReconciliation();
+                } else {
+                    LOG.warn("Reconciliation framework failure for device {}", deviceInfo);
+                    destroyContextChain(deviceInfo);
+                }
+            }
 
+            @Override
+            public void onFailure(Throwable throwable) {
+                LOG.warn("Reconciliation framework failure.");
+                destroyContextChain(deviceInfo);
+            }
+        };
     }
 }
-