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