/*
 * Copyright © 2018 Orange 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.transportpce.renderer.utils;

import java.util.Optional;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.transportpce.common.Timeouts;
import org.opendaylight.transportpce.common.device.DeviceTransaction;
import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;

public final class TransactionUtils {

    private TransactionUtils() {

    }

    public static boolean writeTransaction(DeviceTransactionManager deviceTransactionManager,
                                    String nodeId,
                                    LogicalDatastoreType logicalDatastoreType,
                                    InstanceIdentifier instanceIdentifier,
                                    DataObject object)
            throws ExecutionException, InterruptedException {
        Future<Optional<DeviceTransaction>> deviceTxFuture =
                deviceTransactionManager.getDeviceTransaction(nodeId);
        if (!deviceTxFuture.get().isPresent()) {
            return false;
        }
        DeviceTransaction deviceTx = deviceTxFuture.get().get();
        deviceTx.put(logicalDatastoreType, instanceIdentifier, object, true);
        deviceTx.submit(Timeouts.DEVICE_WRITE_TIMEOUT, Timeouts.DEVICE_WRITE_TIMEOUT_UNIT).get();
        return true;
    }

    public static DataObject readTransaction(DeviceTransactionManager deviceTransactionManager,
                                  String nodeId,
                                  LogicalDatastoreType logicalDatastoreType,
                                  InstanceIdentifier<? extends DataObject> instanceIdentifier)
            throws ExecutionException, InterruptedException {
        Future<Optional<DeviceTransaction>> deviceTxFuture =
                deviceTransactionManager.getDeviceTransaction(nodeId);
        if (!deviceTxFuture.get().isPresent()) {
            return null;
        }
        DeviceTransaction deviceTx = deviceTxFuture.get().get();
        com.google.common.base.Optional<? extends DataObject> readOpt
                = deviceTx.read(logicalDatastoreType, instanceIdentifier).get();
        if (!readOpt.isPresent()) {
            return null;
        }
        return readOpt.get();
    }

}