Search in sources :

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);
}
Also used : Neo4jLabel(com.buschmais.xo.neo4j.api.model.Neo4jLabel) Node(com.buschmais.jqassistant.core.report.api.graph.model.Node) Neo4jNode(com.buschmais.xo.neo4j.api.model.Neo4jNode) Neo4jRelationship(com.buschmais.xo.neo4j.api.model.Neo4jRelationship) Neo4jNode(com.buschmais.xo.neo4j.api.model.Neo4jNode) Identifiable(com.buschmais.jqassistant.core.report.api.graph.model.Identifiable) CompositeObject(com.buschmais.xo.api.CompositeObject) Neo4jRelationship(com.buschmais.xo.neo4j.api.model.Neo4jRelationship) Relationship(com.buschmais.jqassistant.core.report.api.graph.model.Relationship) Collection(java.util.Collection) ReportException(com.buschmais.jqassistant.core.report.api.ReportException) CompositeObject(com.buschmais.xo.api.CompositeObject) Map(java.util.Map) SubGraph(com.buschmais.jqassistant.core.report.api.graph.model.SubGraph)

Aggregations

Node (com.buschmais.jqassistant.core.report.api.graph.model.Node)6 Relationship (com.buschmais.jqassistant.core.report.api.graph.model.Relationship)6 SubGraph (com.buschmais.jqassistant.core.report.api.graph.model.SubGraph)6 CompositeObject (com.buschmais.xo.api.CompositeObject)6 Neo4jNode (com.buschmais.xo.neo4j.api.model.Neo4jNode)6 Neo4jRelationship (com.buschmais.xo.neo4j.api.model.Neo4jRelationship)6 ExecutableRule (com.buschmais.jqassistant.core.rule.api.model.ExecutableRule)4 Test (org.junit.jupiter.api.Test)4 ReportException (com.buschmais.jqassistant.core.report.api.ReportException)2 Neo4jLabel (com.buschmais.xo.neo4j.api.model.Neo4jLabel)2 Collection (java.util.Collection)2 Map (java.util.Map)2 Identifiable (com.buschmais.jqassistant.core.report.api.graph.model.Identifiable)1