Drop <R, I> generic from RIBSupport
[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 io.netty.buffer.ByteBuf;
11 import io.netty.buffer.Unpooled;
12 import java.util.ArrayList;
13 import java.util.Collection;
14 import java.util.Collections;
15 import java.util.List;
16 import java.util.Optional;
17 import java.util.stream.Collectors;
18 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
19 import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction;
20 import org.opendaylight.protocol.bgp.evpn.impl.nlri.EvpnNlriParser;
21 import org.opendaylight.protocol.bgp.parser.spi.PathIdUtil;
22 import org.opendaylight.protocol.bgp.rib.spi.AbstractRIBSupport;
23 import org.opendaylight.protocol.util.ByteArray;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.EvpnSubsequentAddressFamily;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.L2vpnAddressFamily;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.bgp.rib.rib.loc.rib.tables.routes.EvpnRoutesCase;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.evpn.destination.EvpnDestination;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.evpn.routes.EvpnRoutes;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.evpn.routes.evpn.routes.EvpnRoute;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination.evpn._case.DestinationEvpn;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationEvpnCaseBuilder;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.update.attributes.mp.unreach.nlri.withdrawn.routes.destination.type.destination.evpn._case.DestinationEvpnBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
34 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
35 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
36 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
37 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.PathArgument;
38 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
39 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
40 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
41 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
42 import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
43 import org.slf4j.Logger;
44 import org.slf4j.LoggerFactory;
45
46 final class EvpnRibSupport extends AbstractRIBSupport<EvpnRoutesCase, EvpnRoutes, EvpnRoute> {
47     private static final Logger LOG = LoggerFactory.getLogger(EvpnRibSupport.class);
48
49     private static final NodeIdentifier NLRI_ROUTES_LIST = NodeIdentifier.create(EvpnDestination.QNAME);
50
51     EvpnRibSupport(final BindingNormalizedNodeSerializer mappingService) {
52         super(mappingService,
53                 EvpnRoutesCase.class,
54                 EvpnRoutes.class,
55                 EvpnRoute.class,
56                 L2vpnAddressFamily.class,
57                 EvpnSubsequentAddressFamily.class,
58                 DestinationEvpn.QNAME);
59     }
60
61     @Override
62     protected DestinationType buildDestination(final Collection<MapEntryNode> routes) {
63         return new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev200120.update
64                 .attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationEvpnCaseBuilder()
65                 .setDestinationEvpn(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang
66                     .bgp.evpn.rev200120.update.attributes.mp.reach.nlri.advertized.routes.destination.type.destination
67                         .evpn._case.DestinationEvpnBuilder().setEvpnDestination(extractRoutes(routes)).build()).build();
68     }
69
70     @Override
71     protected DestinationType buildWithdrawnDestination(final Collection<MapEntryNode> routes) {
72         return new DestinationEvpnCaseBuilder().setDestinationEvpn(new DestinationEvpnBuilder()
73                 .setEvpnDestination(extractRoutes(routes)).build()).build();
74     }
75
76     private static List<EvpnDestination> extractRoutes(final Collection<MapEntryNode> routes) {
77         return routes.stream().map(EvpnNlriParser::extractEvpnDestination).collect(Collectors.toList());
78     }
79
80     @Override
81     protected Collection<NodeIdentifierWithPredicates> processDestination(final DOMDataTreeWriteTransaction tx,
82                                                                           final YangInstanceIdentifier routesPath,
83                                                                           final ContainerNode destination,
84                                                                           final ContainerNode attributes,
85                                                                           final ApplyRoute function) {
86         if (destination != null) {
87             final Optional<DataContainerChild<? extends PathArgument, ?>> maybeRoutes = destination
88                     .getChild(NLRI_ROUTES_LIST);
89             if (maybeRoutes.isPresent()) {
90                 final DataContainerChild<? extends PathArgument, ?> routes = maybeRoutes.get();
91                 if (routes instanceof UnkeyedListNode) {
92                     final YangInstanceIdentifier base = routesYangInstanceIdentifier(routesPath);
93                     final Collection<UnkeyedListEntryNode> routesList = ((UnkeyedListNode) routes).getValue();
94                     final List<NodeIdentifierWithPredicates> keys = new ArrayList<>(routesList.size());
95                     for (final UnkeyedListEntryNode evpnDest : routesList) {
96                         final NodeIdentifierWithPredicates routeKey = createRouteKey(evpnDest);
97                         function.apply(tx, base, routeKey, evpnDest, attributes);
98                         keys.add(routeKey);
99                     }
100                     return keys;
101                 }
102                 LOG.warn("Routes {} are not a map", routes);
103             }
104         }
105         return Collections.emptyList();
106     }
107
108     private NodeIdentifierWithPredicates createRouteKey(final UnkeyedListEntryNode evpn) {
109         final ByteBuf buffer = Unpooled.buffer();
110         final EvpnDestination dest = EvpnNlriParser.extractRouteKeyDestination(evpn);
111         EvpnNlriParser.serializeNlri(Collections.singletonList(dest), buffer);
112         final Optional<DataContainerChild<? extends PathArgument, ?>> maybePathIdLeaf =
113                 evpn.getChild(routePathIdNid());
114         return PathIdUtil.createNidKey(routeQName(), routeKeyTemplate(),
115                 ByteArray.encodeBase64(buffer), maybePathIdLeaf);
116     }
117 }