diff --git a/src/main/java/com/dispose/manager/impl/DisposeTaskManagerImpl.java b/src/main/java/com/dispose/manager/impl/DisposeTaskManagerImpl.java index 2d34a348..6dff034e 100644 --- a/src/main/java/com/dispose/manager/impl/DisposeTaskManagerImpl.java +++ b/src/main/java/com/dispose/manager/impl/DisposeTaskManagerImpl.java @@ -4,7 +4,6 @@ import com.dispose.common.DisposeCapacityType; import com.dispose.common.DisposeTaskStatus; import com.dispose.common.ErrorCode; import com.dispose.manager.DisposeTaskManager; -import com.dispose.mapper.DeviceTaskMapper; import com.dispose.mapper.DisposeTaskMapper; import com.dispose.pojo.entity.DisposeTask; import com.dispose.pojo.po.MulReturnType; @@ -31,12 +30,6 @@ public class DisposeTaskManagerImpl implements DisposeTaskManager { @Resource private DisposeTaskMapper disposeTaskMapper; - /** - * The Device task mapper. - */ - @Resource - private DeviceTaskMapper deviceTaskMapper; - /** * Gets dispose task. * @@ -130,6 +123,7 @@ public class DisposeTaskManagerImpl implements DisposeTaskManager { public MulReturnType, List> getPagesOfTask(List taskId, Integer startPage, Integer pageSize) { + PageInfo pageInfo; List taskList = new ArrayList<>(); // 设置分页信息 @@ -137,13 +131,15 @@ public class DisposeTaskManagerImpl implements DisposeTaskManager { if(taskId.size() > 0) { taskList.addAll(disposeTaskMapper.selectByIdSet(taskId)); + pageInfo = new PageInfo<>(taskList); } else { - taskList.addAll(disposeTaskMapper.selectPageAll()); + List preList = disposeTaskMapper.selectPageAll(); + pageInfo = new PageInfo<>(preList); + + preList.forEach(k -> taskId.add(k.getId())); + taskList.addAll(disposeTaskMapper.selectByIdSet(taskId)); } - PageInfo pageInfo = new PageInfo<>(taskList); - - return new MulReturnType<>(pageInfo, taskList); } } diff --git a/src/test/java/com/dispose/test/testcase/qa/v200/P2TaskList.java b/src/test/java/com/dispose/test/testcase/qa/v200/P2TaskList.java index 52b417d4..3c9a5461 100644 --- a/src/test/java/com/dispose/test/testcase/qa/v200/P2TaskList.java +++ b/src/test/java/com/dispose/test/testcase/qa/v200/P2TaskList.java @@ -46,7 +46,7 @@ public class P2TaskList { Assert.assertEquals((long) v.getCode(), e.getHttpCode()); //Assert.assertEquals((long) v.getMsgContent().getStatus(), e.getCode()); - Assert.assertEquals((long) v.getMsgContent().getCurPageNumber(), 1); + Assert.assertEquals((long) v.getMsgContent().getCurPageNumber(), 2); Assert.assertEquals((long)v.getMsgContent().getPageSize(), 20); Assert.assertNotNull(v.getMsgContent().getItems());