opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-01 08:52:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

导入链接增加userId

Browse Source
This commit is contained in:
lzy
2023-09-26 16:33:43 +08:00
parent 5db3f82a79
commit 15eec374c5
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ncx/service/impl/ConverterServiceImpl.java
View File

@ -4,6 +4,7 @@ import ai.chat2db.server.domain.core.util.DesUtil;
import ai.chat2db.server.domain.repository.entity.DataSourceDO;
import ai.chat2db.server.domain.repository.mapper.DataSourceMapper;
import ai.chat2db.server.tools.common.util.ConfigUtils;
import ai.chat2db.server.tools.common.util.ContextUtils;
import ai.chat2db.server.web.api.controller.ncx.cipher.CommonCipher;
import ai.chat2db.server.web.api.controller.ncx.dbeaver.DefaultValueEncryptor;
import ai.chat2db.server.web.api.controller.ncx.enums.DataBaseType;
@ -210,10 +211,16 @@ public class ConverterServiceImpl implements ConverterService {
Date dateTime = new Date();
dataSourceDO.setGmtCreate(dateTime);
dataSourceDO.setGmtModified(dateTime);
//插入用户id
dataSourceDO.setUserId(ContextUtils.getUserId());
dataSourceDO.setAlias(configurations.getString("name"));
dataSourceDO.setHost(configuration.getString("host"));
dataSourceDO.setPort(configuration.getString("port"));
dataSourceDO.setUrl(configuration.getString("url"));
//ssh设置为false
SSHInfo sshInfo = new SSHInfo();
sshInfo.setUse(false);
dataSourceDO.setSsh(JSON.toJSONString(sshInfo));
if (null != credentialsJson) {
JSONObject userInfo = credentialsJson.getJSONObject(key);
JSONObject userPassword = userInfo.getJSONObject(connection);
@ -291,6 +298,8 @@ public class ConverterServiceImpl implements ConverterService {
dataSourceDO.setGmtCreate(dateTime);
dataSourceDO.setGmtModified(dateTime);
dataSourceDO.setAlias(rootElement.getAttribute("name"));
//插入用户id
dataSourceDO.setUserId(ContextUtils.getUserId());
// 获取子元素 database-info
Element databaseInfoElement = (Element) rootElement.getElementsByTagName("database-info").item(0);
@ -319,6 +328,10 @@ public class ConverterServiceImpl implements ConverterService {
}
}
//ssh设置为false
SSHInfo sshInfo = new SSHInfo();
sshInfo.setUse(false);
dataSourceDO.setSsh(JSON.toJSONString(sshInfo));
dataSourceDO.setHost(host);
dataSourceDO.setPort(port);
dataSourceDO.setUrl(jdbcUrl);
@ -361,6 +374,8 @@ public class ConverterServiceImpl implements ConverterService {
dataSourceDO.setAlias(resultMap.get("ConnectionName"));
dataSourceDO.setUserName(resultMap.get("UserName"));
dataSourceDO.setType(resultMap.get("ConnType"));
//插入用户id
dataSourceDO.setUserId(ContextUtils.getUserId());
//password 为解密出来的密文，再使用chat2db的加密
DesUtil desUtil = new DesUtil(DesUtil.DES_KEY);
String encryptStr = desUtil.encrypt(password, "CBC");