/*
 * Copyright (c) 2016 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.util;

import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.AsyncFunction;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.common.util.concurrent.SettableFuture;
import java.math.BigInteger;
import java.net.InetSocketAddress;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.ExecutionException;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
import org.opendaylight.openflowplugin.api.ConnectionException;
import org.opendaylight.openflowplugin.api.OFConstants;
import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
import org.opendaylight.openflowplugin.api.openflow.device.Xid;
import org.opendaylight.openflowplugin.api.openflow.device.handlers.MultiMsgCollector;
import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
import org.opendaylight.openflowplugin.impl.common.MultipartRequestInputFactory;
import org.opendaylight.openflowplugin.impl.common.NodeStaticReplyTranslatorUtil;
import org.opendaylight.openflowplugin.impl.rpc.AbstractRequestContext;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IetfInetUtil;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorExecutor;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsData;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsDataBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.group.statistics.rev131111.NodeGroupFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.statistics.rev131111.NodeMeterFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.Capabilities;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.CapabilitiesV10;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortGrouping;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.MultipartReplyBody;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyDescCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyGroupFeaturesCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyMeterFeaturesCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyPortDescCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.MultipartReplyTableFeaturesCase;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.desc._case.MultipartReplyDesc;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.group.features._case.MultipartReplyGroupFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.meter.features._case.MultipartReplyMeterFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.port.desc._case.MultipartReplyPortDesc;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.multipart.reply.multipart.reply.body.multipart.reply.table.features._case.MultipartReplyTableFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsData;
import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsDataBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeatures;
import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.table.features.TableFeaturesKey;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
import org.opendaylight.yangtools.yang.common.RpcError;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class DeviceInitializationUtils {

    private static final Logger LOG = LoggerFactory.getLogger(DeviceInitializationUtils.class);

    private DeviceInitializationUtils() {
        // Hiding implicit constructor
    }

    /**
     * InitializationNodeInformation is good to call only for MASTER otherwise we will have not empty transaction
     * for every Cluster Node (SLAVE too) and we will get race-condition by closing Connection.
     *
     * @param deviceContext
     * @param switchFeaturesMandatory
     * @param convertorExecutor
     */
    public static void initializeNodeInformation(final DeviceContext deviceContext, final boolean switchFeaturesMandatory, final ConvertorExecutor convertorExecutor) throws ExecutionException, InterruptedException {
        Preconditions.checkArgument(deviceContext != null);
        final DeviceState deviceState = Preconditions.checkNotNull(deviceContext.getDeviceState());
        final DeviceInfo deviceInfo = deviceContext.getDeviceInfo();
        final ConnectionContext connectionContext = Preconditions.checkNotNull(deviceContext.getPrimaryConnectionContext());
        final short version = deviceInfo.getVersion();
        LOG.trace("initalizeNodeInformation for node {}", deviceInfo.getNodeId());
        final SettableFuture<Void> returnFuture = SettableFuture.<Void>create();
        addNodeToOperDS(deviceContext, returnFuture);
        final ListenableFuture<List<RpcResult<List<MultipartReply>>>> deviceFeaturesFuture;
        if (OFConstants.OFP_VERSION_1_0 == version) {
            final CapabilitiesV10 capabilitiesV10 = connectionContext.getFeatures().getCapabilitiesV10();

            DeviceStateUtil.setDeviceStateBasedOnV10Capabilities(deviceState, capabilitiesV10);

            deviceFeaturesFuture = createDeviceFeaturesForOF10(deviceContext);
            // create empty tables after device description is processed
            chainTableTrunkWriteOF10(deviceContext, deviceFeaturesFuture);

            final short ofVersion = deviceInfo.getVersion();
            final TranslatorKey translatorKey = new TranslatorKey(ofVersion, PortGrouping.class.getName());
            final MessageTranslator<PortGrouping, FlowCapableNodeConnector> translator = deviceContext.oook()
                    .lookupTranslator(translatorKey);
            final BigInteger dataPathId = deviceContext.getDeviceInfo().getDatapathId();

            for (final PortGrouping port : connectionContext.getFeatures().getPhyPort()) {
                final FlowCapableNodeConnector fcNodeConnector = translator.translate(port, deviceContext.getDeviceInfo(), null);

                final NodeConnectorId nodeConnectorId = NodeStaticReplyTranslatorUtil.nodeConnectorId(
                        dataPathId.toString(), port.getPortNo(), ofVersion);
                final NodeConnectorBuilder ncBuilder = new NodeConnectorBuilder().setId(nodeConnectorId);
                ncBuilder.addAugmentation(FlowCapableNodeConnector.class, fcNodeConnector);
                ncBuilder.addAugmentation(FlowCapableNodeConnectorStatisticsData.class,
                        new FlowCapableNodeConnectorStatisticsDataBuilder().build());
                final NodeConnector connector = ncBuilder.build();
                final InstanceIdentifier<NodeConnector> connectorII = deviceInfo.getNodeInstanceIdentifier().child(
                        NodeConnector.class, connector.getKey());
                try {
                    deviceContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, connectorII, connector);
                } catch (final Exception e) {
                    LOG.debug("initializeNodeInformation: Failed to write node {} to DS ", deviceInfo.getNodeId().toString(),
                            e);
                }

            }
        } else if (OFConstants.OFP_VERSION_1_3 == version) {
            final Capabilities capabilities = connectionContext.getFeatures().getCapabilities();
            LOG.debug("Setting capabilities for device {}", deviceInfo.getNodeId());
            DeviceStateUtil.setDeviceStateBasedOnV13Capabilities(deviceState, capabilities);
            createDeviceFeaturesForOF13(deviceContext, switchFeaturesMandatory, convertorExecutor).get();
        } else {
            throw new ExecutionException(new ConnectionException("Unsupported version " + version));
        }

    }

    private static void addNodeToOperDS(final DeviceContext deviceContext, final SettableFuture<Void> future) {
        Preconditions.checkArgument(deviceContext != null);
        final NodeBuilder nodeBuilder = new NodeBuilder().setId(deviceContext.getDeviceInfo().getNodeId()).setNodeConnector(
                Collections.<NodeConnector>emptyList());
        try {
            deviceContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(),
                    nodeBuilder.build());
        } catch (final Exception e) {
            LOG.warn("addNodeToOperDS: Failed to write node {} to DS ", deviceContext.getDeviceInfo().getNodeId(), e);
            future.cancel(true);
        }
    }

    private static ListenableFuture<List<RpcResult<List<MultipartReply>>>> createDeviceFeaturesForOF10(
            final DeviceContext deviceContext) {
        final ListenableFuture<RpcResult<List<MultipartReply>>> replyDesc = getNodeStaticInfo(MultipartType.OFPMPDESC,
                deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), deviceContext.getDeviceInfo().getVersion());

        return Futures.allAsList(Arrays.asList(replyDesc));
    }

    private static ListenableFuture<List<RpcResult<List<MultipartReply>>>> createDeviceFeaturesForOF13(
            final DeviceContext deviceContext, final boolean switchFeaturesMandatory, final ConvertorExecutor convertorExecutor) {

        final ListenableFuture<RpcResult<List<MultipartReply>>> replyDesc = getNodeStaticInfo(MultipartType.OFPMPDESC,
                deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), deviceContext.getDeviceInfo().getVersion());

        //first process description reply, write data to DS and write consequent data if successful
        return Futures.transform(replyDesc,
                new AsyncFunction<RpcResult<List<MultipartReply>>, List<RpcResult<List<MultipartReply>>>>() {
                    @Override
                    public ListenableFuture<List<RpcResult<List<MultipartReply>>>> apply(
                            final RpcResult<List<MultipartReply>> rpcResult) throws Exception {

                        translateAndWriteReply(MultipartType.OFPMPDESC, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), rpcResult.getResult(), convertorExecutor);

                        final ListenableFuture<RpcResult<List<MultipartReply>>> replyMeterFeature = getNodeStaticInfo(
                                MultipartType.OFPMPMETERFEATURES, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), deviceContext.getDeviceInfo().getVersion());

                        createSuccessProcessingCallback(MultipartType.OFPMPMETERFEATURES, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), replyMeterFeature, convertorExecutor);

                        final ListenableFuture<RpcResult<List<MultipartReply>>> replyGroupFeatures = getNodeStaticInfo(
                                MultipartType.OFPMPGROUPFEATURES, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), deviceContext.getDeviceInfo().getVersion());
                        createSuccessProcessingCallback(MultipartType.OFPMPGROUPFEATURES, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), replyGroupFeatures, convertorExecutor);

                        final ListenableFuture<RpcResult<List<MultipartReply>>> replyTableFeatures;

                        if (deviceContext.isSkipTableFeatures()){
                            replyTableFeatures = RpcResultBuilder.<List<MultipartReply>>success().buildFuture();
                        } else {
                            replyTableFeatures = getNodeStaticInfo(
                                    MultipartType.OFPMPTABLEFEATURES, deviceContext,
                                    deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), deviceContext.getDeviceInfo().getVersion());
                        }
                        createSuccessProcessingCallback(MultipartType.OFPMPTABLEFEATURES, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), replyTableFeatures, convertorExecutor);

                        final ListenableFuture<RpcResult<List<MultipartReply>>> replyPortDescription = getNodeStaticInfo(
                                MultipartType.OFPMPPORTDESC, deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(),
                                deviceContext.getDeviceInfo().getVersion());
                        createSuccessProcessingCallback(MultipartType.OFPMPPORTDESC, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), replyPortDescription, convertorExecutor);
                        if (switchFeaturesMandatory) {
                            return Futures.allAsList(Arrays.asList(replyMeterFeature, replyGroupFeatures,
                                    replyTableFeatures, replyPortDescription));
                        } else {
                            return Futures.successfulAsList(Arrays.asList(replyMeterFeature, replyGroupFeatures,
                                    replyTableFeatures, replyPortDescription));
                        }
                    }
                });

    }

    static void translateAndWriteReply(final MultipartType type, final DeviceContext dContext,
                                       final InstanceIdentifier<Node> nodeII, final Collection<MultipartReply> result,
                                       final ConvertorExecutor convertorExecutor) {
        try {
            result.stream()
                    .map(MultipartReply::getMultipartReplyBody)
                    .forEach(multipartReplyBody -> {
                        if (!(writeDesc(type, multipartReplyBody, dContext, nodeII)
                                || writeTableFeatures(type, multipartReplyBody, dContext, nodeII, convertorExecutor)
                                || writeMeterFeatures(type, multipartReplyBody, dContext, nodeII)
                                || writeGroupFeatures(type, multipartReplyBody, dContext, nodeII)
                                || writePortDesc(type, multipartReplyBody, dContext, nodeII))) {
                            throw new IllegalArgumentException("Unexpected MultipartType " + type);
                        }
                    });
        } catch (final Exception e) {
            LOG.debug("translateAndWriteReply: Failed to write node {} to DS ", dContext.getDeviceInfo().getNodeId().toString(), e);
        }
    }

    private static boolean writeDesc(final MultipartType type,
                                     final MultipartReplyBody body,
                                     final DeviceContext dContext,
                                     final InstanceIdentifier<Node> nodeII) {
        if (!MultipartType.OFPMPDESC.equals(type)) {
            return false;
        }

        Preconditions.checkArgument(body instanceof MultipartReplyDescCase);
        final MultipartReplyDesc replyDesc = ((MultipartReplyDescCase) body).getMultipartReplyDesc();
        final FlowCapableNode fcNode = NodeStaticReplyTranslatorUtil.nodeDescTranslator(replyDesc,
                getIpAddressOf(dContext));
        final InstanceIdentifier<FlowCapableNode> fNodeII = nodeII.augmentation(FlowCapableNode.class);
        dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, fNodeII, fcNode);
        return true;
    }

    private static boolean writeTableFeatures(final MultipartType type,
                                              final MultipartReplyBody body,
                                              final DeviceContext dContext,
                                              final InstanceIdentifier<Node> nodeII,
                                              final ConvertorExecutor convertorExecutor) {
        if (!MultipartType.OFPMPTABLEFEATURES.equals(type)) {
            return false;
        }

        Preconditions.checkArgument(body instanceof MultipartReplyTableFeaturesCase);
        final MultipartReplyTableFeatures tableFeaturesMP = ((MultipartReplyTableFeaturesCase) body)
                .getMultipartReplyTableFeatures();
        final List<TableFeatures> tableFeatures = NodeStaticReplyTranslatorUtil
                .nodeTableFeatureTranslator(tableFeaturesMP, dContext.getDeviceInfo().getVersion(), convertorExecutor);
        for (final TableFeatures tableFeature : tableFeatures) {
            final Short tableId = tableFeature.getTableId();
            final KeyedInstanceIdentifier<TableFeatures, TableFeaturesKey> tableFeaturesII =
                    nodeII.augmentation(FlowCapableNode.class)
                            .child(TableFeatures.class, new TableFeaturesKey(tableId));
            dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, tableFeaturesII, tableFeature);

            // write parent for table statistics
            final KeyedInstanceIdentifier<Table, TableKey> tableII =
                    nodeII.augmentation(FlowCapableNode.class)
                            .child(Table.class, new TableKey(tableId));
            final TableBuilder tableBld = new TableBuilder().setId(tableId)
                    .addAugmentation(FlowTableStatisticsData.class,
                            new FlowTableStatisticsDataBuilder().build());

            dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, tableII, tableBld.build());
        }

        return true;
    }

    private static boolean writeMeterFeatures(final MultipartType type,
                                              final MultipartReplyBody body,
                                              final DeviceContext dContext,
                                              final InstanceIdentifier<Node> nodeII) {
        if (!MultipartType.OFPMPMETERFEATURES.equals(type)) {
            return false;
        }

        Preconditions.checkArgument(body instanceof MultipartReplyMeterFeaturesCase);
        final MultipartReplyMeterFeatures meterFeatures = ((MultipartReplyMeterFeaturesCase) body)
                .getMultipartReplyMeterFeatures();
        final NodeMeterFeatures mFeature = NodeStaticReplyTranslatorUtil
                .nodeMeterFeatureTranslator(meterFeatures);
        final InstanceIdentifier<NodeMeterFeatures> mFeatureII = nodeII
                .augmentation(NodeMeterFeatures.class);
        dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, mFeatureII, mFeature);
        if (0L < mFeature.getMeterFeatures().getMaxMeter().getValue()) {
            dContext.getDeviceState().setMeterAvailable(true);
        }

        return true;
    }

    private static boolean writeGroupFeatures(final MultipartType type,
                                              final MultipartReplyBody body,
                                              final DeviceContext dContext,
                                              final InstanceIdentifier<Node> nodeII) {
        if (!MultipartType.OFPMPGROUPFEATURES.equals(type)) {
            return false;
        }

        Preconditions.checkArgument(body instanceof MultipartReplyGroupFeaturesCase);
        final MultipartReplyGroupFeatures groupFeatures = ((MultipartReplyGroupFeaturesCase) body)
                .getMultipartReplyGroupFeatures();
        final NodeGroupFeatures gFeature = NodeStaticReplyTranslatorUtil
                .nodeGroupFeatureTranslator(groupFeatures);
        final InstanceIdentifier<NodeGroupFeatures> gFeatureII = nodeII
                .augmentation(NodeGroupFeatures.class);
        dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, gFeatureII, gFeature);

        return true;
    }

    private static boolean writePortDesc(final MultipartType type,
                                         final MultipartReplyBody body,
                                         final DeviceContext dContext,
                                         final InstanceIdentifier<Node> nodeII) {
        if (!MultipartType.OFPMPPORTDESC.equals(type)) {
            return false;
        }

        Preconditions.checkArgument(body instanceof MultipartReplyPortDescCase);
        final MultipartReplyPortDesc portDesc = ((MultipartReplyPortDescCase) body)
                .getMultipartReplyPortDesc();
        for (final PortGrouping port : portDesc.getPorts()) {
            final short ofVersion = dContext.getDeviceInfo().getVersion();
            final TranslatorKey translatorKey = new TranslatorKey(ofVersion, PortGrouping.class.getName());
            final MessageTranslator<PortGrouping, FlowCapableNodeConnector> translator = dContext.oook()
                    .lookupTranslator(translatorKey);
            final FlowCapableNodeConnector fcNodeConnector = translator.translate(port, dContext.getDeviceInfo(), null);

            final BigInteger dataPathId = dContext.getPrimaryConnectionContext().getFeatures()
                    .getDatapathId();
            final NodeConnectorId nodeConnectorId = NodeStaticReplyTranslatorUtil.nodeConnectorId(
                    dataPathId.toString(), port.getPortNo(), ofVersion);
            final NodeConnectorBuilder ncBuilder = new NodeConnectorBuilder().setId(nodeConnectorId);
            ncBuilder.addAugmentation(FlowCapableNodeConnector.class, fcNodeConnector);

            ncBuilder.addAugmentation(FlowCapableNodeConnectorStatisticsData.class,
                    new FlowCapableNodeConnectorStatisticsDataBuilder().build());
            final NodeConnector connector = ncBuilder.build();

            final InstanceIdentifier<NodeConnector> connectorII = nodeII.child(NodeConnector.class,
                    connector.getKey());
            dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, connectorII, connector);
        }

        return true;
    }

    private static void createEmptyFlowCapableNodeInDs(final DeviceContext deviceContext) {
        final FlowCapableNodeBuilder flowCapableNodeBuilder = new FlowCapableNodeBuilder();
        final InstanceIdentifier<FlowCapableNode> fNodeII = deviceContext.getDeviceInfo().getNodeInstanceIdentifier()
                .augmentation(FlowCapableNode.class);
        try {
            deviceContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, fNodeII, flowCapableNodeBuilder.build());
        } catch (final Exception e) {
            LOG.debug("createEmptyFlowCapableNodeInDs: Failed to write node {} to DS ", deviceContext.getDeviceInfo().getNodeId().toString(), e);
        }
    }

    private static IpAddress getIpAddressOf(final DeviceContext deviceContext) {

        final InetSocketAddress remoteAddress = deviceContext.getPrimaryConnectionContext().getConnectionAdapter()
                .getRemoteAddress();

        if (remoteAddress == null) {
            LOG.warn("IP address of the node {} cannot be obtained. No connection with switch.", deviceContext
                    .getDeviceInfo().getNodeId());
            return null;
        }
        LOG.info("IP address of switch is: {}", remoteAddress);

        return IetfInetUtil.INSTANCE.ipAddressFor(remoteAddress.getAddress());
    }

    // FIXME : remove after ovs tableFeatures fix
    private static void makeEmptyTables(final DeviceContext dContext, final InstanceIdentifier<Node> nodeII,
                                        final Short nrOfTables) {
        if (LOG.isDebugEnabled()) {
            LOG.debug("About to create {} empty tables.", nrOfTables);
        }
        for (int i = 0; i < nrOfTables; i++) {
            final short tId = (short) i;
            final InstanceIdentifier<Table> tableII = nodeII.augmentation(FlowCapableNode.class).child(Table.class,
                    new TableKey(tId));
            final TableBuilder tableBuilder = new TableBuilder().setId(tId).addAugmentation(
                    FlowTableStatisticsData.class, new FlowTableStatisticsDataBuilder().build());

            try {
                dContext.writeToTransaction(LogicalDatastoreType.OPERATIONAL, tableII, tableBuilder.build());
            } catch (final Exception e) {
                LOG.debug("makeEmptyTables: Failed to write node {} to DS ", dContext.getDeviceInfo().getNodeId().toString(), e);
            }

        }
    }

    static void createSuccessProcessingCallback(final MultipartType type, final DeviceContext deviceContext,
                                                final InstanceIdentifier<Node> nodeII,
                                                final ListenableFuture<RpcResult<List<MultipartReply>>> requestContextFuture,
                                                final ConvertorExecutor convertorExecutor) {
        Futures.addCallback(requestContextFuture, new FutureCallback<RpcResult<List<MultipartReply>>>() {
            @Override
            public void onSuccess(final RpcResult<List<MultipartReply>> rpcResult) {
                final List<MultipartReply> result = rpcResult.getResult();
                if (result != null) {
                    LOG.info("Static node {} info: {} collected", deviceContext.getDeviceInfo().getNodeId(), type);
                    translateAndWriteReply(type, deviceContext, nodeII, result, convertorExecutor);
                } else {
                    for (RpcError rpcError : rpcResult.getErrors()) {
                        LOG.info("Failed to retrieve static node {} info: {}", type, rpcError.getMessage());
                        if (LOG.isTraceEnabled() && Objects.nonNull(rpcError.getCause())) {
                            LOG.trace("Detailed error:", rpcError.getCause());
                        }
                    }
                    if (MultipartType.OFPMPTABLEFEATURES.equals(type)) {
                        makeEmptyTables(deviceContext, nodeII, deviceContext.getPrimaryConnectionContext()
                                .getFeatures().getTables());
                    }
                }
            }

            @Override
            public void onFailure(final Throwable throwable) {
                LOG.info("Request of type {} for static info of node {} failed.", type, nodeII);
            }
        });
    }

    private static ListenableFuture<RpcResult<List<MultipartReply>>> getNodeStaticInfo(final MultipartType type,
                                                                                       final DeviceContext deviceContext,
                                                                                       final InstanceIdentifier<Node> nodeII,
                                                                                       final short version) {

        final OutboundQueue queue = deviceContext.getPrimaryConnectionContext().getOutboundQueueProvider();

        final Long reserved = deviceContext.getDeviceInfo().reserveXidForDeviceMessage();
        final RequestContext<List<MultipartReply>> requestContext = new AbstractRequestContext<List<MultipartReply>>(
                reserved) {
            @Override
            public void close() {
                //NOOP
            }
        };

        final Xid xid = requestContext.getXid();

        LOG.trace("Hooking xid {} to device context - precaution.", reserved);

        final MultiMsgCollector multiMsgCollector = deviceContext.getMultiMsgCollector(requestContext);
        queue.commitEntry(xid.getValue(),
                MultipartRequestInputFactory.makeMultipartRequestInput(xid.getValue(), version, type),
                new FutureCallback<OfHeader>() {
                    @Override
                    public void onSuccess(final OfHeader ofHeader) {
                        if (ofHeader instanceof MultipartReply) {
                            final MultipartReply multipartReply = (MultipartReply) ofHeader;
                            multiMsgCollector.addMultipartMsg(multipartReply);
                        } else if (null != ofHeader) {
                            LOG.info("Unexpected response type received {}.", ofHeader.getClass());
                        } else {
                            multiMsgCollector.endCollecting();
                            LOG.info("Response received is null.");
                        }
                    }

                    @Override
                    public void onFailure(final Throwable t) {
                        LOG.info("Fail response from OutboundQueue for multipart type {}.", type);
                        final RpcResult<List<MultipartReply>> rpcResult = RpcResultBuilder
                                .<List<MultipartReply>>failed().build();
                        requestContext.setResult(rpcResult);
                        if (MultipartType.OFPMPTABLEFEATURES.equals(type)) {
                            makeEmptyTables(deviceContext, nodeII, deviceContext.getPrimaryConnectionContext()
                                    .getFeatures().getTables());
                        }
                        requestContext.close();
                    }
                });

        return requestContext.getFuture();
    }

    static void chainTableTrunkWriteOF10(final DeviceContext deviceContext,
                                         final ListenableFuture<List<RpcResult<List<MultipartReply>>>> deviceFeaturesFuture) {

        try {
            if (LOG.isTraceEnabled()) {
                LOG.trace("Waiting for protocol version 1.0");
            }
            List<RpcResult<List<MultipartReply>>> results = deviceFeaturesFuture.get();
            boolean allSucceeded = true;
            for (final RpcResult<List<MultipartReply>> rpcResult : results) {
                allSucceeded &= rpcResult.isSuccessful();
            }
            if (allSucceeded) {
                if (LOG.isDebugEnabled()) {
                    LOG.debug("Creating empty flow capable node: {}", deviceContext.getDeviceInfo().getLOGValue());
                }
                createEmptyFlowCapableNodeInDs(deviceContext);
                if (LOG.isDebugEnabled()) {
                    LOG.debug("Creating empty tables for {}", deviceContext.getDeviceInfo().getLOGValue());
                }
                makeEmptyTables(deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(),
                        deviceContext.getPrimaryConnectionContext().getFeatures().getTables());
            }
        } catch (InterruptedException | ExecutionException e) {
            LOG.warn("Error occurred in preparation node {} for protocol 1.0", deviceContext.getDeviceInfo().getLOGValue());
            if (LOG.isTraceEnabled()) {
                LOG.trace("Error for node {} : ", deviceContext.getDeviceInfo().getLOGValue(), e);
            }
        }
    }
}