bgp/extensions/route-target/src/main/java/org/opendaylight/protocol/bgp/route/targetcontrain/impl/RouteTargetConstrainRIBSupport.java

   1 /*
   2  * Copyright (c) 2018 AT&T Intellectual Property. 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.route.targetcontrain.impl;
   9
  10 import com.google.common.collect.ImmutableCollection;
  11 import com.google.common.collect.ImmutableSet;
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14 import java.util.ArrayList;
  15 import java.util.Collection;
  16 import java.util.Collections;
  17 import java.util.List;
  18 import java.util.Optional;
  19 import java.util.stream.Collectors;
  20 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
  21 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
  22 import org.opendaylight.protocol.bgp.parser.spi.PathIdUtil;
  23 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupport;
  24 import org.opendaylight.protocol.bgp.route.targetcontrain.impl.nlri.ImmutableRouteTargetConstrainNlriRegistry;
  25 import org.opendaylight.protocol.util.ByteArray;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.AsNumber;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.RouteTargetConstrainSubsequentAddressFamily;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.bgp.rib.rib.loc.rib.tables.routes.RouteTargetConstrainRoutesCase;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.RouteTargetConstrainChoice;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.destination.RouteTargetConstrainDestination;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.destination.RouteTargetConstrainDestinationBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.RouteTargetConstrainRoutes;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.route.target.constrain.routes.route.target.constrain.routes.RouteTargetConstrainRoute;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationRouteTargetConstrainAdvertizedCaseBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.route.target.constrain.advertized._case.DestinationRouteTargetConstrain;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.route.target.constrain.advertized._case.DestinationRouteTargetConstrainBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.route.target.constrain.rev180618.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationRouteTargetConstrainWithdrawnCaseBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.Ipv4AddressFamily;
  40 import org.opendaylight.yangtools.yang.binding.BindingObject;
  41 import org.opendaylight.yangtools.yang.binding.DataObject;
  42 import org.opendaylight.yangtools.yang.common.QName;
  43 import org.opendaylight.yangtools.yang.common.Uint32;
  44 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  46 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  47 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  48 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  49 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerNode;
  50 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  51 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  52 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
  53 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
  54 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
  55 import org.slf4j.Logger;
  56 import org.slf4j.LoggerFactory;
  57
  58 /**
  59  * Route Target Constrains RIBSupport.
  60  *
  61  * @author Claudio D. Gasparini
  62  */
  63 public final class RouteTargetConstrainRIBSupport
  64         extends AbstractRIBSupport<RouteTargetConstrainRoutesCase, RouteTargetConstrainRoutes,
  65         RouteTargetConstrainRoute> {
  66     private static final Logger LOG = LoggerFactory.getLogger(RouteTargetConstrainRIBSupport.class);
  67
  68     private static final NodeIdentifier NLRI_ROUTES_LIST = NodeIdentifier.create(RouteTargetConstrainDestination.QNAME);
  69     private static final String ORIGIN_AS = "origin-as";
  70     private final ImmutableCollection<Class<? extends BindingObject>> cacheableNlriObjects
  71             = ImmutableSet.of(RouteTargetConstrainRoutesCase.class);
  72     private final NodeIdentifier originAsNid;
  73
  74     /**
  75      * Default constructor. Requires the QName of the container augmented under the routes choice
  76      * node in instantiations of the rib grouping. It is assumed that this container is defined by
  77      * the same model which populates it with route grouping instantiation, and by extension with
  78      * the route attributes container.
  79      *
  80      * @param mappingService Serialization service
  81      */
  82     public RouteTargetConstrainRIBSupport(final BindingNormalizedNodeSerializer mappingService) {
  83         super(mappingService,
  84                 RouteTargetConstrainRoutesCase.class,
  85                 RouteTargetConstrainRoutes.class,
  86                 RouteTargetConstrainRoute.class,
  87                 Ipv4AddressFamily.class,
  88                 RouteTargetConstrainSubsequentAddressFamily.class,
  89                 DestinationRouteTargetConstrain.QNAME);
  90         this.originAsNid = new NodeIdentifier(QName.create(routeQName(), ORIGIN_AS).intern());
  91     }
  92
  93     @Override
  94     public ImmutableCollection<Class<? extends BindingObject>> cacheableNlriObjects() {
  95         return this.cacheableNlriObjects;
  96     }
  97
  98     @Override
  99     protected DestinationType buildDestination(final Collection<MapEntryNode> routes) {
 100         return new DestinationRouteTargetConstrainAdvertizedCaseBuilder().setDestinationRouteTargetConstrain(
 101                 new DestinationRouteTargetConstrainBuilder()
 102                         .setRouteTargetConstrainDestination(extractRoutes(routes)).build()).build();
 103     }
 104
 105     @Override
 106     protected DestinationType buildWithdrawnDestination(final Collection<MapEntryNode> routes) {
 107         return new DestinationRouteTargetConstrainWithdrawnCaseBuilder()
 108                 .setDestinationRouteTargetConstrain(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns
 109                         .yang.bgp.route.target.constrain.rev180618.update.attributes.mp.unreach.nlri.withdrawn.routes
 110                         .destination.type.destination.route.target.constrain.withdrawn._case
 111                         .DestinationRouteTargetConstrainBuilder()
 112                         .setRouteTargetConstrainDestination(extractRoutes(routes)).build()).build();
 113     }
 114
 115     private List<RouteTargetConstrainDestination> extractRoutes(final Collection<MapEntryNode> routes) {
 116         return routes.stream().map(this::extractDestination).collect(Collectors.toList());
 117     }
 118
 119     private RouteTargetConstrainDestination extractDestination(final DataContainerNode rtDest) {
 120         final RouteTargetConstrainDestinationBuilder builder = new RouteTargetConstrainDestinationBuilder()
 121                 .setPathId(PathIdUtil.buildPathId(rtDest, routePathIdNid()))
 122                 .setRouteTargetConstrainChoice(extractRouteTargetChoice(rtDest));
 123         final Optional<Object> originAs = NormalizedNodes
 124                 .findNode(rtDest, this.originAsNid).map(NormalizedNode::body);
 125         originAs.ifPresent(o -> builder.setOriginAs(new AsNumber((Uint32) o)));
 126         return builder.build();
 127     }
 128
 129     private RouteTargetConstrainChoice extractRouteTargetChoice(final DataContainerNode route) {
 130         final DataObject nn = this.mappingService.fromNormalizedNode(this.routeDefaultYii, route).getValue();
 131         return ((RouteTargetConstrainRoute) nn).getRouteTargetConstrainChoice();
 132     }
 133
 134     @Override
 135     protected Collection<NodeIdentifierWithPredicates> processDestination(
 136             final DOMDataTreeWriteTransaction tx,
 137             final YangInstanceIdentifier routesPath,
 138             final ContainerNode destination,
 139             final ContainerNode attributes,
 140             final ApplyRoute function) {
 141         if (destination != null) {
 142             final DataContainerChild routes = destination.childByArg(NLRI_ROUTES_LIST);
 143             if (routes != null) {
 144                 if (routes instanceof UnkeyedListNode) {
 145                     final YangInstanceIdentifier base = routesYangInstanceIdentifier(routesPath);
 146                     final Collection<UnkeyedListEntryNode> routesList = ((UnkeyedListNode) routes).body();
 147                     final List<NodeIdentifierWithPredicates> keys = new ArrayList<>(routesList.size());
 148                     for (final UnkeyedListEntryNode rtDest : routesList) {
 149                         final NodeIdentifierWithPredicates routeKey = createRouteKey(rtDest);
 150                         function.apply(tx, base, routeKey, rtDest, attributes);
 151                         keys.add(routeKey);
 152                     }
 153                     return keys;
 154                 }
 155                 LOG.warn("Routes {} are not a map", routes);
 156             }
 157         }
 158         return Collections.emptyList();
 159     }
 160
 161     private NodeIdentifierWithPredicates createRouteKey(final UnkeyedListEntryNode routeTarget) {
 162         final ByteBuf buffer = Unpooled.buffer();
 163         final RouteTargetConstrainDestination dest = extractDestination(routeTarget);
 164         buffer.writeBytes(ImmutableRouteTargetConstrainNlriRegistry.getInstance()
 165                 .serializeRouteTargetConstrain(dest.getRouteTargetConstrainChoice()));
 166         return PathIdUtil.createNidKey(routeQName(), routeKeyTemplate(),
 167                 ByteArray.encodeBase64(buffer), routeTarget.findChildByArg(routePathIdNid()));
 168     }
 169 }