package com.qianwen.smartman.modules.mdc.service.impl;
|
|
import cn.hutool.core.util.StrUtil;
|
import java.time.LocalDate;
|
import java.util.ArrayList;
|
import java.util.Collection;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
import com.qianwen.smartman.common.constant.DateConstant;
|
import com.qianwen.smartman.common.utils.LocalDateTimeUtils;
|
import com.qianwen.smartman.common.utils.LocalDateUtil;
|
import com.qianwen.core.tool.utils.Func;
|
import com.qianwen.smartman.modules.mdc.dto.IntervalDateDto;
|
import com.qianwen.smartman.modules.mdc.enums.StatisticalMethodEnum;
|
import com.qianwen.smartman.modules.mdc.service.IStatisticsService;
|
import com.qianwen.smartman.modules.mdc.utils.MdcConvertUtil;
|
import com.qianwen.smartman.modules.mdc.vo.AnalysisFilterVO;
|
import com.qianwen.smartman.modules.mdc.vo.DateInfo;
|
import com.qianwen.smartman.modules.mdc.vo.SplitFilterListVO;
|
import com.qianwen.smartman.modules.mdc.vo.SplitFilterVO;
|
import org.springframework.stereotype.Service;
|
|
@Service
|
public class StatisticsServiceImpl implements IStatisticsService {
|
@Override // org.springblade.modules.mdc.service.IStatisticsService
|
public SplitFilterListVO organizeQueryConditions(AnalysisFilterVO input) {
|
List<IntervalDateDto> intervalDateDtoList;
|
Map<String, String> shiftKeyNameMap = MdcConvertUtil.keyNameToMap(input.getShiftList());
|
List<DateInfo> result = new ArrayList<>();
|
if (Func.isNotEmpty(input.getIsReverse()) && input.getIsReverse().booleanValue()) {
|
intervalDateDtoList = LocalDateTimeUtils.getIntervalDateReverse(input.getStartDate(), input.getEndDate());
|
} else {
|
intervalDateDtoList = LocalDateTimeUtils.getIntervalDate(input.getStartDate(), input.getEndDate());
|
}
|
if (StatisticalMethodEnum.SHIFT.equals(input.getStatisticalMethod())) {
|
buildShiftData(intervalDateDtoList, shiftKeyNameMap, result);
|
} else if (StatisticalMethodEnum.DAY.equals(input.getStatisticalMethod())) {
|
buildDayData(intervalDateDtoList, result);
|
} else if (StatisticalMethodEnum.WEEK.equals(input.getStatisticalMethod())) {
|
buildWeekData(intervalDateDtoList, result);
|
} else {
|
buildMonthData(intervalDateDtoList, result);
|
}
|
SplitFilterListVO vo = new SplitFilterListVO();
|
vo.setItems( result.stream().map(c -> {
|
SplitFilterVO voItem = new SplitFilterVO();
|
voItem.setTitle(c.getTitle());
|
voItem.setDate(c.getDate());
|
voItem.setMonth(c.getMonth());
|
voItem.setYear(c.getYear());
|
voItem.setShiftIndex(c.getShiftIndex());
|
voItem.setStatisticalMethod(input.getStatisticalMethod());
|
voItem.setWeek(c.getWeek());
|
voItem.setWorkStationList(input.getWorkStationList());
|
return voItem;
|
}).collect(Collectors.toList()));
|
return vo;
|
}
|
|
private void buildMonthData(List<IntervalDateDto> intervalDateDtoList, List<DateInfo> result) {
|
intervalDateDtoList.forEach(intervalDateDto -> {
|
intervalDateDto.getMonthList().forEach(month -> {
|
Integer year = intervalDateDto.getYear();
|
String code = StrUtil.format("{}_{}", new Object[]{year, month});
|
String title = StrUtil.format("{}-{}", new Object[]{year, month});
|
result.add(new DateInfo(code, title, null, null, month, year));
|
});
|
});
|
}
|
|
private void buildWeekData(List<IntervalDateDto> intervalDateDtoList, List<DateInfo> result) {
|
intervalDateDtoList.forEach(intervalDateDto -> {
|
intervalDateDto.getWeekList().forEach(week -> {
|
Integer year = intervalDateDto.getYear();
|
String code = StrUtil.format("{}_{}", new Object[]{year, week});
|
String title = StrUtil.format("{}-{}", new Object[]{year, week});
|
result.add(new DateInfo(code, title, null, week, null, year));
|
});
|
});
|
}
|
|
private void buildDayData(List<IntervalDateDto> intervalDateDtoList, List<DateInfo> result) {
|
IntervalDateDto intervalDateDto = intervalDateDtoList.get(0);
|
List<String> dayList = intervalDateDto.getDayList();
|
dayList.forEach(day -> {
|
LocalDate localDate = LocalDateUtil.stringToLocalDate(day, DateConstant.PATTERN_DATE);
|
Integer week = LocalDateTimeUtils.getWeek(localDate);
|
Integer month = LocalDateTimeUtils.getMonth(localDate);
|
Integer year = LocalDateTimeUtils.getYear(localDate);
|
result.add(new DateInfo(day, day, day, week, month, year));
|
});
|
}
|
|
public void buildShiftData(List<IntervalDateDto> intervalDateDtoList, Map<String, String> shiftKeyNameMap, List<DateInfo> result) {
|
IntervalDateDto intervalDateDto = intervalDateDtoList.get(0);
|
List<String> dayList = intervalDateDto.getDayList();
|
dayList.forEach(day -> {
|
shiftKeyNameMap.forEach((shiftIndex, shiftName) -> {
|
String code = StrUtil.format("{}_{}", new Object[]{day, shiftIndex});
|
String title = StrUtil.format("{}-{}", new Object[]{day, shiftKeyNameMap.get(String.valueOf(shiftIndex))});
|
LocalDate localDate = LocalDateUtil.stringToLocalDate(day, DateConstant.PATTERN_DATE);
|
Integer week = LocalDateTimeUtils.getWeek(localDate);
|
Integer month = LocalDateTimeUtils.getMonth(localDate);
|
Integer year = LocalDateTimeUtils.getYear(localDate);
|
result.add(new DateInfo(code, title, day, week, month, year, Integer.valueOf(shiftIndex)));
|
});
|
});
|
}
|
}
|
