From 0c79bc3a18f72c36f92fd0c6529f127248327f48 Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Thu, 28 Aug 2025 16:47:39 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20SQLBot=20=E6=95=B0=E6=8D=AE=E6=BA=90?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E9=80=9A=E8=BF=87=20AES=20=E5=8A=A0=E5=AF=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../dataset/manage/DatasetSQLBotManage.java   | 42 ++++++++++++++++---
 1 file changed, 37 insertions(+), 5 deletions(-)

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 2ca446433c..c99089a478 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
@@ -20,10 +20,7 @@ import io.dataease.datasource.manage.EngineManage;
 import io.dataease.extensions.datasource.vo.Configuration;
 import io.dataease.extensions.datasource.vo.DatasourceConfiguration;
 import io.dataease.home.manage.DeIndexManage;
-import io.dataease.utils.AuthUtils;
-import io.dataease.utils.CommonBeanFactory;
-import io.dataease.utils.JsonUtil;
-import io.dataease.utils.LogUtil;
+import io.dataease.utils.*;
 import jakarta.annotation.Resource;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.collections4.MapUtils;
@@ -32,6 +29,7 @@ import org.apache.commons.lang3.StringUtils;
 import org.springframework.core.env.Environment;
 import org.springframework.stereotype.Component;
 
+import java.nio.charset.StandardCharsets;
 import java.util.*;
 import java.util.stream.Collectors;
 
@@ -50,6 +48,21 @@ public class DatasetSQLBotManage {
     @Resource
     private Environment environment;
 
+    private String aesKey = "y5txe1mRmS_JpOrUzFzHEu-kIQn3lf7l";
+    private String aesIv = "sqlbot_em_aes_iv";
+
+    private String aesEncrypt(String text) {
+        String iv = aesIv;
+        int len = iv.length();
+        if (len > 16) {
+            iv = iv.substring(0, 16);
+        }
+        if (len < 16) {
+            iv = String.format("%-" + (16 - len) + "s", iv).replace(' ', '0');
+        }
+        return AesUtils.aesEncrypt(text, aesKey, iv);
+    }
+
     private Map<Long, List<DataSetColumnPermissionsDTO>> getColPermission(Long uid, List<Long> roleIds) {
         ColumnPermissionsApi columnPermissionsApi = CommonBeanFactory.getBean(ColumnPermissionsApi.class);
         Objects.requireNonNull(columnPermissionsApi);
@@ -188,7 +201,7 @@ public class DatasetSQLBotManage {
             List<Map<String, Object>> list,
             Map<Long, List<DataSetColumnPermissionsDTO>> colPermissionMap,
             Map<Long, List<DataSetRowPermissionsTreeDTO>> rowPermissionMap
-        ) {
+    ) {
         if (CollectionUtils.isEmpty(vos)) {
             return vos;
         }
@@ -242,9 +255,28 @@ public class DatasetSQLBotManage {
         vo.setSchema(config.getSchema());
         vo.setUser(config.getUsername());
         vo.setPassword(config.getPassword());
+        aesVO(vo);
         return vo;
     }
 
+    private void aesVO(DataSQLBotAssistantVO vo) {
+        if (StringUtils.isNotBlank(vo.getHost())) {
+            vo.setHost(aesEncrypt(vo.getHost()));
+        }
+        if (ObjectUtils.isNotEmpty(vo.getUser())) {
+            vo.setUser(aesEncrypt(vo.getUser()));
+        }
+        if (ObjectUtils.isNotEmpty(vo.getPassword())) {
+            vo.setPassword(aesEncrypt(vo.getPassword()));
+        }
+        if (ObjectUtils.isNotEmpty(vo.getDataBase())) {
+            vo.setDataBase(aesEncrypt(vo.getDataBase()));
+        }
+        if (ObjectUtils.isNotEmpty(vo.getSchema())) {
+            vo.setSchema(aesEncrypt(vo.getSchema()));
+        }
+    }
+
     private SQLBotAssistanTable buildTable(Map<String, Object> row) {
         SQLBotAssistanTable table = new SQLBotAssistanTable();
         table.setName(row.get("table_name").toString());