Визуализация базы данных Neo4j в UbiGraph

MATCH (n)-->(m)
RETURN id(n) as start_id, id(m) as end_id

MATCH (n)
OPTIONAL MATCH (n)-[r]->(m)
RETURN
 id(n) as start_id, head(labels(n)) as start_label, n as start_properties,
 id(m) as end_id, head(labels(m)) as end_label, m as end_properties,
 id(r) as rel_id, type(r) as rel_type, r as rel_properties
 LIMIT 1000

public static final String ALL_NODES_QUERY =
        "MATCH (n) " +
                "OPTIONAL MATCH (n)-[r]->(m) " +
                "RETURN " +
                " id(n) as start_id, labels(n) as start_labels, n as start_properties, " +
                " id(m) as end_id, labels(m) as end_labels, m as end_properties, " +
                " id(r) as rel_id, type(r) as rel_type, r as rel_properties " +
                "LIMIT {1}";

private static void renderNeo4jGraph(UbigraphClient graph, String url, int limit) throws SQLException {
    Connection connection = DriverManager.getConnection(url);
    try (PreparedStatement stmt = connection.prepareStatement(ALL_NODES_QUERY)) {
        stmt.setInt(1, limit);
        try (ResultSet rs = stmt.executeQuery()) {
            while (rs.next()) {
                createUbiVertex(rs, "start", graph, false);
                createUbiVertex(rs, "end", graph, false);
                createUbiEdge(rs, "start_id", "end_id", "rel", graph);
            }
        }
    }
}

private static int createUbiVertex(ResultSet rs, String prefix, UbigraphClient graph, boolean renderProps) throws SQLException {
    int id = rs.getInt(prefix + "_id");
    String label = label((Collection<String>) rs.getObject(prefix + "_labels"));
    Map props = renderProps ? (Map) rs.getObject(prefix + "_properties") : null;

    int success = graph.newVertex(id);
    if (success==-1) return success;
    if (label!=null) graph.changeVertexStyle(id, style(graph, label));
    if (props != null && props.containsKey("name") && asList("Player", "Country", "WorldCup").contains(label)) {
        graph.setVertexAttribute(id, "label", (String) props.get("name"));
    }
    return success;
}

private static int style(UbigraphClient graph, String label) {
    if (!styles.containsKey(label)) {
        Colors colorByLabel = Colors.values()[styles.size() % Colors.values().length];
        int styleId = createVertexStyle(graph, Shape.cube, colorByLabel);
        styles.put(label, styleId);
        return styleId;
    } else {
        return styles.get(label);
    }
}

private static int createVertexStyle(UbigraphClient graph, Shape shape, Colors color) {
    int style = nextStyleId();
    graph.newVertexStyle(style, 0);
    graph.setVertexStyleAttribute(style, "shape", shape.name());
    graph.setVertexStyleAttribute(style, "size", "0.5");
    graph.setVertexStyleAttribute(style, "color", color.toHexValue());
    System.out.println(nextStyleId() + " style = " + style + " color " + color + " shape " + shape);
    return style;
}

enum Shape {
    sphere, cone, cube, torus, dodecahedron, icosahedron, octahedron, tetrahedron, none;
}

enum Colors {
    red(Color.RED), green(Color.GREEN), blue(Color.BLUE), orange(Color.ORANGE), yellow(Color.YELLOW), white(Color.WHITE);
    private final Color color;

    Colors(Color color) { this.color = color; }
    public String toHexValue() {
        return "#" + Integer.toHexString(color.getRGB() & 0x00FFFFFF);
    }
}