Bump upstreams
[bgpcep.git] / pcep / topology / topology-provider / src / main / java / org / opendaylight / bgpcep / pcep / topology / provider / ServerSessionManager.java
index 310f574da439aa8e97499decff7c4fa1f1f3b6f3..8fde9947f136d0487108c749294a10ca93706241 100644 (file)
@@ -9,20 +9,19 @@ package org.opendaylight.bgpcep.pcep.topology.provider;
 
 import static java.util.Objects.requireNonNull;
 
-import com.google.common.annotations.VisibleForTesting;
 import com.google.common.util.concurrent.FluentFuture;
 import com.google.common.util.concurrent.FutureCallback;
 import com.google.common.util.concurrent.ListenableFuture;
 import com.google.common.util.concurrent.MoreExecutors;
 import com.google.common.util.concurrent.SettableFuture;
-import io.netty.util.HashedWheelTimer;
 import io.netty.util.Timeout;
+import java.lang.invoke.MethodHandles;
+import java.lang.invoke.VarHandle;
 import java.net.InetAddress;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.TimeUnit;
-import java.util.concurrent.atomic.AtomicBoolean;
 import org.checkerframework.checker.lock.qual.GuardedBy;
 import org.eclipse.jdt.annotation.NonNull;
 import org.eclipse.jdt.annotation.Nullable;
@@ -33,7 +32,6 @@ import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
 import org.opendaylight.protocol.pcep.PCEPSessionListenerFactory;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.graph.rev220720.graph.topology.GraphKey;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.SrpIdNumber;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.stats.rev171113.PcepSessionState;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.AddLspArgs;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.EnsureLspOperationalInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.OperationResult;
@@ -43,13 +41,11 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.TopologyTypes1Builder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.TriggerSyncArgs;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.UpdateLspArgs;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.topology.pcep.type.TopologyPcepBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev220730.network.topology.topology.topology.types.TopologyPcepBuilder;
 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.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.NodeKey;
 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.KeyedInstanceIdentifier;
 import org.opendaylight.yangtools.yang.common.ErrorType;
@@ -67,28 +63,37 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
 
     private static final Logger LOG = LoggerFactory.getLogger(ServerSessionManager.class);
     private static final long DEFAULT_HOLD_STATE_NANOS = TimeUnit.MINUTES.toNanos(5);
+    private static final VarHandle CLOSED;
+
+    static {
+        try {
+            CLOSED = MethodHandles.lookup().findVarHandle(ServerSessionManager.class, "closed", boolean.class);
+        } catch (NoSuchFieldException | IllegalAccessException e) {
+            throw new ExceptionInInitializerError(e);
+        }
+    }
 
     private final @NonNull KeyedInstanceIdentifier<Topology, TopologyKey> topology;
     private final @NonNull PCEPTopologyProviderDependencies dependencies;
-    private final @NonNull HashedWheelTimer timer = new HashedWheelTimer();
+    private final @Nullable GraphKey graphKey;
 
-    @VisibleForTesting
-    final AtomicBoolean isClosed = new AtomicBoolean(false);
     @GuardedBy("this")
     private final Map<NodeId, TopologySessionListener> nodes = new HashMap<>();
     @GuardedBy("this")
     private final Map<NodeId, TopologyNodeState> state = new HashMap<>();
 
+    private volatile long updateInterval;
     private volatile short rpcTimeout;
+    private volatile boolean closed;
 
-    private final GraphKey graphKey;
-
-    ServerSessionManager(final KeyedInstanceIdentifier<Topology, TopologyKey> instanceIdentifier,
-            final PCEPTopologyProviderDependencies dependencies, final short rpcTimeout, final GraphKey graphKey) {
+    ServerSessionManager(final KeyedInstanceIdentifier<Topology, TopologyKey> topology,
+            final PCEPTopologyProviderDependencies dependencies, final GraphKey graphKey,
+            final short rpcTimeout, final long updateInterval) {
         this.dependencies = requireNonNull(dependencies);
-        topology = requireNonNull(instanceIdentifier);
+        this.topology = requireNonNull(topology);
+        this.graphKey = graphKey;
         this.rpcTimeout = rpcTimeout;
-        this.graphKey = requireNonNull(graphKey);
+        this.updateInterval = updateInterval;
     }
 
     // Initialize the operational view of the topology.
@@ -108,28 +113,32 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
             @Override
             public void onSuccess(final CommitInfo result) {
                 LOG.info("PCEP Topology {} created successfully.", topologyId());
-                isClosed.set(false);
+                closed = false;
                 future.set(Boolean.TRUE);
             }
 
             @Override
             public void onFailure(final Throwable failure) {
                 LOG.error("Failed to create PCEP Topology {}.", topologyId(), failure);
-                isClosed.set(true);
+                closed = true;
                 future.set(Boolean.FALSE);
             }
         }, MoreExecutors.directExecutor());
 
         // Register this new topology to PCE Server
         final PceServerProvider server = dependencies.getPceServerProvider();
