*/
package org.opendaylight.bgpcep.pcep.topology.provider;
-import com.google.common.base.Optional;
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 com.google.common.util.concurrent.MoreExecutors;
+import java.net.InetAddress;
+import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
-import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicReference;
+import javax.annotation.concurrent.GuardedBy;
+import org.opendaylight.controller.config.yang.pcep.topology.provider.PCEPTopologyProviderRuntimeMXBean;
+import org.opendaylight.controller.config.yang.pcep.topology.provider.PCEPTopologyProviderRuntimeRegistration;
+import org.opendaylight.controller.config.yang.pcep.topology.provider.PCEPTopologyProviderRuntimeRegistrator;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
-import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
-import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
-import org.opendaylight.controller.md.sal.common.api.TransactionStatus;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
-import org.opendaylight.protocol.framework.SessionListenerFactory;
+import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
+import org.opendaylight.protocol.pcep.PCEPPeerProposal;
+import org.opendaylight.protocol.pcep.PCEPSession;
import org.opendaylight.protocol.pcep.PCEPSessionListener;
+import org.opendaylight.protocol.pcep.PCEPSessionListenerFactory;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.AddLspArgs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.EnsureLspOperationalInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.OperationResult;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.RemoveLspArgs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.TopologyTypes1;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.TopologyTypes1Builder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.TriggerSyncArgs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.UpdateLspArgs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.topology.pcep.type.TopologyPcepBuilder;
-import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
-import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopologyBuilder;
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.TopologyId;
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.TopologyBuilder;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
-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.TopologyTypesBuilder;
-import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
-import org.opendaylight.yangtools.yang.common.RpcResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
*
*/
-final class ServerSessionManager implements SessionListenerFactory<PCEPSessionListener>, AutoCloseable, TopologySessionRPCs {
+final class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessionRPCs,
+ PCEPTopologyProviderRuntimeMXBean, PCEPPeerProposal {
private static final Logger LOG = LoggerFactory.getLogger(ServerSessionManager.class);
private static final long DEFAULT_HOLD_STATE_NANOS = TimeUnit.MINUTES.toNanos(5);
+ @GuardedBy("this")
private final Map<NodeId, TopologySessionListener> nodes = new HashMap<>();
+ @GuardedBy("this")
private final Map<NodeId, TopologyNodeState> state = new HashMap<>();
private final TopologySessionListenerFactory listenerFactory;
private final InstanceIdentifier<Topology> topology;
- private final DataBroker dataProvider;
-
- public ServerSessionManager(final DataBroker dataProvider, final InstanceIdentifier<Topology> topology,
- final TopologySessionListenerFactory listenerFactory) {
- this.dataProvider = Preconditions.checkNotNull(dataProvider);
+ private final DataBroker broker;
+ private final PCEPStatefulPeerProposal peerProposal;
+ private final AtomicBoolean isClosed = new AtomicBoolean(false);
+ private final short rpcTimeout;
+ private final AtomicReference<PCEPTopologyProviderRuntimeRegistration> runtimeRootRegistration = new AtomicReference<>();
+
+ public ServerSessionManager(final DataBroker broker, final InstanceIdentifier<Topology> topology,
+ final TopologySessionListenerFactory listenerFactory, final short rpcTimeout) {
+ this.broker = Preconditions.checkNotNull(broker);
this.topology = Preconditions.checkNotNull(topology);
this.listenerFactory = Preconditions.checkNotNull(listenerFactory);
+ this.peerProposal = PCEPStatefulPeerProposal.createStatefulPeerProposal(this.broker, this.topology);
+ this.rpcTimeout = rpcTimeout;
+ }
- // FIXME: should migrated to transaction chain
- final ReadWriteTransaction tx = dataProvider.newReadWriteTransaction();
-
- // Make sure the topology does not exist
- final Optional<?> c;
- try {
- c = tx.read(LogicalDatastoreType.OPERATIONAL, topology).get();
- } catch (InterruptedException | ExecutionException e) {
- throw new IllegalStateException("Failed to ensure topology presence", e);
- }
- Preconditions.checkArgument(!c.isPresent(), "Topology %s already exists", topology);
-
- // create empty network-topology is not exists
- tx.put(LogicalDatastoreType.OPERATIONAL, InstanceIdentifier.builder(NetworkTopology.class).build(), new NetworkTopologyBuilder().build());
- // Now create the base topology
- final TopologyKey k = InstanceIdentifier.keyOf(topology);
- tx.put(LogicalDatastoreType.OPERATIONAL, topology, new TopologyBuilder().setKey(k).setTopologyId(k.getTopologyId()).setTopologyTypes(
- new TopologyTypesBuilder().addAugmentation(TopologyTypes1.class,
- new TopologyTypes1Builder().setTopologyPcep(new TopologyPcepBuilder().build()).build()).build()).setNode(
- new ArrayList<Node>()).build());
-
- Futures.addCallback(tx.commit(), new FutureCallback<RpcResult<TransactionStatus>>() {
+ /**
+ * Create Base Topology
+ *
+ * @throws TransactionCommitFailedException exception
+ */
+ synchronized ListenableFuture<Void> instantiateServiceInstance() {
+ final TopologyKey key = InstanceIdentifier.keyOf(this.topology);
+ final TopologyId topologyId = key.getTopologyId();
+ final WriteTransaction tx = this.broker.newWriteOnlyTransaction();
+ tx.put(LogicalDatastoreType.OPERATIONAL, this.topology, new TopologyBuilder().setKey(key)
+ .setTopologyId(topologyId).setTopologyTypes(new TopologyTypesBuilder()
+ .addAugmentation(TopologyTypes1.class, new TopologyTypes1Builder().setTopologyPcep(
+ new TopologyPcepBuilder().build()).build()).build())
+ .setNode(new ArrayList<>()).build(), true);
+ final ListenableFuture<Void> future = tx.submit();
+ Futures.addCallback(future, new FutureCallback<Void>() {
@Override
- public void onSuccess(final RpcResult<TransactionStatus> result) {
- LOG.trace("Topology {} created successfully", topology);
+ public void onSuccess(final Void result) {
+ LOG.debug("PCEP Topology {} created successfully.", topologyId.getValue());
}
@Override
public void onFailure(final Throwable t) {
- LOG.error("Failed to create topology {}", topology, t);
+ LOG.error("Failed to create PCEP Topology {}.", topologyId.getValue(), t);
}
- });
+ }, MoreExecutors.directExecutor());
+ return future;
}
- public void releaseNodeState(final TopologyNodeState nodeState) {
- LOG.debug("Node {} unbound", nodeState.getNodeId());
- this.nodes.remove(nodeState.getNodeId());
- nodeState.released();
+ private static NodeId createNodeId(final InetAddress addr) {
+ return new NodeId("pcc://" + addr.getHostAddress());
}
- synchronized TopologyNodeState takeNodeState(final NodeId id, final TopologySessionListener sessionListener) {
- LOG.debug("Node {} bound to listener {}", id, sessionListener);
+ synchronized void releaseNodeState(final TopologyNodeState nodeState, final PCEPSession session, final boolean persistNode) {
+ this.nodes.remove(createNodeId(session.getRemoteAddress()));
+ if (nodeState != null) {
+ LOG.debug("Node {} unbound", nodeState.getNodeId());
+ nodeState.released(persistNode);
+ }
+ }
+
+ synchronized TopologyNodeState takeNodeState(final InetAddress address, final TopologySessionListener sessionListener, final boolean retrieveNode) {
+ final NodeId id = createNodeId(address);
+ if (this.isClosed.get()) {
+ LOG.error("Server Session Manager is closed. Unable to create topology node {} with listener {}", id, sessionListener);
+ return null;
+ }
+ LOG.debug("Node {} requested by listener {}", id, sessionListener);
TopologyNodeState ret = this.state.get(id);
+
if (ret == null) {
- ret = new TopologyNodeState(id, DEFAULT_HOLD_STATE_NANOS);
+ ret = new TopologyNodeState(this.broker, this.topology, id, DEFAULT_HOLD_STATE_NANOS);
+ LOG.debug("Created topology node {} for id {} at {}", ret, id, ret.getNodeId());
this.state.put(id, ret);
}
+ // FIXME: else check for conflicting session
+ ret.taken(retrieveNode);
this.nodes.put(id, sessionListener);
- ret.taken();
+ LOG.debug("Node {} bound to listener {}", id, sessionListener);
return ret;
}
@Override
public PCEPSessionListener getSessionListener() {
- return listenerFactory.createTopologySessionListener(this);
+ return this.listenerFactory.createTopologySessionListener(this);
}
- @Override
- public synchronized ListenableFuture<OperationResult> addLsp(final AddLspArgs input) {
+ private synchronized TopologySessionListener checkSessionPresence(final NodeId nodeId) {
// Get the listener corresponding to the node
- final TopologySessionListener l = this.nodes.get(input.getNode());
+ final TopologySessionListener l = this.nodes.get(nodeId);
if (l == null) {
- LOG.debug("Session for node {} not found", input.getNode());
- return OperationResults.UNSENT.future();
+ LOG.debug("Session for node {} not found", nodeId);
+ return null;
}
+ return l;
+ }
- return l.addLsp(input);
+ @Override
+ public synchronized ListenableFuture<OperationResult> addLsp(final AddLspArgs input) {
+ final TopologySessionListener l = checkSessionPresence(input.getNode());
+ return l != null ? l.addLsp(input) : OperationResults.UNSENT.future();
}
@Override
public synchronized ListenableFuture<OperationResult> removeLsp(final RemoveLspArgs input) {
- // Get the listener corresponding to the node
- final TopologySessionListener l = this.nodes.get(input.getNode());
- if (l == null) {
- LOG.debug("Session for node {} not found", input.getNode());
- return OperationResults.UNSENT.future();
- }
-
- return l.removeLsp(input);
+ final TopologySessionListener l = checkSessionPresence(input.getNode());
+ return l != null ? l.removeLsp(input) : OperationResults.UNSENT.future();
}
@Override
public synchronized ListenableFuture<OperationResult> updateLsp(final UpdateLspArgs input) {
- // Get the listener corresponding to the node
- final TopologySessionListener l = this.nodes.get(input.getNode());
- if (l == null) {
- LOG.debug("Session for node {} not found", input.getNode());
- return OperationResults.UNSENT.future();
- }
-
- return l.updateLsp(input);
+ final TopologySessionListener l = checkSessionPresence(input.getNode());
+ return l != null ? l.updateLsp(input) : OperationResults.UNSENT.future();
}
@Override
public synchronized ListenableFuture<OperationResult> ensureLspOperational(final EnsureLspOperationalInput input) {
- // Get the listener corresponding to the node
- final TopologySessionListener l = this.nodes.get(input.getNode());
- if (l == null) {
- LOG.debug("Session for node {} not found", input.getNode());
- return OperationResults.UNSENT.future();
- }
-
- return l.ensureLspOperational(input);
+ final TopologySessionListener l = checkSessionPresence(input.getNode());
+ return l != null ? l.ensureLspOperational(input) : OperationResults.UNSENT.future();
}
- InstanceIdentifier<Topology> getTopology() {
- return topology;
+ @Override
+ public synchronized ListenableFuture<OperationResult> triggerSync(final TriggerSyncArgs input) {
+ final TopologySessionListener l = checkSessionPresence(input.getNode());
+ return l != null ? l.triggerSync(input) : OperationResults.UNSENT.future();
}
- WriteTransaction beginTransaction() {
- return dataProvider.newWriteOnlyTransaction();
+ synchronized ListenableFuture<Void> closeServiceInstance() {
+ if (this.isClosed.getAndSet(true)) {
+ LOG.error("Session Manager has already been closed.");
+ Futures.immediateFuture(null);
+ }
+ final PCEPTopologyProviderRuntimeRegistration runtimeReg = this.runtimeRootRegistration.getAndSet(null);
+ if (runtimeReg != null) {
+ runtimeReg.close();
+ }
+ for (final TopologySessionListener sessionListener : this.nodes.values()) {
+ sessionListener.close();
+ }
+ this.nodes.clear();
+ for (final TopologyNodeState nodeState : this.state.values()) {
+ nodeState.close();
+ }
+ this.state.clear();
+ final WriteTransaction t = this.broker.newWriteOnlyTransaction();
+ t.delete(LogicalDatastoreType.OPERATIONAL, this.topology);
+ final ListenableFuture<Void> future = t.submit();
+ Futures.addCallback(future, new FutureCallback<Void>() {
+ @Override
+ public void onSuccess(final Void result) {
+ LOG.debug("Topology {} removed", ServerSessionManager.this.topology);
+ }
+
+ @Override
+ public void onFailure(final Throwable t) {
+ LOG.warn("Failed to remove Topology {}", ServerSessionManager.this.topology, t);
+ }
+ }, MoreExecutors.directExecutor());
+ return future;
}
- ReadWriteTransaction rwTransaction() {
- return dataProvider.newReadWriteTransaction();
+ synchronized void setRuntimeRootRegistrator(final PCEPTopologyProviderRuntimeRegistrator runtimeRootRegistrator) {
+ if (!this.runtimeRootRegistration.compareAndSet(null, runtimeRootRegistrator.register(this))) {
+ LOG.error("Runtime root registration has been set before.");
+ }
}
- <T extends DataObject> ListenableFuture<Optional<T>> readOperationalData(final InstanceIdentifier<T> id) {
- final ReadTransaction t = dataProvider.newReadOnlyTransaction();
- return t.read(LogicalDatastoreType.OPERATIONAL, id);
+ PCEPTopologyProviderRuntimeRegistration getRuntimeRootRegistration() {
+ return this.runtimeRootRegistration.get();
}
@Override
- public void close() throws InterruptedException, ExecutionException {
- final WriteTransaction t = this.dataProvider.newWriteOnlyTransaction();
- t.delete(LogicalDatastoreType.OPERATIONAL, this.topology);
- t.submit().get();
+ public void setPeerSpecificProposal(final InetSocketAddress address, final TlvsBuilder openBuilder) {
+ Preconditions.checkNotNull(address);
+ this.peerProposal.setPeerProposal(createNodeId(address.getAddress()), openBuilder);
+ }
+
+ short getRpcTimeout() {
+ return this.rpcTimeout;
}
-}
+}
\ No newline at end of file