Route Constrain policies
[bgpcep.git] / bgp / openconfig-rp-statement / src / main / java / org / opendaylight / protocol / bgp / openconfig / routing / policy / statement / conditions / MatchCommunitySetHandler.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.conditions;
10
11 import java.util.Collections;
12 import java.util.List;
13 import org.apache.commons.lang3.StringUtils;
14 import org.opendaylight.controller.md.sal.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.condition.BgpConditionsPolicy;
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.bgp.match.conditions.MatchCommunitySet;
21 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
22 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.policy.types.rev151009.MatchSetOptionsType;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.Attributes;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path.attributes.attributes.Communities;
25
26 /**
27  * Match a set of Communities (ALL, ANY, INVERT).
28  */
29 public final class MatchCommunitySetHandler
30         extends AbstractCommunityHandler implements BgpConditionsPolicy<MatchCommunitySet, List<Communities>> {
31     public MatchCommunitySetHandler(final DataBroker databroker) {
32         super(databroker);
33     }
34
35
36     @Override
37     public boolean matchImportCondition(
38             final Class<? extends AfiSafiType> afiSafi,
39             final RouteEntryBaseAttributes routeEntryInfo,
40             final BGPRouteEntryImportParameters routeEntryImportParameters,
41             final List<Communities> communities,
42             final MatchCommunitySet conditions) {
43         return matchCondition(communities, conditions.getCommunitySet(), conditions.getMatchSetOptions());
44     }
45
46     @Override
47     public boolean matchExportCondition(
48             final Class<? extends AfiSafiType> afiSafi,
49             final RouteEntryBaseAttributes routeEntryInfo,
50             final BGPRouteEntryExportParameters routeEntryExportParameters,
51             final List<Communities> communities,
52             final MatchCommunitySet conditions) {
53         return matchCondition(communities, conditions.getCommunitySet(), conditions.getMatchSetOptions());
54     }
55
56     @Override
57     public List<Communities> getConditionParameter(final Attributes attributes) {
58         return attributes.getCommunities();
59     }
60
61     private boolean matchCondition(
62             final List<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path
63                     .attributes.attributes.Communities> communities,
64             final String communitySetName,
65             final MatchSetOptionsType matchSetOptions) {
66
67         final String setKey = StringUtils
68                 .substringBetween(communitySetName, "=\"", "\"");
69         final List<Communities> communityFilter = this.communitySets.getUnchecked(setKey);
70
71         if (communityFilter == null || communityFilter.isEmpty()) {
72             return false;
73         }
74
75         List<Communities> commAttributeList;
76         if (communities == null) {
77             commAttributeList = Collections.emptyList();
78         } else {
79             commAttributeList = communities;
80         }
81
82         if (matchSetOptions.equals(MatchSetOptionsType.ALL)) {
83             return commAttributeList.containsAll(communityFilter)
84                     && communityFilter.containsAll(commAttributeList);
85         }
86         final boolean noneInCommon = Collections.disjoint(commAttributeList, communityFilter);
87
88         if (matchSetOptions.equals(MatchSetOptionsType.ANY)) {
89             return !noneInCommon;
90         }
91         //(matchSetOptions.equals(MatchSetOptionsType.INVERT))
92         return noneInCommon;
93     }
94 }