/*
 * Copyright (c) 2013 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.bgpcep.pcep.topology.provider;

import io.netty.util.concurrent.EventExecutor;
import io.netty.util.concurrent.FutureListener;
import io.netty.util.concurrent.GlobalEventExecutor;
import io.netty.util.concurrent.Promise;

import java.net.InetAddress;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;

import javax.annotation.concurrent.GuardedBy;

import org.opendaylight.bgpcep.programming.spi.ExecutionResult;
import org.opendaylight.bgpcep.programming.spi.InstructionExecutor;
import org.opendaylight.bgpcep.programming.spi.InstructionScheduler;
import org.opendaylight.bgpcep.programming.spi.SuccessfulRpcResult;
import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
import org.opendaylight.controller.sal.binding.api.data.DataProviderService;
import org.opendaylight.protocol.framework.SessionListenerFactory;
import org.opendaylight.protocol.pcep.PCEPSession;
import org.opendaylight.protocol.pcep.PCEPSessionListener;
import org.opendaylight.protocol.pcep.PCEPTerminationReason;
import org.opendaylight.protocol.pcep.TerminationReason;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcerr;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcinitiateBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcupdBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.MessageHeader;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PcrptMessage;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.PlspId;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ProtocolVersion;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.SrpIdNumber;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.object.TlvsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.Tlvs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.PcinitiateMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.pcinitiate.message.RequestsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcinitiate.message.pcinitiate.message.requests.LspBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.Reports;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.pcrpt.message.reports.Lsp;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcupd.message.PcupdMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcupd.message.pcupd.message.UpdatesBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcupd.message.pcupd.message.updates.PathBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.srp.object.Srp;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.srp.object.Srp.Flags;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.srp.object.SrpBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.stateful.capability.tlv.Stateful;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.symbolic.path.name.tlv.SymbolicPathName;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.symbolic.path.name.tlv.SymbolicPathNameBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev130930.InstructionStatus;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev130930.instruction.status.changed.Details;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev130930.submit.instruction.output.result.FailureBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.programming.rev130930.submit.instruction.output.result.failure.Failure;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.AddLspInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.AddLspOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.AddLspOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.NetworkTopologyPcepService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.Node1;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.Node1Builder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.PccSyncState;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.RemoveLspInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.RemoveLspOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.RemoveLspOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspOutputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.PathComputationClientBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLsps;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLspsKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.StatefulTlvBuilder;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
import org.opendaylight.yangtools.yang.binding.DataContainer;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.common.RpcResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;

/**
 *
 */
final class ServerSessionManager implements SessionListenerFactory<PCEPSessionListener>, NetworkTopologyPcepService {
	private static String createNodeId(final InetAddress addr) {
		return "pcc://" + addr.getHostAddress();
	}

	private final class SessionListener implements PCEPSessionListener {
		private final Map<SrpIdNumber, Promise<ExecutionResult<Details>>> waitingRequests = new HashMap<>();
		private final Map<SrpIdNumber, Promise<ExecutionResult<Details>>> sendingRequests = new HashMap<>();
		private final Map<PlspId, SymbolicPathName> lsps = new HashMap<>();
		private PathComputationClientBuilder pccBuilder;
		private InstanceIdentifier<Node1> topologyAugment;
		private InstanceIdentifier<Node> topologyNode;
		private Node1Builder topologyAugmentBuilder;
		private boolean ownsTopology = false;
		private boolean synced = false;
		private PCEPSession session;
		private long requestId = 1;
		private NodeId nodeId;

		final Node topologyNode(final DataModificationTransaction trans, final InetAddress address) {
			final String pccId = createNodeId(address);
			final Topology topo = (Topology) trans.readOperationalData(ServerSessionManager.this.topology);

			for (final Node n : topo.getNode()) {
				LOG.debug("Matching topology node {} to id {}", n, pccId);
				if (n.getNodeId().getValue().equals(pccId)) {
					return n;
				}
			}

			/*
			 * We failed to find a matching node. Let's create a dynamic one
			 * and note that we are the owner (so we clean it up afterwards).
			 */
			final NodeId id = new NodeId(pccId);
			final NodeKey nk = new NodeKey(id);
			final InstanceIdentifier<Node> nti = InstanceIdentifier.builder(ServerSessionManager.this.topology).node(Node.class, nk).toInstance();

			final Node ret = new NodeBuilder().setKey(nk).setNodeId(id).build();

			trans.putRuntimeData(nti, ret);
			this.ownsTopology = true;
			this.topologyNode = nti;
			this.nodeId = id;
			return ret;
		}

