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
   9 package org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.actions;
  10
  11 import java.util.ArrayList;
  12 import java.util.List;
  13 import java.util.stream.Collectors;
  14 import org.apache.commons.lang3.StringUtils;
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
  17 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.action.BgpActionPolicy;
  18 import org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.AbstractCommunityHandler;
  19 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
  20 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
  21 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.BgpSetCommunityOptionType;
  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.SetCommunity;
  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.SetCommunityMethod;
  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.Inline;
  25 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;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.AttributesBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Communities;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.CommunitiesBuilder;
  30
  31 /**
  32  * Prepend / Replace / Remove a set of communities.
  33  */
  34 public final class SetCommunityHandler extends AbstractCommunityHandler implements BgpActionPolicy<SetCommunity> {
  35     public SetCommunityHandler(final DataBroker dataBroker) {
  36         super(dataBroker);
  37     }
  38
  39     @Override
  40     public Attributes applyImportAction(
  41             final RouteEntryBaseAttributes routeEntryInfo,
  42             final BGPRouteEntryImportParameters routeEntryImportParameters,
  43             final Attributes attributes,
  44             final SetCommunity bgpActions) {
  45         return setComm(attributes, bgpActions.getSetCommunityMethod(), bgpActions.getOptions());
  46     }
  47
  48     @Override
  49     public Attributes applyExportAction(
  50             final RouteEntryBaseAttributes routeEntryInfo,
  51             final BGPRouteEntryExportParameters routeEntryExportParameters,
  52             final Attributes attributes,
  53             final SetCommunity bgpActions) {
  54         return setComm(attributes, bgpActions.getSetCommunityMethod(), bgpActions.getOptions());
  55     }
  56
  57     private Attributes setComm(
  58             final Attributes attributes,
  59             final SetCommunityMethod setCommunityMethod,
  60             final BgpSetCommunityOptionType options) {
  61         if (setCommunityMethod instanceof Inline) {
  62             final Inline inline = (Inline) setCommunityMethod;
  63             final List<Communities> list = inline.getCommunities()
  64                     .stream().map(ge -> new CommunitiesBuilder().setAsNumber(ge.getAsNumber())
  65                             .setSemantics(ge.getSemantics()).build()).collect(Collectors.toList());
  66             return inlineSetComm(attributes, list, options);
  67         }
  68         return referenceSetComm(attributes, ((Reference) setCommunityMethod).getCommunitySetRef(), options);
  69     }
  70
  71     private Attributes referenceSetComm(
  72             final Attributes attributes,
  73             final String communitySetName,
  74             final BgpSetCommunityOptionType options) {
  75         final String setKey = StringUtils.substringBetween(communitySetName, "=\"", "\"");
  76         return inlineSetComm(attributes, this.communitySets.getUnchecked(setKey), options);
  77     }
  78
  79     private Attributes inlineSetComm(
  80             final Attributes attributes,
  81             final List<Communities> actionCommunities,
  82             final BgpSetCommunityOptionType options) {
  83
  84         final AttributesBuilder newAtt = new AttributesBuilder(attributes);
  85
  86         if (options.equals(BgpSetCommunityOptionType.REPLACE)) {
  87             return newAtt.setCommunities(actionCommunities).build();
  88         }
  89
  90         final List<Communities> actualComm;
  91         if (attributes.getCommunities() != null) {
  92             actualComm = new ArrayList<>(attributes.getCommunities());
  93         } else {
  94             actualComm = new ArrayList<>();
  95         }
  96
  97         switch (options) {
  98             case ADD:
  99                 actualComm.addAll(actionCommunities);
 100                 break;
 101             case REMOVE:
 102                 actualComm.removeAll(actionCommunities);
 103                 break;
 104             default:
 105                 throw new IllegalArgumentException("Option Type not Recognized!");
 106         }
 107
 108         return newAtt.setCommunities(actualComm).build();
 109     }
 110 }