bgp/linkstate/src/main/java/org/opendaylight/protocol/bgp/linkstate/LinkstateRIBSupport.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.linkstate;
   9
  10 import com.google.common.base.Optional;
  11 import com.google.common.collect.ImmutableCollection;
  12 import com.google.common.collect.ImmutableSet;
  13 import io.netty.buffer.ByteBuf;
  14 import io.netty.buffer.Unpooled;
  15 import java.util.Collection;
  16 import java.util.List;
  17 import java.util.stream.Collectors;
  18 import javax.annotation.Nonnull;
  19 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
  20 import org.opendaylight.protocol.bgp.linkstate.nlri.LinkstateNlriParser;
  21 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupport;
  22 import org.opendaylight.protocol.util.ByteArray;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateAddressFamily;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.LinkstateSubsequentAddressFamily;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.bgp.rib.rib.loc.rib.tables.routes.LinkstateRoutesCase;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.linkstate.destination.CLinkstateDestination;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.linkstate.routes.LinkstateRoutes;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.linkstate.routes.linkstate.routes.LinkstateRoute;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationLinkstateCaseBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.linkstate._case.DestinationLinkstate;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.linkstate._case.DestinationLinkstateBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.destination.DestinationType;
  33 import org.opendaylight.yangtools.yang.binding.DataObject;
  34 import org.opendaylight.yangtools.yang.common.QName;
  35 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  36 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  37 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
  38 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
  39 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  40 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  41 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  42 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
  43 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
  44 import org.slf4j.Logger;
  45 import org.slf4j.LoggerFactory;
  46
  47 final class LinkstateRIBSupport extends AbstractRIBSupport {
  48     private static final Logger LOG = LoggerFactory.getLogger(LinkstateRIBSupport.class);
  49     private static final QName ROUTE_KEY = QName.create(LinkstateRoute.QNAME, "route-key").intern();
  50     private static final LinkstateRIBSupport SINGLETON = new LinkstateRIBSupport();
  51     private final NodeIdentifier route = new NodeIdentifier(LinkstateRoute.QNAME);
  52     private final NodeIdentifier nlriRoutesList = new NodeIdentifier(CLinkstateDestination.QNAME);
  53
  54     private LinkstateRIBSupport() {
  55         super(LinkstateRoutesCase.class, LinkstateRoutes.class, LinkstateRoute.class, LinkstateAddressFamily.class, LinkstateSubsequentAddressFamily.class,
  56             DestinationLinkstate.QNAME);
  57     }
  58
  59     static LinkstateRIBSupport getInstance() {
  60         return SINGLETON;
  61     }
  62
  63     @Override
  64     public ImmutableCollection<Class<? extends DataObject>> cacheableAttributeObjects() {
  65         return ImmutableSet.of();
  66     }
  67
  68     @Override
  69     public ImmutableCollection<Class<? extends DataObject>> cacheableNlriObjects() {
  70         return ImmutableSet.of();
  71     }
  72
  73     @Override
  74     public boolean isComplexRoute() {
  75         return true;
  76     }
  77
  78     @Override
  79     protected void processDestination(final DOMDataWriteTransaction tx, final YangInstanceIdentifier routesPath,
  80         final ContainerNode destination, final ContainerNode attributes, final ApplyRoute function) {
  81         if (destination != null) {
  82             final Optional<DataContainerChild<? extends PathArgument, ?>> maybeRoutes = destination.getChild(this.nlriRoutesList);
  83             if (maybeRoutes.isPresent()) {
  84                 final DataContainerChild<? extends PathArgument, ?> routes = maybeRoutes.get();
  85                 if (routes instanceof UnkeyedListNode) {
  86                     final YangInstanceIdentifier base = routesPath.node(routesContainerIdentifier()).node(this.route);
  87                     for (final UnkeyedListEntryNode e : ((UnkeyedListNode) routes).getValue()) {
  88                         final NodeIdentifierWithPredicates routeKey = createRouteKey(e);
  89                         function.apply(tx, base, routeKey, e, attributes);
  90                     }
  91                 } else {
  92                     LOG.warn("Routes {} are not a map", routes);
  93                 }
  94             }
  95         }
  96     }
  97
  98     private NodeIdentifierWithPredicates createRouteKey(final UnkeyedListEntryNode linkstate) {
  99         final ByteBuf buffer = Unpooled.buffer();
 100         final CLinkstateDestination cLinkstateDestination = LinkstateNlriParser.extractLinkstateDestination(linkstate);
 101         LinkstateNlriParser.serializeNlri(cLinkstateDestination, buffer);
 102
 103         return new NodeIdentifierWithPredicates(LinkstateRoute.QNAME, ROUTE_KEY, ByteArray.readAllBytes(buffer));
 104     }
 105
 106     @Nonnull
 107     @Override
 108     protected DestinationType buildDestination(@Nonnull final Collection<MapEntryNode> routes) {
 109         return new DestinationLinkstateCaseBuilder().setDestinationLinkstate(
 110             new DestinationLinkstateBuilder().setCLinkstateDestination(extractRoutes(routes)).build()).build();
 111     }
 112
 113     @Nonnull
 114     @Override
 115     protected DestinationType buildWithdrawnDestination(@Nonnull final Collection<MapEntryNode> routes) {
 116         return new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationLinkstateCaseBuilder().setDestinationLinkstate(
 117             new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev150210.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.destination.linkstate._case.DestinationLinkstateBuilder().
 118                 setCLinkstateDestination(extractRoutes(routes)).build()).build();
 119     }
 120
 121     private List<CLinkstateDestination> extractRoutes(final Collection<MapEntryNode> routes) {
 122         return routes.stream().map(LinkstateNlriParser::extractLinkstateDestination).collect(Collectors.toList());
 123     }
 124 }