From 01f4e7f1ac47968bab5a5c4aec7595ecfc2d4df5 Mon Sep 17 00:00:00 2001
From: liulin <lin.liu@aliyun.com>
Date: 星期一, 12 八月 2024 23:28:17 +0800
Subject: [PATCH] gauss导入导出
---
src/test/java/com/lunhan/xxx/host/GenCodeGauss.java | 11 ++++++++---
src/test/java/com/lunhan/xxx/host/DataBaseExportTest.java | 28 ++++++++++++++++++++++++++++
2 files changed, 36 insertions(+), 3 deletions(-)
diff --git a/src/test/java/com/lunhan/xxx/host/DataBaseExportTest.java b/src/test/java/com/lunhan/xxx/host/DataBaseExportTest.java
new file mode 100644
index 0000000..ce3d57f
--- /dev/null
+++ b/src/test/java/com/lunhan/xxx/host/DataBaseExportTest.java
@@ -0,0 +1,28 @@
+package com.lunhan.xxx.host;
+
+import com.lunhan.xxx.common.util.StringUtil;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+import java.util.List;
+
+public class DataBaseExportTest {
+ public static void main(String[] args) {
+ String dbName = "water_trade_center";
+
+ Connection CONN;
+ //获得连接
+ try {
+ CONN = DriverManager.getConnection("jdbc:postgresql://113.250.189.120:57654/" + dbName, "lunhan", "lunhan.20240330");
+ } catch (SQLException e) {
+ System.out.println(e.toString());
+ return;
+ }
+
+ List<String> listSql = GenCodeGauss.export(CONN, dbName);
+ String sql = StringUtil.join(listSql, "\n\n");
+
+ System.out.println(sql);
+ }
+}
diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
index 89ff2a8..97cbbb3 100644
--- a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
+++ b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
@@ -1602,7 +1602,7 @@
Map<String, String> mapColumnDesc = new HashMap<>();
int idx = 0;
- list.add("-- \"" + tableName + "\" - \"" + tableDesc + "\"\nDROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n");
+ list.add("-- " + tableName + " - " + tableDesc + "\nDROP TABLE IF EXISTS \"public\".\"" + tableName + "\";\n");
// 遍历列，构建sql
for (Map<String, Object> column : listColumn) {
@@ -1619,8 +1619,10 @@
case "char":
case "varchar":
case "nvarchar":
- String length = column.get("length").toString();
- sql += "(" + length + ")";
+ if (Objects.nonNull(column.get("length"))) {
+ String length = column.get("length").toString();
+ sql += "(" + length + ")";
+ }
if ("1".equals(isNullable)) {
sql += " NULL DEFAULT NULL";
} else {
@@ -1956,4 +1958,7 @@
}
return listTable;
}
+ public static List<String> export(Connection connect, String dbSchema) {
+ return GenCodeGauss.export(connect, dbSchema, Boolean.TRUE, Boolean.TRUE);
+ }
}
--
Gitblit v1.9.3