		@Override
		public synchronized void onSessionUp(final PCEPSession session) {
			/*
			 * The session went up. Look up the router in Inventory model,
			 * create it if it is not there (marking that fact for later
			 * deletion), and mark it as synchronizing. Also create it in
			 * the topology model, with empty LSP list.
			 */
			final InetAddress peerAddress = session.getRemoteAddress();
			final DataModificationTransaction trans = ServerSessionManager.this.dataProvider.beginTransaction();

			final Node topoNode = topologyNode(trans, peerAddress);
			LOG.debug("Peer {} resolved to topology node {}", peerAddress, topoNode);

			// Our augmentation in the topology node
			this.pccBuilder = new PathComputationClientBuilder();

			final Tlvs tlvs = session.getRemoteTlvs();
			final Stateful stateful = tlvs.getStateful();
			if (stateful != null) {
				this.pccBuilder.setStatefulTlv(new StatefulTlvBuilder(tlvs).build());
				this.pccBuilder.setStateSync(PccSyncState.InitialResync);
			}

			this.topologyAugmentBuilder = new Node1Builder().setPathComputationClient(this.pccBuilder.build());
			this.topologyAugment = InstanceIdentifier.builder(this.topologyNode).node(Node1.class).toInstance();
			trans.putRuntimeData(this.topologyAugment, this.topologyAugmentBuilder.build());

			// All set, commit the modifications
			final Future<RpcResult<TransactionStatus>> s = trans.commit();

			/*
			 * FIXME: once this Future is listenable, attach to it so we can
			 *        do cleanup if the commit fails. For now we force a commit.
			 */
			try {
				s.get();
			} catch (InterruptedException | ExecutionException e) {
				LOG.error("Failed to update internal state for session {}, terminating it", session, e);
				session.close(TerminationReason.Unknown);
				return;
			}

			ServerSessionManager.this.nodes.put(this.nodeId, this);
			this.session = session;
			LOG.info("Session with {} attached to topology node {}", session.getRemoteAddress(), topoNode.getNodeId());
		}

