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

Support for creating database and schema

Browse Source
This commit is contained in:
SwallowGG
2023-10-24 23:13:18 +08:00
parent a68ea74783
commit 300ec51b5a
5 changed files with 38 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLDBManage.java
View File

@ -4,17 +4,22 @@ import java.sql.Connection;
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.StringUtils;
public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
@Override
public void connectDatabase(Connection connection, String database) {
//try {
// SQLExecutor.getInstance().execute(connection,"SELECT pg_database_size('"+database+"');");
//} catch (SQLException e) {
// throw new RuntimeException(e);
//}
try {
ConnectInfo connectInfo = Chat2DBContext.getConnectInfo();
if (!StringUtils.isEmpty(connectInfo.getSchemaName())) {
SQLExecutor.getInstance().execute(connection, "SET search_path TO \"" + connectInfo.getSchemaName() + "\"");
}
} catch (Exception e) {
}
}
@Override
@ -26,7 +31,7 @@ public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
}
connectInfo.setUrl(url);
return super.getConnection(connectInfo);
return super.getConnection(connectInfo);
}
@ -53,8 +58,8 @@ public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
@Override
public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
String sql = "DROP TABLE "+ tableName;
SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
String sql = "DROP TABLE " + tableName;
SQLExecutor.getInstance().executeSql(connection, sql, resultSet -> null);
}
}