From 798ffd9065d72d76bb48a69ae91e81a19be8f22e Mon Sep 17 00:00:00 2001
From: 朱杰 <693337446@qq.com>
Date: 星期六, 22 五月 2021 23:29:15 +0800
Subject: [PATCH] fix 修复代码生成器中表字段取消必填无法更新问题

---
 ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java |   20 ++++++++++++++++++++
 1 files changed, 20 insertions(+), 0 deletions(-)

diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java
index 81d0d47..69421cc 100644
--- a/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java
+++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/util/VelocityUtils.java
@@ -129,6 +129,10 @@
     {
         List<String> templates = new ArrayList<String>();
         templates.add("vm/java/domain.java.vm");
+        templates.add("vm/java/vo.java.vm");
+        templates.add("vm/java/queryBo.java.vm");
+        templates.add("vm/java/addBo.java.vm");
+        templates.add("vm/java/editBo.java.vm");
         templates.add("vm/java/mapper.java.vm");
         templates.add("vm/java/service.java.vm");
         templates.add("vm/java/serviceImpl.java.vm");
@@ -176,6 +180,22 @@
         {
             fileName = StrUtil.format("{}/domain/{}.java", javaPath, className);
         }
+        if (template.contains("vo.java.vm"))
+        {
+            fileName = StrUtil.format("{}/vo/{}Vo.java", javaPath, className);
+        }
+        if (template.contains("queryBo.java.vm"))
+        {
+            fileName = StrUtil.format("{}/bo/{}QueryBo.java", javaPath, className);
+        }
+        if (template.contains("addBo.java.vm"))
+        {
+            fileName = StrUtil.format("{}/bo/{}AddBo.java", javaPath, className);
+        }
+        if (template.contains("editBo.java.vm"))
+        {
+            fileName = StrUtil.format("{}/bo/{}EditBo.java", javaPath, className);
+        }
         if (template.contains("sub-domain.java.vm") && StrUtil.equals(GenConstants.TPL_SUB, genTable.getTplCategory()))
         {
             fileName = StrUtil.format("{}/domain/{}.java", javaPath, genTable.getSubTable().getClassName());

--
Gitblit v1.9.3