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

sqlite-database-export

Browse Source
This commit is contained in:
zgq
2024-03-21 13:27:40 +08:00
parent 8bfdbc344c
commit c85616c455
1 changed files with 0 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteDBManage.java
View File

@ -17,31 +17,10 @@ public class SqliteDBManage extends DefaultDBManage implements DBManage {
StringBuilder sqlBuilder = new StringBuilder();
exportTables(connection, databaseName, sqlBuilder, containData);
exportViews(connection, databaseName, sqlBuilder);
// exportProcedures(connection, sqlBuilder);
exportTriggers(connection, sqlBuilder);
// exportFunctions(connection, databaseName, sqlBuilder);
return sqlBuilder.toString();
}
private void exportFunctions(Connection connection, String databaseName, StringBuilder sqlBuilder) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getFunctions(databaseName, null, null)) {
while (resultSet.next()) {
exportFunction(connection, resultSet.getString("FUNCTION_NAME"), sqlBuilder);
}
}
}
private void exportFunction(Connection connection, String functionName, StringBuilder sqlBuilder) throws SQLException {
String sql = String.format("SHOW CREATE FUNCTION %s;", functionName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
sqlBuilder.append("DROP FUNCTION IF EXISTS ").append(functionName).append(";").append("\n")
.append(resultSet.getString("Create Function")).append(";").append("\n");
}
}
}
private void exportTables(Connection connection, String databaseName, StringBuilder sqlBuilder, boolean containData) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) {
@ -110,26 +89,6 @@ public class SqliteDBManage extends DefaultDBManage implements DBManage {
}
}
private void exportProcedures(Connection connection, StringBuilder sqlBuilder) throws SQLException {
String sql = "SHOW PROCEDURE STATUS WHERE Db = DATABASE()";
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) {
exportProcedure(connection, resultSet.getString("Name"), sqlBuilder);
}
}
}
private void exportProcedure(Connection connection, String procedureName, StringBuilder sqlBuilder) throws SQLException {
String sql = String.format("show create procedure %s ", procedureName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
sqlBuilder.append("DROP PROCEDURE IF EXISTS ").append(format(procedureName)).append(";").append("\n")
.append("delimiter ;;").append("\n").append(resultSet.getString("Create Procedure")).append(";;")
.append("\n").append("delimiter ;").append("\n");
}
}
}
private void exportTriggers(Connection connection, StringBuilder sqlBuilder) throws SQLException {
String sql = "SELECT * FROM sqlite_master WHERE type = 'trigger';";
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {