From d0cc511bbcaf12a5e0a462a5a6484b27c294bcb5 Mon Sep 17 00:00:00 2001
From: 雪忆天堂 <1455668754@qq.com>
Date: Tue, 25 May 2021 16:27:41 +0800
Subject: [PATCH] 代码生成优化
---
ruoyi-modules/ruoyi-gen/src/main/resources/vm/java/serviceImpl.java.vm | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 58 insertions(+), 1 deletions(-)
diff --git a/ruoyi-modules/ruoyi-gen/src/main/resources/vm/java/serviceImpl.java.vm b/ruoyi-modules/ruoyi-gen/src/main/resources/vm/java/serviceImpl.java.vm
index 9e70bd2..2e928e0 100644
--- a/ruoyi-modules/ruoyi-gen/src/main/resources/vm/java/serviceImpl.java.vm
+++ b/ruoyi-modules/ruoyi-gen/src/main/resources/vm/java/serviceImpl.java.vm
@@ -3,12 +3,18 @@
import java.util.List;
#foreach ($column in $columns)
#if($column.javaField == 'createTime' || $column.javaField == 'updateTime')
-import com.ruoyi.common.core.utils.DateUtils
+import com.ruoyi.common.core.utils.DateUtils;
#break
#end
#end
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
+#if($table.sub)
+import java.util.ArrayList;
+import com.ruoyi.common.core.utils.StringUtils;
+import org.springframework.transaction.annotation.Transactional;
+import ${packageName}.domain.${subClassName};
+#end
import ${packageName}.mapper.${ClassName}Mapper;
import ${packageName}.domain.${ClassName};
import ${packageName}.service.I${ClassName}Service;
@@ -55,6 +61,9 @@
* @param ${className} ${functionName}
* @return 结果
*/
+#if($table.sub)
+ @Transactional
+#end
@Override
public int insert${ClassName}(${ClassName} ${className})
{
@@ -63,7 +72,13 @@
${className}.setCreateTime(DateUtils.getNowDate());
#end
#end
+#if($table.sub)
+ int rows = ${className}Mapper.insert${ClassName}(${className});
+ insert${subClassName}(${className});
+ return rows;
+#else
return ${className}Mapper.insert${ClassName}(${className});
+#end
}
/**
@@ -72,6 +87,9 @@
* @param ${className} ${functionName}
* @return 结果
*/
+#if($table.sub)
+ @Transactional
+#end
@Override
public int update${ClassName}(${ClassName} ${className})
{
@@ -79,6 +97,10 @@
#if($column.javaField == 'updateTime')
${className}.setUpdateTime(DateUtils.getNowDate());
#end
+#end
+#if($table.sub)
+ ${className}Mapper.delete${subClassName}By${subTableFkClassName}(${className}.get${pkColumn.capJavaField}());
+ insert${subClassName}(${className});
#end
return ${className}Mapper.update${ClassName}(${className});
}
@@ -89,9 +111,15 @@
* @param ${pkColumn.javaField}s 需要删除的${functionName}ID
* @return 结果
*/
+#if($table.sub)
+ @Transactional
+#end
@Override
public int delete${ClassName}ByIds(${pkColumn.javaType}[] ${pkColumn.javaField}s)
{
+#if($table.sub)
+ ${className}Mapper.delete${subClassName}By${subTableFkClassName}s(${pkColumn.javaField}s);
+#end
return ${className}Mapper.delete${ClassName}ByIds(${pkColumn.javaField}s);
}
@@ -104,6 +132,35 @@
@Override
public int delete${ClassName}ById(${pkColumn.javaType} ${pkColumn.javaField})
{
+#if($table.sub)
+ ${className}Mapper.delete${subClassName}By${subTableFkClassName}(${pkColumn.javaField});
+#end
return ${className}Mapper.delete${ClassName}ById(${pkColumn.javaField});
}
+#if($table.sub)
+
+ /**
+ * 新增${subTable.functionName}信息
+ *
+ * @param ${className} ${functionName}对象
+ */
+ public void insert${subClassName}(${ClassName} ${className})
+ {
+ List<${subClassName}> ${subclassName}List = ${className}.get${subClassName}List();
+ Long ${pkColumn.javaField} = ${className}.get${pkColumn.capJavaField}();
+ if (StringUtils.isNotNull(${subclassName}List))
+ {
+ List<${subClassName}> list = new ArrayList<${subClassName}>();
+ for (${subClassName} ${subclassName} : ${subclassName}List)
+ {
+ ${subclassName}.set${subTableFkClassName}(${pkColumn.javaField});
+ list.add(${subclassName});
+ }
+ if (list.size() > 0)
+ {
+ ${className}Mapper.batch${subClassName}(list);
+ }
+ }
+ }
+#end
}
--
Gitblit v1.9.3