use of com.buschmais.jqassistant.core.report.api.graph.model.Node in project jqa-core-framework by buschmais.
the class SubGraphFactory method convert.
private <I extends Identifiable> I convert(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 = convert(virtualObject.get(START_NODE));
Node endNode = convert(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--);
subgraph.setParent((Node) convert(subgraph, virtualObject.get(PARENT)));
addSubGraphChildren(subgraph, virtualObject, NODES, subgraph.getNodes());
addSubGraphChildren(subgraph, virtualObject, RELATIONSHIPS, subgraph.getRelationships());
return (I) subgraph;
}
}
} else if (value instanceof CompositeObject) {
CompositeObject compositeObject = (CompositeObject) value;
I identifiable = convert(compositeObject.getDelegate());
identifiable.setLabel(ReportHelper.getLabel(value));
return 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((Node) convert(neo4jRelationship.getStartNode()));
relationship.setEndNode((Node) convert(neo4jRelationship.getEndNode()));
relationship.getProperties().putAll(neo4jRelationship.getProperties());
return (I) relationship;
}
throw new ReportException("Element type not supported: " + value);
}
use of com.buschmais.jqassistant.core.report.api.graph.model.Node in project jqa-core-framework by buschmais.
the class SubGraphFactoryTest method subGraph.
@Test
public void subGraph() throws ReportException {
Map<String, Object> virtualGraph = //
MapBuilder.<String, Object>builder().entry("role", //
"graph").entry("label", //
"Virtual Graph").entry("parent", //
getNeo4jNode(0l)).entry("nodes", //
singletonList(getNeo4jNode(1l))).entry("relationships", //
singletonList(getNeo4jRelationship(1l, "TEST"))).build();
MapBuilder<String, Object> rowBuilder = MapBuilder.builder();
Map<String, Object> row = rowBuilder.entry("graph", virtualGraph).build();
Result<ExecutableRule> result = Result.builder().rows(singletonList(row)).build();
SubGraph graph = factory.createSubGraph(result);
assertThat(graph.getParent()).isNull();
assertThat(graph.getNodes()).isEmpty();
assertThat(graph.getRelationships()).isEmpty();
Map<Long, SubGraph> subGraphs = graph.getSubGraphs();
assertThat(subGraphs).hasSize(1);
SubGraph subGraph = subGraphs.get(-2l);
assertThat(subGraph.getId()).isEqualTo(-2l);
assertThat(subGraph.getLabel()).isEqualTo("Virtual Graph");
Node parent = subGraph.getParent();
assertThat(parent).isNotNull();
assertThat(parent.getId()).isEqualTo(0l);
Map<Long, Node> nodes = subGraph.getNodes();
assertThat(nodes).hasSize(1);
Node node1 = nodes.get(1l);
assertThat(node1.getId()).isEqualTo(1l);
Map<Long, Relationship> relationships = subGraph.getRelationships();
assertThat(relationships).hasSize(1);
Relationship relationship1 = relationships.get(1l);
assertThat(relationship1.getId()).isEqualTo(1l);
}
use of com.buschmais.jqassistant.core.report.api.graph.model.Node in project jqa-core-framework by buschmais.
the class SubGraphFactoryTest method compositeObject.
@Test
public void compositeObject() throws ReportException {
CompositeObject compositeObject = mock(CompositeObject.class);
Neo4jNode neo4jNode = getNeo4jNode(1l, "Test");
doReturn(neo4jNode).when(compositeObject).getDelegate();
MapBuilder<String, Object> builder = MapBuilder.builder();
builder.entry("nodes", singletonList(compositeObject));
Result<ExecutableRule> result = Result.builder().rows(singletonList(builder.build())).build();
SubGraph graph = factory.createSubGraph(result);
Map<Long, Node> nodes = graph.getNodes();
assertThat(nodes).isNotEmpty();
Node node = nodes.get(1l);
assertThat(node).isNotNull();
assertThat(node.getId()).isEqualTo(1l);
assertThat(node.getLabels()).containsExactly("Test");
}
use of com.buschmais.jqassistant.core.report.api.graph.model.Node in project jqa-core-framework by buschmais.
the class SubGraphFactoryTest method virtualNode.
@Test
public void virtualNode() throws ReportException {
Map<Object, Object> properties = MapBuilder.builder().entry("key", "value").build();
Map<String, Object> virtualNode = //
MapBuilder.<String, Object>builder().entry("role", //
"node").entry("label", //
"Virtual Node").entry("labels", //
singletonList("Test")).entry("properties", //
properties).build();
MapBuilder<String, Object> builder = MapBuilder.builder();
builder.entry("nodes", singletonList(virtualNode));
Result<ExecutableRule> result = Result.builder().rows(singletonList(builder.build())).build();
SubGraph graph = factory.createSubGraph(result);
Map<Long, Node> nodes = graph.getNodes();
assertThat(nodes.size()).isEqualTo(1);
Node node = nodes.get(-1l);
assertThat(node).isNotNull();
assertThat(node.getId()).isEqualTo(-1l);
assertThat(node.getLabel()).isEqualTo("Virtual Node");
assertThat(node.getLabels()).containsExactly("Test");
assertThat(node.getProperties()).isEqualTo(properties);
}
use of com.buschmais.jqassistant.core.report.api.graph.model.Node in project jqa-core-framework by buschmais.
the class SubGraphFactoryTest method nodeAndRelationship.
@Test
public void nodeAndRelationship() throws ReportException {
MapBuilder<String, Object> builder = MapBuilder.builder();
Map<String, Object> nodeProperties = MapBuilder.<String, Object>builder().entry("nodeKey", "value").build();
builder.entry("node", getNeo4jNode(1l, nodeProperties, "Test1", "Test2"));
Map<String, Object> relationshipProperties = MapBuilder.<String, Object>builder().entry("relationshipKey", "value").build();
builder.entry("relation", getNeo4jRelationship(1l, relationshipProperties, "TEST"));
Result<ExecutableRule> result = Result.builder().rows(singletonList(builder.build())).build();
SubGraph graph = factory.createSubGraph(result);
assertThat(graph.getId()).isEqualTo(-1l);
assertThat(graph.getParent()).isNull();
assertThat(graph.getSubGraphs()).isEmpty();
Map<Long, Node> nodes = graph.getNodes();
assertThat(nodes).hasSize(1);
Node node = nodes.get(1l);
assertThat(node.getId()).isEqualTo(1l);
assertThat(node.getLabels()).containsExactly("Test1", "Test2");
assertThat(node.getProperties()).isEqualTo(nodeProperties);
Map<Long, Relationship> relationships = graph.getRelationships();
assertThat(relationships).hasSize(1);
Relationship relationship = relationships.get(1l);
assertThat(relationship.getId()).isEqualTo(1l);
assertThat(relationship.getType()).isEqualTo("TEST");
assertThat(relationship.getProperties()).isEqualTo(relationshipProperties);
}
Aggregations