bgp/openconfig-rp-statement/src/main/java/org/opendaylight/protocol/bgp/openconfig/routing/policy/statement/actions/SetCommunityHandler.java

   1 /*
   2  * Copyright (c) 2018 AT&T Intellectual Property. 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.openconfig.routing.policy.statement.actions;
   9
  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;
  29
  30 /**
  31  * Prepend / Replace / Remove a set of communities.
  32  */
  33 public final class SetCommunityHandler extends AbstractCommunityHandler implements BgpActionPolicy<SetCommunity> {
  34     public SetCommunityHandler(final DataBroker dataBroker) {
  35         super(dataBroker);
  36     }
  37
  38     @Override
  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());
  45     }
  46
  47     @Override
  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());
  54     }
  55
  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);
  66         }
  67         return referenceSetComm(attributes, ((Reference) setCommunityMethod).getCommunitySetRef(), options);
  68     }
  69
  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);
  76     }
  77
  78     private Attributes inlineSetComm(
  79             final Attributes attributes,
  80             final List<Communities> actionCommunities,
  81             final BgpSetCommunityOptionType options) {
  82
  83         final AttributesBuilder newAtt = new AttributesBuilder(attributes);
  84
  85         if (options.equals(BgpSetCommunityOptionType.REPLACE)) {
  86             return newAtt.setCommunities(actionCommunities).build();
  87         }
  88
  89         final List<Communities> actualComm;
  90         if (attributes.getCommunities() != null) {
  91             actualComm = new ArrayList<>(attributes.getCommunities());
  92         } else {
  93             actualComm = new ArrayList<>();
  94         }
  95
  96         switch (options) {
  97             case ADD:
  98                 actualComm.addAll(actionCommunities);
  99                 break;
 100             case REMOVE:
 101                 actualComm.removeAll(actionCommunities);
 102                 break;
 103             default:
 104                 throw new IllegalArgumentException("Option Type not Recognized!");
 105         }
 106
 107         return newAtt.setCommunities(actualComm).build();
 108     }
 109 }