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
8 package org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.actions;
10 import java.util.ArrayList;
11 import java.util.List;
12 import java.util.stream.Collectors;
13 import org.apache.commons.lang3.StringUtils;
14 import org.opendaylight.mdsal.binding.api.DataBroker;
15 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
16 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.action.BgpActionPolicy;
17 import org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.AbstractCommunityHandler;
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.http.openconfig.net.yang.bgp.policy.rev151009.BgpSetCommunityOptionType;
21 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.routing.policy.policy.definitions.policy.definition.statements.statement.actions.bgp.actions.SetCommunity;
22 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.routing.policy.policy.definitions.policy.definition.statements.statement.actions.bgp.actions.set.community.SetCommunityMethod;
23 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.routing.policy.policy.definitions.policy.definition.statements.statement.actions.bgp.actions.set.community.set.community.method.Inline;
24 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.routing.policy.policy.definitions.policy.definition.statements.statement.actions.bgp.actions.set.community.set.community.method.Reference;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.AttributesBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Communities;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.CommunitiesBuilder;
31 * Prepend / Replace / Remove a set of communities.
33 public final class SetCommunityHandler extends AbstractCommunityHandler implements BgpActionPolicy<SetCommunity> {
34 public SetCommunityHandler(final DataBroker dataBroker) {
39 public Attributes applyImportAction(
40 final RouteEntryBaseAttributes routeEntryInfo,
41 final BGPRouteEntryImportParameters routeEntryImportParameters,
42 final Attributes attributes,
43 final SetCommunity bgpActions) {
44 return setComm(attributes, bgpActions.getSetCommunityMethod(), bgpActions.getOptions());
48 public Attributes applyExportAction(
49 final RouteEntryBaseAttributes routeEntryInfo,
50 final BGPRouteEntryExportParameters routeEntryExportParameters,
51 final Attributes attributes,
52 final SetCommunity bgpActions) {
53 return setComm(attributes, bgpActions.getSetCommunityMethod(), bgpActions.getOptions());
56 private Attributes setComm(
57 final Attributes attributes,
58 final SetCommunityMethod setCommunityMethod,
59 final BgpSetCommunityOptionType options) {
60 if (setCommunityMethod instanceof Inline) {
61 final Inline inline = (Inline) setCommunityMethod;
62 final List<Communities> list = inline.getCommunities()
63 .stream().map(ge -> new CommunitiesBuilder().setAsNumber(ge.getAsNumber())
64 .setSemantics(ge.getSemantics()).build()).collect(Collectors.toList());
65 return inlineSetComm(attributes, list, options);
67 return referenceSetComm(attributes, ((Reference) setCommunityMethod).getCommunitySetRef(), options);
70 private Attributes referenceSetComm(
71 final Attributes attributes,
72 final String communitySetName,
73 final BgpSetCommunityOptionType options) {
74 final String setKey = StringUtils.substringBetween(communitySetName, "=\"", "\"");
75 return inlineSetComm(attributes, this.communitySets.getUnchecked(setKey), options);
78 private Attributes inlineSetComm(
79 final Attributes attributes,
80 final List<Communities> actionCommunities,
81 final BgpSetCommunityOptionType options) {
83 final AttributesBuilder newAtt = new AttributesBuilder(attributes);
85 if (options.equals(BgpSetCommunityOptionType.REPLACE)) {
86 return newAtt.setCommunities(actionCommunities).build();
89 final List<Communities> actualComm;
90 if (attributes.getCommunities() != null) {
91 actualComm = new ArrayList<>(attributes.getCommunities());
93 actualComm = new ArrayList<>();
98 actualComm.addAll(actionCommunities);
101 actualComm.removeAll(actionCommunities);
104 throw new IllegalArgumentException("Option Type not Recognized!");
107 return newAtt.setCommunities(actualComm).build();