bgp/openconfig-rp-statement/src/main/java/org/opendaylight/protocol/bgp/openconfig/routing/policy/statement/actions/SetOriginatorIdPrependHandler.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 org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
  12 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.action.BgpActionAugPolicy;
  13 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
  14 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
  15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.Attributes;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.AttributesBuilder;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev171207.path.attributes.attributes.OriginatorIdBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp._default.policy.rev180329.SetOriginatorIdPrepend;
  20
  21 /**
  22  * Prepend Originator Id.
  23  */
  24 public final class SetOriginatorIdPrependHandler implements BgpActionAugPolicy<SetOriginatorIdPrepend> {
  25     @Override
  26     public Attributes applyImportAction(
  27             final RouteEntryBaseAttributes routeEntryInfo,
  28             final BGPRouteEntryImportParameters routeEntryImportParameters,
  29             final Attributes attributes,
  30             final SetOriginatorIdPrepend bgpActions) {
  31
  32         final Ipv4Address defOri = bgpActions.getSetOriginatorIdPrepend().getOriginatorId();
  33         return prependOriginatorId(attributes, defOri == null
  34                 ? routeEntryInfo.getOriginatorId() : defOri);
  35     }
  36
  37     private Attributes prependOriginatorId(final Attributes attributes, final Ipv4Address originatorId) {
  38         if (attributes.getOriginatorId() != null) {
  39             return attributes;
  40         }
  41         final AttributesBuilder newAtt = new AttributesBuilder(attributes);
  42         return newAtt.setOriginatorId(new OriginatorIdBuilder().setOriginator(originatorId).build()).build();
  43     }
  44
  45     @Override
  46     public Attributes applyExportAction(
  47             final RouteEntryBaseAttributes routeEntryInfo,
  48             final BGPRouteEntryExportParameters routeEntryExportParameters,
  49             final Attributes attributes,
  50             final SetOriginatorIdPrepend bgpActions) {
  51         final Ipv4Address defOri = bgpActions.getSetOriginatorIdPrepend().getOriginatorId();
  52         return prependOriginatorId(attributes, defOri == null
  53                 ? routeEntryInfo.getOriginatorId() : defOri);
  54     }
  55 }