@Override
public void serialize(
final TinkerGraph graph,
final JsonGenerator jsonGenerator,
final SerializerProvider serializerProvider)
throws IOException {
jsonGenerator.writeStartObject();
jsonGenerator.writeFieldName(GraphSONTokens.VERTICES);
jsonGenerator.writeStartArray();
final Iterator<Vertex> vertices = graph.vertices();
while (vertices.hasNext()) {
serializerProvider.defaultSerializeValue(vertices.next(), jsonGenerator);
}
jsonGenerator.writeEndArray();
jsonGenerator.writeFieldName(GraphSONTokens.EDGES);
jsonGenerator.writeStartArray();
final Iterator<Edge> edges = graph.edges();
while (edges.hasNext()) {
serializerProvider.defaultSerializeValue(edges.next(), jsonGenerator);
}
jsonGenerator.writeEndArray();
jsonGenerator.writeEndObject();
}
public TestCreateAndSaveGraph() {
// Create graph
graph = new MGraph();
ingestPairs(10);
// ingestPairsWithCalculatedIDs(10);
// Create index:
((TinkerGraph) graph).createKeyIndex("firstname", Vertex.class);
((TinkerGraph) graph).createKeyIndex("lastname", Vertex.class);
System.out.println(((TinkerGraph) graph).getIndexedKeys(Vertex.class) + "\n");
// Save graph
GraphUtils.saveGraph(graph, GRAPH_FILE);
}
@Override
public TinkerGraph deserialize(
JsonParser jsonParser, DeserializationContext deserializationContext)
throws IOException, JsonProcessingException {
final TinkerGraph graph = TinkerGraph.open();
final List<Map<String, Object>> edges;
final List<Map<String, Object>> vertices;
if (!jsonParser.getCurrentToken().isStructStart()) {
if (!jsonParser.getCurrentName().equals(GraphSONTokens.VERTICES))
throw new IOException(String.format("Expected a '%s' key", GraphSONTokens.VERTICES));
jsonParser.nextToken();
vertices =
(List<Map<String, Object>>)
deserializationContext.readValue(jsonParser, ArrayList.class);
jsonParser.nextToken();
if (!jsonParser.getCurrentName().equals(GraphSONTokens.EDGES))
throw new IOException(String.format("Expected a '%s' key", GraphSONTokens.EDGES));
jsonParser.nextToken();
edges =
(List<Map<String, Object>>)
deserializationContext.readValue(jsonParser, ArrayList.class);
} else {
final Map<String, Object> graphData =
deserializationContext.readValue(jsonParser, HashMap.class);
vertices = (List<Map<String, Object>>) graphData.get(GraphSONTokens.VERTICES);
edges = (List<Map<String, Object>>) graphData.get(GraphSONTokens.EDGES);
}
for (Map<String, Object> vertexData : vertices) {
final DetachedVertex detached =
new DetachedVertex(
vertexData.get(GraphSONTokens.ID),
vertexData.get(GraphSONTokens.LABEL).toString(),
(Map<String, Object>) vertexData.get(GraphSONTokens.PROPERTIES));
detached.attach(Attachable.Method.getOrCreate(graph));
}
for (Map<String, Object> edgeData : edges) {
final DetachedEdge detached =
new DetachedEdge(
edgeData.get(GraphSONTokens.ID),
edgeData.get(GraphSONTokens.LABEL).toString(),
(Map<String, Object>) edgeData.get(GraphSONTokens.PROPERTIES),
Pair.with(
edgeData.get(GraphSONTokens.OUT),
edgeData.get(GraphSONTokens.OUT_LABEL).toString()),
Pair.with(
edgeData.get(GraphSONTokens.IN),
edgeData.get(GraphSONTokens.IN_LABEL).toString()));
detached.attach(Attachable.Method.getOrCreate(graph));
}
return graph;
}
@Override
public void serializeWithType(
final TinkerGraph graph,
final JsonGenerator jsonGenerator,
final SerializerProvider serializerProvider,
final TypeSerializer typeSerializer)
throws IOException {
jsonGenerator.writeStartObject();
jsonGenerator.writeStringField(GraphSONTokens.CLASS, TinkerGraph.class.getName());
jsonGenerator.writeFieldName(GraphSONTokens.VERTICES);
jsonGenerator.writeStartArray();
jsonGenerator.writeString(ArrayList.class.getName());
jsonGenerator.writeStartArray();
final Iterator<Vertex> vertices = graph.vertices();
while (vertices.hasNext()) {
GraphSONUtil.writeWithType(
vertices.next(), jsonGenerator, serializerProvider, typeSerializer);
}
jsonGenerator.writeEndArray();
jsonGenerator.writeEndArray();
jsonGenerator.writeFieldName(GraphSONTokens.EDGES);
jsonGenerator.writeStartArray();
jsonGenerator.writeString(ArrayList.class.getName());
jsonGenerator.writeStartArray();
final Iterator<Edge> edges = graph.edges();
while (edges.hasNext()) {
GraphSONUtil.writeWithType(edges.next(), jsonGenerator, serializerProvider, typeSerializer);
}
jsonGenerator.writeEndArray();
jsonGenerator.writeEndArray();
jsonGenerator.writeEndObject();
}
@Override
public TinkerGraph read(
final Kryo kryo, final Input input, final Class<TinkerGraph> tinkerGraphClass) {
final TinkerGraph graph = TinkerGraph.open();
final int len = input.readInt();
final byte[] bytes = input.readBytes(len);
try (final ByteArrayInputStream stream = new ByteArrayInputStream(bytes)) {
GryoReader.build().mapper(() -> kryo).create().readGraph(stream, graph);
} catch (Exception io) {
throw new RuntimeException(io);
}
return graph;
}
