浏览代码

10/10 novelty

xiexiang 1 年之前
父节点
当前提交
12a032fca8
共有 1 个文件被更改,包括 11 次插入8 次删除
  1. 11 8
      src/main/java/cn/cslg/pas/service/novelty/NoveltySearchRecordService.java

+ 11 - 8
src/main/java/cn/cslg/pas/service/novelty/NoveltySearchRecordService.java

@@ -62,9 +62,8 @@ public class NoveltySearchRecordService extends ServiceImpl<RetrieveRecordMapper
     public Records querySearchRecord(QuerySearchRecordDTO querySearchRecordDTO) throws IOException {
         Long total = null;
         Records records = new Records();
-        Long size = querySearchRecordDTO.getSize() != null ? querySearchRecordDTO.getSize() : 10;
-        Long current = querySearchRecordDTO.getCurrent() != null ? querySearchRecordDTO.getCurrent() : 0;
-
+        Long size = querySearchRecordDTO.getSize();
+        Long current = querySearchRecordDTO.getCurrent();
         Integer projectId = querySearchRecordDTO.getProjectId();
         List<RetrieveRecord> retrieveRecordList = new ArrayList<>();
         List<RetrieveRecordVO> retrieveRecordVOS = new ArrayList<>();
@@ -83,14 +82,18 @@ public class NoveltySearchRecordService extends ServiceImpl<RetrieveRecordMapper
                 queryWrapper.in(RetrieveRecord::getId, retrieveIds)
                         .orderByDesc(RetrieveRecord::getRetrieveTime)
                         .orderByDesc(RetrieveRecord::getCreateTime);
-
-                IPage<RetrieveRecord> retrieveRecords = this.page(new Page<>(current, size), queryWrapper);
-                retrieveRecordList = retrieveRecords.getRecords();
-
+                if (current != null && size != null) {
+                    IPage<RetrieveRecord> retrieveRecords = this.page(new Page<>(current, size), queryWrapper);
+                    retrieveRecordList = retrieveRecords.getRecords();
+                    total = retrieveRecords.getTotal();
+                } else {
+                    retrieveRecordList = this.list(queryWrapper);
+                    total = (long) retrieveRecordList.size();
+                }
                 if (!retrieveRecordList.isEmpty()) {
                     retrieveRecordVOS = this.loadRetrieveRecords(retrieveRecordList, assoRetrieveRecordProjects);
                 }
-                total = retrieveRecords.getTotal();
+
             }
         }