bgp/rib-impl/src/main/java/org/opendaylight/protocol/bgp/rib/impl/AdjRibInWriter.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.protocol.bgp.rib.impl;
   9
  10 import com.google.common.base.Preconditions;
  11 import com.google.common.collect.ImmutableMap;
  12 import com.google.common.collect.ImmutableMap.Builder;
  13 import java.util.Collection;
  14 import java.util.Collections;
  15 import java.util.Map;
  16 import java.util.Map.Entry;
  17 import java.util.Set;
  18 import javax.annotation.Nonnull;
  19 import javax.annotation.concurrent.NotThreadSafe;
  20 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  21 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  22 import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain;
  23 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContext;
  24 import org.opendaylight.protocol.bgp.rib.impl.spi.RIBSupportContextRegistry;
  25 import org.opendaylight.protocol.bgp.rib.spi.RibSupportUtils;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpReachNlri;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.update.attributes.MpUnreachNlri;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.PeerRole;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.Peer;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.AdjRibIn;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.AdjRibOut;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.bgp.rib.rib.peer.EffectiveRibIn;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Attributes;
  37 import org.opendaylight.yangtools.yang.common.QName;
  38 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  39 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.InstanceIdentifierBuilder;
  40 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  41 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  42 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  43 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
  44 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  45 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  46 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  47 import org.opendaylight.yangtools.yang.data.impl.schema.builder.api.DataContainerNodeBuilder;
  48 import org.slf4j.Logger;
  49 import org.slf4j.LoggerFactory;
  50
  51 /**
  52  * Writer of Adjacency-RIB-In for a single peer. An instance of this object
  53  * is attached to each {@link BGPPeer} and {@link ApplicationPeer}.
  54  */
  55 @NotThreadSafe
  56 final class AdjRibInWriter {
  57     private static final Logger LOG = LoggerFactory.getLogger(AdjRibInWriter.class);
  58
  59     private static final LeafNode<Boolean> ATTRIBUTES_UPTODATE_FALSE = ImmutableNodes.leafNode(QName.create(Attributes.QNAME, "uptodate"), Boolean.FALSE);
  60     private static final LeafNode<Boolean> ATTRIBUTES_UPTODATE_TRUE = ImmutableNodes.leafNode(ATTRIBUTES_UPTODATE_FALSE.getNodeType(), Boolean.TRUE);
  61     private static final QName PEER_ID_QNAME = QName.cachedReference(QName.create(Peer.QNAME, "peer-id"));
  62     private static final QName PEER_ROLE_QNAME = QName.cachedReference(QName.create(Peer.QNAME, "peer-role"));
  63     private static final NodeIdentifier ADJRIBIN = new NodeIdentifier(AdjRibIn.QNAME);
  64     private static final NodeIdentifier ADJRIBOUT = new NodeIdentifier(AdjRibOut.QNAME);
  65     private static final NodeIdentifier EFFRIBIN = new NodeIdentifier(EffectiveRibIn.QNAME);
  66     private static final NodeIdentifier PEER_ID = new NodeIdentifier(PEER_ID_QNAME);
  67     private static final NodeIdentifier PEER_ROLE = new NodeIdentifier(PEER_ROLE_QNAME);
  68     private static final NodeIdentifier TABLES = new NodeIdentifier(Tables.QNAME);
  69
  70     // FIXME: is there a utility method to construct this?
  71     private static final ContainerNode EMPTY_ADJRIBIN = Builders.containerBuilder().withNodeIdentifier(ADJRIBIN).addChild(ImmutableNodes.mapNodeBuilder(Tables.QNAME).build()).build();
  72     private static final ContainerNode EMPTY_EFFRIBIN = Builders.containerBuilder().withNodeIdentifier(EFFRIBIN).addChild(ImmutableNodes.mapNodeBuilder(Tables.QNAME).build()).build();
  73     private static final ContainerNode EMPTY_ADJRIBOUT = Builders.containerBuilder().withNodeIdentifier(ADJRIBOUT).addChild(ImmutableNodes.mapNodeBuilder(Tables.QNAME).build()).build();
  74
  75     private final Map<TablesKey, TableContext> tables;
  76     private final YangInstanceIdentifier peerPath;
  77     private final YangInstanceIdentifier ribPath;
  78     private final DOMTransactionChain chain;
  79     private final PeerId peerId;
  80     private final String role;
  81
  82     private AdjRibInWriter(final YangInstanceIdentifier ribPath, final DOMTransactionChain chain, final PeerId peerId, final String role, final YangInstanceIdentifier peerPath, final Map<TablesKey, TableContext> tables) {
  83         this.ribPath = Preconditions.checkNotNull(ribPath);
  84         this.chain = Preconditions.checkNotNull(chain);
  85         this.tables = Preconditions.checkNotNull(tables);
  86         this.role = Preconditions.checkNotNull(role);
  87         this.peerPath = peerPath;
  88         this.peerId = peerId;
  89     }
  90
  91     // We could use a codec, but this should be fine, too
  92     private static String roleString(final PeerRole role) {
  93         switch (role) {
  94         case Ebgp:
  95             return "ebgp";
  96         case Ibgp:
  97             return "ibgp";
  98         case RrClient:
  99             return "rr-client";
 100         default:
 101             throw new IllegalArgumentException("Unhandled role " + role);
 102         }
 103     }
 104
 105     /**
 106      * Create a new writer using a transaction chain.
 107      *
 108      * @param role peer's role
 109      * @param chain transaction chain
 110      * @return A fresh writer instance
 111      */
 112     static AdjRibInWriter create(@Nonnull final YangInstanceIdentifier ribId, @Nonnull final PeerRole role, @Nonnull final DOMTransactionChain chain) {
 113         return new AdjRibInWriter(ribId, chain, null, roleString(role), null, Collections.<TablesKey, TableContext>emptyMap());
 114     }
 115
 116     /**
 117      * Transform this writer to a new writer, which is in charge of specified tables.
 118      * Empty tables are created for new entries and old tables are deleted. Once this
 119      * method returns, the old instance must not be reasonably used.
 120      *
 121      * @param newPeerId new peer BGP identifier
 122      * @param registry RIB extension registry
 123      * @param tableTypes New tables, must not be null
 124      * @return New writer
 125      */
 126     AdjRibInWriter transform(final PeerId newPeerId, final RIBSupportContextRegistry registry, final Set<TablesKey> tableTypes, final boolean isAppPeer) {
 127         final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
 128
 129         final YangInstanceIdentifier newPeerPath;
 130         if (!newPeerId.equals(this.peerId)) {
 131             newPeerPath = this.createEmptyPeerStructure(newPeerId, isAppPeer, tx);
 132         } else {
 133             newPeerPath = this.peerPath;
 134
 135             // Wipe tables which are not present in the new types
 136             for (final Entry<TablesKey, TableContext> e : this.tables.entrySet()) {
 137                 if (!tableTypes.contains(e.getKey())) {
 138                     e.getValue().removeTable(tx);
 139                 }
 140             }
 141         }
 142
 143         // Now create new table instances, potentially creating their empty entries
 144         final Builder<TablesKey, TableContext> tb = ImmutableMap.builder();
 145         for (final TablesKey k : tableTypes) {
 146             TableContext ctx = this.tables.get(k);
 147             if (ctx == null) {
 148                 final RIBSupportContext rs = registry.getRIBSupportContext(k);
 149                 // TODO: Use returned value once Instance Identifier builder allows for it.
 150                 final NodeIdentifierWithPredicates key = RibSupportUtils.toYangTablesKey(k);
 151                 if (rs == null) {
 152                     LOG.warn("No support for table type {}, skipping it", k);
 153                     continue;
 154                 }
 155                 // install tables for adj-ribs-out (we do not need TableContext for that
 156                 if (!isAppPeer) {
 157                     rs.clearTable(tx, newPeerPath.node(EMPTY_ADJRIBOUT.getIdentifier()).node(TABLES).node(key));
 158                 }
 159
 160                 // We will use table keys very often, make sure they are optimized
 161                 final InstanceIdentifierBuilder idb = YangInstanceIdentifier.builder(newPeerPath.node(EMPTY_ADJRIBIN.getIdentifier()).node(TABLES));
 162                 idb.nodeWithKey(key.getNodeType(), key.getKeyValues());
 163
 164                 // install tables for adj-ribs-in
 165                 ctx = new TableContext(rs, idb.build());
 166                 ctx.clearTable(tx);
 167             }
 168             tx.merge(LogicalDatastoreType.OPERATIONAL, ctx.getTableId().node(Attributes.QNAME).node(ATTRIBUTES_UPTODATE_FALSE.getNodeType()), ATTRIBUTES_UPTODATE_FALSE);
 169             LOG.debug("Created table instance {}", ctx.getTableId());
 170             tb.put(k, ctx);
 171         }
 172
 173         tx.submit();
 174
 175         return new AdjRibInWriter(this.ribPath, this.chain, newPeerId, this.role, newPeerPath, tb.build());
 176     }
 177
 178     private YangInstanceIdentifier createEmptyPeerStructure(final PeerId newPeerId, final boolean isAppPeer, final DOMDataWriteTransaction tx) {
 179         if (this.peerId != null) {
 180             // Wipe old peer data completely
 181             tx.delete(LogicalDatastoreType.OPERATIONAL, this.ribPath.node(Peer.QNAME).node(new NodeIdentifierWithPredicates(Peer.QNAME, PEER_ID_QNAME, this.peerId.getValue())));
 182         }
 183         // Install new empty peer structure
 184         final NodeIdentifierWithPredicates peerKey = IdentifierUtils.domPeerId(newPeerId);
 185         final YangInstanceIdentifier newPeerPath = this.ribPath.node(Peer.QNAME).node(peerKey);
 186
 187         final DataContainerNodeBuilder<NodeIdentifierWithPredicates, MapEntryNode> pb = Builders.mapEntryBuilder();
 188         pb.withNodeIdentifier(peerKey);
 189         pb.withChild(ImmutableNodes.leafNode(PEER_ID, newPeerId.getValue()));
 190         pb.withChild(ImmutableNodes.leafNode(PEER_ROLE, this.role));
 191         pb.withChild(EMPTY_ADJRIBIN);
 192         pb.withChild(EMPTY_EFFRIBIN);
 193         if (!isAppPeer) {
 194             pb.withChild(EMPTY_ADJRIBOUT);
 195         }
 196         tx.put(LogicalDatastoreType.OPERATIONAL, newPeerPath, pb.build());
 197         LOG.debug("New peer {} structure installed.", newPeerPath);
 198         return newPeerPath;
 199     }
 200
 201     /**
 202      * Clean all routes in specified tables
 203      *
 204      * @param tableTypes Tables to clean.
 205      */
 206     void cleanTables(final Collection<TablesKey> tableTypes) {
 207         final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
 208
 209         for (final TablesKey k : tableTypes) {
 210             LOG.debug("Clearing table {}", k);
 211             this.tables.get(k).clearTable(tx);
 212         }
 213
 214         tx.submit();
 215     }
 216
 217     void markTableUptodate(final TablesKey tableTypes) {
 218         final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
 219
 220         final TableContext ctx = this.tables.get(tableTypes);
 221         tx.merge(LogicalDatastoreType.OPERATIONAL, ctx.getTableId().node(Attributes.QNAME).node(ATTRIBUTES_UPTODATE_TRUE.getNodeType()), ATTRIBUTES_UPTODATE_TRUE);
 222
 223         tx.submit();
 224     }
 225
 226     void updateRoutes(final MpReachNlri nlri, final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Attributes attributes) {
 227         final TablesKey key = new TablesKey(nlri.getAfi(), nlri.getSafi());
 228         final TableContext ctx = this.tables.get(key);
 229         if (ctx == null) {
 230             LOG.debug("No table for {}, not accepting NLRI {}", key, nlri);
 231             return;
 232         }
 233
 234         final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
 235         ctx.writeRoutes(tx, nlri, attributes);
 236         LOG.trace("Write routes {}", nlri);
 237         tx.submit();
 238     }
 239
 240     void removeRoutes(final MpUnreachNlri nlri) {
 241         final TablesKey key = new TablesKey(nlri.getAfi(), nlri.getSafi());
 242         final TableContext ctx = this.tables.get(key);
 243         if (ctx == null) {
 244             LOG.debug("No table for {}, not accepting NLRI {}", key, nlri);
 245             return;
 246         }
 247         LOG.trace("Removing routes {}", nlri);
 248         final DOMDataWriteTransaction tx = this.chain.newWriteOnlyTransaction();
 249         ctx.removeRoutes(tx, nlri);
 250         tx.submit();
 251     }
 252
 253 }