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