/**
 * 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.applications.frsync.impl;

import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.FutureCallback;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import java.util.concurrent.Semaphore;
import java.util.concurrent.TimeUnit;
import javax.annotation.Nullable;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.openflowplugin.applications.frsync.SemaphoreKeeper;
import org.opendaylight.openflowplugin.applications.frsync.SyncReactor;
import org.opendaylight.openflowplugin.applications.frsync.util.PathUtil;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Decorator for NodeId level syncup locking.
 */
public class SyncReactorGuardDecorator implements SyncReactor {

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

    private final SyncReactor delegate;
    private final SemaphoreKeeper<InstanceIdentifier<FlowCapableNode>> semaphoreKeeper;

    public SyncReactorGuardDecorator(SyncReactor delegate,
            SemaphoreKeeper<InstanceIdentifier<FlowCapableNode>> semaphoreKeeper) {
        this.delegate = delegate;
        this.semaphoreKeeper = semaphoreKeeper;
    }

    public ListenableFuture<Boolean> syncup(final InstanceIdentifier<FlowCapableNode> flowcapableNodePath,
                                            final FlowCapableNode configTree, final FlowCapableNode operationalTree,
                                            final LogicalDatastoreType dsType) throws InterruptedException {
        final NodeId nodeId = PathUtil.digNodeId(flowcapableNodePath);
        LOG.trace("syncup guard {}", nodeId.getValue());

        final long stampBeforeGuard = System.nanoTime();
        final Semaphore guard = summonGuardAndAcquire(flowcapableNodePath);//TODO handle InteruptedException

        try {
            final long stampAfterGuard = System.nanoTime();
            if (LOG.isDebugEnabled()) {
                LOG.debug("syncup start {} waiting:{} guard:{} thread:{}", nodeId.getValue(),
                        formatNanos(stampAfterGuard - stampBeforeGuard),
                        guard, threadName());
            }

            final ListenableFuture<Boolean> endResult =
                    delegate.syncup(flowcapableNodePath, configTree, operationalTree, dsType);//TODO handle InteruptedException

            Futures.addCallback(endResult, new FutureCallback<Boolean>() {
                @Override
                public void onSuccess(@Nullable final Boolean result) {
                    if (LOG.isDebugEnabled()) {
                        final long stampFinished = System.nanoTime();
                        LOG.debug("syncup finished {} took:{} rpc:{} wait:{} guard:{} permits thread:{}", nodeId.getValue(),
                                formatNanos(stampFinished - stampBeforeGuard),
                                formatNanos(stampFinished - stampAfterGuard),
                                formatNanos(stampAfterGuard - stampBeforeGuard),
                                guard.availablePermits(), threadName());
                    }

                    releaseGuardForNodeId(guard);
                }

                @Override
                public void onFailure(final Throwable t) {
                    if (LOG.isDebugEnabled()) {
                        final long stampFinished = System.nanoTime();
                        LOG.warn("syncup failed {} took:{} rpc:{} wait:{} guard:{} permits thread:{}", nodeId.getValue(),
                                formatNanos(stampFinished - stampBeforeGuard),
                                formatNanos(stampFinished - stampAfterGuard),
                                formatNanos(stampAfterGuard - stampBeforeGuard),
                                guard.availablePermits(), threadName());
                    }

                    releaseGuardForNodeId(guard);
                }
            });
            return endResult;
        } catch(InterruptedException e) {
            releaseGuardForNodeId(guard);
            throw e;
        }
    }

    private String formatNanos(long nanos) {
        return "'" + TimeUnit.NANOSECONDS.toMillis(nanos) + " ms'";
    }

    /**
     * Get guard and lock for node.
     * @param flowcapableNodePath II of node for which guard should be acquired
     * @return semaphore guard
     */
    private Semaphore summonGuardAndAcquire(final InstanceIdentifier<FlowCapableNode> flowcapableNodePath)
            throws InterruptedException {
        final Semaphore guard = Preconditions.checkNotNull(semaphoreKeeper.summonGuard(flowcapableNodePath),
                "no guard for " + flowcapableNodePath);

        if (LOG.isDebugEnabled()) {
            final NodeId nodeId = PathUtil.digNodeId(flowcapableNodePath);
            try {
                LOG.debug("syncup summon {} guard:{} thread:{}", nodeId.getValue(), guard, threadName());
            } catch (Exception e) {
                LOG.error("error logging guard after summon before aquiring {}", nodeId);
            }
        }

        guard.acquire();
        return guard;
    }

    /**
     * Unlock and release guard.
     * @param guard semaphore guard which should be unlocked
     */
    private void releaseGuardForNodeId(final Semaphore guard) {
        if (guard == null) {
            return;
        }
        guard.release();
    }

    private static String threadName() {
        final Thread currentThread = Thread.currentThread();
        return currentThread.getName();
    }

}