|
@@ -49,19 +49,24 @@ public class NoveltyReportTemplateService extends ServiceImpl<NoveltyReportTempl
|
|
|
// }
|
|
|
|
|
|
public Records queryTemplate(QueryTemplateDTO queryTemplateDTO) throws IOException {
|
|
|
+ Integer temId = queryTemplateDTO.getTemplateId();
|
|
|
+ Long size = queryTemplateDTO.getSize();
|
|
|
+ Long current = queryTemplateDTO.getCurrent();
|
|
|
Records records = new Records();
|
|
|
Long total = null;
|
|
|
List<QueryTemplateVO> queryTemplateVOS = new ArrayList<>();
|
|
|
List<NoveltyReportTemplate> noveltyReportTemplates = new ArrayList<>();
|
|
|
- if (queryTemplateDTO.getTemplateId() != null) {
|
|
|
- NoveltyReportTemplate noveltyReportTemplate = this.getById(queryTemplateDTO.getTemplateId());
|
|
|
- noveltyReportTemplates.add(noveltyReportTemplate);
|
|
|
- } else {
|
|
|
- LambdaQueryWrapper<NoveltyReportTemplate> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
- queryWrapper.orderByDesc(NoveltyReportTemplate::getCreateTime);
|
|
|
- IPage<NoveltyReportTemplate> noveltyReportTemplatePage = this.page(new Page<>(queryTemplateDTO.getCurrent(), queryTemplateDTO.getSize()), queryWrapper);
|
|
|
+ LambdaQueryWrapper<NoveltyReportTemplate> queryWrapper = new LambdaQueryWrapper<>();
|
|
|
+ if (temId != null) {
|
|
|
+ queryWrapper.eq(NoveltyReportTemplate::getId, temId);
|
|
|
+ }
|
|
|
+ if (size != null && current != null) {
|
|
|
+ IPage<NoveltyReportTemplate> noveltyReportTemplatePage = this.page(new Page<>(current, size), queryWrapper);
|
|
|
noveltyReportTemplates = noveltyReportTemplatePage.getRecords();
|
|
|
total = noveltyReportTemplatePage.getTotal();
|
|
|
+ } else {
|
|
|
+ noveltyReportTemplates = this.list();
|
|
|
+ total = (long) noveltyReportTemplates.size();
|
|
|
}
|
|
|
if (!noveltyReportTemplates.isEmpty()) {
|
|
|
queryTemplateVOS = this.loadTemplateVOS(noveltyReportTemplates);
|