From 866c69e70638ba22672520aecedad0d76fe3ca51 Mon Sep 17 00:00:00 2001
From: 疯狂的狮子li <15040126243@163.com>
Date: 星期日, 04 四月 2021 11:27:08 +0800
Subject: [PATCH] fix 代码生成器问题

---
 ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm     |    5 +++++
 ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm |    2 +-
 2 files changed, 6 insertions(+), 1 deletions(-)

diff --git a/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm b/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm
index 865ede4..b98b42f 100644
--- a/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm
+++ b/ruoyi-generator/src/main/resources/vm/java/queryBo.java.vm
@@ -12,6 +12,11 @@
 #foreach ($import in $importList)
 import ${import};
 #end
+#if($table.crud || $table.sub)
+import com.ruoyi.common.core.domain.BaseEntity;
+#elseif($table.tree)
+import com.ruoyi.common.core.domain.TreeEntity;
+#end
 
 /**
  * ${functionName}鍒嗛〉鏌ヨ瀵硅薄 ${tableName}
diff --git a/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm b/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm
index 8665705..cbf9d32 100644
--- a/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm
+++ b/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm
@@ -32,7 +32,7 @@
     @Override
     public ${ClassName}Vo queryById(${pkColumn.javaType} ${pkColumn.javaField}){
         ${ClassName} db = this.baseMapper.selectById(${pkColumn.javaField});
-        return BeanUtil.toBean(db, ${ClassName}Vo.class)
+        return BeanUtil.toBean(db, ${ClassName}Vo.class);
     }
 
     @Override

--
Gitblit v1.9.3