admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update 优化代码生成 insertByBo 方法 针对返回主键优化

Browse Source
This commit is contained in:
疯狂的狮子li 2021-09-09 15:24:33 +08:00
parent 61edd5588e
commit 5bb7b0dda5
3 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ruoyi-demo/src/main/java/com/ruoyi/demo/service/impl/TestDemoServiceImpl.java
View File

@ -75,7 +75,11 @@ public class TestDemoServiceImpl extends ServicePlusImpl<TestDemoMapper, TestDem
public Boolean insertByBo(TestDemoBo bo) { public Boolean insertByBo(TestDemoBo bo) {
TestDemo add = BeanUtil.toBean(bo, TestDemo.class); TestDemo add = BeanUtil.toBean(bo, TestDemo.class);
validEntityBeforeSave(add); validEntityBeforeSave(add);
return save(add); boolean flag = save(add);
if (flag) {
bo.setId(add.getId());
}
return flag;
} }
@Override @Override

6
ruoyi-demo/src/main/java/com/ruoyi/demo/service/impl/TestTreeServiceImpl.java
View File

@ -54,7 +54,11 @@ public class TestTreeServiceImpl extends ServicePlusImpl<TestTreeMapper, TestTre
public Boolean insertByBo(TestTreeBo bo) { public Boolean insertByBo(TestTreeBo bo) {
TestTree add = BeanUtil.toBean(bo, TestTree.class); TestTree add = BeanUtil.toBean(bo, TestTree.class);
validEntityBeforeSave(add); validEntityBeforeSave(add);
return save(add); boolean flag = save(add);
if (flag) {
bo.setId(add.getId());
}
return flag;
} }
@Override @Override

7
ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm
View File

@ -79,7 +79,12 @@ public class ${ClassName}ServiceImpl extends ServicePlusImpl<${ClassName}Mapper,
public Boolean insertByBo(${ClassName}Bo bo) { public Boolean insertByBo(${ClassName}Bo bo) {
${ClassName} add = BeanUtil.toBean(bo, ${ClassName}.class); ${ClassName} add = BeanUtil.toBean(bo, ${ClassName}.class);
validEntityBeforeSave(add); validEntityBeforeSave(add);
return save(add); boolean flag = save(add);
#set($pk=$pkColumn.javaField.substring(0,1).toUpperCase() + ${pkColumn.javaField.substring(1)})
if (flag) {
bo.set$pk(add.get$pk());
}
return flag;
} }
@Override @Override