Fix EffectiveRibInWriter.writeTable()
[bgpcep.git] / bgp / extensions / evpn / src / main / java / org / opendaylight / protocol / bgp / evpn / impl / EvpnRibSupport.java
1 /*
2  * Copyright (c) 2016 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.evpn.impl;
9
10 import static com.google.common.base.Verify.verify;
11
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.controller.md.sal.dom.api.DOMDataWriteTransaction;
21 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
22 import org.opendaylight.protocol.bgp.evpn.impl.nlri.EvpnNlriParser;
23 import org.opendaylight.protocol.bgp.parser.spi.PathIdUtil;
24 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupport;
25 import org.opendaylight.protocol.util.ByteArray;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.EvpnSubsequentAddressFamily;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.L2vpnAddressFamily;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.bgp.rib.rib.loc.rib.tables.routes.EvpnRoutesCase;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.bgp.rib.rib.loc.rib.tables.routes.EvpnRoutesCaseBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.destination.EvpnDestination;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.routes.EvpnRoutes;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.routes.EvpnRoutesBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.routes.evpn.routes.EvpnRoute;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.routes.evpn.routes.EvpnRouteBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.evpn.routes.evpn.routes.EvpnRouteKey;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.evpn._case.DestinationEvpn;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationEvpnCaseBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.destination.evpn._case.DestinationEvpnBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.PathId;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
43 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
44 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
45 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
46 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
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.MapEntryNode;
50 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
51 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
52 import org.slf4j.Logger;
53 import org.slf4j.LoggerFactory;
54
55 final class EvpnRibSupport extends AbstractRIBSupport<EvpnRoutesCase, EvpnRoutes, EvpnRoute, EvpnRouteKey> {
56     private static final Logger LOG = LoggerFactory.getLogger(EvpnRibSupport.class);
57
58     private static final NodeIdentifier NLRI_ROUTES_LIST = NodeIdentifier.create(EvpnDestination.QNAME);
59     private static final EvpnRoutes EMPTY_CONTAINER
60             =  new EvpnRoutesBuilder().setEvpnRoute(Collections.emptyList()).build();
61     private static final EvpnRoutesCase EMPTY_CASE =
62             new EvpnRoutesCaseBuilder().setEvpnRoutes(EMPTY_CONTAINER).build();
63     private static EvpnRibSupport SINGLETON;
64
65     private EvpnRibSupport(final BindingNormalizedNodeSerializer mappingService) {
66         super(mappingService,
67                 EvpnRoutesCase.class,
68                 EvpnRoutes.class,
69                 EvpnRoute.class,
70                 L2vpnAddressFamily.class,
71                 EvpnSubsequentAddressFamily.class,
72                 DestinationEvpn.QNAME);
73     }
74
75     static synchronized EvpnRibSupport getInstance(final BindingNormalizedNodeSerializer mappingService) {
76         if (SINGLETON == null) {
77             SINGLETON = new EvpnRibSupport(mappingService);
78         }
79         return SINGLETON;
80     }
81
82     @Override
83     protected DestinationType buildDestination(final Collection<MapEntryNode> routes) {
84         return new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.update
85                 .attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationEvpnCaseBuilder()
86                 .setDestinationEvpn(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn
87                         .rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination
88                         .evpn._case.DestinationEvpnBuilder().setEvpnDestination(extractRoutes(routes)).build()).build();
89     }
90
91     @Override
92     protected DestinationType buildWithdrawnDestination(final Collection<MapEntryNode> routes) {
93         return new DestinationEvpnCaseBuilder().setDestinationEvpn(new DestinationEvpnBuilder()
94                 .setEvpnDestination(extractRoutes(routes)).build()).build();
95     }
96
97     private static List<EvpnDestination> extractRoutes(final Collection<MapEntryNode> routes) {
98         return routes.stream().map(EvpnNlriParser::extractEvpnDestination).collect(Collectors.toList());
99     }
100
101     @Override
102     protected Collection<NodeIdentifierWithPredicates> processDestination(final DOMDataWriteTransaction tx,
103                                                                           final YangInstanceIdentifier routesPath,
104                                                                           final ContainerNode destination,
105                                                                           final ContainerNode attributes,
106                                                                           final ApplyRoute function) {
107         if (destination != null) {
108             final Optional<DataContainerChild<? extends PathArgument, ?>> maybeRoutes = destination
109                     .getChild(NLRI_ROUTES_LIST);
110             if (maybeRoutes.isPresent()) {
111                 final DataContainerChild<? extends PathArgument, ?> routes = maybeRoutes.get();
112                 if (routes instanceof UnkeyedListNode) {
113                     final YangInstanceIdentifier base = routesYangInstanceIdentifier(routesPath);
114                     final Collection<UnkeyedListEntryNode> routesList = ((UnkeyedListNode) routes).getValue();
115                     final List<NodeIdentifierWithPredicates> keys = new ArrayList<>(routesList.size());
116                     for (final UnkeyedListEntryNode evpnDest : routesList) {
117                         final NodeIdentifierWithPredicates routeKey = createRouteKey(evpnDest);
118                         function.apply(tx, base, routeKey, evpnDest, attributes);
119                         keys.add(routeKey);
120                     }
121                     return keys;
122                 } else {
123                     LOG.warn("Routes {} are not a map", routes);
124                 }
125             }
126         }
127         return Collections.emptyList();
128     }
129
130     private NodeIdentifierWithPredicates createRouteKey(final UnkeyedListEntryNode evpn) {
131         final ByteBuf buffer = Unpooled.buffer();
132         final EvpnDestination dest = EvpnNlriParser.extractRouteKeyDestination(evpn);
133         EvpnNlriParser.serializeNlri(Collections.singletonList(dest), buffer);
134         final Optional<DataContainerChild<? extends PathArgument, ?>> maybePathIdLeaf =
135                 evpn.getChild(routePathIdNid());
136         return PathIdUtil.createNidKey(routeQName(), routeKeyQName(),
137                 pathIdQName(), ByteArray.encodeBase64(buffer), maybePathIdLeaf);
138     }
139
140     @Override
141     public EvpnRoute createRoute(final EvpnRoute route, final EvpnRouteKey key, final Attributes attributes) {
142         final EvpnRouteBuilder builder;
143         if (route != null) {
144             builder = new EvpnRouteBuilder(route);
145         } else {
146             builder = new EvpnRouteBuilder();
147         }
148         return builder.withKey(key).setAttributes(attributes).build();
149     }
150
151     @Override
152     public EvpnRoutesCase emptyRoutesCase() {
153         return EMPTY_CASE;
154     }
155
156     @Override
157     public EvpnRoutes emptyRoutesContainer() {
158         return EMPTY_CONTAINER;
159     }
160
161     @Override
162     public EvpnRouteKey createRouteListKey(final PathId pathId, final String routeKey) {
163         return new EvpnRouteKey(pathId, routeKey);
164     }
165
166     @Override
167     public PathId extractPathId(final EvpnRouteKey routeListKey) {
168         return routeListKey.getPathId();
169     }
170
171     @Override
172     public String extractRouteKey(final EvpnRouteKey routeListKey) {
173         return routeListKey.getRouteKey();
174     }
175
176     @Override
177     public List<EvpnRoute> extractAdjRibInRoutes(Routes routes) {
178         verify(routes instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329
179             .bgp.rib.rib.peer.adj.rib.in.tables.routes.EvpnRoutesCase, "Unrecognized routes %s", routes);
180         return ((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329
181                 .bgp.rib.rib.peer.adj.rib.in.tables.routes.EvpnRoutesCase) routes).getEvpnRoutes().nonnullEvpnRoute();
182     }
183 }