From 70ef8133f71ffa1b4902e1abbf2ea265edeeca57 Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Mon, 26 Jan 2026 13:53:40 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E5=AF=B9=E6=8E=A5=20SQLBot=20=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E6=8F=90=E4=BE=9B=E6=98=AF=E5=90=A6=E5=9B=BA=E5=AE=9A?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E6=BA=90=20ID=20=E7=9A=84=E9=85=8D=E7=BD=AE?=
 =?UTF-8?q?=E9=A1=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/dataset/manage/DatasetSQLBotManage.java     | 6 ++++++
 .../io/dataease/api/dataset/vo/DataSQLBotAssistantVO.java   | 2 ++
 2 files changed, 8 insertions(+)

diff --git a/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetSQLBotManage.java b/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetSQLBotManage.java
index 6e7fe62548..6a792f25dd 100644
--- a/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetSQLBotManage.java
+++ b/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetSQLBotManage.java
@@ -97,6 +97,9 @@ public class DatasetSQLBotManage {
     @Value("${dataease.sqlbot.log:false}")
     private boolean sqlbotApiLog;
 
+    @Value("${dataease.sqlbot.ds-id-fixed:false}")
+    private boolean dsIdFixed;
+
     @Resource
     private ExtChartViewMapper extChartViewMapper;
 
@@ -546,6 +549,9 @@ public class DatasetSQLBotManage {
         vo.setSchema(config.getSchema());
         vo.setUser(config.getUsername());
         vo.setPassword(config.getPassword());
+        if (dsIdFixed) {
+            vo.setId(Long.parseLong(row.get("cd_id").toString()));
+        }
         row.put("cd_configuration", config_json);
         Map<String, Object> rowData = buildRowData(row, 0);
         rowData.put("id", Long.parseLong(row.get("cd_id").toString()));
diff --git a/sdk/api/api-base/src/main/java/io/dataease/api/dataset/vo/DataSQLBotAssistantVO.java b/sdk/api/api-base/src/main/java/io/dataease/api/dataset/vo/DataSQLBotAssistantVO.java
index 4b9c4ad1ca..0da0701ae9 100644
--- a/sdk/api/api-base/src/main/java/io/dataease/api/dataset/vo/DataSQLBotAssistantVO.java
+++ b/sdk/api/api-base/src/main/java/io/dataease/api/dataset/vo/DataSQLBotAssistantVO.java
@@ -11,6 +11,8 @@ import java.util.Map;
 @Data
 public class DataSQLBotAssistantVO implements Serializable {
 
+    private Long id;
+
     private String name;
 
     private String type;