diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/model/document/DocumentTree.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/model/document/DocumentTree.java index 64077a54..375a9080 100644 --- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/model/document/DocumentTree.java +++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/model/document/DocumentTree.java @@ -40,7 +40,10 @@ public class DocumentTree { public TextBlock buildTextBlock() { - return allEntriesInOrder().map(Entry::getNode).filter(SemanticNode::isLeaf).map(SemanticNode::getLeafTextBlock).collect(new TextBlockCollector()); + return allEntriesInOrder().map(Entry::getNode) + .filter(SemanticNode::isLeaf) + .map(SemanticNode::getLeafTextBlock) + .collect(new TextBlockCollector()); } @@ -89,8 +92,8 @@ public class DocumentTree { if (treeId.isEmpty()) { return root != null; } - Entry entry = root.children.get(treeId.get(0)); - for (int id : treeId.subList(1, treeId.size())) { + Entry entry = root; + for (int id : treeId) { if (id >= entry.children.size() || 0 > id) { return false; } @@ -114,13 +117,16 @@ public class DocumentTree { public Stream childNodes(List treeId) { - return getEntryById(treeId).children.stream().map(Entry::getNode); + return getEntryById(treeId).children.stream() + .map(Entry::getNode); } public Stream childNodesOfType(List treeId, NodeType nodeType) { - return getEntryById(treeId).children.stream().filter(entry -> entry.node.getType().equals(nodeType)).map(Entry::getNode); + return getEntryById(treeId).children.stream() + .filter(entry -> entry.node.getType().equals(nodeType)) + .map(Entry::getNode); } @@ -199,26 +205,32 @@ public class DocumentTree { public Stream allEntriesInOrder() { - return Stream.of(root).flatMap(DocumentTree::flatten); + return Stream.of(root) + .flatMap(DocumentTree::flatten); } public Stream allSubEntriesInOrder(List parentId) { - return getEntryById(parentId).children.stream().flatMap(DocumentTree::flatten); + return getEntryById(parentId).children.stream() + .flatMap(DocumentTree::flatten); } @Override public String toString() { - return String.join("\n", allEntriesInOrder().map(Entry::toString).toList()); + return String.join("\n", + allEntriesInOrder().map(Entry::toString) + .toList()); } private static Stream flatten(Entry entry) { - return Stream.concat(Stream.of(entry), entry.children.stream().flatMap(DocumentTree::flatten)); + return Stream.concat(Stream.of(entry), + entry.children.stream() + .flatMap(DocumentTree::flatten)); }