From 2f624ab5f41f15a4a1093867e4519b7ddc7ccedf Mon Sep 17 00:00:00 2001 From: RuoYi <yzz_ivy@163.com> Date: 星期五, 01 三月 2024 14:33:09 +0800 Subject: [PATCH] 代码生成新增创建表结构功能 --- ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java | 22 ++++++++++++++++------ 1 files changed, 16 insertions(+), 6 deletions(-) diff --git a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java index 1620995..18231eb 100644 --- a/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java +++ b/ruoyi-generator/src/main/java/com/ruoyi/generator/service/GenTableServiceImpl.java @@ -27,7 +27,6 @@ import com.ruoyi.common.constant.GenConstants; import com.ruoyi.common.core.text.CharsetKit; import com.ruoyi.common.exception.ServiceException; -import com.ruoyi.common.utils.SecurityUtils; import com.ruoyi.common.utils.StringUtils; import com.ruoyi.generator.domain.GenTable; import com.ruoyi.generator.domain.GenTableColumn; @@ -151,15 +150,26 @@ } /** + * 创建表 + * + * @param sql 创建表语句 + * @return 结果 + */ + @Override + public boolean createTable(String sql) + { + return genTableMapper.createTable(sql) == 0; + } + + /** * 导入表结构 * * @param tableList 导入表列表 */ @Override @Transactional - public void importGenTable(List<GenTable> tableList) + public void importGenTable(List<GenTable> tableList, String operName) { - String operName = SecurityUtils.getUsername(); try { for (GenTable table : tableList) @@ -206,7 +216,7 @@ VelocityContext context = VelocityUtils.prepareContext(table); // 获取模板列表 - List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory()); + List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType()); for (String template : templates) { // 渲染模板 @@ -254,7 +264,7 @@ VelocityContext context = VelocityUtils.prepareContext(table); // 获取模板列表 - List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory()); + List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType()); for (String template : templates) { if (!StringUtils.containsAny(template, "sql.vm", "api.js.vm", "index.vue.vm", "index-tree.vue.vm")) @@ -367,7 +377,7 @@ VelocityContext context = VelocityUtils.prepareContext(table); // 获取模板列表 - List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory()); + List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory(), table.getTplWebType()); for (String template : templates) { // 渲染模板 -- Gitblit v1.9.2