From ae199e407461dc60d2645116c875188af931489c Mon Sep 17 00:00:00 2001
From: suyue <2016494681@qq.com>
Date: Fri, 16 Aug 2024 19:21:07 +0800
Subject: [PATCH] export current page CSV

---
 .../server/web/api/controller/rdb/RdbDmlExportController.java  | 2 +-
 .../web/api/controller/rdb/request/DataExportRequest.java      | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlExportController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlExportController.java
index 0f554a2c..35be1079 100644
--- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlExportController.java
+++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlExportController.java
@@ -84,7 +84,7 @@ public class RdbDmlExportController {
         ExportTypeEnum exportType = EasyEnumUtils.getEnum(ExportTypeEnum.class, request.getExportType());
         String sql;
         if (exportSize == ExportSizeEnum.CURRENT_PAGE) {
-            sql = request.getSql();
+            sql = request.getOriginalSql() + " LIMIT " + request.getPageSize() + " OFFSET " + (request.getPageNo() - 1) * request.getPageSize();
         } else {
             sql = request.getOriginalSql();
         }
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/request/DataExportRequest.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/request/DataExportRequest.java
index c4f02e7d..747040ab 100644
--- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/request/DataExportRequest.java
+++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/request/DataExportRequest.java
@@ -23,6 +23,9 @@ public class DataExportRequest extends DataSourceBaseRequest {
      */
     private String originalSql;
 
+    private Integer pageNo;
+    private Integer pageSize;
+
     /**
      * export type
      *