From fbc3f392552c14074e1295e4c6e74c5cb87ff8b5 Mon Sep 17 00:00:00 2001 From: liulin <lin.liu@aliyun.com> Date: 星期六, 03 八月 2024 14:09:24 +0800 Subject: [PATCH] mysql迁移数据库到gauss --- src/test/java/com/lunhan/xxx/host/GenCodeMysql.java | 24 ++++++++++++------------ 1 files changed, 12 insertions(+), 12 deletions(-) diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java b/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java index 9fd7533..472669c 100644 --- a/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java +++ b/src/test/java/com/lunhan/xxx/host/GenCodeMysql.java @@ -257,7 +257,7 @@ System.out.println("conn is null."); return; } - Map<String, String> tables = getTables(DB_NAME); + Map<String, String> tables = getTables(CONN, DB_NAME); for (Map.Entry<String, String> table : tables.entrySet()) { String tableName = table.getKey(); String tableDesc = table.getValue(); @@ -265,7 +265,7 @@ if (ONLY_TABLES.size() > 0 && BooleanUtils.isFalse(ONLY_TABLES.contains(tableName))) { continue; } - List<Map<String, Object>> columns = getColumns(tableName, DB_NAME); + List<Map<String, Object>> columns = getColumns(CONN, tableName, DB_NAME); genPO(columns, tableDesc); genVO(columns, tableDesc); genColumnsEnum(columns, tableDesc); @@ -1783,12 +1783,12 @@ //#endregion - public static Map<String, String> getTables(String dbSchema) { + public static Map<String, String> getTables(Connection connect, String dbSchema) { Map<String, String> result = new LinkedHashMap<>(); String sql = " SELECT TABLE_NAME ,TABLE_COMMENT " + " FROM information_schema.`TABLES` " + " WHERE TABLE_SCHEMA = ? "; - try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (PreparedStatement pStmt = connect.prepareStatement(sql)) { pStmt.setString(1, dbSchema); try (ResultSet rs = pStmt.executeQuery()) { while (rs.next()) { @@ -1803,7 +1803,7 @@ return result; } - public static List<Map<String, Object>> getColumns(String tableName, String dbSchema) { + public static List<Map<String, Object>> getColumns(Connection connect, String tableName, String dbSchema) { List<Map<String, Object>> result = new ArrayList<>(); String sql = "SELECT `TABLE_NAME` AS'objName', `COLUMN_NAME`AS'name', `DATA_TYPE` AS'type', `CHARACTER_MAXIMUM_LENGTH` AS'length', `COLUMN_COMMENT` AS'description',\n" + "`COLUMN_DEFAULT` AS'default',NUMERIC_PRECISION AS'numeric_precision',NUMERIC_SCALE AS'numeric_scale',DATETIME_PRECISION AS'datetime_precision',\n" + @@ -1812,7 +1812,7 @@ "CASE `IS_NULLABLE` WHEN 'YES' THEN CONVERT(1,UNSIGNED) ELSE CONVERT(0,UNSIGNED) END AS'isNullable'\n" + "FROM `information_schema`.`COLUMNS`\n" + "WHERE `table_name` = ? AND `table_schema` = ?; "; - try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (PreparedStatement pStmt = connect.prepareStatement(sql)) { pStmt.setString(1, tableName); pStmt.setString(2, dbSchema); try (ResultSet rs = pStmt.executeQuery()) { @@ -1836,9 +1836,9 @@ return result; } - public static List<Map<String, Object>> getListAll(String sql) { + public static List<Map<String, Object>> getListAll(Connection connect, String sql) { List<Map<String, Object>> result = new ArrayList<>(); - try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (PreparedStatement pStmt = connect.prepareStatement(sql)) { try (ResultSet rs = pStmt.executeQuery()) { //获取键名 ResultSetMetaData md = rs.getMetaData(); @@ -1958,10 +1958,10 @@ } } - public static Map<String, String> getPK(String tableName) { + public static Map<String, String> getPK(Connection connect, String tableName) { Map<String, String> result = new LinkedHashMap<>(); String sql = "SHOW INDEX FROM ? WHERE Key_name='PRIMARY'"; - try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (PreparedStatement pStmt = connect.prepareStatement(sql)) { pStmt.setString(1, tableName); try (ResultSet rs = pStmt.executeQuery()) { while (rs.next()) { @@ -1976,10 +1976,10 @@ return result; } - public static Map<String, String> getIndex(String tableName) { + public static Map<String, String> getIndex(Connection connect, String tableName) { Map<String, String> result = new LinkedHashMap<>(); String sql = "SHOW INDEX FROM ? WHERE Key_name!='PRIMARY'"; - try (PreparedStatement pStmt = CONN.prepareStatement(sql)) { + try (PreparedStatement pStmt = connect.prepareStatement(sql)) { pStmt.setString(1, tableName); try (ResultSet rs = pStmt.executeQuery()) { while (rs.next()) { -- Gitblit v1.9.3