Code Clean Up
[bgpcep.git] / bgp / flowspec / src / main / java / org / opendaylight / protocol / bgp / flowspec / l3vpn / AbstractFlowspecL3vpnRIBSupport.java
1 /*
2  * Copyright (c) 2016 Brocade Communications 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.flowspec.l3vpn;
9
10 import static org.opendaylight.bgp.concepts.RouteDistinguisherUtil.extractRouteDistinguisher;
11
12 import com.google.common.collect.Iterables;
13 import java.util.Collection;
14 import javax.annotation.Nonnull;
15
16 import org.opendaylight.protocol.bgp.flowspec.AbstractFlowspecRIBSupport;
17 import org.opendaylight.protocol.bgp.parser.spi.PathIdUtil;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev150807.FlowspecL3vpnSubsequentAddressFamily;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.PathId;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.destination.DestinationType;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.Route;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.tables.Routes;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.RouteDistinguisher;
25 import org.opendaylight.yangtools.yang.binding.DataObject;
26 import org.opendaylight.yangtools.yang.common.QName;
27 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
28 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
29
30 public abstract class AbstractFlowspecL3vpnRIBSupport<T extends AbstractFlowspecL3vpnNlriParser> extends AbstractFlowspecRIBSupport<T> {
31     private final NodeIdentifier routeDistinguisherNID;
32
33     protected AbstractFlowspecL3vpnRIBSupport(
34         final Class<? extends Routes> cazeClass,
35         final Class<? extends DataObject> containerClass,
36         final Class<? extends Route> listClass,
37         final QName dstContainerClassQName,
38         final Class<? extends AddressFamily> afiClass,
39         final T flowspecNlriParser
40     ) {
41         super(cazeClass, containerClass, listClass, afiClass, FlowspecL3vpnSubsequentAddressFamily.class, dstContainerClassQName, flowspecNlriParser);
42         this.routeDistinguisherNID = new NodeIdentifier(QName.create(routeQName(), "route-distinguisher").intern());
43     }
44
45     @Nonnull
46     @Override
47     protected DestinationType buildDestination(@Nonnull final Collection<MapEntryNode> routes) {
48         final MapEntryNode routesCont = Iterables.getOnlyElement(routes);
49         final PathId pathId = PathIdUtil.buildPathId(routesCont, routePathIdNid());
50         final RouteDistinguisher rd = extractRouteDistinguisher(routesCont, this.routeDistinguisherNID);
51         return this.nlriParser.createAdvertizedRoutesDestinationType(
52             new Object[] {rd, this.nlriParser.extractFlowspec(routesCont)},
53             pathId
54         );
55     }
56
57     @Nonnull
58     @Override
59     protected DestinationType buildWithdrawnDestination(@Nonnull final Collection<MapEntryNode> routes) {
60         final MapEntryNode routesCont = Iterables.getOnlyElement(routes);
61         final PathId pathId = PathIdUtil.buildPathId(routesCont, routePathIdNid());
62             final RouteDistinguisher rd = extractRouteDistinguisher(routesCont, this.routeDistinguisherNID);
63         return this.nlriParser.createWithdrawnDestinationType(
64             new Object[] {rd, this.nlriParser.extractFlowspec(Iterables.getOnlyElement(routes))},
65             pathId
66         );
67     }
68 }