		@GuardedBy("this")
		private void tearDown(final PCEPSession session) {
			this.session = null;
			ServerSessionManager.this.nodes.remove(this.nodeId);

			final DataModificationTransaction trans = ServerSessionManager.this.dataProvider.beginTransaction();

			// The session went down. Undo all the Topology changes we have done.
			trans.removeRuntimeData(this.topologyAugment);
			if (this.ownsTopology) {
				trans.removeRuntimeData(this.topologyNode);
			}

			/*
			 * FIXME: once this Future is listenable, attach to it so we can
			 *        do cleanup if the commit fails. For now we force a commit.
			 */
			final Future<RpcResult<TransactionStatus>> s = trans.commit();
			try {
				s.get();
			} catch (InterruptedException | ExecutionException e) {
				LOG.error("Failed to cleanup internal state for session {}", session, e);
			}

			// Clear all requests which have not been sent to the peer: they result in cancellation
			for (final Entry<SrpIdNumber, Promise<ExecutionResult<Details>>> e : this.sendingRequests.entrySet()) {
				LOG.debug("Request {} was not sent when session went down, cancelling the instruction", e.getKey());
				e.getValue().setSuccess(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
			}
			this.sendingRequests.clear();

			// CLear all requests which have not been acked by the peer: they result in failure
			for (final Entry<SrpIdNumber, Promise<ExecutionResult<Details>>> e : this.waitingRequests.entrySet()) {
				LOG.info("Request {} was incomplete when session went down, failing the instruction", e.getKey());
				e.getValue().setSuccess(new ExecutionResult<Details>(InstructionStatus.Failed, null));
			}
			this.waitingRequests.clear();
		}

		@Override
		public synchronized void onSessionDown(final PCEPSession session, final Exception e) {
			LOG.warn("Session {} went down unexpectedly", e);
			tearDown(session);
		}

		@Override
		public synchronized void onSessionTerminated(final PCEPSession session, final PCEPTerminationReason reason) {
			LOG.info("Session {} terminated by peer with reason {}", session, reason);
			tearDown(session);
		}

		private InstanceIdentifier<ReportedLsps> lspIdentifier(final SymbolicPathName name) {
			return InstanceIdentifier.builder(this.topologyAugment).node(ReportedLsps.class, new ReportedLspsKey(name.getPathName())).toInstance();
		}

		@Override
		public synchronized void onMessage(final PCEPSession session, final Message message) {
			if (!(message instanceof PcrptMessage)) {
				LOG.info("Unhandled message {} on session {}", message, session);
				session.sendMessage(unhandledMessageError);
			}

			final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrpt.message.PcrptMessage rpt = ((PcrptMessage) message).getPcrptMessage();

			final DataModificationTransaction trans = ServerSessionManager.this.dataProvider.beginTransaction();

			for (final Reports r : rpt.getReports()) {
				final Lsp lsp = r.getLsp();

				if (lsp.isSync() && !this.synced) {
					// Update synchronization flag
					this.synced = true;
					this.topologyAugmentBuilder.setPathComputationClient(this.pccBuilder.setStateSync(PccSyncState.Synchronized).build());
					trans.putRuntimeData(this.topologyAugment, this.topologyAugmentBuilder.build());
					LOG.debug("Session {} achieved synchronized state", session);
				}

				final Srp srp = r.getSrp();
				if (srp != null) {
					final SrpIdNumber id = srp.getOperationId();
					if (id.getValue() != 0) {
						switch (lsp.getOperational()) {
						case Active:
						case Down:
						case Up:
							final Promise<ExecutionResult<Details>> p = this.waitingRequests.remove(id);
							if (p != null) {
								LOG.debug("Request {} resulted in LSP operational state {}", id, lsp.getOperational());
								p.setSuccess(new ExecutionResult<Details>(InstructionStatus.Successful, null));
							} else {
								LOG.warn("Request ID {} not found in outstanding DB", id);
							}
							break;
						case GoingDown:
						case GoingUp:
							// These are transitive states, so
							break;
						}
					}
				}

				final PlspId id = lsp.getPlspId();
				if (lsp.isRemove()) {
					final SymbolicPathName name = this.lsps.remove(id);
					if (name != null) {
						trans.removeRuntimeData(lspIdentifier(name));
					}

					LOG.debug("LSP {} removed", lsp);
				} else {
					if (!this.lsps.containsKey(id)) {
						LOG.debug("PLSPID {} not known yet, looking for a symbolic name", id);

						final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.object.Tlvs tlvs = r.getLsp().getTlvs();
						final SymbolicPathName name = tlvs.getSymbolicPathName();
						if (name == null) {
							LOG.error("PLSPID {} seen for the first time, not reporting the LSP");
							// TODO: what should we do here?
							continue;
						}
					}

					final SymbolicPathName name = this.lsps.get(id);
					trans.putRuntimeData(lspIdentifier(name), lsp);

					LOG.debug("LSP {} updated");
				}
			}

			/*
			 * FIXME: once this Future is listenable, attach to it so we can
			 *        do cleanup if the commit fails. For now we force a commit.
			 */
			final Future<RpcResult<TransactionStatus>> s = trans.commit();
			try {
				s.get();
			} catch (InterruptedException | ExecutionException e) {
				LOG.error("Failed to update internal state for session {}, closing it", session, e);
				session.close(TerminationReason.Unknown);
			}
		}

		private synchronized SrpIdNumber nextRequest() {
			return new SrpIdNumber(this.requestId++);
		}

		private synchronized void messageSendingComplete(final SrpIdNumber requestId, final io.netty.util.concurrent.Future<Void> future) {
			final Promise<ExecutionResult<Details>> promise = this.sendingRequests.remove(requestId);

			if (future.isSuccess()) {
				this.waitingRequests.put(requestId, promise);
			} else {
				LOG.info("Failed to send request {}, instruction cancelled", requestId, future.cause());
				promise.setSuccess(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
			}
		}

		private synchronized io.netty.util.concurrent.Future<ExecutionResult<Details>> sendMessage(final Message message,
				final SrpIdNumber requestId) {
			final io.netty.util.concurrent.Future<Void> f = this.session.sendMessage(message);
			final Promise<ExecutionResult<Details>> ret = exec.newPromise();

			this.sendingRequests.put(requestId, ret);

			f.addListener(new FutureListener<Void>() {
				@Override
				public void operationComplete(final io.netty.util.concurrent.Future<Void> future) {
					messageSendingComplete(requestId, future);
				}
			});

			return ret;
		}
	}

	private static final Logger LOG = LoggerFactory.getLogger(ServerSessionManager.class);
	private static final Pcerr unhandledMessageError = new PcerrBuilder().setPcerrMessage(
			new PcerrMessageBuilder().setErrorType(null).build()).build();
	private static final MessageHeader messageHeader = new MessageHeader() {
		private final ProtocolVersion version = new ProtocolVersion((short) 1);

		@Override
		public Class<? extends DataContainer> getImplementedInterface() {
			return MessageHeader.class;
		}

		@Override
		public ProtocolVersion getVersion() {
			return this.version;
		}
	};
	private static final EventExecutor exec = GlobalEventExecutor.INSTANCE;
	private final Map<NodeId, SessionListener> nodes = new HashMap<>();
	private final InstanceIdentifier<Topology> topology;
	private final DataProviderService dataProvider;
	private final InstructionScheduler scheduler;

	public ServerSessionManager(final InstructionScheduler scheduler, final DataProviderService dataProvider,
			final InstanceIdentifier<Topology> topology) {
		this.dataProvider = Preconditions.checkNotNull(dataProvider);
		this.topology = Preconditions.checkNotNull(topology);
		this.scheduler = Preconditions.checkNotNull(scheduler);
	}

	@Override
	public PCEPSessionListener getSessionListener() {
		return new SessionListener();
	}

	private synchronized io.netty.util.concurrent.Future<ExecutionResult<Details>> realAddLsp(final AddLspInput input) {
		// Get the listener corresponding to the node
		final SessionListener l = this.nodes.get(input.getNode());
		if (l == null) {
			LOG.debug("Session for node {} not found", input.getNode());
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final InstanceIdentifier<ReportedLsps> lsp = InstanceIdentifier.builder(l.topologyAugment).node(ReportedLsps.class,
				new ReportedLspsKey(input.getName())).toInstance();
		if (this.dataProvider.readOperationalData(lsp) != null) {
			LOG.debug("Node {} already contains lsp {} at {}", input.getNode(), input.getName(), lsp);
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final RequestsBuilder rb = new RequestsBuilder(input.getArguments());
		rb.setSrp(new SrpBuilder().setOperationId(l.nextRequest()).setProcessingRule(Boolean.TRUE).build());
		rb.setLsp(new LspBuilder().setAdministrative(input.getArguments().isAdministrative()).setTlvs(
				new TlvsBuilder().setSymbolicPathName(new SymbolicPathNameBuilder().setPathName(input.getName()).build()).build()).build());

		final PcinitiateMessageBuilder ib = new PcinitiateMessageBuilder(messageHeader);
		ib.setRequests(ImmutableList.of(rb.build()));

		return l.sendMessage(new PcinitiateBuilder().setPcinitiateMessage(ib.build()).build(), rb.getSrp().getOperationId());
	}

	private synchronized io.netty.util.concurrent.Future<ExecutionResult<Details>> realRemoveLsp(final RemoveLspInput input) {
		// Get the listener corresponding to the node
		final SessionListener l = this.nodes.get(input.getNode());
		if (l == null) {
			LOG.debug("Session for node {} not found", input.getNode());
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final InstanceIdentifier<ReportedLsps> lsp = InstanceIdentifier.builder(l.topologyAugment).node(ReportedLsps.class,
				new ReportedLspsKey(input.getName())).toInstance();
		final ReportedLsps rep = (ReportedLsps) this.dataProvider.readOperationalData(lsp);
		if (rep == null) {
			LOG.debug("Node {} does not contain LSP {}", input.getNode(), input.getName());
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final RequestsBuilder rb = new RequestsBuilder();
		rb.setSrp(new SrpBuilder().setOperationId(l.nextRequest()).setProcessingRule(Boolean.TRUE).setFlags(new Flags(Boolean.TRUE)).build());
		rb.setLsp(new LspBuilder().setRemove(Boolean.TRUE).setPlspId(rep.getLsp().getPlspId()).setDelegate(Boolean.TRUE).build());

		final PcinitiateMessageBuilder ib = new PcinitiateMessageBuilder(messageHeader);
		ib.setRequests(ImmutableList.of(rb.build()));
		return l.sendMessage(new PcinitiateBuilder().setPcinitiateMessage(ib.build()).build(), rb.getSrp().getOperationId());
	}

	private synchronized io.netty.util.concurrent.Future<ExecutionResult<Details>> realUpdateLsp(final UpdateLspInput input) {
		// Get the listener corresponding to the node
		final SessionListener l = this.nodes.get(input.getNode());
		if (l == null) {
			LOG.debug("Session for node {} not found", input.getNode());
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final InstanceIdentifier<ReportedLsps> lsp = InstanceIdentifier.builder(l.topologyAugment).node(ReportedLsps.class,
				new ReportedLspsKey(input.getName())).toInstance();
		final ReportedLsps rep = (ReportedLsps) this.dataProvider.readOperationalData(lsp);
		if (rep == null) {
			LOG.debug("Node {} does not contain LSP {}", input.getNode(), input.getName());
			return exec.newSucceededFuture(new ExecutionResult<Details>(InstructionStatus.Cancelled, null));
		}

		final UpdatesBuilder rb = new UpdatesBuilder();
		rb.setSrp(new SrpBuilder().setOperationId(l.nextRequest()).setProcessingRule(Boolean.TRUE).build());
		rb.setLsp(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcupd.message.pcupd.message.updates.LspBuilder().setPlspId(
				rep.getLsp().getPlspId()).setDelegate(Boolean.TRUE).build());
		rb.setPath(new PathBuilder(input.getArguments()).build());

		final PcupdMessageBuilder ub = new PcupdMessageBuilder(messageHeader);
		ub.setUpdates(ImmutableList.of(rb.build()));
		return l.sendMessage(new PcupdBuilder().setPcupdMessage(ub.build()).build(), rb.getSrp().getOperationId());
	}

	@Override
	public Future<RpcResult<AddLspOutput>> addLsp(final AddLspInput input) {
		Preconditions.checkArgument(input.getNode() != null);
		Preconditions.checkArgument(input.getName() != null);

		final InstructionExecutor e = new InstructionExecutor() {
			@Override
			public io.netty.util.concurrent.Future<ExecutionResult<Details>> execute() {
				return realAddLsp(input);
			}
		};

		final Failure f = this.scheduler.submitInstruction(input, e);
		final AddLspOutputBuilder b = new AddLspOutputBuilder();
		if (f != null) {
			b.setResult(new FailureBuilder().setFailure(f).build());
		}

		final RpcResult<AddLspOutput> res = SuccessfulRpcResult.create(b.build());
		return exec.newSucceededFuture(res);
	}

	@Override
	public Future<RpcResult<RemoveLspOutput>> removeLsp(final RemoveLspInput input) {
		Preconditions.checkArgument(input.getNode() != null);
		Preconditions.checkArgument(input.getName() != null);

		final InstructionExecutor e = new InstructionExecutor() {
			@Override
			public io.netty.util.concurrent.Future<ExecutionResult<Details>> execute() {
				return realRemoveLsp(input);
			}
		};

		final Failure f = this.scheduler.submitInstruction(input, e);
		final RemoveLspOutputBuilder b = new RemoveLspOutputBuilder();
		if (f != null) {
			b.setResult(new FailureBuilder().setFailure(f).build());
		}

		final RpcResult<RemoveLspOutput> res = SuccessfulRpcResult.create(b.build());
		return exec.newSucceededFuture(res);
	}

	@Override
	public Future<RpcResult<UpdateLspOutput>> updateLsp(final UpdateLspInput input) {
		Preconditions.checkArgument(input.getNode() != null);
		Preconditions.checkArgument(input.getName() != null);

		final InstructionExecutor e = new InstructionExecutor() {
			@Override
			public io.netty.util.concurrent.Future<ExecutionResult<Details>> execute() {
				return realUpdateLsp(input);
			}
		};

		final Failure f = this.scheduler.submitInstruction(input, e);
		final UpdateLspOutputBuilder b = new UpdateLspOutputBuilder();
		if (f != null) {
			b.setResult(new FailureBuilder().setFailure(f).build());
		}

		final RpcResult<UpdateLspOutput> res = SuccessfulRpcResult.create(b.build());
		return exec.newSucceededFuture(res);
	}
}