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 org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
13 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.action.BgpActionAugPolicy;
14 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
15 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
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.ClusterIdBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.ClusterIdentifier;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.odl.bgp._default.policy.rev180109.SetClusterIdPrepend;
25 public final class SetClusterIdPrependHandler implements BgpActionAugPolicy<SetClusterIdPrepend> {
27 public Attributes applyImportAction(
28 final RouteEntryBaseAttributes routeEntryInfo,
29 final BGPRouteEntryImportParameters importParameters,
30 final Attributes attributes,
31 final SetClusterIdPrepend bgpActions) {
32 final ClusterIdentifier clusterIdLocal = importParameters.getFromClusterId() == null
33 ? routeEntryInfo.getClusterId() : importParameters.getFromClusterId();
34 return prependClusterId(attributes, clusterIdLocal);
37 private Attributes prependClusterId(final Attributes attributes, final ClusterIdentifier clusterId) {
38 final AttributesBuilder newAtt = new AttributesBuilder(attributes);
39 final List<ClusterIdentifier> newClusterList = new ArrayList<>();
40 newClusterList.add(clusterId);
41 if (attributes.getClusterId() != null && !attributes.getClusterId().getCluster().isEmpty()) {
42 final List<ClusterIdentifier> oldList = attributes.getClusterId().getCluster();
43 newClusterList.addAll(oldList);
45 return newAtt.setClusterId(new ClusterIdBuilder().setCluster(newClusterList).build()).build();
49 public Attributes applyExportAction(
50 final RouteEntryBaseAttributes routeEntryInfo,
51 final BGPRouteEntryExportParameters exportParameters,
52 final Attributes attributes,
53 final SetClusterIdPrepend bgpActions) {
54 final ClusterIdentifier clusterIdLocal = exportParameters.getFromClusterId() == null
55 ? routeEntryInfo.getClusterId() : exportParameters.getFromClusterId();
56 return prependClusterId(attributes, clusterIdLocal);