bgp/linkstate/src/main/java/org/opendaylight/protocol/bgp/linkstate/LinkstateAdjRIBsIn.java

   1 /*
   2  * Copyright (c) 2013 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 java.util.Comparator;
  11
  12 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  13 import org.opendaylight.protocol.bgp.rib.spi.AbstractAdjRIBsIn;
  14 import org.opendaylight.protocol.bgp.rib.spi.Peer;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.LinkstateDestination;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.linkstate.destination.CLinkstateDestination;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.LinkstateRoute;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.LinkstateRouteBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.LinkstateRouteKey;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.ObjectType;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.LinkBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.NodeBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.Prefix;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.PrefixBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.link.LinkDescriptorsBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.link.LocalNodeDescriptorsBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.link.RemoteNodeDescriptorsBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.linkstate.rev130918.loc.rib.tables.routes.linkstate.routes.linkstate.route.object.type.node.NodeDescriptorsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.PathAttributes;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.update.path.attributes.MpReachNlri;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.update.path.attributes.MpUnreachNlri;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.Tables;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.route.AttributesBuilder;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36
  37 import com.google.common.base.Preconditions;
  38
  39 final class LinkstateAdjRIBsIn extends AbstractAdjRIBsIn<CLinkstateDestination, LinkstateRoute> {
  40         private static abstract class LinkstateRIBEntryData extends RIBEntryData<CLinkstateDestination, LinkstateRoute> {
  41                 protected LinkstateRIBEntryData(final PathAttributes attributes) {
  42                         super(attributes);
  43                 }
  44
  45                 protected abstract ObjectType createObject(CLinkstateDestination key);
  46
  47                 @Override
  48                 protected final LinkstateRoute getDataObject(final CLinkstateDestination key) {
  49                         final LinkstateRouteBuilder builder = new LinkstateRouteBuilder();
  50
  51                         builder.setIdentifier(key.getIdentifier());
  52                         builder.setProtocolId(key.getProtocolId());
  53                         builder.setDistinguisher(key.getDistinguisher());
  54                         builder.setAttributes(new AttributesBuilder(getPathAttributes()).build());
  55                         builder.setObjectType(Preconditions.checkNotNull(createObject(key)));
  56
  57                         return builder.build();
  58                 }
  59         }
  60
  61         LinkstateAdjRIBsIn(final Comparator<PathAttributes> comparator, final TablesKey key) {
  62                 super(comparator, key);
  63         }
  64
  65         @Override
  66         public InstanceIdentifier<?> identifierForKey(final InstanceIdentifier<Tables> basePath, final CLinkstateDestination key) {
  67                 return InstanceIdentifier.builder(basePath).node(LinkstateRoute.class, new LinkstateRouteKey(LinkstateNlriParser.serializeNlri(key))).toInstance();
  68         }
  69
  70         @Override
  71         public void addRoutes(
  72                         final DataModificationTransaction trans,
  73                         final Peer peer,
  74                         final MpReachNlri nlri,
  75                         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.PathAttributes attributes) {
  76                 final CLinkstateDestination key = (CLinkstateDestination) ((LinkstateDestination) nlri.getAdvertizedRoutes().getDestinationType()).getCLinkstateDestination();
  77
  78                 RIBEntryData<CLinkstateDestination, LinkstateRoute> data = null;
  79                 switch (key.getNlriType()) {
  80                 case Ipv4Prefix:
  81                 case Ipv6Prefix:
  82                         data = new LinkstateRIBEntryData(attributes) {
  83                                 @Override
  84                                 protected Prefix createObject(final CLinkstateDestination key) {
  85                                         return new PrefixBuilder(key.getPrefixDescriptors()).build();
  86                                 }
  87                         };
  88                         break;
  89                 case Link:
  90                         data = new LinkstateRIBEntryData(attributes) {
  91                                 @Override
  92                                 protected ObjectType createObject(final CLinkstateDestination key) {
  93                                         final LinkBuilder b = new LinkBuilder();
  94
  95                                         b.setLinkDescriptors(new LinkDescriptorsBuilder(key.getLinkDescriptors()).build());
  96                                         b.setLocalNodeDescriptors(new LocalNodeDescriptorsBuilder(key.getLocalNodeDescriptors()).build());
  97                                         b.setRemoteNodeDescriptors(new RemoteNodeDescriptorsBuilder(key.getRemoteNodeDescriptors()).build());
  98
  99                                         return b.build();
 100                                 }
 101                         };
 102                         break;
 103                 case Node:
 104                         data = new LinkstateRIBEntryData(attributes) {
 105                                 @Override
 106                                 protected ObjectType createObject(final CLinkstateDestination key) {
 107                                         return new NodeBuilder().setNodeDescriptors(new NodeDescriptorsBuilder(key.getLocalNodeDescriptors()).build()).build();
 108                                 }
 109                         };
 110                         break;
 111                 }
 112
 113                 if (data == null) {
 114                         throw new IllegalStateException("Unhandled link-state NLRI type " + key.getNlriType());
 115                 }
 116                 super.add(trans, peer, key, data);
 117         }
 118
 119         @Override
 120         public void removeRoutes(final DataModificationTransaction trans, final Peer peer, final MpUnreachNlri nlri) {
 121                 final CLinkstateDestination key = (CLinkstateDestination) ((LinkstateDestination) nlri.getWithdrawnRoutes().getDestinationType()).getCLinkstateDestination();
 122
 123                 super.remove(trans, peer, key);
 124         }
 125 }