local-setup/src/main/java/onap/aai/dto/ModelGenerator.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2018-2019 Nokia Intellectual Property. All rights reserved.
   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  * ============LICENSE_END=========================================================
  19  */
  20 package onap.aai.dto;
  21
  22 import static onap.aai.util.Resources.readerFrom;
  23
  24 import java.io.IOException;
  25 import java.io.UncheckedIOException;
  26 import java.util.stream.Stream;
  27 import org.apache.commons.csv.CSVFormat;
  28 import org.apache.commons.csv.CSVParser;
  29 import org.apache.commons.csv.CSVRecord;
  30
  31 public class ModelGenerator {
  32
  33     public static Stream<Model> generate(String fileName) {
  34         try (CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(readerFrom(fileName))) {
  35             return parser.getRecords().stream().map(ModelGenerator::csvToModel);
  36         } catch (IOException e) {
  37             throw new UncheckedIOException(e);
  38         }
  39     }
  40
  41     private static Model csvToModel(CSVRecord csvRecord) {
  42         return new Model(
  43             csvRecord.get("model-name"),
  44             csvRecord.get("model-invariant-id"),
  45             csvRecord.get("model-version-id")
  46         );
  47     }
  48 }