From 836db3f8dc9f8de27e2c93d73b6e9225ac1a75e5 Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Fri, 12 Sep 2025 11:29:46 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E5=AF=B9=E6=8E=A5=20SQLBot=20=E5=88=97?=
 =?UTF-8?q?=E6=9D=83=E9=99=90=E5=A4=B1=E6=95=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/dataset/manage/DatasetSQLBotManage.java    | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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 4520b840ab..fa047fd529 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
@@ -113,7 +113,7 @@ public class DatasetSQLBotManage {
         DataSetColumnPermissionsDTO dataSetColumnPermissionsDTO = new DataSetColumnPermissionsDTO();
         dataSetColumnPermissionsDTO.setAuthTargetId(uid);
         dataSetColumnPermissionsDTO.setAuthTargetType("user");
-        dataSetColumnPermissionsDTO.setEnable(true);
+        // dataSetColumnPermissionsDTO.setEnable(true);
         List<DataSetColumnPermissionsDTO> dataSetColumnPermissionsDTOS = columnPermissionsApi.list(dataSetColumnPermissionsDTO);
 
         if (CollectionUtils.isNotEmpty(roleIds)) {
@@ -364,6 +364,11 @@ public class DatasetSQLBotManage {
             if (ObjectUtils.isEmpty(fields)) {
                 DEException.throwException(Translator.get("i18n_no_column_permission"));
             }
+            if (sqlbotFields.size() > fields.size()) {
+                Set<Long> fieldIdSet = fields.stream().map(DatasetTableFieldDTO::getId).collect(Collectors.toSet());
+                List<SQLBotAssistantField> filterSqlbotFields = sqlbotFields.stream().filter(item -> fieldIdSet.contains(item.getFieldId())).collect(Collectors.toList());
+                table.setFields(filterSqlbotFields);
+            }
         }
         buildFieldName(sqlMap, originFields);
         Map<Long, DatasourceSchemaDTO> dsMap = (Map<Long, DatasourceSchemaDTO>) sqlMap.get("dsMap");