opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-28 10:12:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

support for mongodb

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-11-26 16:14:15 +08:00
parent 5549908aaa
commit f1176af805
3 changed files with 124 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
chat2db-server/chat2db-plugins/chat2db-mongodb/src/main/java/ai/chat2db/plugin/mongodb/MongodbManage.java
View File

@ -5,17 +5,25 @@ import java.sql.SQLException;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.sql.Chat2DBContext;
import ai.chat2db.spi.sql.ConnectInfo;
import ai.chat2db.spi.sql.SQLExecutor;
import org.apache.commons.lang3.ObjectUtils;
import org.springframework.util.StringUtils;
public class MongodbManage extends DefaultDBManage implements DBManage {
@Override
public void connectDatabase(Connection connection, String database) {
if (StringUtils.isEmpty(database)) {
ConnectInfo connectInfo = Chat2DBContext.getConnectInfo();
if (ObjectUtils.anyNull(connectInfo) || StringUtils.isEmpty(connectInfo.getSchemaName())) {
return;
}
String schemaName = connectInfo.getSchemaName();
if (StringUtils.isEmpty(schemaName)) {
return;
}
try {
SQLExecutor.getInstance().execute(connection, "use " + database + ";");
SQLExecutor.getInstance().execute(connection, "use " + schemaName + ";");
} catch (SQLException e) {
throw new RuntimeException(e);
}