Fix bugs on filters
[policy/models.git] / models-pdp / src / main / java / org / onap / policy / models / pdp / concepts / PdpGroupFilter.java
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2019 Nordix Foundation.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.models.pdp.concepts;
22
23 import java.util.List;
24 import java.util.stream.Collectors;
25
26 import lombok.Builder;
27 import lombok.Data;
28 import lombok.NonNull;
29
30 import org.apache.commons.lang3.ObjectUtils;
31 import org.onap.policy.models.base.PfObjectFilter;
32 import org.onap.policy.models.pdp.enums.PdpState;
33 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyIdentifier;
34 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicyTypeIdentifier;
35
36 /**
37  * Filter class for searches for {@link PdpGroup} instances. If any fields are null, they are ignored.
38  *
39  * @author Liam Fallon (liam.fallon@est.tech)
40  */
41 @Builder
42 @Data
43 public class PdpGroupFilter implements PfObjectFilter<PdpGroup> {
44     public static final String LATEST_VERSION = "LATEST";
45
46     // Regular expression
47     private String name;
48
49     // Regular Expression, set to LATEST_VERRSION to get the latest version
50     private String version;
51
52     private PdpState groupState;
53
54     // Regular expression
55     private String pdpType;
56
57     // Set regular expressions on fields to match policy type names and versions
58     private ToscaPolicyTypeIdentifier policyType;
59
60     // Set regular expressions on fields to match policy names and versions
61     private ToscaPolicyIdentifier policy;
62
63     private PdpState pdpState;
64
65     @Override
66     public List<PdpGroup> filter(@NonNull final List<PdpGroup> originalList) {
67
68         // @formatter:off
69         List<PdpGroup> returnList = originalList.stream()
70                 .filter(p -> filterOnRegexp(p.getName(),    name))
71                 .filter(p -> version.equals(LATEST_VERSION) || filterOnRegexp(p.getVersion(), version))
72                 .filter(p -> ObjectUtils.compare(p.getPdpGroupState(), groupState) == 0)
73                 .filter(p -> filterOnPdpType(p, pdpType))
74                 .filter(p -> filterOnPolicyType(p, policyType))
75                 .filter(p -> filterOnPolicy(p, policy))
76                 .filter(p -> filterOnPdpState(p, pdpState))
77                 .collect(Collectors.toList());
78         // @formatter:off
79
80         if (LATEST_VERSION.equals(version)) {
81             returnList = this.latestVersionFilter(returnList);
82         }
83
84         return returnList;
85     }
86
87     /**
88      * Filter PDP groups on PDP type.
89      *
90      * @param pdpGroup the PDP group to check
91      * @param pdpType the PDP type to check for
92      * @return true if the filter should let this PDP group through
93      */
94     private boolean filterOnPdpType(final PdpGroup pdpGroup, final String pdpType) {
95         if (pdpType == null) {
96             return true;
97         }
98
99         for (PdpSubGroup pdpSubGroup: pdpGroup.getPdpSubgroups()) {
100             if (pdpSubGroup.getPdpType().equals(pdpType)) {
101                 return true;
102             }
103         }
104
105         return false;
106     }
107
108     /**
109      * Filter PDP groups on policy type.
110      *
111      * @param pdpGroup the PDP group to check
112      * @param policyTypeFilter the policy type regular expressions to check for
113      * @return true if the filter should let this PDP group through
114      */
115     private boolean filterOnPolicyType(final PdpGroup pdpGroup, final ToscaPolicyTypeIdentifier policyTypeFiler) {
116         if (policyTypeFiler == null) {
117             return true;
118         }
119
120         for (PdpSubGroup pdpSubGroup: pdpGroup.getPdpSubgroups()) {
121             for (ToscaPolicyTypeIdentifier foundPolicyType : pdpSubGroup.getSupportedPolicyTypes()) {
122                 if (foundPolicyType.getName().matches(policyTypeFiler.getName())
123                         && foundPolicyType.getVersion().matches(policyTypeFiler.getVersion())) {
124                     return true;
125                 }
126             }
127         }
128
129         return false;
130     }
131
132     /**
133      * Filter PDP groups on policy.
134      *
135      * @param pdpGroup the PDP group to check
136      * @param policyFilter the policy regular expressions to check for
137      * @return true if the filter should let this PDP group through
138      */
139     private boolean filterOnPolicy(final PdpGroup pdpGroup, final ToscaPolicyIdentifier policyFiler) {
140         if (policyFiler == null) {
141             return true;
142         }
143
144         for (PdpSubGroup pdpSubGroup: pdpGroup.getPdpSubgroups()) {
145             for (ToscaPolicyIdentifier foundPolicy : pdpSubGroup.getPolicies()) {
146                 if (foundPolicy.getName().matches(policyFiler.getName())
147                         && foundPolicy.getVersion().matches(policyFiler.getVersion())) {
148                     return true;
149                 }
150             }
151         }
152
153         return false;
154     }
155
156     /**
157      * Filter PDP groups on PDP state.
158      *
159      * @param pdpGroup the PDP group to check
160      * @param policyFilter the policy regular expressions to check for
161      * @return true if the filter should let this PDP group through
162      */
163     private boolean filterOnPdpState(final PdpGroup pdpGroup, final PdpState pdpState) {
164         if (pdpState == null) {
165             return true;
166         }
167
168         for (PdpSubGroup pdpSubGroup: pdpGroup.getPdpSubgroups()) {
169             for (Pdp pdp : pdpSubGroup.getPdpInstances()) {
170                 if (pdpState.equals(pdp.getPdpState())) {
171                     return true;
172                 }
173             }
174         }
175
176         return false;
177     }
178 }