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