-        if (server != null) {
+        if (server != null && graphKey != null) {
             server.registerPcepTopology(topology, graphKey);
         }
         return future;
     }
 
+    final boolean isClosed() {
+        return closed;
+    }
+
     final synchronized FluentFuture<? extends CommitInfo> stop() {
-        if (isClosed.getAndSet(true)) {
+        if (!CLOSED.compareAndSet(this, false, true)) {
             LOG.error("Session Manager has already been closed.");
             return CommitInfo.emptyFluentFuture();
         }
@@ -146,12 +155,6 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
         }
         state.clear();
 
-        // Stop the timer
-        final var cancelledTasks = timer.stop().size();
-        if (cancelledTasks != 0) {
-            LOG.warn("Stopped timer with {} remaining tasks", cancelledTasks);
-        }
-
         // Un-Register Pcep Topology into PCE Server
         final PceServerProvider server = dependencies.getPceServerProvider();
         if (server != null) {
@@ -177,7 +180,7 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
 
     final synchronized void releaseNodeState(final TopologyNodeState nodeState, final InetAddress peerAddress,
             final boolean persistNode) {
-        if (isClosed.get()) {
+        if (isClosed()) {
             LOG.error("Session Manager has already been closed.");
             return;
         }
@@ -193,7 +196,7 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
     final synchronized TopologyNodeState takeNodeState(final InetAddress address,
             final TopologySessionListener sessionListener, final boolean retrieveNode) {
         final NodeId id = createNodeId(address);
-        if (isClosed.get()) {
+        if (isClosed()) {
             LOG.error("Server Session Manager is closed. Unable to create topology node {} with listener {}", id,
                 sessionListener);
             return null;
@@ -223,7 +226,8 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
     // Non-final for testing
     @Override
     public PCEPTopologySessionListener getSessionListener() {
-        return new PCEPTopologySessionListener(this);
+        return new PCEPTopologySessionListener(dependencies.getStateRegistry(), this,
+            dependencies.getPceServerProvider());
     }
 
     private synchronized TopologySessionListener checkSessionPresence(final NodeId nodeId) {
@@ -283,27 +287,27 @@ class ServerSessionManager implements PCEPSessionListenerFactory, TopologySessio
     final @Nullable Timeout newRpcTimeout(final RpcTimeout task, final SrpIdNumber requestId) {
         final short localTimeout = rpcTimeout;
         return localTimeout <= 0 ? null
-            : timer.newTimeout(ignored -> task.run(requestId), localTimeout, TimeUnit.SECONDS);
+            : dependencies.getTimer().newTimeout(ignored -> task.run(requestId), localTimeout, TimeUnit.SECONDS);
+    }
+
+    final long updateInterval() {
+        return isClosed() ? 0 : updateInterval;
     }
 
     final void setRpcTimeout(final short rpcTimeout) {
         this.rpcTimeout = rpcTimeout;
     }
 
+    final void setUpdateInterval(final long updateInterval) {
+        this.updateInterval = updateInterval;
+    }
+
     final void tearDownSessions(final List<InetAddress> outdatedNodes) {
         for (var address : outdatedNodes) {
             tearDownSession(new TearDownSessionInputBuilder().setNode(createNodeId(address)).build());
         }
     }
 
-    final void bind(final KeyedInstanceIdentifier<Node, NodeKey> nodeId, final PcepSessionState sessionState) {
-        dependencies.getStateRegistry().bind(nodeId, sessionState);
-    }
-
-    final void unbind(final KeyedInstanceIdentifier<Node, NodeKey> nodeId) {
-        dependencies.getStateRegistry().unbind(nodeId);
-    }
-
     final PCEPTopologyProviderDependencies getPCEPTopologyProviderDependencies() {
         return dependencies;
     }