fix(sirius): Fixed find page all error

yizzuide · Nov 22, 2023 · a0326a4 · a0326a4
1 parent fd95005
commit a0326a4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Milkomeda/src/main/java/com/github/yizzuide/milkomeda/sirius/PageableService.java b/Milkomeda/src/main/java/com/github/yizzuide/milkomeda/sirius/PageableService.java
@@ -232,7 +232,13 @@ public UniformPage<T> selectByPage(UniformQueryPageData<T> queryPageData, Map<St
         List<T> records;
         // 如果页记录数为-1，则不分页
         if (page.getSize() == -1) {
+            String sqlSelect = queryWrapper.getSqlSelect();
+            queryWrapper.select("*");
             Long totalSize = this.baseMapper.selectCount(queryWrapper);
+            if (totalSize > 10000) {
+                throw new UnsupportedOperationException("Record row over more then 10000");
+            }
+            queryWrapper.select(sqlSelect);
             records = this.baseMapper.selectList(queryWrapper);
             uniformPage.setTotalSize(totalSize);
             uniformPage.setPageCount(1L);