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

support for mongodb

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-11-26 14:57:24 +08:00
parent 536ad6f53f
commit 5549908aaa
9 changed files with 267 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,8 +1,30 @@
package ai.chat2db.plugin.mongodb;
import java.sql.Connection;
import java.sql.SQLException;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.sql.SQLExecutor;
import org.springframework.util.StringUtils;
public class MongodbManage extends DefaultDBManage implements DBManage {
@Override
public void connectDatabase(Connection connection, String database) {
if (StringUtils.isEmpty(database)) {
return;
}
try {
SQLExecutor.getInstance().execute(connection, "use " + database + ";");
} catch (SQLException e) {
throw new RuntimeException(e);
}
}
@Override
public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
String sql = " db. " + tableName + ".drop();";
SQLExecutor.getInstance().executeSql(connection, sql, resultSet -> null);
}
}