From cfbbc7264eb4745f3de364b86c43390276dc845d Mon Sep 17 00:00:00 2001
From: RuoYi <yzz_ivy@163.com>
Date: 星期日, 13 二月 2022 21:03:49 +0800
Subject: [PATCH] 代码生成同步保留必填/类型选项

---
 src/main/java/com/ruoyi/common/constant/GenConstants.java                 |    3 +++
 src/main/java/com/ruoyi/project/tool/gen/util/GenUtils.java               |    1 +
 src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java |   11 ++++++++++-
 3 files changed, 14 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/ruoyi/common/constant/GenConstants.java b/src/main/java/com/ruoyi/common/constant/GenConstants.java
index 1d5d884..e7382b3 100644
--- a/src/main/java/com/ruoyi/common/constant/GenConstants.java
+++ b/src/main/java/com/ruoyi/common/constant/GenConstants.java
@@ -109,6 +109,9 @@
     /** 妯$硦鏌ヨ */
     public static final String QUERY_LIKE = "LIKE";
 
+    /** 鐩哥瓑鏌ヨ */
+    public static final String QUERY_EQ = "EQ";
+
     /** 闇�瑕� */
     public static final String REQUIRE = "1";
 }
diff --git a/src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java b/src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java
index c2200a1..3fb0c95 100644
--- a/src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java
+++ b/src/main/java/com/ruoyi/project/tool/gen/service/GenTableServiceImpl.java
@@ -304,8 +304,17 @@
                 column.setColumnId(prevColumn.getColumnId());
                 if (column.isList())
                 {
-                    // 濡傛灉鏄垪琛紝缁х画淇濈暀瀛楀吀绫诲瀷
+                    // 濡傛灉鏄垪琛紝缁х画淇濈暀鏌ヨ鏂瑰紡/瀛楀吀绫诲瀷閫夐」
                     column.setDictType(prevColumn.getDictType());
+                    column.setQueryType(prevColumn.getQueryType());
+                }
+                if (StringUtils.isNotEmpty(prevColumn.getIsRequired()) && !column.isPk()
+                        && (column.isInsert() || column.isEdit())
+                        && ((column.isUsableColumn()) || (!column.isSuperColumn())))
+                {
+                    // 濡傛灉鏄�(鏂板/淇敼&闈炰富閿�/闈炲拷鐣ュ強鐖跺睘鎬�)锛岀户缁繚鐣欏繀濉�/鏄剧ず绫诲瀷閫夐」
+                    column.setIsRequired(prevColumn.getIsRequired());
+                    column.setHtmlType(prevColumn.getHtmlType());
                 }
                 genTableColumnMapper.updateGenTableColumn(column);
             }
diff --git a/src/main/java/com/ruoyi/project/tool/gen/util/GenUtils.java b/src/main/java/com/ruoyi/project/tool/gen/util/GenUtils.java
index ed92f85..b5065aa 100644
--- a/src/main/java/com/ruoyi/project/tool/gen/util/GenUtils.java
+++ b/src/main/java/com/ruoyi/project/tool/gen/util/GenUtils.java
@@ -42,6 +42,7 @@
         column.setJavaField(StringUtils.toCamelCase(columnName));
         // 璁剧疆榛樿绫诲瀷
         column.setJavaType(GenConstants.TYPE_STRING);
+        column.setQueryType(GenConstants.QUERY_EQ);
 
         if (arraysContains(GenConstants.COLUMNTYPE_STR, dataType) || arraysContains(GenConstants.COLUMNTYPE_TEXT, dataType))
         {

--
Gitblit v1.9.3