From 08eca779084a82643e27aba15d64df9393dedf11 Mon Sep 17 00:00:00 2001
From: liulin <lin.liu@aliyun.com>
Date: 星期六, 03 八月 2024 22:36:38 +0800
Subject: [PATCH] mysql迁移数据库到gauss

---
 src/test/java/com/lunhan/xxx/host/GenCodeGauss.java |   87 +++++++++++++++++++++++++------------------
 1 files changed, 51 insertions(+), 36 deletions(-)

diff --git a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
index 589d63b..dc3f35a 100644
--- a/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
+++ b/src/test/java/com/lunhan/xxx/host/GenCodeGauss.java
@@ -377,8 +377,10 @@
         content.append(ConstantFactory.STR_NEWLINE);
         content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
         content.append(ConstantFactory.STR_NEWLINE);
-        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
-        content.append(ConstantFactory.STR_NEWLINE);
+        if (!isData) {
+            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+            content.append(ConstantFactory.STR_NEWLINE);
+        }
         content.append("}");
 
         writeFile(GenCodeGauss.OutSet.PO, modelName + GenCodeGauss.SuffixSet.PO + ".java", content.toString());
@@ -427,7 +429,6 @@
         content.append(ConstantFactory.STR_NEWLINE);
         if (isData) {
             content.append("@Data");
-            content.append(ConstantFactory.STR_NEWLINE);
         }
         content.append(ConstantFactory.STR_NEWLINE);
         content.append("public class ");
@@ -466,11 +467,17 @@
         String modelName = underline2Camel(tableName, true);
         String searchName = "Search" + modelName;
         String tableNamePackage = tableName.toLowerCase().replaceAll("_", "");
+        boolean isData = Objects.equals(SET_PO_Getter, 2);
 
         StringBuilder content = new StringBuilder();
         content.append("package ");
         content.append(GenCodeGauss.PackageSet.SEARCH + ConstantFactory.STR_SEMICOLON);
         content.append(ConstantFactory.STR_NEWLINE);
+        if (isData) {
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("import lombok.Data;");
+            content.append(ConstantFactory.STR_NEWLINE);
+        }
         if (SET_SEARCH_IMPORT.length > 0) {
             content.append(ConstantFactory.STR_NEWLINE);
             for (String s : SET_SEARCH_IMPORT) {
@@ -484,36 +491,42 @@
                 .replaceAll("\\{\\#\\=author\\}", AUTHOR)
                 .replaceAll("\\{\\#\\=date\\}", LocalDateTimeUtil.todayStr().replaceAll("\\-", "/"))
         );
+        if (isData) {
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("@Data");
+        }
         content.append(ConstantFactory.STR_NEWLINE);
         content.append("public class " + searchName + " extends SearchBasicDTO {");
         content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t@Override");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\tpublic String toString() {");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\"keywords='\" + getKeywords() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", listId =\" + getListId() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", status =\" + getStatus() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", listStatus =\" + getListStatus() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", lastId =\" + getListId() +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t\t'}';");
-        content.append(ConstantFactory.STR_NEWLINE);
-        content.append("\t}");
+        if (!isData) {
+            content.append("\t@Override");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\tpublic String toString() {");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\treturn \"" + searchName + "{\" + \"'\" +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\"keywords='\" + getKeywords() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", listId =\" + getListId() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", status =\" + getStatus() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", listStatus =\" + getListStatus() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", createTimeRange='\" + getCreateTimeRange() + \"'\" +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", createTimeStart =\" + getCreateTimeEnd() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", createTimeEnd =\" + getCreateTimeEnd() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", lastId =\" + getListId() +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t\t\", orderBy='\" + getOrderBy() + \"'\" +");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t\t'}';");
+            content.append(ConstantFactory.STR_NEWLINE);
+            content.append("\t}");
+        }
         content.append(ConstantFactory.STR_NEWLINE);
         content.append("}");
 
@@ -739,7 +752,6 @@
         content.append(ConstantFactory.STR_NEWLINE);
         if (isData) {
             content.append("@Data");
-            content.append(ConstantFactory.STR_NEWLINE);
         }
         content.append(ConstantFactory.STR_NEWLINE);
         content.append("public class ");
@@ -748,8 +760,10 @@
         content.append(ConstantFactory.STR_NEWLINE);
         content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
         content.append(ConstantFactory.STR_NEWLINE);
-        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
-        content.append(ConstantFactory.STR_NEWLINE);
+        if (!isData) {
+            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+            content.append(ConstantFactory.STR_NEWLINE);
+        }
         content.append("}");
 
         if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + createDTOName + ".java").exists()) {
@@ -831,7 +845,6 @@
         content.append(ConstantFactory.STR_NEWLINE);
         if (isData) {
             content.append("@Data");
-            content.append(ConstantFactory.STR_NEWLINE);
         }
         content.append(ConstantFactory.STR_NEWLINE);
         content.append("public class ");
@@ -840,8 +853,10 @@
         content.append(ConstantFactory.STR_NEWLINE);
         content.append(StringUtil.join(listColumn, ConstantFactory.STR_NEWLINE));
         content.append(ConstantFactory.STR_NEWLINE);
-        content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
-        content.append(ConstantFactory.STR_NEWLINE);
+        if (!isData) {
+            content.append(StringUtil.join(listGetterSetter, ConstantFactory.STR_NEWLINE));
+            content.append(ConstantFactory.STR_NEWLINE);
+        }
         content.append("}");
 
         if (new File(GenCodeGauss.OutSet.RequestDTO + tableNamePackage + "/" + modifyDTOName + ".java").exists()) {

--
Gitblit v1.9.3