/*
 * 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.Iterator;
import java.util.List;
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.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);

    /**
     * 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
     * @return future - recommended to have blocking call for this future
     */
    public static ListenableFuture<Void> initializeNodeInformation(final DeviceContext deviceContext, final boolean switchFeaturesMandatory) {
        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, deviceState);
            // 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("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);
            deviceFeaturesFuture = createDeviceFeaturesForOF13(deviceContext, deviceState, switchFeaturesMandatory);
        } else {
            deviceFeaturesFuture = Futures.immediateFailedFuture(new ConnectionException("Unsupported version "
                    + version));
        }

        Futures.addCallback(deviceFeaturesFuture, new FutureCallback<List<RpcResult<List<MultipartReply>>>>() {
            @Override
            public void onSuccess(final List<RpcResult<List<MultipartReply>>> result) {
                LOG.debug("All init data for node {} is in submited.", deviceInfo.getNodeId());
                returnFuture.set(null);
            }

            @Override
            public void onFailure(final Throwable t) {
                // FIXME : remove session
                LOG.trace("Device capabilities gathering future failed.");
                LOG.trace("more info in exploration failure..", t);
                LOG.debug("All init data for node {} was not submited correctly - connection has to go down.", deviceInfo.getNodeId());
                returnFuture.setException(t);
            }
        });
        return returnFuture;
    }

    private static void addNodeToOperDS(final DeviceContext deviceContext, final SettableFuture<Void> future) {
        Preconditions.checkArgument(deviceContext != null);
        final DeviceState deviceState = deviceContext.getDeviceState();
        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("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 DeviceState deviceState) {
        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 DeviceState deviceState, final boolean switchFeaturesMandatory) {

        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());

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

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

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

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

                        final ListenableFuture<RpcResult<List<MultipartReply>>> replyPortDescription = getNodeStaticInfo(
                                MultipartType.OFPMPPORTDESC, deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(),
                                deviceContext.getDeviceInfo().getVersion());
                        createSuccessProcessingCallback(MultipartType.OFPMPPORTDESC, deviceContext,
                                deviceContext.getDeviceInfo().getNodeInstanceIdentifier(), replyPortDescription);
                        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) {
        try {
            for (final MultipartReply reply : result) {
                final MultipartReplyBody body = reply.getMultipartReplyBody();
                switch (type) {
                    case OFPMPDESC:
                        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);
                        break;

                    case OFPMPTABLEFEATURES:
                        Preconditions.checkArgument(body instanceof MultipartReplyTableFeaturesCase);
                        final MultipartReplyTableFeatures tableFeaturesMP = ((MultipartReplyTableFeaturesCase) body)
                                .getMultipartReplyTableFeatures();
                        final List<TableFeatures> tableFeatures = NodeStaticReplyTranslatorUtil
                                .nodeTableFeatureTranslator(tableFeaturesMP);
                        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());
                        }
                        break;

                    case OFPMPMETERFEATURES:
                        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);
                        }
                        break;

                    case OFPMPGROUPFEATURES:
                        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);
                        break;

                    case OFPMPPORTDESC:
                        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);
                        }

                        break;

                    default:
                        throw new IllegalArgumentException("Unnexpected MultipartType " + type);
                }
            }
        } catch (final Exception e) {
            LOG.debug("Failed to write node {} to DS ", dContext.getDeviceInfo().getNodeId().toString(), e);
        }
    }

    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("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) {
        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("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) {
        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);
                } else {
                    final Iterator<RpcError> rpcErrorIterator = rpcResult.getErrors().iterator();
                    while (rpcErrorIterator.hasNext()) {
                        final RpcError rpcError = rpcErrorIterator.next();
                        LOG.info("Failed to retrieve static node {} info: {}", type, rpcError.getMessage());
                        if (null != 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.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) {
        Futures.addCallback(deviceFeaturesFuture, new FutureCallback<List<RpcResult<List<MultipartReply>>>>() {
            @Override
            public void onSuccess(final List<RpcResult<List<MultipartReply>>> results) {
                boolean allSucceeded = true;
                for (final RpcResult<List<MultipartReply>> rpcResult : results) {
                    allSucceeded &= rpcResult.isSuccessful();
                }
                if (allSucceeded) {
                    createEmptyFlowCapableNodeInDs(deviceContext);
                    makeEmptyTables(deviceContext, deviceContext.getDeviceInfo().getNodeInstanceIdentifier(),
                            deviceContext.getPrimaryConnectionContext().getFeatures().getTables());
                }
            }

            @Override
            public void onFailure(final Throwable t) {
                //NOOP
            }
        });
    }
}