智慧粮库
blame | history | patch | commit | commitdiff | ignore whitespace
fuyongxin
2026-06-03 1fd34350ea68fe820c4b62881b13c9f01082e6c0 
 ruoyi-modules/ruoyi-gen/src/main/java/com/ruoyi/gen/service/GenTableServiceImpl.java


@@ -1,13 +1,19 @@


package com.ruoyi.gen.service;







import java.io.ByteArrayOutputStream;



import java.io.File;



import java.io.IOException;



import java.io.StringWriter;



import java.util.Arrays;



import java.util.HashMap;



import java.util.LinkedHashMap;



import java.util.List;



import java.util.Map;



import java.util.function.Function;



import java.util.stream.Collectors;



import java.util.zip.ZipEntry;



import java.util.zip.ZipOutputStream;



import org.apache.commons.io.FileUtils;



import org.apache.commons.io.IOUtils;



import org.apache.velocity.Template;



import org.apache.velocity.VelocityContext;



@@ -17,11 +23,12 @@


import org.springframework.beans.factory.annotation.Autowired;



import org.springframework.stereotype.Service;



import org.springframework.transaction.annotation.Transactional;



import com.alibaba.fastjson.JSON;



import com.alibaba.fastjson.JSONObject;



import com.alibaba.fastjson2.JSON;



import com.alibaba.fastjson2.JSONObject;



import com.ruoyi.common.core.constant.Constants;



import com.ruoyi.common.core.constant.GenConstants;



import com.ruoyi.common.core.exception.CustomException;



import com.ruoyi.common.core.exception.ServiceException;



import com.ruoyi.common.core.text.CharsetKit;



import com.ruoyi.common.core.utils.StringUtils;



import com.ruoyi.common.security.utils.SecurityUtils;



import com.ruoyi.gen.domain.GenTable;



@@ -99,13 +106,24 @@


    }







    /**



     * 查询所有表信息



     * 


     * @return 表信息集合



     */



    @Override



    public List<GenTable> selectGenTableAll()



    {



        return genTableMapper.selectGenTableAll();



    }







    /**



     * 修改业务



     * 



     * @param genTable 业务信息



     * @return 结果



     */



    @Override



    @Transactional



    @Transactional(rollbackFor = Exception.class)



    public void updateGenTable(GenTable genTable)



    {



        String options = JSON.toJSONString(genTable.getParams());



@@ -113,9 +131,9 @@


        int row = genTableMapper.updateGenTable(genTable);



        if (row > 0)



        {



            for (GenTableColumn cenTableColumn : genTable.getColumns())



            for (GenTableColumn genTableColumn : genTable.getColumns())



            {



                genTableColumnMapper.updateGenTableColumn(cenTableColumn);



                genTableColumnMapper.updateGenTableColumn(genTableColumn);



            }



        }



    }



