Migrate to MD-SAL APIs
[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 package org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.conditions;
9
10 import java.util.Collections;
11 import java.util.List;
12 import org.apache.commons.lang3.StringUtils;
13 import org.opendaylight.mdsal.binding.api.DataBroker;
14 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.RouteEntryBaseAttributes;
15 import org.opendaylight.protocol.bgp.openconfig.routing.policy.spi.policy.condition.BgpConditionsPolicy;
16 import org.opendaylight.protocol.bgp.openconfig.routing.policy.statement.AbstractCommunityHandler;
17 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryExportParameters;
18 import org.opendaylight.protocol.bgp.rib.spi.policy.BGPRouteEntryImportParameters;
19 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.policy.rev151009.bgp.match.conditions.MatchCommunitySet;
20 import org.opendaylight.yang.gen.v1.http.openconfig.net.yang.bgp.types.rev151009.AfiSafiType;
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.rev180329.path.attributes.Attributes;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.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     @Override
35     public boolean matchImportCondition(
36             final Class<? extends AfiSafiType> afiSafi,
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 Class<? extends AfiSafiType> afiSafi,
47             final RouteEntryBaseAttributes routeEntryInfo,
48             final BGPRouteEntryExportParameters routeEntryExportParameters,
49             final List<Communities> communities,
50             final MatchCommunitySet conditions) {
51         return matchCondition(communities, conditions.getCommunitySet(), conditions.getMatchSetOptions());
52     }
53
54     @Override
55     public List<Communities> getConditionParameter(final Attributes attributes) {
56         return attributes.getCommunities();
57     }
58
59     private boolean matchCondition(
60             final List<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.path
61                     .attributes.attributes.Communities> communities,
62             final String communitySetName,
63             final MatchSetOptionsType matchSetOptions) {
64
65         final String setKey = StringUtils
66                 .substringBetween(communitySetName, "=\"", "\"");
67         final List<Communities> communityFilter = this.communitySets.getUnchecked(setKey);
68
69         if (communityFilter == null || communityFilter.isEmpty()) {
70             return false;
71         }
72
73         List<Communities> commAttributeList;
74         if (communities == null) {
75             commAttributeList = Collections.emptyList();
76         } else {
77             commAttributeList = communities;
78         }
79
80         if (matchSetOptions.equals(MatchSetOptionsType.ALL)) {
81             return commAttributeList.containsAll(communityFilter)
82                     && communityFilter.containsAll(commAttributeList);
83         }
84         final boolean noneInCommon = Collections.disjoint(commAttributeList, communityFilter);
85
86         if (matchSetOptions.equals(MatchSetOptionsType.ANY)) {
87             return !noneInCommon;
88         }
89         //(matchSetOptions.equals(MatchSetOptionsType.INVERT))
90         return noneInCommon;
91     }
92 }