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