Add bug fixes and tests for filters
[policy/models.git] / models-tosca / src / main / java / org / onap / policy / models / tosca / authorative / concepts / ToscaPolicyTypeFilter.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.tosca.authorative.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.onap.policy.models.base.PfObjectFilter;
31
32 /**
33  * Filter class for searches for {@link ToscaPolicyType} instances.
34  * If any fields are null, they are ignored.
35  *
36  * @author Liam Fallon (liam.fallon@est.tech)
37  */
38 @Builder
39 @Data
40 public class ToscaPolicyTypeFilter implements PfObjectFilter<ToscaPolicyType> {
41     public static final String LATEST_VERSION = "LATEST";
42
43     // Regular expression
44     private String name;
45
46     // Regular Expression, set to LATEST_VERRSION to get the latest version
47     private String version;
48
49     @Override
50     public List<ToscaPolicyType> filter(@NonNull final List<ToscaPolicyType> originalList) {
51
52         // @formatter:off
53         List<ToscaPolicyType> returnList = originalList.stream()
54                 .filter(p -> filterString(p.getName(), name))
55                 .filter(p -> (version != null && LATEST_VERSION.equals(version))
56                         || filterString(p.getVersion(), version))
57                 .collect(Collectors.toList());
58         // @formatter:off
59
60         if (LATEST_VERSION.equals(version)) {
61             return this.latestVersionFilter(returnList);
62         }
63         else {
64             return returnList;
65         }
66     }
67 }