Fix EffectiveRibInWriter.writeTable()
[bgpcep.git] / bgp / extensions / inet / src / main / java / org / opendaylight / protocol / bgp / inet / IPv4RIBSupport.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.inet;
9
10 import static com.google.common.base.Verify.verify;
11
12 import java.util.ArrayList;
13 import java.util.Collection;
14 import java.util.Collections;
15 import java.util.List;
16 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
17 import org.opendaylight.protocol.bgp.parser.spi.PathIdUtil;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Prefix;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCase;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.bgp.rib.rib.loc.rib.tables.routes.Ipv4RoutesCaseBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.DestinationIpv4;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.DestinationIpv4Builder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.destination.ipv4.Ipv4Prefixes;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.prefixes.destination.ipv4.Ipv4PrefixesBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.Ipv4Routes;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.Ipv4RoutesBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4Route;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4RouteBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.ipv4.routes.ipv4.routes.Ipv4RouteKey;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.update.attributes.mp.reach.nlri.advertized.routes.destination.type.DestinationIpv4CaseBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.Ipv4PrefixAndPathId;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.PathId;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.destination.DestinationType;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
37 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
38 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodes;
39
40 /**
41  * Class supporting IPv4 unicast RIBs.
42  */
43 final class IPv4RIBSupport extends AbstractIPRibSupport<Ipv4RoutesCase, Ipv4Routes, Ipv4Route, Ipv4RouteKey> {
44
45     private static final Ipv4Routes EMPTY_CONTAINER
46             = new Ipv4RoutesBuilder().setIpv4Route(Collections.emptyList()).build();
47     private static final Ipv4RoutesCase EMPTY_CASE = new Ipv4RoutesCaseBuilder().setIpv4Routes(EMPTY_CONTAINER).build();
48     private static IPv4RIBSupport SINGLETON = null;
49
50     private IPv4RIBSupport(final BindingNormalizedNodeSerializer mappingService) {
51         super(
52                 mappingService,
53                 Ipv4PrefixAndPathId.class,
54                 Ipv4AddressFamily.class,
55                 Ipv4RoutesCase.class,
56                 Ipv4Routes.class,
57                 Ipv4Route.class,
58                 DestinationIpv4.QNAME,
59                 Ipv4Prefixes.QNAME);
60     }
61
62     static synchronized IPv4RIBSupport getInstance(final BindingNormalizedNodeSerializer mappingService) {
63         if (SINGLETON == null) {
64             SINGLETON = new IPv4RIBSupport(mappingService);
65         }
66         return SINGLETON;
67     }
68
69     private List<Ipv4Prefixes> extractPrefixes(final Collection<MapEntryNode> routes) {
70         final List<Ipv4Prefixes> prefs = new ArrayList<>(routes.size());
71         for (final MapEntryNode route : routes) {
72             final String prefix = (String) NormalizedNodes.findNode(route, routePrefixIdentifier()).get().getValue();
73             final Ipv4PrefixesBuilder prefixBuilder = new Ipv4PrefixesBuilder().setPrefix(new Ipv4Prefix(prefix));
74             prefixBuilder.setPathId(PathIdUtil.buildPathId(route, routePathIdNid()));
75             prefs.add(prefixBuilder.build());
76         }
77         return prefs;
78     }
79
80     @Override
81     protected DestinationType buildDestination(final Collection<MapEntryNode> routes) {
82         return new DestinationIpv4CaseBuilder().setDestinationIpv4(new DestinationIpv4Builder()
83                 .setIpv4Prefixes(extractPrefixes(routes)).build()).build();
84     }
85
86     @Override
87     protected DestinationType buildWithdrawnDestination(final Collection<MapEntryNode> routes) {
88         return new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.update
89                 .attributes.mp.unreach.nlri.withdrawn.routes.destination.type.DestinationIpv4CaseBuilder()
90                 .setDestinationIpv4(new DestinationIpv4Builder().setIpv4Prefixes(extractPrefixes(routes))
91                         .build()).build();
92     }
93
94     @Override
95     public Ipv4Route createRoute(final Ipv4Route route, final Ipv4RouteKey key, final Attributes attributes) {
96         final Ipv4RouteBuilder builder;
97         if (route != null) {
98             builder = new Ipv4RouteBuilder(route);
99         } else {
100             builder = new Ipv4RouteBuilder();
101         }
102         builder.withKey(key).setAttributes(attributes);
103         return builder.build();
104     }
105
106     @Override
107     public Ipv4RoutesCase emptyRoutesCase() {
108         return EMPTY_CASE;
109     }
110
111     @Override
112     public Ipv4Routes emptyRoutesContainer() {
113         return EMPTY_CONTAINER;
114     }
115
116     @Override
117     public Ipv4RouteKey createRouteListKey(final PathId pathId, final String routeKey) {
118         return new Ipv4RouteKey(pathId, routeKey);
119     }
120
121     @Override
122     public PathId extractPathId(final Ipv4RouteKey routeListKey) {
123         return routeListKey.getPathId();
124     }
125
126     @Override
127     public String extractRouteKey(final Ipv4RouteKey routeListKey) {
128         return routeListKey.getRouteKey();
129     }
130
131     @Override
132     public List<Ipv4Route> extractAdjRibInRoutes(Routes routes) {
133         verify(routes instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329
134             .bgp.rib.rib.peer.adj.rib.in.tables.routes.Ipv4RoutesCase, "Unrecognized routes %s", routes);
135         return ((org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.inet.rev180329.bgp.rib.rib.peer
136                 .adj.rib.in.tables.routes.Ipv4RoutesCase) routes).getIpv4Routes().nonnullIpv4Route();
137     }
138 }