@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 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;
}
