cps-ri/src/main/java/org/onap/cps/spi/impl/CpsAdminPersistenceServiceImpl.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2020-2022 Nordix Foundation.
   4  * Modifications Copyright (C) 2020-2022 Bell Canada.
   5  * Modifications Copyright (C) 2021 Pantheon.tech
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  *
  19  * SPDX-License-Identifier: Apache-2.0
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.cps.spi.impl;
  24
  25 import java.util.Collection;
  26 import java.util.Collections;
  27 import java.util.List;
  28 import java.util.stream.Collectors;
  29 import javax.transaction.Transactional;
  30 import lombok.RequiredArgsConstructor;
  31 import lombok.extern.slf4j.Slf4j;
  32 import org.onap.cps.spi.CpsAdminPersistenceService;
  33 import org.onap.cps.spi.entities.AnchorEntity;
  34 import org.onap.cps.spi.entities.DataspaceEntity;
  35 import org.onap.cps.spi.entities.SchemaSetEntity;
  36 import org.onap.cps.spi.entities.YangResourceModuleReference;
  37 import org.onap.cps.spi.exceptions.AlreadyDefinedException;
  38 import org.onap.cps.spi.exceptions.DataspaceInUseException;
  39 import org.onap.cps.spi.exceptions.DataspaceNotFoundException;
  40 import org.onap.cps.spi.exceptions.ModuleNamesNotFoundException;
  41 import org.onap.cps.spi.model.Anchor;
  42 import org.onap.cps.spi.repository.AnchorRepository;
  43 import org.onap.cps.spi.repository.DataspaceRepository;
  44 import org.onap.cps.spi.repository.SchemaSetRepository;
  45 import org.onap.cps.spi.repository.YangResourceRepository;
  46 import org.springframework.dao.DataIntegrityViolationException;
  47 import org.springframework.stereotype.Component;
  48
  49 @Slf4j
  50 @Component
  51 @RequiredArgsConstructor
  52 public class CpsAdminPersistenceServiceImpl implements CpsAdminPersistenceService {
  53
  54     private final DataspaceRepository dataspaceRepository;
  55     private final AnchorRepository anchorRepository;
  56     private final SchemaSetRepository schemaSetRepository;
  57     private final YangResourceRepository yangResourceRepository;
  58
  59     @Override
  60     public void createDataspace(final String dataspaceName) {
  61         try {
  62             dataspaceRepository.save(new DataspaceEntity(dataspaceName));
  63         } catch (final DataIntegrityViolationException e) {
  64             throw AlreadyDefinedException.forDataspace(dataspaceName, e);
  65         }
  66     }
  67
  68     @Override
  69     public void deleteDataspace(final String dataspaceName) {
  70         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  71         final int numberOfAssociatedAnchors = anchorRepository.countByDataspace(dataspaceEntity);
  72         if (numberOfAssociatedAnchors != 0) {
  73             throw new DataspaceInUseException(dataspaceName,
  74                 String.format("Dataspace contains %d anchor(s)", numberOfAssociatedAnchors));
  75         }
  76         final int numberOfAssociatedSchemaSets = schemaSetRepository.countByDataspace(dataspaceEntity);
  77         if (numberOfAssociatedSchemaSets != 0) {
  78             throw new DataspaceInUseException(dataspaceName,
  79                 String.format("Dataspace contains %d schemaset(s)", numberOfAssociatedSchemaSets));
  80         }
  81         dataspaceRepository.delete(dataspaceEntity);
  82     }
  83
  84     @Override
  85     public void createAnchor(final String dataspaceName, final String schemaSetName, final String anchorName) {
  86         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  87         final var schemaSetEntity =
  88             schemaSetRepository.getByDataspaceAndName(dataspaceEntity, schemaSetName);
  89         final var anchorEntity = AnchorEntity.builder()
  90             .name(anchorName)
  91             .dataspace(dataspaceEntity)
  92             .schemaSet(schemaSetEntity)
  93             .build();
  94         try {
  95             anchorRepository.save(anchorEntity);
  96         } catch (final DataIntegrityViolationException e) {
  97             throw AlreadyDefinedException.forAnchor(anchorName, dataspaceName, e);
  98         }
  99     }
 100
 101     @Override
 102     public Collection<Anchor> getAnchors(final String dataspaceName) {
 103         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 104         final Collection<AnchorEntity> anchorEntities = anchorRepository.findAllByDataspace(dataspaceEntity);
 105         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toSet());
 106     }
 107
 108     @Override
 109     public Collection<Anchor> getAnchors(final String dataspaceName, final String schemaSetName) {
 110         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 111         final SchemaSetEntity schemaSetEntity = schemaSetRepository.getByDataspaceAndName(
 112             dataspaceEntity, schemaSetName);
 113         return anchorRepository.findAllBySchemaSet(schemaSetEntity)
 114             .stream().map(CpsAdminPersistenceServiceImpl::toAnchor)
 115             .collect(Collectors.toSet());
 116     }
 117
 118     @Override
 119     public Collection<Anchor> queryAnchors(final String dataspaceName, final Collection<String> inputModuleNames) {
 120         try {
 121             validateDataspaceAndModuleNames(dataspaceName, inputModuleNames);
 122         } catch (DataspaceNotFoundException | ModuleNamesNotFoundException  e) {
 123             log.info("Module search encountered unknown dataspace or modulename, treating this as nothing found");
 124             return Collections.emptySet();
 125         }
 126
 127         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 128         final Collection<AnchorEntity> anchorEntities = anchorRepository
 129             .getAnchorsByDataspaceIdAndModuleNames(dataspaceEntity.getId(), inputModuleNames, inputModuleNames.size());
 130         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toSet());
 131     }
 132
 133     @Override
 134     public Anchor getAnchor(final String dataspaceName, final String anchorName) {
 135         return toAnchor(getAnchorEntity(dataspaceName, anchorName));
 136     }
 137
 138     @Transactional
 139     @Override
 140     public void deleteAnchor(final String dataspaceName, final String anchorName) {
 141         final var anchorEntity = getAnchorEntity(dataspaceName, anchorName);
 142         anchorRepository.delete(anchorEntity);
 143     }
 144
 145     private AnchorEntity getAnchorEntity(final String dataspaceName, final String anchorName) {
 146         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 147         return anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
 148     }
 149
 150     private static Anchor toAnchor(final AnchorEntity anchorEntity) {
 151         return Anchor.builder()
 152             .name(anchorEntity.getName())
 153             .dataspaceName(anchorEntity.getDataspace().getName())
 154             .schemaSetName(anchorEntity.getSchemaSet().getName())
 155             .build();
 156     }
 157
 158     private void validateDataspaceAndModuleNames(final String dataspaceName,
 159         final Collection<String> inputModuleNames) {
 160         final Collection<String> retrievedModuleReferences =
 161             yangResourceRepository.findAllModuleReferencesByDataspaceAndModuleNames(dataspaceName, inputModuleNames)
 162                 .stream().map(YangResourceModuleReference::getModuleName)
 163                 .collect(Collectors.toList());
 164         if (retrievedModuleReferences.isEmpty()) {
 165             verifyDataspaceName(dataspaceName);
 166         }
 167         if (inputModuleNames.size() > retrievedModuleReferences.size()) {
 168             final List<String> moduleNamesNotFound = inputModuleNames.stream()
 169                 .filter(moduleName -> !retrievedModuleReferences.contains(moduleName))
 170                 .collect(Collectors.toList());
 171             if (!moduleNamesNotFound.isEmpty()) {
 172                 throw new ModuleNamesNotFoundException(dataspaceName, moduleNamesNotFound);
 173             }
 174         }
 175     }
 176
 177     private void verifyDataspaceName(final String dataspaceName) {
 178         dataspaceRepository.getByName(dataspaceName);
 179     }
 180 }