2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2022-2023 Nordix Foundation
4 * Modifications Copyright (C) 2023 TechMahindra Ltd.
5 * ================================================================================
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
10 * http://www.apache.org/licenses/LICENSE-2.0
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
18 * SPDX-License-Identifier: Apache-2.0
19 * ============LICENSE_END=========================================================
22 package org.onap.cps.ncmp.api.impl.inventory;
24 import static org.onap.cps.ncmp.api.impl.ncmppersistence.NcmpPersistence.NCMP_DATASPACE_NAME;
25 import static org.onap.cps.ncmp.api.impl.ncmppersistence.NcmpPersistence.NCMP_DMI_REGISTRY_ANCHOR;
26 import static org.onap.cps.ncmp.api.impl.ncmppersistence.NcmpPersistence.NCMP_DMI_REGISTRY_PARENT;
27 import static org.onap.cps.spi.FetchDescendantsOption.INCLUDE_ALL_DESCENDANTS;
28 import static org.onap.cps.spi.FetchDescendantsOption.OMIT_DESCENDANTS;
30 import java.util.Collection;
31 import java.util.Collections;
32 import java.util.HashSet;
33 import java.util.List;
35 import java.util.stream.Collectors;
36 import lombok.RequiredArgsConstructor;
37 import org.onap.cps.ncmp.api.impl.inventory.enums.PropertyType;
38 import org.onap.cps.ncmp.api.impl.trustlevel.TrustLevel;
39 import org.onap.cps.ncmp.api.impl.trustlevel.TrustLevelFilter;
40 import org.onap.cps.spi.CpsDataPersistenceService;
41 import org.onap.cps.spi.FetchDescendantsOption;
42 import org.onap.cps.spi.model.DataNode;
43 import org.springframework.stereotype.Component;
45 @RequiredArgsConstructor
47 public class CmHandleQueriesImpl implements CmHandleQueries {
49 private static final String DESCENDANT_PATH = "//";
50 private static final String ANCESTOR_CM_HANDLES = "/ancestor::cm-handles";
51 private final CpsDataPersistenceService cpsDataPersistenceService;
52 private final Map<String, TrustLevel> trustLevelPerCmHandle;
55 public Collection<String> queryCmHandleAdditionalProperties(final Map<String, String> privatePropertyQueryPairs) {
56 return queryCmHandleAnyProperties(privatePropertyQueryPairs, PropertyType.ADDITIONAL);
60 public Collection<String> queryCmHandlePublicProperties(final Map<String, String> publicPropertyQueryPairs) {
61 return queryCmHandleAnyProperties(publicPropertyQueryPairs, PropertyType.PUBLIC);
65 public Collection<String> queryCmHandlesByTrustLevel(final Map<String, String> trustLevelPropertyQueryPairs) {
66 final String trustLevelProperty = trustLevelPropertyQueryPairs.values().iterator().next();
67 final TrustLevel targetTrustLevel = TrustLevel.valueOf(trustLevelProperty);
69 final TrustLevelFilter trustLevelFilter = new TrustLevelFilter(targetTrustLevel, trustLevelPerCmHandle);
70 return trustLevelFilter.getAllCmHandleIdsByTargetTrustLevel();
74 public List<DataNode> queryCmHandlesByState(final CmHandleState cmHandleState) {
75 return queryCmHandleDataNodesByCpsPath("//state[@cm-handle-state=\"" + cmHandleState + "\"]",
76 INCLUDE_ALL_DESCENDANTS);
80 public List<DataNode> queryCmHandleDataNodesByCpsPath(final String cpsPath,
81 final FetchDescendantsOption fetchDescendantsOption) {
82 return cpsDataPersistenceService.queryDataNodes(NCMP_DATASPACE_NAME, NCMP_DMI_REGISTRY_ANCHOR,
83 cpsPath + ANCESTOR_CM_HANDLES, fetchDescendantsOption);
87 public boolean cmHandleHasState(final String cmHandleId, final CmHandleState requiredCmHandleState) {
88 final DataNode stateDataNode = getCmHandleState(cmHandleId);
89 final String cmHandleStateAsString = (String) stateDataNode.getLeaves().get("cm-handle-state");
90 return CmHandleState.valueOf(cmHandleStateAsString).equals(requiredCmHandleState);
94 public List<DataNode> queryCmHandlesByOperationalSyncState(final DataStoreSyncState dataStoreSyncState) {
95 return queryCmHandleDataNodesByCpsPath("//state/datastores" + "/operational[@sync-state=\""
96 + dataStoreSyncState + "\"]", FetchDescendantsOption.OMIT_DESCENDANTS);
100 public Collection<String> getCmHandleIdsByDmiPluginIdentifier(final String dmiPluginIdentifier) {
101 final Collection<String> cmHandleIds = new HashSet<>();
102 for (final ModelledDmiServiceLeaves modelledDmiServiceLeaf : ModelledDmiServiceLeaves.values()) {
103 for (final DataNode cmHandleAsDataNode: getCmHandlesByDmiPluginIdentifierAndDmiProperty(
105 modelledDmiServiceLeaf.getLeafName())) {
106 cmHandleIds.add(cmHandleAsDataNode.getLeaves().get("id").toString());
112 private Collection<String> collectCmHandleIdsFromDataNodes(final Collection<DataNode> dataNodes) {
113 return dataNodes.stream().map(dataNode -> (String) dataNode.getLeaves().get("id")).collect(Collectors.toSet());
116 private Collection<String> queryCmHandleAnyProperties(
117 final Map<String, String> propertyQueryPairs,
118 final PropertyType propertyType) {
119 if (propertyQueryPairs.isEmpty()) {
120 return Collections.emptySet();
122 Collection<String> cmHandleIds = null;
123 for (final Map.Entry<String, String> publicPropertyQueryPair : propertyQueryPairs.entrySet()) {
124 final String cpsPath = DESCENDANT_PATH + propertyType.getYangContainerName() + "[@name=\""
125 + publicPropertyQueryPair.getKey()
126 + "\" and @value=\"" + publicPropertyQueryPair.getValue() + "\"]";
128 final Collection<DataNode> dataNodes = queryCmHandleDataNodesByCpsPath(cpsPath, OMIT_DESCENDANTS);
129 if (cmHandleIds == null) {
130 cmHandleIds = collectCmHandleIdsFromDataNodes(dataNodes);
132 final Collection<String> cmHandleIdsToRetain = collectCmHandleIdsFromDataNodes(dataNodes);
133 cmHandleIds.retainAll(cmHandleIdsToRetain);
135 if (cmHandleIds.isEmpty()) {
142 private List<DataNode> getCmHandlesByDmiPluginIdentifierAndDmiProperty(final String dmiPluginIdentifier,
143 final String dmiProperty) {
144 return cpsDataPersistenceService.queryDataNodes(NCMP_DATASPACE_NAME, NCMP_DMI_REGISTRY_ANCHOR,
145 NCMP_DMI_REGISTRY_PARENT + "/cm-handles[@" + dmiProperty + "='" + dmiPluginIdentifier + "']",
149 private DataNode getCmHandleState(final String cmHandleId) {
150 final String xpath = NCMP_DMI_REGISTRY_PARENT + "/cm-handles[@id='" + cmHandleId + "']/state";
151 return cpsDataPersistenceService.getDataNodes(NCMP_DATASPACE_NAME, NCMP_DMI_REGISTRY_ANCHOR,
152 xpath, OMIT_DESCENDANTS).iterator().next();