Rework metadata storage to retain metadata through PCEP session flaps
[bgpcep.git] / pcep / topology-provider / src / main / java / org / opendaylight / bgpcep / pcep / topology / provider / TopologyNodeState.java
1 /*
2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.bgpcep.pcep.topology.provider;
9
10 import java.util.Collection;
11 import java.util.HashMap;
12 import java.util.Iterator;
13 import java.util.Map;
14
15 import javax.annotation.concurrent.ThreadSafe;
16
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SymbolicPathName;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.lsp.metadata.Metadata;
19 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
20
21 import com.google.common.base.Preconditions;
22
23 @ThreadSafe
24 final class TopologyNodeState {
25         private final Map<SymbolicPathName, Metadata> metadata = new HashMap<>();
26         private final long holdStateNanos;
27         private final NodeId nodeId;
28         private long lastReleased = 0;
29
30         public TopologyNodeState(final NodeId nodeId, final long holdStateNanos) {
31                 Preconditions.checkArgument(holdStateNanos >= 0);
32                 this.nodeId = Preconditions.checkNotNull(nodeId);
33                 this.holdStateNanos = holdStateNanos;
34         }
35
36         public NodeId getNodeId() {
37                 return nodeId;
38         }
39
40         public synchronized Metadata getLspMetadata(final SymbolicPathName name) {
41                 return metadata.get(name);
42         }
43
44         public synchronized void setLspMetadata(final SymbolicPathName name, final Metadata value) {
45                 if (value == null) {
46                         metadata.remove(name);
47                 } else {
48                         metadata.put(name, value);
49                 }
50         }
51
52         public synchronized void removeLspMetadata(final SymbolicPathName name) {
53                 metadata.remove(name);
54         }
55
56         public synchronized void cleanupExcept(final Collection<SymbolicPathName> values) {
57                 final Iterator<SymbolicPathName> it = metadata.keySet().iterator();
58                 while (it.hasNext()) {
59                         if (!values.contains(it.next())) {
60                                 it.remove();
61                         }
62                 }
63         }
64
65         public synchronized void released() {
66                 lastReleased = System.nanoTime();
67         }
68
69         public synchronized void taken() {
70                 final long now = System.nanoTime();
71
72                 if (now - lastReleased > holdStateNanos) {
73                         metadata.clear();
74                 }
75         }
76 }