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 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.Set;
  28 import java.util.stream.Collectors;
  29 import javax.transaction.Transactional;
  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.exceptions.AlreadyDefinedException;
  34 import org.onap.cps.spi.exceptions.ModuleNamesNotFoundException;
  35 import org.onap.cps.spi.model.Anchor;
  36 import org.onap.cps.spi.repository.AnchorRepository;
  37 import org.onap.cps.spi.repository.DataspaceRepository;
  38 import org.onap.cps.spi.repository.FragmentRepository;
  39 import org.onap.cps.spi.repository.SchemaSetRepository;
  40 import org.onap.cps.spi.repository.YangResourceRepository;
  41 import org.springframework.beans.factory.annotation.Autowired;
  42 import org.springframework.dao.DataIntegrityViolationException;
  43 import org.springframework.stereotype.Component;
  44
  45 @Component
  46 public class CpsAdminPersistenceServiceImpl implements CpsAdminPersistenceService {
  47
  48     @Autowired
  49     private DataspaceRepository dataspaceRepository;
  50
  51     @Autowired
  52     private AnchorRepository anchorRepository;
  53
  54     @Autowired
  55     private SchemaSetRepository schemaSetRepository;
  56
  57     @Autowired
  58     private FragmentRepository fragmentRepository;
  59
  60     @Autowired
  61     private YangResourceRepository yangResourceRepository;
  62
  63     @Override
  64     public void createDataspace(final String dataspaceName) {
  65         try {
  66             dataspaceRepository.save(new DataspaceEntity(dataspaceName));
  67         } catch (final DataIntegrityViolationException e) {
  68             throw AlreadyDefinedException.forDataspace(dataspaceName, e);
  69         }
  70     }
  71
  72     @Override
  73     public void createAnchor(final String dataspaceName, final String schemaSetName, final String anchorName) {
  74         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  75         final var schemaSetEntity =
  76             schemaSetRepository.getByDataspaceAndName(dataspaceEntity, schemaSetName);
  77         final var anchorEntity = AnchorEntity.builder()
  78             .name(anchorName)
  79             .dataspace(dataspaceEntity)
  80             .schemaSet(schemaSetEntity)
  81             .build();
  82         try {
  83             anchorRepository.save(anchorEntity);
  84         } catch (final DataIntegrityViolationException e) {
  85             throw AlreadyDefinedException.forAnchor(anchorName, dataspaceName, e);
  86         }
  87     }
  88
  89     @Override
  90     public Collection<Anchor> getAnchors(final String dataspaceName) {
  91         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
  92         final Collection<AnchorEntity> anchorEntities = anchorRepository.findAllByDataspace(dataspaceEntity);
  93         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toList());
  94     }
  95
  96     @Override
  97     public Collection<Anchor> queryAnchors(final String dataspaceName, final Collection<String> inputModuleNames) {
  98         validateDataspaceAndModuleNames(dataspaceName, inputModuleNames);
  99         final DataspaceEntity dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 100         final Collection<AnchorEntity> anchorEntities = anchorRepository
 101             .getAnchorsByDataspaceIdAndModuleNames(dataspaceEntity.getId(), inputModuleNames, inputModuleNames.size());
 102         return anchorEntities.stream().map(CpsAdminPersistenceServiceImpl::toAnchor).collect(Collectors.toSet());
 103     }
 104
 105     @Override
 106     public Anchor getAnchor(final String dataspaceName, final String anchorName) {
 107         return toAnchor(getAnchorEntity(dataspaceName, anchorName));
 108     }
 109
 110     @Transactional
 111     @Override
 112     public void deleteAnchor(final String dataspaceName, final String anchorName) {
 113         final var anchorEntity = getAnchorEntity(dataspaceName, anchorName);
 114         fragmentRepository.deleteByAnchorIn(Set.of(anchorEntity));
 115         anchorRepository.delete(anchorEntity);
 116     }
 117
 118     private AnchorEntity getAnchorEntity(final String dataspaceName, final String anchorName) {
 119         final var dataspaceEntity = dataspaceRepository.getByName(dataspaceName);
 120         return anchorRepository.getByDataspaceAndName(dataspaceEntity, anchorName);
 121     }
 122
 123     private static Anchor toAnchor(final AnchorEntity anchorEntity) {
 124         return Anchor.builder()
 125             .name(anchorEntity.getName())
 126             .dataspaceName(anchorEntity.getDataspace().getName())
 127             .schemaSetName(anchorEntity.getSchemaSet().getName())
 128             .build();
 129     }
 130
 131     private void validateDataspaceAndModuleNames(final String dataspaceName,
 132         final Collection<String> inputModuleNames) {
 133         final Collection<String> retrievedModuleNames =
 134             yangResourceRepository.findAllModuleReferences(dataspaceName, inputModuleNames)
 135                 .stream().map(module -> module.getModuleName())
 136                 .collect(Collectors.toList());
 137         if (retrievedModuleNames.isEmpty()) {
 138             dataspaceRepository.getByName(dataspaceName);
 139         }
 140         if (inputModuleNames.size() > retrievedModuleNames.size()) {
 141             final List<String> moduleNamesNotFound = inputModuleNames.stream()
 142                 .filter(moduleName -> !retrievedModuleNames.contains(moduleName))
 143                 .collect(Collectors.toList());
 144             if (!moduleNamesNotFound.isEmpty()) {
 145                 throw new ModuleNamesNotFoundException(dataspaceName, moduleNamesNotFound);
 146             }
 147         }
 148     }
 149 }