pcep/topology/topology-provider/src/main/java/org/opendaylight/bgpcep/pcep/topology/provider/PCEPStatefulPeerProposal.java

   1 /*
   2  * Copyright (c) 2015 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 static com.google.common.base.Verify.verifyNotNull;
  11
  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;
  16 import java.util.Map;
  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;
  44
  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;
  50
  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);
  56         }
  57
  58         final void remove(final DataTreeModification<?> modification) {
  59             map.remove(extractNodeId(modification));
  60         }
  61
  62         final void update(final DataTreeModification<?> modification, final V value) {
  63             final var nodeId = extractNodeId(modification);
  64             if (value != null) {
  65                 map.put(nodeId, value);
  66             } else {
  67                 map.remove(nodeId);
  68             }
  69         }
  70
  71         private static @NonNull NodeId extractNodeId(final DataTreeModification<?> modification) {
  72             return verifyNotNull(modification.getRootPath().getRootIdentifier().firstKeyOf(Node.class)).getNodeId();
  73         }
  74     }
  75
  76     @VisibleForTesting
  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));
  82         }
  83
  84         @Override
  85         public void onDataTreeChanged(final Collection<DataTreeModification<PcepNodeSyncConfig>> changes) {
  86             for (var change : changes) {
  87                 final var config = change.getRootNode().getDataAfter();
  88                 if (config != null) {
  89                     update(change, config.getSpeakerEntityIdValue());
  90                 } else {
  91                     remove(change);
  92                 }
  93             }
  94         }
  95     }
  96
  97     @VisibleForTesting
  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));
 103         }
 104
 105         @Override
 106         public void onDataTreeChanged(final Collection<DataTreeModification<LspDbVersion>> changes) {
 107             for (var change : changes) {
 108                 update(change, change.getRootNode().getDataAfter());
 109             }
 110         }
 111     }
 112
 113     private final LspDbVersionListener lspDbVersions;
 114     private final SpeakerIdListener speakerIds;
 115
 116     PCEPStatefulPeerProposal(final DataBroker dataBroker, final InstanceIdentifier<Topology> topologyId) {
 117         lspDbVersions = new LspDbVersionListener(dataBroker, topologyId);
 118         speakerIds = new SpeakerIdListener(dataBroker, topologyId);
 119     }
 120
 121     @Override
 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) {
 126             return;
 127         }
 128         final var stateful = statefulTlv.getStateful();
 129         if (stateful == null || stateful.augmentation(Stateful1.class) == null) {
 130             return;
 131         }
 132
 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) {
 137             // Nothing to add
 138             return;
 139         }
 140
 141         final Tlvs3Builder syncBuilder = new Tlvs3Builder();
 142         if (dbVersion != null) {
 143             syncBuilder.setLspDbVersion(dbVersion);
 144         }
 145         if (speakerId != null) {
 146             syncBuilder.setSpeakerEntityId(new SpeakerEntityIdBuilder().setSpeakerEntityIdValue(speakerId).build());
 147         }
 148         openBuilder.addAugmentation(syncBuilder.build()).build();
 149     }
 150
 151     @Override
 152     protected void removeRegistration() {
 153         lspDbVersions.reg.close();
 154         speakerIds.reg.close();
 155     }
 156 }