asdc-tests/src/main/java/org/openecomp/sdc/ci/tests/utils/graph/GraphFileUtils.java

   1 package org.openecomp.sdc.ci.tests.utils.graph;
   2
   3 import com.thinkaurelius.titan.core.TitanVertex;
   4 import org.apache.commons.io.FileUtils;
   5 import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
   6
   7 import java.io.IOException;
   8 import java.nio.charset.Charset;
   9 import java.nio.file.Files;
  10 import java.nio.file.Path;
  11 import java.nio.file.Paths;
  12 import java.util.ArrayList;
  13 import java.util.List;
  14
  15 public class GraphFileUtils {
  16
  17     public static final String TEMP_FILES_PATH = "src/main/resources/ci/tempFiles/%s.txt";
  18
  19     public static void writeVerticesUIDToFile(String fileName, Iterable<TitanVertex> vertices) throws IOException {
  20         Path path = Paths.get(String.format(TEMP_FILES_PATH, fileName));
  21         Files.deleteIfExists(path);
  22         Path file = Files.createFile(path);
  23         final String newLine = System.getProperty("line.separator");
  24         for (TitanVertex vertex : vertices) {
  25             FileUtils.writeStringToFile(file.toFile(), String.valueOf(vertex.id()) + newLine, Charset.defaultCharset(), true);
  26         }
  27     }
  28
  29     public static List<String> getVerticesIdsFromFile(String fileName) throws IOException {
  30         List<String> verticesUids = new ArrayList<>();
  31         Files.lines(Paths.get(String.format(TEMP_FILES_PATH, fileName))).forEach(verticesUids::add);
  32         return verticesUids;
  33     }
  34
  35     private static String getUid(TitanVertex titanVertex) {
  36         return (String )titanVertex.value(GraphPropertiesDictionary.UNIQUE_ID.getProperty());
  37     }
  38
  39 }