2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.bgpcep.pcep.topology.provider;
10 import static com.google.common.base.Verify.verifyNotNull;
12 import com.google.common.annotations.VisibleForTesting;
13 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
14 import java.net.InetSocketAddress;
15 import java.util.Collection;
17 import java.util.concurrent.ConcurrentHashMap;
18 import org.eclipse.jdt.annotation.NonNull;
19 import org.opendaylight.mdsal.binding.api.ClusteredDataTreeChangeListener;
20 import org.opendaylight.mdsal.binding.api.DataBroker;
21 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
22 import org.opendaylight.mdsal.binding.api.DataTreeModification;
23 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
24 import org.opendaylight.protocol.pcep.PCEPPeerProposal;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.PathComputationClient1;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Stateful1;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.Tlvs3Builder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.lsp.db.version.tlv.LspDbVersion;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev200720.speaker.entity.id.tlv.SpeakerEntityIdBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.config.rev200120.pcep.node.config.SessionConfig;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.Tlvs1;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.open.object.open.TlvsBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.config.rev181109.PcepNodeConfig;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev200120.Node1;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev200120.pcep.client.attributes.PathComputationClient;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.sync.optimizations.config.rev181109.PcepNodeSyncConfig;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
40 import org.opendaylight.yangtools.concepts.AbstractRegistration;
41 import org.opendaylight.yangtools.concepts.Registration;
42 import org.opendaylight.yangtools.yang.binding.DataObject;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
45 final class PCEPStatefulPeerProposal extends AbstractRegistration implements PCEPPeerProposal {
46 private abstract static class AbstractListener<D extends DataObject, V>
47 implements ClusteredDataTreeChangeListener<D> {
48 final Map<NodeId, V> map = new ConcurrentHashMap<>();
49 final Registration reg;
51 @SuppressFBWarnings(value = "MC_OVERRIDABLE_METHOD_CALL_IN_CONSTRUCTOR",
52 justification = "Stateless specializations in this nest")
53 AbstractListener(final DataBroker dataBroker, final @NonNull LogicalDatastoreType datastore,
54 final @NonNull InstanceIdentifier<D> wildcard) {
55 reg = dataBroker.registerDataTreeChangeListener(DataTreeIdentifier.create(datastore, wildcard), this);
58 final void remove(final DataTreeModification<?> modification) {
59 map.remove(extractNodeId(modification));
62 final void update(final DataTreeModification<?> modification, final V value) {
63 final var nodeId = extractNodeId(modification);
65 map.put(nodeId, value);
71 private static @NonNull NodeId extractNodeId(final DataTreeModification<?> modification) {
72 return verifyNotNull(modification.getRootPath().getRootIdentifier().firstKeyOf(Node.class)).getNodeId();
77 static final class SpeakerIdListener extends AbstractListener<PcepNodeSyncConfig, byte[]> {
78 SpeakerIdListener(final DataBroker dataBroker, final InstanceIdentifier<Topology> topologyId) {
79 super(dataBroker, LogicalDatastoreType.CONFIGURATION, topologyId.child(Node.class)
80 .augmentation(PcepNodeConfig.class).child(SessionConfig.class)
81 .augmentation(PcepNodeSyncConfig.class));
85 public void onDataTreeChanged(final Collection<DataTreeModification<PcepNodeSyncConfig>> changes) {
86 for (var change : changes) {
87 final var config = change.getRootNode().getDataAfter();
89 update(change, config.getSpeakerEntityIdValue());
98 static final class LspDbVersionListener extends AbstractListener<LspDbVersion, LspDbVersion> {
99 LspDbVersionListener(final DataBroker dataBroker, final InstanceIdentifier<Topology> topologyId) {
100 super(dataBroker, LogicalDatastoreType.OPERATIONAL, topologyId.child(Node.class)
101 .augmentation(Node1.class).child(PathComputationClient.class)
102 .augmentation(PathComputationClient1.class).child(LspDbVersion.class));
106 public void onDataTreeChanged(final Collection<DataTreeModification<LspDbVersion>> changes) {
107 for (var change : changes) {
108 update(change, change.getRootNode().getDataAfter());
113 private final LspDbVersionListener lspDbVersions;
114 private final SpeakerIdListener speakerIds;
116 PCEPStatefulPeerProposal(final DataBroker dataBroker, final InstanceIdentifier<Topology> topologyId) {
117 lspDbVersions = new LspDbVersionListener(dataBroker, topologyId);
118 speakerIds = new SpeakerIdListener(dataBroker, topologyId);
122 public void setPeerSpecificProposal(final InetSocketAddress address, final TlvsBuilder openBuilder) {
123 // Check if we are dealing with synchronization optimization
124 final var statefulTlv = openBuilder.augmentation(Tlvs1.class);
125 if (statefulTlv == null) {
128 final var stateful = statefulTlv.getStateful();
129 if (stateful == null || stateful.augmentation(Stateful1.class) == null) {
133 final var nodeId = ServerSessionManager.createNodeId(address.getAddress());
134 final var dbVersion = lspDbVersions.map.get(nodeId);
135 final var speakerId = speakerIds.map.get(nodeId);
136 if (speakerId == null && dbVersion == null) {
141 final Tlvs3Builder syncBuilder = new Tlvs3Builder();
142 if (dbVersion != null) {
143 syncBuilder.setLspDbVersion(dbVersion);
145 if (speakerId != null) {
146 syncBuilder.setSpeakerEntityId(new SpeakerEntityIdBuilder().setSpeakerEntityIdValue(speakerId).build());
148 openBuilder.addAugmentation(syncBuilder.build()).build();
152 protected void removeRegistration() {
153 lspDbVersions.reg.close();
154 speakerIds.reg.close();