@@ -127,7 +145,7 @@


     * @return 结果



     */



    @Override



    @Transactional



    @Transactional(rollbackFor = Exception.class)



    public void deleteGenTableByIds(Long[] tableIds)



    {



        genTableMapper.deleteGenTableByIds(tableIds);



@@ -140,8 +158,8 @@


     * @param tableList 导入表列表



     */



    @Override



    @Transactional



    public void importGenTable(List<GenTable> tableList)



    @Transactional(rollbackFor = Exception.class)



    public void importGenTable(List<GenTable> tableList, String tplWebType)



    {



        String operName = SecurityUtils.getUsername();



        try



@@ -149,6 +167,7 @@


            for (GenTable table : tableList)



            {



                String tableName = table.getTableName();



                table.setTplWebType(tplWebType);



                GenUtils.initTable(table, operName);



                int row = genTableMapper.insertGenTable(table);



                if (row > 0)



@@ -165,7 +184,7 @@


        }



        catch (Exception e)



        {



            throw new CustomException("导入失败：" + e.getMessage());



            throw new ServiceException("导入失败：" + e.getMessage());



        }



    }







@@ -175,20 +194,22 @@


     * @param tableId 表编号



     * @return 预览数据列表



     */



    @Override



    public Map<String, String> previewCode(Long tableId)



    {



        Map<String, String> dataMap = new LinkedHashMap<>();



        // 查询表信息



        GenTable table = genTableMapper.selectGenTableById(tableId);



        // 查询列信息



        List<GenTableColumn> columns = table.getColumns();



        setPkColumn(table, columns);



        // 设置主子表信息



        setSubTable(table);



        // 设置主键列信息



        setPkColumn(table);



        VelocityInitializer.initVelocity();







        VelocityContext context = VelocityUtils.prepareContext(table);







        // 获取模板列表



        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());



        List<String> templates = VelocityUtils.getTemplateList(table);



        for (String template : templates)



        {



            // 渲染模板



@@ -201,35 +222,133 @@


    }







    /**



     * 生成代码



     * 生成代码（下载方式）



     * 



     * @param tableName 表名称



     * @return 数据



     */



    @Override



    public byte[] generatorCode(String tableName)



    public byte[] downloadCode(String tableName)



    {



        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();



        ZipOutputStream zip = new ZipOutputStream(outputStream);



        generatorCode(tableName, zip);



        IOUtils.closeQuietly(zip);



        return outputStream.toByteArray();



        return downloadCode(new String[] { tableName });



    }







    /**



     * 批量生成代码



     * 生成代码（自定义路径）



     * 


     * @param tableName 表名称



     */



    @Override



    public void generatorCode(String tableName)



    {



        // 查询表信息



        GenTable table = genTableMapper.selectGenTableByName(tableName);



        // 设置主子表信息



        setSubTable(table);



        // 设置主键列信息



        setPkColumn(table);







        VelocityInitializer.initVelocity();







        VelocityContext context = VelocityUtils.prepareContext(table);







        // 获取模板列表



        List<String> templates = VelocityUtils.getTemplateList(table);



        for (String template : templates)



        {



            if (!StringUtils.containsAny(template, "sql.vm", "api.js.vm", "api.ts.vm", "type.ts.vm", "index.ts.vm", "index.vue.vm", "index-tree.vue.vm", "view.vue.vm"))



            {



                // 渲染模板



                StringWriter sw = new StringWriter();



                Template tpl = Velocity.getTemplate(template, Constants.UTF8);



                tpl.merge(context, sw);



                try



                {



                    String path = getGenPath(table, template);



                    FileUtils.writeStringToFile(new File(path), sw.toString(), CharsetKit.UTF_8);



                }



                catch (IOException e)



                {



                    throw new ServiceException("渲染模板失败，表名：" + table.getTableName());



                }



            }



        }



    }







    /**



     * 同步数据库



     * 


     * @param tableName 表名称



     */



    @Override



    @Transactional(rollbackFor = Exception.class)



    public void synchDb(String tableName)



    {



        GenTable table = genTableMapper.selectGenTableByName(tableName);



        List<GenTableColumn> tableColumns = table.getColumns();



        Map<String, GenTableColumn> tableColumnMap = tableColumns.stream().collect(Collectors.toMap(GenTableColumn::getColumnName, Function.identity()));







        List<GenTableColumn> dbTableColumns = genTableColumnMapper.selectDbTableColumnsByName(tableName);



        if (StringUtils.isEmpty(dbTableColumns))



        {



            throw new ServiceException("同步数据失败，原表结构不存在");



        }



        List<String> dbTableColumnNames = dbTableColumns.stream().map(GenTableColumn::getColumnName).collect(Collectors.toList());







        dbTableColumns.forEach(column -> {



            GenUtils.initColumnField(column, table);



            if (tableColumnMap.containsKey(column.getColumnName()))



            {



                GenTableColumn prevColumn = tableColumnMap.get(column.getColumnName());



                column.setColumnId(prevColumn.getColumnId());



                if (column.isList())



                {



                    // 如果是列表，继续保留查询方式/字典类型选项



                    column.setDictType(prevColumn.getDictType());



                    column.setQueryType(prevColumn.getQueryType());



                }



                if (StringUtils.isNotEmpty(prevColumn.getIsRequired()) && !column.isPk()



                        && (column.isInsert() || column.isEdit())



                        && ((column.isUsableColumn()) || (!column.isSuperColumn())))



                {



                    // 如果是(新增/修改&非主键/非忽略及父属性)，继续保留必填/显示类型选项



                    column.setIsRequired(prevColumn.getIsRequired());



                    column.setHtmlType(prevColumn.getHtmlType());



                }



                genTableColumnMapper.updateGenTableColumn(column);



            }



            else



            {



                genTableColumnMapper.insertGenTableColumn(column);



            }



        });







        List<GenTableColumn> delColumns = tableColumns.stream().filter(column -> !dbTableColumnNames.contains(column.getColumnName())).collect(Collectors.toList());



        if (StringUtils.isNotEmpty(delColumns))



        {



            genTableColumnMapper.deleteGenTableColumns(delColumns);



        }



    }







    /**



     * 批量生成代码（下载方式）



     * 



     * @param tableNames 表数组



     * @return 数据



     */



    @Override



    public byte[] generatorCode(String[] tableNames)



    public byte[] downloadCode(String[] tableNames)



    {



        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();



        ZipOutputStream zip = new ZipOutputStream(outputStream);



        Map<String, StringBuffer> typeFiles = new HashMap<>();



        for (String tableName : tableNames)



        {



            generatorCode(tableName, zip);



            generatorCode(tableName, zip, typeFiles);



        }



        for (Map.Entry<String, StringBuffer> entry : typeFiles.entrySet())



        {



            writeToZip(zip, entry.getKey(), entry.getValue().toString());



        }



        IOUtils.closeQuietly(zip);



        return outputStream.toByteArray();



@@ -238,39 +357,67 @@


    /**



     * 查询表信息并生成代码



     */



    private void generatorCode(String tableName, ZipOutputStream zip)



    private void generatorCode(String tableName, ZipOutputStream zip, Map<String, StringBuffer> typeFiles)



    {



        // 查询表信息



        GenTable table = genTableMapper.selectGenTableByName(tableName);



        // 查询列信息



        List<GenTableColumn> columns = table.getColumns();



        setPkColumn(table, columns);



        // 设置主子表信息



        setSubTable(table);



        // 设置主键列信息



        setPkColumn(table);







        VelocityInitializer.initVelocity();







        VelocityContext context = VelocityUtils.prepareContext(table);







        // 获取模板列表



        List<String> templates = VelocityUtils.getTemplateList(table.getTplCategory());



        List<String> templates = VelocityUtils.getTemplateList(table);



        for (String template : templates)



        {



            // 渲染模板



            StringWriter sw = new StringWriter();



            Template tpl = Velocity.getTemplate(template, Constants.UTF8);



            tpl.merge(context, sw);



            try



            String fileName = VelocityUtils.getFileName(template, table);



            // index-bak.ts 模版，追加内容



            if (fileName.contains("index-bak.ts"))



            {



                // 添加到zip



                zip.putNextEntry(new ZipEntry(VelocityUtils.getFileName(template, table)));



                IOUtils.write(sw.toString(), zip, Constants.UTF8);



                IOUtils.closeQuietly(sw);



            zip.flush();



                zip.closeEntry();



                if (!typeFiles.containsKey(fileName))



                {



                    typeFiles.put(fileName, new StringBuffer(sw.toString()));



                }



                else



                {



                    Arrays.stream(sw.toString().split("\n")).filter(line -> line.startsWith("export * from")).forEach(line -> typeFiles.get(fileName).append("\n").append(line));



                }



            }



            catch (IOException e)



            else



            {



                log.error("渲染模板失败，表名：" + table.getTableName(), e);



                // 其他文件正常添加



                writeToZip(zip, fileName, sw.toString());



            }



        }



    }







    /**



     * 将字符串内容写入ZIP输出流



     * 


     * @param zip ZIP输出流



     * @param fileName ZIP条目名称（即文件名）



     * @param content 要写入的内容



     */



    private void writeToZip(ZipOutputStream zip, String fileName, String content)



    {



        try



        {



            zip.putNextEntry(new ZipEntry(fileName));



            IOUtils.write(content, zip, Constants.UTF8);



            zip.flush();



            zip.closeEntry();



        }



        catch (IOException e)



        {



            log.error("写入ZIP文件失败，文件名: " + fileName, e);



        }



    }







@@ -285,18 +432,29 @@


        if (GenConstants.TPL_TREE.equals(genTable.getTplCategory()))



        {



            String options = JSON.toJSONString(genTable.getParams());



            JSONObject paramsObj = JSONObject.parseObject(options);



            JSONObject paramsObj = JSON.parseObject(options);



            if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_CODE)))



            {



                throw new CustomException("树编码字段不能为空");



                throw new ServiceException("树编码字段不能为空");



            }



            else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_PARENT_CODE)))



            {



                throw new CustomException("树父编码字段不能为空");



                throw new ServiceException("树父编码字段不能为空");



            }



            else if (StringUtils.isEmpty(paramsObj.getString(GenConstants.TREE_NAME)))



            {



                throw new CustomException("树名称字段不能为空");



                throw new ServiceException("树名称字段不能为空");



            }



        }



        else if (GenConstants.TPL_SUB.equals(genTable.getTplCategory()))



        {



            if (StringUtils.isEmpty(genTable.getSubTableName()))



            {



                throw new ServiceException("关联子表的表名不能为空");



            }



            else if (StringUtils.isEmpty(genTable.getSubTableFkName()))



            {



                throw new ServiceException("子表关联的外键名不能为空");



            }



        }



    }



