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

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2020 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.List;
  27 import java.util.stream.Collectors;
  28 import javax.transaction.Transactional;
  29 import lombok.AllArgsConstructor;
  30 import org.onap.cps.spi.CpsAdminPersistenceService;
  31 import org.onap.cps.spi.entities.AnchorEntity;
  32 import org.onap.cps.spi.entities.DataspaceEntity;
  33 import org.onap.cps.spi.entities.SchemaSetEntity;
  34 import org.onap.cps.spi.entities.YangResourceModuleReference;
  35 import org.onap.cps.spi.exceptions.AlreadyDefinedException;
  36 import org.onap.cps.spi.exceptions.DataspaceInUseException;
  37 import org.onap.cps.spi.exceptions.ModuleNamesNotFoundException;
  38 import org.onap.cps.spi.model.Anchor;
  39 import org.onap.cps.spi.repository.AnchorRepository;
  40 import org.onap.cps.spi.repository.DataspaceRepository;
  41 import org.onap.cps.spi.repository.SchemaSetRepository;
  42 import org.onap.cps.spi.repository.YangResourceRepository;
  43 import org.springframework.dao.DataIntegrityViolationException;
  44 import org.springframework.stereotype.Component;
  45
  46 @Component
  47 @AllArgsConstructor
  48 public class CpsAdminPersistenceServiceImpl implements CpsAdminPersistenceService {
  49
  50     private final DataspaceRepository dataspaceRepository;
  51     private final AnchorRepository anchorRepository;
  52     private final SchemaSetRepository schemaSetRepository;
  53     private final YangResourceRepository yangResourceRepository;
  54
  55     @Override
  56     public void createDataspace(final String dataspaceName) {
  57         try {
  58             dataspaceRepository.save(new DataspaceEntity(dataspaceName));
  59         } catch (final DataIntegrityViolationException e) {
  60             throw AlreadyDefinedException.forDataspace(dataspaceName, e);
  61         }
  62     }
  63
  64     @Override
  65     public void deleteDataspace(final String dataspaceName) {
  66         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  67         final int numberOfAssociatedAnchors = anchorRepository.countByDataspace(dataspaceEntity);
  68         if (numberOfAssociatedAnchors != 0) {
  69             throw new DataspaceInUseException(dataspaceName,
  70                 String.format("Dataspace contains %d anchor(s)", numberOfAssociatedAnchors));
  71         }
  72         final int numberOfAssociatedSchemaSets = schemaSetRepository.countByDataspace(dataspaceEntity);
  73         if (numberOfAssociatedSchemaSets != 0) {
  74             throw new DataspaceInUseException(dataspaceName,
  75                 String.format("Dataspace contains %d schemaset(s)", numberOfAssociatedSchemaSets));
  76         }
  77         dataspaceRepository.delete(dataspaceEntity);
  78     }
  79
  80     @Override
  81     public void createAnchor(final String dataspaceName, final String schemaSetName, final String anchorName) {
  82         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  83         final var schemaSetEntity =
  84             schemaSetRepository.getByDataspaceAndName(dataspaceEntity, schemaSetName);
  85         final var anchorEntity = AnchorEntity.builder()
  86             .name(anchorName)
  87             .dataspace(dataspaceEntity)
  88             .schemaSet(schemaSetEntity)
  89             .build();
  90         try {
  91             anchorRepository.save(anchorEntity);
  92         } catch (final DataIntegrityViolationException e) {
  93             throw AlreadyDefinedException.forAnchor(anchorName, dataspaceName, e);
  94         }
  95     }
  96
  97     @Override
  98     public Collection<Anchor> getAnchors(final String dataspaceName) {
  99         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 100         final Collection<AnchorEntity> anchorEntities = anchorRepository.findAllByDataspace(dataspaceEntity);
 101         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toSet());
 102     }
 103
 104     @Override
 105     public Collection<Anchor> getAnchors(final String dataspaceName, final String schemaSetName) {
 106         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 107         final SchemaSetEntity schemaSetEntity = schemaSetRepository.getByDataspaceAndName(
 108             dataspaceEntity, schemaSetName);
 109         return anchorRepository.findAllBySchemaSet(schemaSetEntity)
 110             .stream().map(CpsAdminPersistenceServiceImpl::toAnchor)
 111             .collect(Collectors.toSet());
 112     }
 113
 114     @Override
 115     public Collection<Anchor> queryAnchors(final String dataspaceName, final Collection<String> inputModuleNames) {
 116         validateDataspaceAndModuleNames(dataspaceName, inputModuleNames);
 117         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 118         final Collection<AnchorEntity> anchorEntities = anchorRepository
 119             .getAnchorsByDataspaceIdAndModuleNames(dataspaceEntity.getId(), inputModuleNames, inputModuleNames.size());
 120         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toSet());
 121     }
 122
 123     @Override
 124     public Anchor getAnchor(final String dataspaceName, final String anchorName) {
 125         return toAnchor(getAnchorEntity(dataspaceName, anchorName));
 126     }
 127
 128     @Transactional
 129     @Override
 130     public void deleteAnchor(final String dataspaceName, final String anchorName) {
 131         final var anchorEntity = getAnchorEntity(dataspaceName, anchorName);
 132         anchorRepository.delete(anchorEntity);
 133     }
 134
 135     private AnchorEntity getAnchorEntity(final String dataspaceName, final String anchorName) {
 136         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 137         return anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
 138     }
 139
 140     private static Anchor toAnchor(final AnchorEntity anchorEntity) {
 141         return Anchor.builder()
 142             .name(anchorEntity.getName())
 143             .dataspaceName(anchorEntity.getDataspace().getName())
 144             .schemaSetName(anchorEntity.getSchemaSet().getName())
 145             .build();
 146     }
 147
 148     private void validateDataspaceAndModuleNames(final String dataspaceName,
 149         final Collection<String> inputModuleNames) {
 150         final Collection<String> retrievedModuleNames =
 151             yangResourceRepository.findAllModuleReferences(dataspaceName, inputModuleNames)
 152                 .stream().map(YangResourceModuleReference::getModuleName)
 153                 .collect(Collectors.toList());
 154         if (retrievedModuleNames.isEmpty()) {
 155             dataspaceRepository.getByName(dataspaceName);
 156         }
 157         if (inputModuleNames.size() > retrievedModuleNames.size()) {
 158             final List<String> moduleNamesNotFound = inputModuleNames.stream()
 159                 .filter(moduleName -> !retrievedModuleNames.contains(moduleName))
 160                 .collect(Collectors.toList());
 161             if (!moduleNamesNotFound.isEmpty()) {
 162                 throw new ModuleNamesNotFoundException(dataspaceName, moduleNamesNotFound);
 163             }
 164         }
 165     }
 166 }