|
|
|
@ -36,12 +36,11 @@ import java.io.IOException;
|
|
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
|
import java.util.ArrayList;
|
|
|
|
|
import java.util.LinkedHashMap;
|
|
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @author x
|
|
|
|
|
* @website https://el-admin.vip
|
|
|
|
|
* @description 服务实现
|
|
|
|
|
* @author x
|
|
|
|
|
* @date 2020-09-09
|
|
|
|
|
**/
|
|
|
|
|
@Service
|
|
|
|
@ -53,16 +52,7 @@ public class TaskRecordServiceImpl implements TaskRecordService {
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public Map<String,Object> queryAll(TaskRecordQueryCriteria criteria, Pageable pageable){
|
|
|
|
|
// 查询出一级
|
|
|
|
|
criteria.setParentId(0L);
|
|
|
|
|
Page<TaskRecord> page =
|
|
|
|
|
taskRecordRepository.findAll
|
|
|
|
|
((root, criteriaQuery, criteriaBuilder) ->
|
|
|
|
|
QueryHelp.getPredicate(root, criteria, criteriaBuilder), pageable);
|
|
|
|
|
|
|
|
|
|
// 找出所有
|
|
|
|
|
List<TaskRecord> all = taskRecordRepository.findAll();
|
|
|
|
|
page.getContent().stream().peek(taskRecord -> taskRecord.setChildren(getChildrens(taskRecord, all))).collect(Collectors.toList());
|
|
|
|
|
Page<TaskRecord> page = taskRecordRepository.findAll((root, criteriaQuery, criteriaBuilder) -> QueryHelp.getPredicate(root,criteria,criteriaBuilder),pageable);
|
|
|
|
|
return PageUtil.toPage(page.map(taskRecordMapper::toDto));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -80,6 +70,7 @@ public class TaskRecordServiceImpl implements TaskRecordService {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
@Transactional(rollbackFor = Exception.class)
|
|
|
|
|
public TaskRecordDto create(TaskRecord resources) {
|
|
|
|
@ -125,14 +116,4 @@ public class TaskRecordServiceImpl implements TaskRecordService {
|
|
|
|
|
}
|
|
|
|
|
FileUtil.downloadExcel(list, response);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 递归查找所有菜单的子菜单
|
|
|
|
|
*/
|
|
|
|
|
private List<TaskRecord> getChildrens(TaskRecord taskRecord, List<TaskRecord> all) {
|
|
|
|
|
return all.stream().filter(categoryEntity -> categoryEntity.getParentId().equals(taskRecord.getId())).peek(categoryEntity -> {
|
|
|
|
|
//1、找到子任务
|
|
|
|
|
categoryEntity.setChildren(getChildrens(categoryEntity, all));
|
|
|
|
|
}).collect(Collectors.toList());
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|