Merge pull request #7540 from gdicristofaro/8277-fix

8277 code cleanup
This commit is contained in:
eugene7646 2022-01-28 14:18:13 -05:00 committed by GitHub
commit 0b34e6aa3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -362,7 +362,7 @@ public class AnalysisResultDAO extends BlackboardArtifactDAO {
? TreeDisplayCount.INDETERMINATE ? TreeDisplayCount.INDETERMINATE
: TreeDisplayCount.getDeterminate(entry.getValue().getLeft()); : TreeDisplayCount.getDeterminate(entry.getValue().getLeft());
return getTreeItem(entry.getKey(), null, dataSourceId, displayCount, entry.getValue().getRight()); return new AnalysisResultTreeItem(entry.getKey(), null, dataSourceId, displayCount, entry.getValue().getRight());
}) })
.sorted(Comparator.comparing(countRow -> countRow.getDisplayName())) .sorted(Comparator.comparing(countRow -> countRow.getDisplayName()))
.collect(Collectors.toList()); .collect(Collectors.toList());
@ -1107,7 +1107,12 @@ public class AnalysisResultDAO extends BlackboardArtifactDAO {
displayCount displayCount
); );
} else { } else {
return getTreeItem(arEvt.getArtifactType(), arEvt.getConfiguration(), arEvt.getDataSourceId(), displayCount, null); return getConfigTreeItem(
arEvt.getArtifactType(),
arEvt.getDataSourceId(),
arEvt.getConfiguration(),
StringUtils.isBlank(arEvt.getConfiguration()) ? arEvt.getArtifactType().getDisplayName() : arEvt.getConfiguration(),
displayCount);
} }
} }
@ -1134,11 +1139,6 @@ public class AnalysisResultDAO extends BlackboardArtifactDAO {
displayCount); displayCount);
} }
private TreeItemDTO<AnalysisResultSearchParam> getTreeItem(BlackboardArtifact.Type type, String configuration,
Long dataSourceId, TreeDisplayCount displayCount, Boolean hasChildren) {
return new AnalysisResultTreeItem(type, configuration, dataSourceId, displayCount, hasChildren);
}
@Override @Override
Set<? extends DAOEvent> handleIngestComplete() { Set<? extends DAOEvent> handleIngestComplete() {
return SubDAOUtils.getIngestCompleteEvents(this.treeCounts, (arEvt, count) -> getTreeItem(arEvt, count)); return SubDAOUtils.getIngestCompleteEvents(this.treeCounts, (arEvt, count) -> getTreeItem(arEvt, count));