Examples with Relationship com.buschmais.jqassistant.core.report.api.graph.model.Relationship used on opensource projects

Example 6 with Relationship

use of com.buschmais.jqassistant.core.report.api.graph.model.Relationship in project jqa-core-framework by buschmais.

the class SubGraphFactory method toIdentifiable.

/**
 * Convert a given value to an {@link Identifiable} element of a
 * {@link SubGraph}.
 *
 * @param value
 *            The value.
 * @return The {@link Identifiable}.
 * @throws ReportException
 *             If value that cannot be converted to an {@link Identifiable}.
 */
public <I extends Identifiable> I toIdentifiable(Object value) throws ReportException {
    if (value == null) {
        return null;
    } else if (value instanceof Map) {
        Map<String, Object> virtualObject = (Map) value;
        Object role = virtualObject.get(ROLE);
        if (role != null) {
            Map<String, Object> properties = (Map<String, Object>) virtualObject.get(PROPERTIES);
            switch(role.toString().toLowerCase()) {
                case NODE:
                    Node node = new Node();
                    node.setId(nodeId--);
                    Collection<String> labels = (Collection<String>) virtualObject.get(LABELS);
                    node.getLabels().addAll(labels);
                    node.getProperties().putAll(properties);
                    node.setLabel((String) virtualObject.get(LABEL));
                    return (I) node;
                case RELATIONSHIP:
                    Relationship relationship = new Relationship();
                    relationship.setId(relationshipId--);
                    Node startNode = toIdentifiable(virtualObject.get(START_NODE));
                    Node endNode = toIdentifiable(virtualObject.get(END_NODE));
                    String type = (String) virtualObject.get(TYPE);
                    relationship.setType(type);
                    relationship.setStartNode(startNode);
                    relationship.setEndNode(endNode);
                    relationship.getProperties().putAll(properties);
                    relationship.setLabel((String) virtualObject.get(LABEL));
                    if (startNode == null || endNode == null || type == null) {
                        throw new ReportException("The virtual relationship does not contain either start node, end node or type: " + relationship);
                    }
                    return (I) relationship;
                case GRAPH:
                    SubGraph subgraph = new SubGraph();
                    subgraph.setId(subgraphId--);
                    Node parent = toIdentifiable(virtualObject.get(PARENT));
                    subgraph.setParent(parent);
                    subgraph.setLabel((String) virtualObject.get(LABEL));
                    addToValueGraph(subgraph, virtualObject.get(NODES));
                    addToValueGraph(subgraph, virtualObject.get(RELATIONSHIPS));
                    return (I) subgraph;
            }
        }
    } else if (value instanceof CompositeObject) {
        CompositeObject compositeObject = (CompositeObject) value;
        Identifiable identifiable = toIdentifiable(compositeObject.getDelegate());
        identifiable.setLabel(ReportHelper.getLabel(value));
        return (I) identifiable;
    } else if (value instanceof Neo4jNode) {
        Neo4jNode<Neo4jLabel, ?, ?, ?> neo4jNode = (Neo4jNode<Neo4jLabel, ?, ?, ?>) value;
        Node node = new Node();
        node.setId(neo4jNode.getId());
        for (Neo4jLabel label : neo4jNode.getLabels()) {
            node.getLabels().add(label.getName());
        }
        node.getProperties().putAll(neo4jNode.getProperties());
        return (I) node;
    } else if (value instanceof Neo4jRelationship) {
        Neo4jRelationship<?, ?, ?, ?, ?> neo4jRelationship = (Neo4jRelationship) value;
        Relationship relationship = new Relationship();
        relationship.setId(neo4jRelationship.getId());
        relationship.setType(neo4jRelationship.getType().getName());
        relationship.setStartNode(toIdentifiable(neo4jRelationship.getStartNode()));
        relationship.setEndNode(toIdentifiable(neo4jRelationship.getEndNode()));
        relationship.getProperties().putAll(neo4jRelationship.getProperties());
        return (I) relationship;
    }
    throw new ReportException("Element type not supported: " + value);
}