@@ -305,11 +463,10 @@


     * 设置主键列信息



     * 



     * @param table 业务表信息



     * @param columns 业务字段列表



     */



    public void setPkColumn(GenTable table, List<GenTableColumn> columns)



    public void setPkColumn(GenTable table)



    {



        for (GenTableColumn column : columns)



        for (GenTableColumn column : table.getColumns())



        {



            if (column.isPk())



            {



@@ -319,7 +476,36 @@


        }



        if (StringUtils.isNull(table.getPkColumn()))



        {



            table.setPkColumn(columns.get(0));



            table.setPkColumn(table.getColumns().get(0));



        }



        if (GenConstants.TPL_SUB.equals(table.getTplCategory()))



        {



            for (GenTableColumn column : table.getSubTable().getColumns())



            {



                if (column.isPk())



                {



                    table.getSubTable().setPkColumn(column);



                    break;



                }



            }



            if (StringUtils.isNull(table.getSubTable().getPkColumn()))



            {



                table.getSubTable().setPkColumn(table.getSubTable().getColumns().get(0));



            }



        }



    }







    /**



     * 设置主子表信息



     * 


     * @param table 业务表信息



     */



    public void setSubTable(GenTable table)



    {



        String subTableName = table.getSubTableName();



        if (StringUtils.isNotEmpty(subTableName))



        {



            table.setSubTable(genTableMapper.selectGenTableByName(subTableName));



        }



    }







@@ -330,15 +516,39 @@


     */



    public void setTableFromOptions(GenTable genTable)



    {



        JSONObject paramsObj = JSONObject.parseObject(genTable.getOptions());



        JSONObject paramsObj = JSON.parseObject(genTable.getOptions());



        if (StringUtils.isNotNull(paramsObj))



        {



            String treeCode = paramsObj.getString(GenConstants.TREE_CODE);



            String treeParentCode = paramsObj.getString(GenConstants.TREE_PARENT_CODE);



            String treeName = paramsObj.getString(GenConstants.TREE_NAME);



            Long parentMenuId = paramsObj.getLongValue(GenConstants.PARENT_MENU_ID);



            String parentMenuName = paramsObj.getString(GenConstants.PARENT_MENU_NAME);



            boolean isView = paramsObj.getBooleanValue(GenConstants.GEN_VIEW);







            genTable.setTreeCode(treeCode);



            genTable.setTreeParentCode(treeParentCode);



            genTable.setTreeName(treeName);



            genTable.setParentMenuId(parentMenuId);



            genTable.setParentMenuName(parentMenuName);



            genTable.setView(isView);



        }



    }



}






    /**



     * 获取代码生成地址



     * 


     * @param table 业务表信息



     * @param template 模板文件路径



     * @return 生成地址



     */



    public static String getGenPath(GenTable table, String template)



    {



        String genPath = table.getGenPath();



        if (StringUtils.equals(genPath, "/"))



        {



            return System.getProperty("user.dir") + File.separator + "src" + File.separator + VelocityUtils.getFileName(template, table);



        }



        return genPath + File.separator + VelocityUtils.getFileName(template, table);



    }



}