From 5bb7b0dda5a29f6aa2326c40de24a0f4b3e5740c Mon Sep 17 00:00:00 2001 From: 疯狂的狮子li <15040126243@163.com> Date: 星期四, 09 九月 2021 15:24:33 +0800 Subject: [PATCH] update 优化代码生成 insertByBo 方法 针对返回主键优化 --- ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm | 7 ++++++- 1 files changed, 6 insertions(+), 1 deletions(-) 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 32a9ac0..b9a45c6 100644 --- a/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm +++ b/ruoyi-generator/src/main/resources/vm/java/serviceImpl.java.vm @@ -79,7 +79,12 @@ public Boolean insertByBo(${ClassName}Bo bo) { ${ClassName} add = BeanUtil.toBean(bo, ${ClassName}.class); 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 -- Gitblit v1.9.3