2 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
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
9 package org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.actions;
11 import com.google.common.collect.ImmutableMap;
12 import java.util.List;
14 import java.util.function.Function;
15 import java.util.stream.Collectors;
16 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
17 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.action.BgpActionAugPolicy;
18 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
19 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.Attributes;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.AttributesBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.ExtendedCommunities;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.UnrecognizedAttributes;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev200120.path.attributes.attributes.UnrecognizedAttributesKey;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.ExtendedCommunity;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp._default.policy.rev200120.NonTransitiveAttributesFilter;
29 * Removes non transitive attributes.
31 public final class NonTransitiveAttributesFilterHandler implements BgpActionAugPolicy<NonTransitiveAttributesFilter> {
32 private static final NonTransitiveAttributesFilterHandler INSTANCE = new NonTransitiveAttributesFilterHandler();
34 private NonTransitiveAttributesFilterHandler() {
38 public static NonTransitiveAttributesFilterHandler getInstance() {
43 public Attributes applyImportAction(
44 final RouteEntryBaseAttributes routeEntryInfo,
45 final BGPRouteEntryImportParameters routeEntryImportParameters,
46 final Attributes attributes,
47 final NonTransitiveAttributesFilter bgpActions) {
48 return filterAttributes(attributes);
51 private static Attributes filterAttributes(final Attributes attributes) {
52 final AttributesBuilder builder = new AttributesBuilder()
53 .setCNextHop(attributes.getCNextHop())
54 .setOrigin(attributes.getOrigin())
55 .setAsPath(attributes.getAsPath())
56 .setCommunities(attributes.getCommunities());
58 final Map<UnrecognizedAttributesKey, UnrecognizedAttributes> oldAtt = attributes.getUnrecognizedAttributes();
60 // TODO: consider using Maps.filterValues(attributes.getUnrecognizedAttributes(),
61 // UnrecognizedAttributes::isTransitive)) ?
62 builder.setUnrecognizedAttributes(attributes.getUnrecognizedAttributes().values().stream()
63 .filter(UnrecognizedAttributes::isTransitive)
64 .collect(ImmutableMap.toImmutableMap(UnrecognizedAttributes::key, Function.identity())));
66 final List<ExtendedCommunities> oldExt = attributes.getExtendedCommunities();
68 builder.setExtendedCommunities(oldExt.stream()
69 .filter(ExtendedCommunity::isTransitive)
70 .collect(Collectors.toList()));
72 return builder.build();
76 public Attributes applyExportAction(final RouteEntryBaseAttributes routeEntryInfo,
77 final BGPRouteEntryExportParameters routeEntryExportParameters,
78 final Attributes attributes,
79 final NonTransitiveAttributesFilter bgpActions) {
80 return filterAttributes(attributes);