Merge pull request CodePhiliaX#1241 from openai0229/sqlite-database-e…

…xport Sqlite database export
Murant7 · Mar 22, 2024 · 7a05bd0 · 7a05bd0
2 parents b659b69 + c85616c
commit 7a05bd0
Show file tree

Hide file tree

Showing 2 changed files with 106 additions and 3 deletions.
diff --git a/...chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteDBManage.java b/...chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteDBManage.java
@@ -3,5 +3,108 @@
 import ai.chat2db.spi.DBManage;
 import ai.chat2db.spi.jdbc.DefaultDBManage;
 
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.ResultSetMetaData;
+import java.sql.SQLException;
+import java.util.Objects;
+
 public class SqliteDBManage extends DefaultDBManage implements DBManage {
+
+
+    @Override
+    public String exportDatabase(Connection connection, String databaseName, String schemaName, boolean containData) throws SQLException {
+        StringBuilder sqlBuilder = new StringBuilder();
+        exportTables(connection, databaseName, sqlBuilder, containData);
+        exportViews(connection, databaseName, sqlBuilder);
+        exportTriggers(connection, sqlBuilder);
+        return sqlBuilder.toString();
+    }
+
+    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()) {
+                exportTable(connection, resultSet.getString("TABLE_NAME"), sqlBuilder, containData);
+            }
+        }
+    }
+
+
+    private void exportTable(Connection connection, String tableName, StringBuilder sqlBuilder, boolean containData) throws SQLException {
+        String sql = String.format("SELECT sql FROM sqlite_master WHERE type='table' AND name='%s'", tableName);
+        try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
+            if (resultSet.next()) {
+                sqlBuilder.append("DROP TABLE IF EXISTS ").append(format(tableName)).append(";").append("\n")
+                        .append(resultSet.getString("sql")).append(";").append("\n");
+                if (containData) {
+                    exportTableData(connection, tableName, sqlBuilder);
+                }
+            }
+        }
+    }
+
+    private String format(String tableName) {
+        return "\""+tableName+"\"";
+    }
+
+
+    private void exportTableData(Connection connection, String tableName, StringBuilder sqlBuilder) throws SQLException {
+        String sql = String.format("select * from %s", tableName);
+        try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
+            ResultSetMetaData metaData = resultSet.getMetaData();
+            while (resultSet.next()) {
+                sqlBuilder.append("INSERT INTO ").append(tableName).append(" VALUES (");
+                for (int i = 1; i <= metaData.getColumnCount(); i++) {
+                    String value = resultSet.getString(i);
+                    if (Objects.isNull(value)) {
+                        sqlBuilder.append("NULL");
+                    } else {
+                        sqlBuilder.append("'").append(value).append("'");
+                    }
+                    if (i < metaData.getColumnCount()) {
+                        sqlBuilder.append(", ");
+                    }
+                }
+                sqlBuilder.append(");\n");
+            }
+            sqlBuilder.append("\n");
+        }
+    }
+
+    private void exportViews(Connection connection, String databaseName, StringBuilder sqlBuilder) throws SQLException {
+        try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"VIEW"})) {
+            while (resultSet.next()) {
+                exportView(connection, resultSet.getString("TABLE_NAME"), sqlBuilder);
+            }
+        }
+    }
+
+    private void exportView(Connection connection, String viewName, StringBuilder sqlBuilder) throws SQLException {
+        String sql = String.format("SELECT * FROM sqlite_master WHERE type = 'view' and name='%s';", viewName);
+        try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
+            if (resultSet.next()) {
+                sqlBuilder.append("DROP VIEW IF EXISTS ").append(format(viewName)).append(";").append("\n")
+                        .append(resultSet.getString("sql")).append(";").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)) {
+            while (resultSet.next()) {
+                String triggerName = resultSet.getString("name");
+                exportTrigger(connection, triggerName, sqlBuilder);
+            }
+        }
+    }
+
+    private void exportTrigger(Connection connection, String triggerName, StringBuilder sqlBuilder) throws SQLException {
+        String sql = String.format("SELECT * FROM sqlite_master WHERE type = 'trigger' and name='%s';", triggerName);
+        try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
+            if (resultSet.next()) {
+                sqlBuilder.append(resultSet.getString("sql")).append("\n");
+            }
+        }
+    }
 }
diff --git a/...chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteMetaData.java b/...chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteMetaData.java
@@ -21,7 +21,7 @@
 import java.util.stream.Collectors;
 
 public class SqliteMetaData extends DefaultMetaService implements MetaData {
-    public static  String  VIEW_DDL_SQL="SELECT * FROM sqlite_master WHERE type = 'view' and name='%s';";
+    private static  String  VIEW_DDL_SQL="SELECT * FROM sqlite_master WHERE type = 'view' and name='%s';";
     @Override
     public Table view(Connection connection, String databaseName, String schemaName, String viewName) {
         Table view = new Table();
@@ -35,8 +35,8 @@ public Table view(Connection connection, String databaseName, String schemaName,
         return view;
     }
 
-    public static final String TRIGGER_LIST_SQL = "SELECT * FROM sqlite_master WHERE type = 'trigger';";
-    public static  String TRIGGER_DDL_SQL = "SELECT * FROM sqlite_master WHERE type = 'trigger' and name='%s';";
+    private static final String TRIGGER_LIST_SQL = "SELECT * FROM sqlite_master WHERE type = 'trigger';";
+    private static  String TRIGGER_DDL_SQL = "SELECT * FROM sqlite_master WHERE type = 'trigger' and name='%s';";
 
     @Override
     public List<Trigger> triggers(Connection connection, String databaseName, String schemaName) {