cps/cps-service/src/main/java/org/onap/cps/api/impl/CpServiceImpl.java

   1 /*
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2020 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  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License.
  15  *
  16  *  SPDX-License-Identifier: Apache-2.0
  17  *  ============LICENSE_END=========================================================
  18  */
  19
  20 package org.onap.cps.api.impl;
  21
  22
  23 import java.io.BufferedWriter;
  24 import java.io.File;
  25 import java.io.FileWriter;
  26 import java.io.IOException;
  27 import java.util.Iterator;
  28 import java.util.ServiceLoader;
  29 import org.onap.cps.api.CpService;
  30 import org.onap.cps.spi.ModelPersistencyService;
  31 import org.opendaylight.yangtools.yang.model.api.Module;
  32 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  33 import org.opendaylight.yangtools.yang.model.parser.api.YangParser;
  34 import org.opendaylight.yangtools.yang.model.parser.api.YangParserException;
  35 import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;
  36 import org.opendaylight.yangtools.yang.model.repo.api.StatementParserMode;
  37 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
  38 import org.slf4j.Logger;
  39 import org.slf4j.LoggerFactory;
  40 import org.springframework.beans.factory.annotation.Autowired;
  41 import org.springframework.stereotype.Component;
  42
  43 @Component
  44 public class CpServiceImpl implements CpService {
  45
  46     private static final Logger LOGGER = LoggerFactory.getLogger(CpServiceImpl.class);
  47
  48     private static final YangParserFactory PARSER_FACTORY;
  49
  50     static {
  51         final Iterator<YangParserFactory> it =
  52             ServiceLoader.load(YangParserFactory.class).iterator();
  53         if (!it.hasNext()) {
  54             throw new IllegalStateException("No YangParserFactory found");
  55         }
  56         PARSER_FACTORY = it.next();
  57     }
  58
  59     @Autowired
  60     private ModelPersistencyService modelPersistencyService;
  61
  62     @Override
  63     public SchemaContext parseAndValidateModel(final String yangModelContent)
  64         throws IOException, YangParserException {
  65         final File tempFile = File.createTempFile("yang", ".yang");
  66         try (BufferedWriter writer = new BufferedWriter(new FileWriter(tempFile))) {
  67             writer.write(yangModelContent);
  68         } catch (final IOException e) {
  69             LOGGER.error("Unable to write to temporary file {}", e.getMessage());
  70         }
  71         return parseAndValidateModel(tempFile);
  72     }
  73
  74     @Override
  75     public SchemaContext parseAndValidateModel(final File yangModelFile) throws IOException, YangParserException {
  76         final YangTextSchemaSource yangTextSchemaSource = YangTextSchemaSource.forFile(yangModelFile);
  77         final YangParser yangParser = PARSER_FACTORY.createParser(StatementParserMode.DEFAULT_MODE);
  78         yangParser.addSource(yangTextSchemaSource);
  79         return yangParser.buildEffectiveModel();
  80     }
  81
  82     @Override
  83     public void storeSchemaContext(final SchemaContext schemaContext) {
  84         for (final Module module : schemaContext.getModules()) {
  85             modelPersistencyService.storeModule(module.getName(), module.toString(),
  86                 module.getRevision().toString());
  87         }
  88     }
  89 }