From 0a3d5fd5d47e90333de9ff316641e50d43e87b68 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子Li <15040126243@163.com>
Date: 星期二, 23 七月 2024 17:05:46 +0800
Subject: [PATCH] update 优化 代码生成分页实现 避免数据误传等问题

---
 ruoyi-modules/ruoyi-generator/src/main/java/org/dromara/generator/service/GenTableServiceImpl.java |   11 ++++-------
 1 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/ruoyi-modules/ruoyi-generator/src/main/java/org/dromara/generator/service/GenTableServiceImpl.java b/ruoyi-modules/ruoyi-generator/src/main/java/org/dromara/generator/service/GenTableServiceImpl.java
index 48ae8b2..cf9d2d0 100644
--- a/ruoyi-modules/ruoyi-generator/src/main/java/org/dromara/generator/service/GenTableServiceImpl.java
+++ b/ruoyi-modules/ruoyi-generator/src/main/java/org/dromara/generator/service/GenTableServiceImpl.java
@@ -123,9 +123,6 @@
         // 鑾峰彇鏌ヨ鏉′欢
         String tableName = genTable.getTableName();
         String tableComment = genTable.getTableComment();
-        // 鑾峰彇鍒嗛〉鍙傛暟
-        Integer pageNum = pageQuery.getPageNum();
-        Integer pageSize = pageQuery.getPageSize();
 
         LinkedHashMap<String, Table<?>> tablesMap = ServiceProxy.metadata().tables();
         if (CollUtil.isEmpty(tablesMap)) {
@@ -170,10 +167,10 @@
                 return gen;
             }).toList();
 
-        // 鍒涘缓鍒嗛〉瀵硅薄锛屽苟璁剧疆鎬昏褰曟暟
-        IPage<GenTable> page = new Page<>(pageNum, pageSize, tables.size());
-        // 浣跨敤CollUtil杩涜鍒嗛〉锛屽苟璁剧疆鍒嗛〉璁板綍
-        page.setRecords(CollUtil.page(pageNum - 1, pageSize, tables));
+        IPage<GenTable> page = pageQuery.build();
+        page.setTotal(tables.size());
+        // 鎵嬪姩鍒嗛〉 set鏁版嵁
+        page.setRecords(CollUtil.page((int) page.getCurrent() - 1, (int) page.getSize(), tables));
         return TableDataInfo.build(page);
     }
 

--
Gitblit v1.9.3