use of de.symeda.sormas.api.disease.DiseaseConfigurationFacade in project SORMAS-Project by hzi-braunschweig.
the class CaseStatisticsFacadeEjb method queryCaseCount.
@SuppressWarnings("unchecked")
@Override
public List<StatisticsCaseCountDto> queryCaseCount(StatisticsCaseCriteria caseCriteria, StatisticsCaseAttribute rowGrouping, StatisticsCaseSubAttribute rowSubGrouping, StatisticsCaseAttribute columnGrouping, StatisticsCaseSubAttribute columnSubGrouping, boolean includePopulation, boolean includeZeroValues, Integer populationReferenceYear) {
// case counts
Pair<String, List<Object>> caseCountQueryAndParams = buildCaseCountQuery(caseCriteria, rowGrouping, rowSubGrouping, columnGrouping, columnSubGrouping);
Query caseCountQuery = em.createNativeQuery(caseCountQueryAndParams.getKey().toString());
for (int i = 0; i < caseCountQueryAndParams.getValue().size(); i++) {
caseCountQuery.setParameter(i + 1, caseCountQueryAndParams.getValue().get(i));
}
Function<Integer, RegionReferenceDto> regionProvider = id -> regionFacade.getRegionReferenceById(id);
Function<Integer, DistrictReferenceDto> districtProvider = id -> districtFacade.getDistrictReferenceById(id);
Function<Integer, CommunityReferenceDto> communityProvider = id -> communityFacade.getCommunityReferenceById(id);
Function<Integer, FacilityReferenceDto> healthFacilityProvider = id -> facilityFacade.getFacilityReferenceById(id);
List<StatisticsCaseCountDto> caseCountResults = ((Stream<Object[]>) caseCountQuery.getResultStream()).map(result -> {
Object rowKey = "".equals(result[1]) ? null : result[1];
Object columnKey = "".equals(result[2]) ? null : result[2];
return new StatisticsCaseCountDto(result[0] != null ? ((Number) result[0]).intValue() : null, null, StatisticsHelper.buildGroupingKey(rowKey, rowGrouping, rowSubGrouping, regionProvider, districtProvider, communityProvider, healthFacilityProvider), StatisticsHelper.buildGroupingKey(columnKey, columnGrouping, columnSubGrouping, regionProvider, districtProvider, communityProvider, healthFacilityProvider));
}).collect(Collectors.toList());
if (includeZeroValues) {
List<StatisticsGroupingKey> allRowKeys;
if (rowGrouping != null) {
allRowKeys = (List<StatisticsGroupingKey>) caseCriteria.getFilterValuesForGrouping(rowGrouping, rowSubGrouping);
if (allRowKeys == null) {
allRowKeys = StatisticsHelper.getAttributeGroupingKeys(rowGrouping, rowSubGrouping, diseaseConfigurationFacade, caseFacade, regionFacade, districtFacade);
}
} else {
allRowKeys = Arrays.asList((StatisticsGroupingKey) null);
}
List<StatisticsGroupingKey> allColumnKeys;
if (columnGrouping != null) {
allColumnKeys = (List<StatisticsGroupingKey>) caseCriteria.getFilterValuesForGrouping(columnGrouping, columnSubGrouping);
if (allColumnKeys == null) {
allColumnKeys = StatisticsHelper.getAttributeGroupingKeys(columnGrouping, columnSubGrouping, diseaseConfigurationFacade, caseFacade, regionFacade, districtFacade);
}
} else {
allColumnKeys = Arrays.asList((StatisticsGroupingKey) null);
}
for (StatisticsGroupingKey rowKey : allRowKeys) {
for (StatisticsGroupingKey columnKey : allColumnKeys) {
StatisticsCaseCountDto zeroDto = new StatisticsCaseCountDto(0, null, rowKey, columnKey);
if (!caseCountResults.contains(zeroDto)) {
caseCountResults.add(zeroDto);
}
}
}
}
// population
if (includePopulation) {
Pair<String, List<Object>> populationQueryAndParams = buildPopulationQuery(caseCriteria, rowGrouping, rowSubGrouping, columnGrouping, columnSubGrouping, populationReferenceYear);
Query populationQuery = em.createNativeQuery(populationQueryAndParams.getKey().toString());
for (int i = 0; i < populationQueryAndParams.getValue().size(); i++) {
populationQuery.setParameter(i + 1, populationQueryAndParams.getValue().get(i));
}
List<StatisticsCaseCountDto> populationResults = ((Stream<Object[]>) populationQuery.getResultStream()).map(result -> {
Object rowKey = "".equals(result[1]) ? null : result[1];
Object columnKey = "".equals(result[2]) ? null : result[2];
return new StatisticsCaseCountDto(null, result[0] != null ? ((Number) result[0]).intValue() : null, StatisticsHelper.buildGroupingKey(rowKey, rowGrouping, rowSubGrouping, regionProvider, districtProvider, communityProvider, healthFacilityProvider), StatisticsHelper.buildGroupingKey(columnKey, columnGrouping, columnSubGrouping, regionProvider, districtProvider, communityProvider, healthFacilityProvider));
}).collect(Collectors.toList());
boolean rowIsPopulation = rowGrouping != null && rowGrouping.isPopulationData();
boolean columnIsPopulation = columnGrouping != null && columnGrouping.isPopulationData();
if (!populationResults.isEmpty()) {
assert ((populationResults.get(0).getRowKey() != null) == rowIsPopulation);
assert ((populationResults.get(0).getColumnKey() != null) == columnIsPopulation);
}
// add the population data to the case counts
// when a key is not a population data key, we use null instead
StatisticsCaseCountDto searchDto = new StatisticsCaseCountDto(null, null, null, null);
for (StatisticsCaseCountDto caseCountResult : caseCountResults) {
if (rowIsPopulation) {
searchDto.setRowKey(caseCountResult.getRowKey());
}
if (columnIsPopulation) {
searchDto.setColumnKey(caseCountResult.getColumnKey());
}
int index = populationResults.indexOf(searchDto);
if (index >= 0) {
caseCountResult.setPopulation(populationResults.get(index).getPopulation());
}
}
}
return caseCountResults;
}
Aggregations