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

export function

Browse Source
This commit is contained in:
zgq
2024-03-09 14:15:40 +08:00
parent b9cefb2bd7
commit 8ed44a3724
1 changed files with 39 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java
View File

@ -15,25 +15,43 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
StringBuilder sqlBuilder = new StringBuilder(); StringBuilder sqlBuilder = new StringBuilder();
exportTables(connection, databaseName, sqlBuilder, containData); exportTables(connection, databaseName, sqlBuilder, containData);
exportViews(connection, databaseName, sqlBuilder); exportViews(connection, databaseName, sqlBuilder);
exportProcedures(connection, databaseName, sqlBuilder); exportProcedures(connection, sqlBuilder);
exportTriggers(connection, sqlBuilder); exportTriggers(connection, sqlBuilder);
exportFunctions(connection, databaseName, sqlBuilder);
return sqlBuilder.toString(); 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 { 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"})) { try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) { while (resultSet.next()) {
String tableName = resultSet.getString("TABLE_NAME"); exportTable(connection, resultSet.getString("TABLE_NAME"), sqlBuilder, containData);
exportTable(connection, tableName, sqlBuilder, containData);
} }
} }
} }
private void exportTable(Connection connection, String tableName, StringBuilder sqlBuilder, boolean containData) throws SQLException { private void exportTable(Connection connection, String tableName, StringBuilder sqlBuilder, boolean containData) throws SQLException {
String TABLE_DDL_SQL = "show create table %s "; String sql = String.format("show create table %s ", tableName);
String sql = String.format(TABLE_DDL_SQL, tableName); try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) {
if (resultSet.next()) { if (resultSet.next()) {
sqlBuilder.append("DROP TABLE IF EXISTS ").append(format(tableName)).append(";").append("\n") sqlBuilder.append("DROP TABLE IF EXISTS ").append(format(tableName)).append(";").append("\n")
.append(resultSet.getString("Create Table")).append(";").append("\n"); .append(resultSet.getString("Create Table")).append(";").append("\n");
@ -45,9 +63,8 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
} }
private void exportTableData(Connection connection, String tableName, StringBuilder sqlBuilder) throws SQLException { private void exportTableData(Connection connection, String tableName, StringBuilder sqlBuilder) throws SQLException {
String TABLE_QUERY_SQL = "select * from %s"; String sql = String.format("select * from %s", tableName);
String sql = String.format(TABLE_QUERY_SQL, tableName); try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) {
ResultSetMetaData metaData = resultSet.getMetaData(); ResultSetMetaData metaData = resultSet.getMetaData();
while (resultSet.next()) { while (resultSet.next()) {
sqlBuilder.append("INSERT INTO ").append(tableName).append(" VALUES ("); sqlBuilder.append("INSERT INTO ").append(tableName).append(" VALUES (");
@ -71,16 +88,14 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
private void exportViews(Connection connection, String databaseName, StringBuilder sqlBuilder) throws SQLException { private void exportViews(Connection connection, String databaseName, StringBuilder sqlBuilder) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"VIEW"})) { try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"VIEW"})) {
while (resultSet.next()) { while (resultSet.next()) {
String viewName = resultSet.getString("TABLE_NAME"); exportView(connection, resultSet.getString("TABLE_NAME"), sqlBuilder);
exportView(connection, viewName, sqlBuilder);
} }
} }
} }
private void exportView(Connection connection, String viewName, StringBuilder sqlBuilder) throws SQLException { private void exportView(Connection connection, String viewName, StringBuilder sqlBuilder) throws SQLException {
String VIEW_DDL_SQL = "show create view %s "; String sql = String.format("show create view %s ", viewName);
String sql = String.format(VIEW_DDL_SQL, viewName); try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) {
if (resultSet.next()) { if (resultSet.next()) {
sqlBuilder.append("DROP VIEW IF EXISTS ").append(format(viewName)).append(";").append("\n") sqlBuilder.append("DROP VIEW IF EXISTS ").append(format(viewName)).append(";").append("\n")
.append(resultSet.getString("Create View")).append(";").append("\n"); .append(resultSet.getString("Create View")).append(";").append("\n");
@ -88,19 +103,18 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
} }
} }
private void exportProcedures(Connection connection, String databaseName, StringBuilder sqlBuilder) throws SQLException { private void exportProcedures(Connection connection, StringBuilder sqlBuilder) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getProcedures(databaseName, null, null)) { String sql = "SHOW PROCEDURE STATUS WHERE Db = DATABASE()";
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) { while (resultSet.next()) {
String procedureName = resultSet.getString("PROCEDURE_NAME"); exportProcedure(connection, resultSet.getString("Name"), sqlBuilder);
exportProcedure(connection, procedureName, sqlBuilder);
} }
} }
} }
private void exportProcedure(Connection connection, String procedureName, StringBuilder sqlBuilder) throws SQLException { private void exportProcedure(Connection connection, String procedureName, StringBuilder sqlBuilder) throws SQLException {
String PROCEDURE_DDL_SQL = "show create procedure %s "; String sql = String.format("show create procedure %s ", procedureName);
String sql = String.format(PROCEDURE_DDL_SQL, procedureName); try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) {
if (resultSet.next()) { if (resultSet.next()) {
sqlBuilder.append("DROP PROCEDURE IF EXISTS ").append(format(procedureName)).append(";").append("\n") sqlBuilder.append("DROP PROCEDURE IF EXISTS ").append(format(procedureName)).append(";").append("\n")
.append("delimiter ;;").append("\n").append(resultSet.getString("Create Procedure")).append(";;") .append("delimiter ;;").append("\n").append(resultSet.getString("Create Procedure")).append(";;")
@ -110,8 +124,8 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
} }
private void exportTriggers(Connection connection, StringBuilder sqlBuilder) throws SQLException { private void exportTriggers(Connection connection, StringBuilder sqlBuilder) throws SQLException {
String sql ="SHOW TRIGGERS"; String sql = "SHOW TRIGGERS";
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) { try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
while (resultSet.next()) { while (resultSet.next()) {
String triggerName = resultSet.getString("Trigger"); String triggerName = resultSet.getString("Trigger");
exportTrigger(connection, triggerName, sqlBuilder); exportTrigger(connection, triggerName, sqlBuilder);
@ -120,9 +134,8 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
} }
private void exportTrigger(Connection connection, String triggerName, StringBuilder sqlBuilder) throws SQLException { private void exportTrigger(Connection connection, String triggerName, StringBuilder sqlBuilder) throws SQLException {
String TRIGGER_DDL_SQL = "show create trigger %s "; String sql = String.format("show create trigger %s ", triggerName);
String sql = String.format(TRIGGER_DDL_SQL, triggerName); try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
try (Statement statement = connection.createStatement(); ResultSet resultSet = statement.executeQuery(sql)) {
if (resultSet.next()) { if (resultSet.next()) {
sqlBuilder.append("DROP TRIGGER IF EXISTS ").append(format(triggerName)).append(";").append("\n") sqlBuilder.append("DROP TRIGGER IF EXISTS ").append(format(triggerName)).append(";").append("\n")
.append("delimiter ;;").append("\n").append(resultSet.getString("SQL Original Statement")).append(";;") .append("delimiter ;;").append("\n").append(resultSet.getString("SQL Original Statement")).append(";;")